use crate::pkg::{db, resolve, types};
use pubgrub::{Dependencies, DependencyProvider, Ranges};
use rusqlite::params;
use rustc_hash::FxHashMap;
use semver::Version;
use std::cell::RefCell;
use std::fmt::Display;
use thiserror::Error;
#[derive(Clone, Debug, Eq, PartialEq, Hash)]
pub struct PkgName {
pub name: String,
pub sub_package: Option<String>,
pub repo: String,
pub registry: String,
pub explicit_source: Option<String>,
}
impl Display for PkgName {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
if let Some(source) = &self.explicit_source {
return write!(f, "{}", source);
}
if let Some(sub) = &self.sub_package {
write!(f, "#{}@{}/{}:{}", self.registry, self.repo, self.name, sub)
} else {
write!(f, "#{}@{}/{}", self.registry, self.repo, self.name)
}
}
}
#[derive(Clone, Debug, Eq, PartialEq, Ord, PartialOrd, Hash)]
pub struct SemVersion(pub Version);
impl Display for SemVersion {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
write!(f, "{}", self.0)
}
}
#[derive(Error, Debug)]
pub enum ZoiSolverError {
#[error("Dependency error: {0}")]
Dependency(String),
#[error("Version error: {0}")]
Version(String),
#[error("Anyhow error: {0}")]
Anyhow(#[from] anyhow::Error),
#[error("Other error: {0}")]
Other(String),
}
pub struct ZoiDependencyProvider {
pub root_deps: FxHashMap<PkgName, Ranges<SemVersion>>,
pub initial_sources: Vec<String>,
pub quiet: bool,
pub yes: bool,
pub all_optional: bool,
pub mini_index: Option<crate::pkg::mini_resolve::MiniRegistryIndex>,
pub deps_cache:
RefCell<FxHashMap<(PkgName, SemVersion), FxHashMap<PkgName, Ranges<SemVersion>>>>,
pub chosen_cache:
RefCell<FxHashMap<(PkgName, SemVersion), (Vec<String>, Vec<String>, Vec<String>)>>,
}
pub fn semver_to_range(req_str: &str) -> Ranges<SemVersion> {
let req_str = req_str.trim_start_matches('@').trim_start_matches('v');
if let Ok(version) = Version::parse(req_str) {
return Ranges::singleton(SemVersion(version));
}
if let Ok(req) = semver::VersionReq::parse(req_str) {
let mut range = Ranges::full();
for comparator in &req.comparators {
let v = SemVersion(Version {
major: comparator.major,
minor: comparator.minor.unwrap_or(0),
patch: comparator.patch.unwrap_or(0),
pre: comparator.pre.clone(),
build: semver::BuildMetadata::EMPTY,
});
let comp_range = match comparator.op {
semver::Op::Exact => Ranges::singleton(v),
semver::Op::Greater => Ranges::strictly_higher_than(v),
semver::Op::GreaterEq => Ranges::higher_than(v),
semver::Op::Less => Ranges::strictly_lower_than(v),
semver::Op::LessEq => Ranges::lower_than(v),
semver::Op::Tilde => {
let next_minor = SemVersion(Version {
major: comparator.major,
minor: comparator.minor.unwrap_or(0) + 1,
patch: 0,
pre: semver::Prerelease::EMPTY,
build: semver::BuildMetadata::EMPTY,
});
Ranges::higher_than(v).intersection(&Ranges::strictly_lower_than(next_minor))
}
semver::Op::Caret => {
let next = if comparator.major > 0 {
Version {
major: comparator.major + 1,
minor: 0,
patch: 0,
pre: semver::Prerelease::EMPTY,
build: semver::BuildMetadata::EMPTY,
}
} else if let Some(minor) = comparator.minor {
if minor > 0 {
Version {
major: 0,
minor: minor + 1,
patch: 0,
pre: semver::Prerelease::EMPTY,
build: semver::BuildMetadata::EMPTY,
}
} else {
Version {
major: 0,
minor: 0,
patch: comparator.patch.unwrap_or(0) + 1,
pre: semver::Prerelease::EMPTY,
build: semver::BuildMetadata::EMPTY,
}
}
} else {
Version {
major: 1,
minor: 0,
patch: 0,
pre: semver::Prerelease::EMPTY,
build: semver::BuildMetadata::EMPTY,
}
};
Ranges::higher_than(v)
.intersection(&Ranges::strictly_lower_than(SemVersion(next)))
}
_ => Ranges::full(),
};
range = range.intersection(&comp_range);
}
return range;
}
Ranges::full()
}
impl ZoiDependencyProvider {
pub fn new(
root_deps: FxHashMap<PkgName, Ranges<SemVersion>>,
initial_sources: Vec<String>,
quiet: bool,
yes: bool,
all_optional: bool,
) -> Result<Self, anyhow::Error> {
let mini_index = if crate::utils::is_mini_mode() {
Some(crate::pkg::mini_resolve::fetch_registry_index()?)
} else {
None
};
Ok(Self {
root_deps,
initial_sources,
quiet,
yes,
all_optional,
mini_index,
deps_cache: RefCell::new(FxHashMap::default()),
chosen_cache: RefCell::new(FxHashMap::default()),
})
}
fn semver_to_range(&self, req_str: &str) -> Ranges<SemVersion> {
semver_to_range(req_str)
}
fn source_matches_package(&self, package: &PkgName, source: &str) -> bool {
if let Some(explicit_source) = &package.explicit_source {
let explicit_base = explicit_source
.rsplit_once('@')
.map(|(base, _)| base)
.unwrap_or(explicit_source.as_str());
let source_base = source
.rsplit_once('@')
.map(|(base, _)| base)
.unwrap_or(source);
return explicit_base == source_base;
}
let Ok(req) = resolve::parse_source_string(source) else {
return false;
};
if req.name != package.name || req.sub_package != package.sub_package {
return false;
}
let Ok(resolved_source) = resolve::resolve_source(source, true, true) else {
return false;
};
resolved_source.repo_name.unwrap_or_default() == package.repo
&& resolved_source
.registry_handle
.unwrap_or_else(|| "zoidberg".to_string())
== package.registry
}
pub fn get_versions(&self, package: &PkgName) -> Result<Vec<SemVersion>, ZoiSolverError> {
let mut all_versions = Vec::new();
if let Some(index) = &self.mini_index
&& let Some(pkg_info) = index.packages.get(&package.name)
&& let Ok(v) = Version::parse(pkg_info.version.trim_start_matches('v'))
{
all_versions.push(SemVersion(v));
}
if let Ok(version_strings) =
db::get_all_versions(&package.registry, &package.name, &package.repo)
{
for v_str in version_strings {
if let Ok(v) = Version::parse(&v_str) {
all_versions.push(SemVersion(v));
}
}
}
for source in &self.initial_sources {
if self.source_matches_package(package, source)
&& let Ok(req) = resolve::parse_source_string(source)
&& let Some(v_spec) = req.version_spec
{
let v_clean = v_spec.trim_start_matches('@').trim_start_matches('v');
if let Ok(v) = Version::parse(v_clean) {
all_versions.push(SemVersion(v));
}
}
}
let source_str = package.explicit_source.clone().unwrap_or_else(|| {
if let Some(sub) = &package.sub_package {
format!(
"#{}@{}/{}:{}",
package.registry, package.repo, package.name, sub
)
} else {
format!("#{}@{}/{}", package.registry, package.repo, package.name)
}
});
if let Ok(resolved) = resolve::resolve_source(&source_str, true, true) {
let path_str = resolved.path.to_string_lossy();
if let Ok(pkg) = crate::pkg::lua::parser::parse_lua_package(&path_str, None, true) {
if let Some(v_str) = &pkg.version {
let v_clean = v_str.trim_start_matches('v');
if let Ok(v) = Version::parse(v_clean) {
all_versions.push(SemVersion(v));
}
}
if let Some(versions_map) = &pkg.versions {
for channel in versions_map.keys() {
if let Ok(v_str) = resolve::resolve_channel(versions_map, channel) {
let v_clean = v_str.trim_start_matches('v');
if let Ok(v) = Version::parse(v_clean) {
all_versions.push(SemVersion(v));
}
}
}
}
}
}
if all_versions.is_empty()
&& let Ok(conn) = db::open_connection(&package.registry)
{
let mut stmt = conn
.prepare("SELECT version FROM packages WHERE name = ?1")
.map_err(|e| ZoiSolverError::Other(e.to_string()))?;
let rows = stmt
.query_map(params![package.name], |row| row.get::<_, Option<String>>(0))
.map_err(|e| ZoiSolverError::Other(e.to_string()))?;
for v_res in rows {
if let Ok(Some(v_str)) = v_res {
let v_clean = v_str.trim_start_matches('v');
if let Ok(v) = Version::parse(v_clean) {
all_versions.push(SemVersion(v));
}
}
}
}
all_versions.sort();
all_versions.dedup();
Ok(all_versions)
}
}
impl DependencyProvider for ZoiDependencyProvider {
type P = PkgName;
type V = SemVersion;
type VS = Ranges<SemVersion>;
type M = String;
type Priority = i32;
type Err = ZoiSolverError;
fn get_dependencies(
&self,
package: &Self::P,
version: &Self::V,
) -> Result<Dependencies<Self::P, Self::VS, Self::M>, Self::Err> {
if package.name == "$root" {
return Ok(Dependencies::Available(
pubgrub::DependencyConstraints::from_iter(self.root_deps.clone()),
));
}
let cache_key = (package.clone(), version.clone());
if let Some(cached) = self.deps_cache.borrow().get(&cache_key) {
return Ok(Dependencies::Available(
pubgrub::DependencyConstraints::from_iter(cached.clone()),
));
}
let version_str = version.to_string();
let dependencies_opt = db::get_package_dependencies(
&package.registry,
&package.name,
&version_str,
package.sub_package.as_deref(),
&package.repo,
)
.ok()
.flatten();
let package_deps = if let Some(deps_json) = dependencies_opt
&& !deps_json.is_empty()
{
serde_json::from_str::<types::Dependencies>(&deps_json).ok()
} else {
let source = package
.explicit_source
.clone()
.unwrap_or_else(|| format!("{}@{}", package, version_str));
let pkg_res = resolve::resolve_package_and_version(&source, self.quiet, self.yes);
match pkg_res {
Ok((pkg, _, _, _, _, _)) => pkg.dependencies,
Err(_) => None,
}
};
let mut deps = FxHashMap::default();
let mut chosen_opts = Vec::new();
let mut chosen_opts_opt = Vec::new();
let mut all_req = Vec::new();
if let Some(dependencies) = package_deps
&& let Some(runtime) = &dependencies.runtime
{
let (req_deps, co, coo) =
crate::pkg::install::resolver::collect_dependencies_for_group(
runtime,
package.sub_package.as_deref(),
Some("runtime"),
self.yes,
self.all_optional,
)
.map_err(|e| ZoiSolverError::Dependency(e.to_string()))?;
chosen_opts = co;
chosen_opts_opt = coo;
all_req = req_deps.clone();
for dep_str in req_deps {
let dep_req = crate::pkg::dependencies::parse_dependency_string(&dep_str)
.map_err(|e| ZoiSolverError::Dependency(e.to_string()))?;
if dep_req.manager == "zoi" {
let req = resolve::parse_source_string(dep_req.package)
.map_err(|e| ZoiSolverError::Dependency(e.to_string()))?;
let resolved_dep =
resolve::resolve_source(dep_req.package, self.quiet, self.yes)
.map_err(|e| ZoiSolverError::Dependency(e.to_string()))?;
let dep_name = PkgName {
name: req.name,
sub_package: req.sub_package,
repo: resolved_dep.repo_name.unwrap_or_default(),
registry: resolved_dep
.registry_handle
.unwrap_or_else(|| "zoidberg".to_string()),
explicit_source: matches!(
resolved_dep.source_type,
resolve::SourceType::LocalFile
| resolve::SourceType::Url
| resolve::SourceType::GitRepo(_)
)
.then(|| dep_req.package.to_string()),
};
let range = if req.version_spec.is_some() {
let resolved_version =
resolve::resolve_requested_version_spec(dep_req.package, true, true)
.map_err(|e| ZoiSolverError::Dependency(e.to_string()))?
.ok_or_else(|| {
ZoiSolverError::Dependency(format!(
"version spec missing for '{}'",
dep_req.package
))
})?;
self.semver_to_range(&resolved_version)
} else {
Ranges::full()
};
deps.insert(dep_name, range);
}
}
}
self.deps_cache
.borrow_mut()
.insert(cache_key.clone(), deps.clone());
self.chosen_cache
.borrow_mut()
.insert(cache_key, (chosen_opts, chosen_opts_opt, all_req));
Ok(Dependencies::Available(
pubgrub::DependencyConstraints::from_iter(deps),
))
}
fn choose_version(
&self,
package: &Self::P,
versions: &pubgrub::Ranges<Self::V>,
) -> Result<Option<Self::V>, Self::Err> {
if package.name == "$root" {
return Ok(Some(SemVersion(Version::new(0, 0, 0))));
}
let all_versions = self.get_versions(package)?;
let best_version = all_versions.into_iter().rfind(|v| versions.contains(v));
Ok(best_version)
}
fn prioritize(
&self,
_package: &Self::P,
_range: &Self::VS,
_stats: &pubgrub::PackageResolutionStatistics,
) -> Self::Priority {
0
}
}