use std::{
collections::BTreeMap,
fs,
io::{self, Seek},
path::Path,
};
use fs_lock::FileLock;
use miette::Diagnostic;
use thiserror::Error as ThisError;
use crate::{
binstall_crates_v1::{Error as BinstallCratesV1Error, Records as BinstallCratesV1Records},
cargo_crates_v1::{CratesToml, CratesTomlParseError},
crate_info::CrateInfo,
helpers::create_if_not_exist,
CompactString, Version,
};
#[derive(Debug, Diagnostic, ThisError)]
#[non_exhaustive]
pub enum ManifestsError {
#[error("failed to parse binstall crates-v1 manifest: {0}")]
#[diagnostic(transparent)]
BinstallCratesV1(#[from] BinstallCratesV1Error),
#[error("failed to parse cargo v1 manifest: {0}")]
#[diagnostic(transparent)]
CargoManifestV1(#[from] CratesTomlParseError),
#[error("I/O error: {0}")]
Io(#[from] io::Error),
}
pub struct Manifests {
binstall: BinstallCratesV1Records,
cargo_crates_v1: FileLock,
installed_crates: BTreeMap<CompactString, Version>,
}
impl Manifests {
pub fn open_exclusive(cargo_roots: &Path) -> Result<Self, ManifestsError> {
let binstall_dir = cargo_roots.join("binstall");
fs::create_dir_all(&binstall_dir)?;
let metadata_path = binstall_dir.join("crates-v1.json");
let mut binstall = BinstallCratesV1Records::load_from_path(&metadata_path)?;
let manifest_path = cargo_roots.join(".crates.toml");
let mut cargo_crates_v1 = create_if_not_exist(&manifest_path)?;
let installed_crates = CratesToml::load_from_reader(&mut cargo_crates_v1)
.and_then(CratesToml::collect_into_crates_versions)?;
binstall.retain(|crate_info| installed_crates.contains_key(&crate_info.name));
Ok(Self {
binstall,
cargo_crates_v1,
installed_crates,
})
}
fn rewind_cargo_crates_v1(&mut self) -> Result<(), ManifestsError> {
self.cargo_crates_v1.rewind().map_err(ManifestsError::from)
}
pub fn installed_crates(&self) -> &BTreeMap<CompactString, Version> {
&self.installed_crates
}
pub fn update(mut self, metadata_vec: Vec<CrateInfo>) -> Result<(), ManifestsError> {
self.rewind_cargo_crates_v1()?;
CratesToml::append_to_file(&mut self.cargo_crates_v1, &metadata_vec)?;
for metadata in metadata_vec {
self.binstall.replace(metadata);
}
self.binstall.overwrite()?;
Ok(())
}
}