use crate::{File, Graph, MAX_COMMITS};
use gix_error::Result;
use gix_error::{OptionExt, ResultExt, bail, message, validation};
use std::{
io::{BufRead, BufReader},
path::Path,
};
impl Graph {
pub fn at(path: &Path) -> Result<Self> {
Self::try_from(path)
}
pub fn from_commit_graphs_dir(path: &Path) -> Result<Self> {
let commit_graphs_dir = path;
let chain_file_path = commit_graphs_dir.join("commit-graph-chain");
let chain_file = std::fs::File::open(&chain_file_path).or_raise(|| {
message!(
"Could not open commit-graph chain file at \"{}\"",
chain_file_path.display()
)
})?;
let mut files = Vec::new();
for line in BufReader::new(chain_file).lines() {
let hash = line.or_raise(|| {
message!(
"Could not read from commit-graph file at \"{}\"",
chain_file_path.display()
)
})?;
let graph_file_path = commit_graphs_dir.join(format!("graph-{hash}.graph"));
files.push(File::at(&graph_file_path)?);
}
Self::new(files).or_raise(|| {
message!(
"Could not assemble commit-graph from chain file at \"{}\"",
chain_file_path.display()
)
.corrupted()
})
}
pub fn from_file(path: &Path) -> Result<Self> {
let file = File::at(path)?;
Self::new(vec![file])
.or_raise(|| message!("Could not assemble commit-graph from file at \"{}\"", path.display()).corrupted())
}
pub fn from_info_dir(info_dir: &Path) -> Result<Self> {
Self::from_file(&info_dir.join("commit-graph"))
.or_else(|_| Self::from_commit_graphs_dir(&info_dir.join("commit-graphs")))
}
pub fn new(files: Vec<File>) -> Result<Self> {
let files = nonempty::NonEmpty::from_vec(files)
.ok_or_raise(|| validation("Commit-graph must contain at least one file"))?;
let num_commits: u64 = files.iter().map(|f| u64::from(f.num_commits())).sum();
if num_commits > u64::from(MAX_COMMITS) {
bail!(
"Commit-graph files contain {num_commits} commits altogether, but only {MAX_COMMITS} commits are allowed".validation()
);
}
let mut f1 = files.first();
for f2 in files.tail() {
if f1.object_hash() != f2.object_hash() {
bail!(
"Commit-graph files mismatch: \"{path1}\" uses hash {hash1:?}, but \"{path2}\" uses hash {hash2:?}"
.validation(),
path1 = f1.path().display(),
hash1 = f1.object_hash(),
path2 = f2.path().display(),
hash2 = f2.object_hash(),
);
}
f1 = f2;
}
Ok(Self { files })
}
}
impl TryFrom<&Path> for Graph {
type Error = gix_error::Error;
fn try_from(path: &Path) -> Result<Self> {
let metadata = path
.metadata()
.or_raise(|| message!("Could not access commit-graph path at \"{}\"", path.display()))?;
if metadata.is_file() {
Self::from_file(path)
} else if metadata.is_dir() {
if path.join("commit-graph-chain").is_file() {
Self::from_commit_graphs_dir(path)
} else {
Self::from_info_dir(path)
}
} else {
Err(message!(
"Did not find any files that look like commit graphs at \"{}\"",
path.display()
)
.validation_error())
}
}
}