use std::io::BufReader;
use std::path::PathBuf;
use anyhow::Context;
use fs_err::File;
use crate::manifest::Manifest;
use crate::package_id::PackageId;
use crate::package_index::PackageIndexConfig;
use crate::package_req::PackageReq;
use crate::package_source::PackageContents;
use super::{PackageSourceId, PackageSourceProvider};
#[derive(Clone)]
pub struct TestRegistry {
path: PathBuf,
}
impl TestRegistry {
pub fn new<P: Into<PathBuf>>(path: P) -> Self {
Self { path: path.into() }
}
}
impl PackageSourceProvider for TestRegistry {
fn update(&self) -> anyhow::Result<()> {
Ok(())
}
fn query(&self, package_req: &PackageReq) -> anyhow::Result<Vec<Manifest>> {
let mut package_path = self.path.clone();
package_path.push("index");
package_path.push(package_req.name().scope());
package_path.push(package_req.name().name());
let file = File::open(&package_path)
.with_context(|| format!("could not open package {} from index", package_req.name()))?;
let file = BufReader::new(file);
let manifest_stream: Result<Vec<Manifest>, serde_json::Error> =
serde_json::Deserializer::from_reader(file)
.into_iter::<Manifest>()
.filter(|manifest| {
if let Ok(manifest) = manifest {
package_req.matches(&manifest.package.name, &manifest.package.version)
} else {
true
}
})
.collect();
let versions = manifest_stream.with_context(|| {
format!(
"could not parse package index entry for {}",
package_req.name()
)
})?;
Ok(versions)
}
fn download_package(&self, package_id: &PackageId) -> anyhow::Result<PackageContents> {
let mut package_path = self.path.clone();
package_path.push("contents");
package_path.push(package_id.name().scope());
package_path.push(package_id.name().name());
package_path.push(format!("{}.zip", package_id.version()));
let data = fs_err::read(&package_path)?;
Ok(PackageContents::from_buffer(data))
}
fn fallback_sources(&self) -> anyhow::Result<Vec<PackageSourceId>> {
let config_path = self.path.join("index/config.json");
let contents = fs_err::read_to_string(config_path)?;
let config: PackageIndexConfig = serde_json::from_str(&contents)?;
let sources = config
.fallback_registries
.iter()
.map(|source| self.path.join(source).canonicalize().unwrap())
.map(PackageSourceId::Path)
.collect();
Ok(sources)
}
}