lux_lib/lockfile/

mod.rs

use std::collections::{BTreeMap, HashSet};
use std::fmt::Display;
use std::io::{self, Write};
use std::marker::PhantomData;
use std::ops::{Deref, DerefMut};
use std::{
    collections::HashMap,
    fs::File,
    io::ErrorKind,
    path::{Path, PathBuf},
};

use itertools::Itertools;

use miette::Diagnostic;
use serde::{de, Deserialize, Serialize, Serializer};
use sha2::{Digest, Sha256};
use ssri::Integrity;
use strum_macros::EnumIter;
use thiserror::Error;
use url::Url;

use crate::fs;
use crate::package::{
    PackageName, PackageReq, PackageSpec, PackageVersion, PackageVersionReq,
    PackageVersionReqError, RemotePackageTypeFilterSpec,
};
use crate::remote_package_source::RemotePackageSource;
use crate::rockspec::lua_dependency::LuaDependencySpec;
use crate::rockspec::RockBinaries;
use crate::tree::{EntryType, InstallTree, Tree};

/// Bump this whenever an incompatible change is made to the lockfile format.
/// Lockfiles with a different major version are rejected on load.
const LOCKFILE_VERSION: u64 = 2;
const LOCKFILE_VERSION_STR: &str = "2.0.0";

#[derive(Copy, Debug, PartialEq, Eq, Hash, Clone, PartialOrd, Ord, Default)]
pub enum PinnedState {
    /// Unpinned packages can be updated
    #[default]
    Unpinned,
    /// Pinned packages cannot be updated
    Pinned,
}

impl Display for PinnedState {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match &self {
            PinnedState::Unpinned => "unpinned".fmt(f),
            PinnedState::Pinned => "pinned".fmt(f),
        }
    }
}

impl From<bool> for PinnedState {
    fn from(value: bool) -> Self {
        if value {
            Self::Pinned
        } else {
            Self::Unpinned
        }
    }
}

impl PinnedState {
    pub fn as_bool(&self) -> bool {
        match self {
            Self::Unpinned => false,
            Self::Pinned => true,
        }
    }

    fn is_default(&self) -> bool {
        &Self::default() == self
    }
}

impl Serialize for PinnedState {
    fn serialize<S>(&self, serializer: S) -> std::result::Result<S::Ok, S::Error>
    where
        S: serde::Serializer,
    {
        serializer.serialize_bool(self.as_bool())
    }
}

impl<'de> Deserialize<'de> for PinnedState {
    fn deserialize<D>(deserializer: D) -> std::result::Result<Self, D::Error>
    where
        D: serde::Deserializer<'de>,
    {
        Ok(match bool::deserialize(deserializer)? {
            false => Self::Unpinned,
            true => Self::Pinned,
        })
    }
}

#[derive(Copy, Debug, PartialEq, Eq, Hash, Clone, PartialOrd, Ord, Default)]
pub enum OptState {
    /// A required package
    #[default]
    Required,
    /// An optional package
    Optional,
}

impl OptState {
    pub(crate) fn as_bool(&self) -> bool {
        match self {
            Self::Required => false,
            Self::Optional => true,
        }
    }

    fn is_default(&self) -> bool {
        &Self::default() == self
    }
}

impl From<bool> for OptState {
    fn from(value: bool) -> Self {
        if value {
            Self::Optional
        } else {
            Self::Required
        }
    }
}

impl Display for OptState {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match &self {
            OptState::Required => "required".fmt(f),
            OptState::Optional => "optional".fmt(f),
        }
    }
}

impl Serialize for OptState {
    fn serialize<S>(&self, serializer: S) -> std::result::Result<S::Ok, S::Error>
    where
        S: serde::Serializer,
    {
        serializer.serialize_bool(self.as_bool())
    }
}

impl<'de> Deserialize<'de> for OptState {
    fn deserialize<D>(deserializer: D) -> std::result::Result<Self, D::Error>
    where
        D: serde::Deserializer<'de>,
    {
        Ok(match bool::deserialize(deserializer)? {
            false => Self::Required,
            true => Self::Optional,
        })
    }
}

#[derive(Debug, Serialize, Deserialize, Clone, PartialEq, Eq, Hash, PartialOrd, Ord)]
pub struct LockedPackageSpec {
    name: PackageName,
    version: PackageVersion,
    pinned: PinnedState,
    opt: OptState,
    dependencies: Vec<LockedPackageId>,
    build_dependencies: Vec<LockedPackageId>,
    // TODO: Deserialize this directly into a `LuaPackageReq`
    constraint: Option<String>,
}

/// ID of a local package, a hash that is comprised of:
/// - name
/// - version
/// - pinned state
/// - opt state
/// - lock constraint
#[derive(Debug, PartialEq, Eq, Hash, PartialOrd, Ord, Serialize, Deserialize, Clone)]
pub struct LockedPackageId(String);

impl LockedPackageId {
    pub fn new(
        name: &PackageName,
        version: &PackageVersion,
        pinned: PinnedState,
        opt: OptState,
        constraint: LockConstraint,
    ) -> Self {
        let mut hasher = Sha256::new();

        hasher.update(format!(
            "{}{}{}{}{}",
            name,
            version,
            pinned.as_bool(),
            opt.as_bool(),
            match constraint {
                LockConstraint::Unconstrained => String::default(),
                LockConstraint::Constrained(version_req) => version_req.to_string(),
            },
        ));

        Self(hex::encode(hasher.finalize()))
    }

    /// Constructs a package ID from a hashed string.
    ///
    /// # Safety
    ///
    /// Ensure that the hash you are providing to this function
    /// is not malformed and resolves to a valid package ID for the target
    /// tree you are working with.
    pub unsafe fn from_unchecked(str: String) -> Self {
        Self(str)
    }

    pub fn into_string(self) -> String {
        self.0
    }
}

impl Display for LockedPackageId {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        self.0.fmt(f)
    }
}

impl LockedPackageSpec {
    #[allow(clippy::too_many_arguments)]
    pub(crate) fn new(
        name: &PackageName,
        version: &PackageVersion,
        constraint: LockConstraint,
        dependencies: Vec<LockedPackageId>,
        build_dependencies: Vec<LockedPackageId>,
        pinned: &PinnedState,
        opt: &OptState,
    ) -> Self {
        Self {
            name: name.clone(),
            version: version.clone(),
            pinned: *pinned,
            opt: *opt,
            dependencies,
            build_dependencies,
            constraint: match constraint {
                LockConstraint::Unconstrained => None,
                LockConstraint::Constrained(version_req) => Some(version_req.to_string()),
            },
        }
    }

    pub fn id(&self) -> LockedPackageId {
        LockedPackageId::new(
            self.name(),
            self.version(),
            self.pinned,
            self.opt,
            match &self.constraint {
                None => LockConstraint::Unconstrained,
                Some(_) => self.constraint(),
            },
        )
    }

    pub fn constraint(&self) -> LockConstraint {
        // Safe to unwrap as the data can only end up in the struct as a valid constraint
        unsafe { LockConstraint::try_from(&self.constraint).unwrap_unchecked() }
    }

    pub fn name(&self) -> &PackageName {
        &self.name
    }

    pub fn version(&self) -> &PackageVersion {
        &self.version
    }

    pub fn pinned(&self) -> PinnedState {
        self.pinned
    }

    pub fn opt(&self) -> OptState {
        self.opt
    }

    pub fn dependencies(&self) -> Vec<&LockedPackageId> {
        self.dependencies.iter().collect()
    }

    pub fn build_dependencies(&self) -> Vec<&LockedPackageId> {
        self.build_dependencies.iter().collect()
    }

    pub fn to_package(&self) -> PackageSpec {
        PackageSpec::new(self.name.clone(), self.version.clone())
    }

    pub fn into_package_req(self) -> PackageReq {
        match self.constraint() {
            LockConstraint::Unconstrained => self.name.into(),
            LockConstraint::Constrained(version_req) => PackageReq {
                name: self.name,
                version_req,
            },
        }
    }

    pub(crate) fn as_package_req(&self) -> PackageReq {
        match self.constraint() {
            LockConstraint::Unconstrained => self.name.clone().into(),
            LockConstraint::Constrained(version_req) => PackageReq {
                name: self.name.clone(),
                version_req: version_req.clone(),
            },
        }
    }
}

#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord, Hash, Serialize, Deserialize)]
#[serde(rename_all = "lowercase", tag = "type")]
pub(crate) enum RemotePackageSourceUrl {
    Git {
        url: String,
        #[serde(rename = "ref")]
        checkout_ref: String,
        /// Whether the source repository contains git submodules that must be
        /// fetched for the package to build.
        #[serde(default, skip_serializing_if = "std::ops::Not::not")]
        submodules: bool,
    }, // GitUrl doesn't have all the trait instances we need
    Url {
        #[serde(deserialize_with = "deserialize_url", serialize_with = "serialize_url")]
        url: Url,
    },
    File {
        path: PathBuf,
    },
}

impl Display for RemotePackageSourceUrl {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            RemotePackageSourceUrl::Git {
                url, checkout_ref, ..
            } => format!("{url}@{checkout_ref}").fmt(f),
            RemotePackageSourceUrl::Url { url } => url.fmt(f),
            RemotePackageSourceUrl::File { path } => format!("{}", path.display()).fmt(f),
        }
    }
}

// TODO(vhyrro): Move to `package/local.rs`

/// A package whose source has been downloaded, hashed and which is installed in a tree.
#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord, Hash)]
pub struct LockedPackage {
    pub(crate) spec: LockedPackageSpec,
    pub(crate) source: RemotePackageSource,
    pub(crate) source_url: Option<RemotePackageSourceUrl>,
    pub(crate) hashes: LockedPackageHashes,
}

impl LockedPackage {
    pub(crate) fn repin(mut self, pinned: PinnedState) -> Self {
        self.spec.pinned = pinned;
        self
    }

    #[cfg(test)]
    pub(crate) fn from(
        package: &PackageSpec,
        constraint: LockConstraint,
        source: RemotePackageSource,
        source_url: Option<RemotePackageSourceUrl>,
        hashes: LockedPackageHashes,
    ) -> Self {
        Self::new(
            LockedPackageSpec::new(
                package.name(),
                package.version(),
                constraint,
                Vec::default(),
                Vec::default(),
                &PinnedState::Unpinned,
                &OptState::Required,
            ),
            source,
            source_url,
            hashes,
        )
    }

    pub(crate) fn new(
        spec: LockedPackageSpec,
        source: RemotePackageSource,
        source_url: Option<RemotePackageSourceUrl>,
        hashes: LockedPackageHashes,
    ) -> Self {
        Self {
            spec,
            source,
            source_url,
            hashes,
        }
    }

    pub fn spec(&self) -> &LockedPackageSpec {
        &self.spec
    }

    pub fn into_package_spec(self) -> PackageSpec {
        self.spec.to_package()
    }

    pub fn as_package_spec(&self) -> PackageSpec {
        self.spec.to_package()
    }

    pub fn id(&self) -> LockedPackageId {
        self.spec.id()
    }

    pub fn name(&self) -> &PackageName {
        self.spec.name()
    }

    pub fn version(&self) -> &PackageVersion {
        self.spec.version()
    }

    pub fn pinned(&self) -> PinnedState {
        self.spec.pinned()
    }

    pub fn opt(&self) -> OptState {
        self.spec.opt()
    }

    pub fn dependencies(&self) -> Vec<&LockedPackageId> {
        self.spec.dependencies()
    }

    pub fn build_dependencies(&self) -> Vec<&LockedPackageId> {
        self.spec.build_dependencies()
    }

    pub fn constraint(&self) -> LockConstraint {
        self.spec.constraint()
    }

    pub(crate) fn source(&self) -> &RemotePackageSource {
        &self.source
    }

    pub(crate) fn source_url(&self) -> Option<&RemotePackageSourceUrl> {
        self.source_url.as_ref()
    }

    pub fn hashes(&self) -> &LockedPackageHashes {
        &self.hashes
    }

    pub fn to_package(&self) -> PackageSpec {
        self.spec.to_package()
    }

    pub fn into_package_req(self) -> PackageReq {
        self.spec.into_package_req()
    }
}

#[derive(Debug, Serialize, Deserialize, Clone)]
struct LockedPackageIntermediate {
    name: PackageName,
    version: PackageVersion,
    #[serde(default, skip_serializing_if = "PinnedState::is_default")]
    pinned: PinnedState,
    #[serde(default, skip_serializing_if = "OptState::is_default")]
    opt: OptState,
    #[serde(default, skip_serializing_if = "Vec::is_empty")]
    dependencies: Vec<LockedPackageId>,
    #[serde(default, skip_serializing_if = "Vec::is_empty")]
    build_dependencies: Vec<LockedPackageId>,
    #[serde(default, skip_serializing_if = "Option::is_none")]
    constraint: Option<String>,
    source: RemotePackageSource,
    #[serde(default, skip_serializing_if = "Option::is_none")]
    source_url: Option<RemotePackageSourceUrl>,
    hashes: LockedPackageHashes,
}

impl TryFrom<LockedPackageIntermediate> for LockedPackage {
    type Error = LockConstraintParseError;

    fn try_from(value: LockedPackageIntermediate) -> Result<Self, Self::Error> {
        let constraint = LockConstraint::try_from(&value.constraint)?;
        Ok(Self {
            spec: LockedPackageSpec::new(
                &value.name,
                &value.version,
                constraint,
                value.dependencies,
                value.build_dependencies,
                &value.pinned,
                &value.opt,
            ),
            source: value.source,
            source_url: value.source_url,
            hashes: value.hashes,
        })
    }
}

impl From<&LockedPackage> for LockedPackageIntermediate {
    fn from(value: &LockedPackage) -> Self {
        Self {
            name: value.spec.name.clone(),
            version: value.spec.version.clone(),
            pinned: value.spec.pinned,
            opt: value.spec.opt,
            dependencies: value.spec.dependencies.clone(),
            build_dependencies: value.spec.build_dependencies.clone(),
            constraint: value.spec.constraint.clone(),
            source: value.source.clone(),
            source_url: value.source_url.clone(),
            hashes: value.hashes.clone(),
        }
    }
}

impl<'de> Deserialize<'de> for LockedPackage {
    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
    where
        D: serde::Deserializer<'de>,
    {
        LockedPackage::try_from(LockedPackageIntermediate::deserialize(deserializer)?)
            .map_err(de::Error::custom)
    }
}

impl Serialize for LockedPackage {
    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
    where
        S: serde::Serializer,
    {
        LockedPackageIntermediate::from(self).serialize(serializer)
    }
}

#[derive(Debug, PartialEq, Eq, Clone, Serialize, Deserialize, Hash)]
pub struct LockedPackageHashes {
    pub rockspec: Integrity,
    pub source: Integrity,
}

impl Ord for LockedPackageHashes {
    fn cmp(&self, other: &Self) -> std::cmp::Ordering {
        let a = (self.rockspec.to_hex().1, self.source.to_hex().1);
        let b = (other.rockspec.to_hex().1, other.source.to_hex().1);
        a.cmp(&b)
    }
}

impl PartialOrd for LockedPackageHashes {
    fn partial_cmp(&self, other: &Self) -> Option<std::cmp::Ordering> {
        Some(self.cmp(other))
    }
}

#[derive(Clone, Debug, PartialEq, Eq, Hash, Default)]
pub enum LockConstraint {
    #[default]
    Unconstrained,
    Constrained(PackageVersionReq),
}

impl<'de> Deserialize<'de> for LockConstraint {
    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
    where
        D: serde::Deserializer<'de>,
    {
        let str = String::deserialize(deserializer)?;
        match str.as_str() {
            "*" => Ok(LockConstraint::Unconstrained),
            _ => Ok(LockConstraint::Constrained(
                str.parse().map_err(serde::de::Error::custom)?,
            )),
        }
    }
}

impl LockConstraint {
    pub fn to_string_opt(&self) -> Option<String> {
        match self {
            LockConstraint::Unconstrained => None,
            LockConstraint::Constrained(req) => Some(req.to_string()),
        }
    }

    fn matches_version_req(&self, req: &PackageVersionReq) -> bool {
        match self {
            LockConstraint::Unconstrained => req.is_any(),
            LockConstraint::Constrained(package_version_req) => package_version_req == req,
        }
    }
}

impl From<PackageVersionReq> for LockConstraint {
    fn from(value: PackageVersionReq) -> Self {
        if value.is_any() {
            Self::Unconstrained
        } else {
            Self::Constrained(value)
        }
    }
}

#[derive(Error, Debug, Diagnostic)]
#[non_exhaustive]
pub enum LockConstraintParseError {
    #[error("Invalid constraint in LuaPackage")]
    #[diagnostic(forward(0))]
    LockConstraintParseError(#[from] PackageVersionReqError),
}

impl TryFrom<&Option<String>> for LockConstraint {
    type Error = LockConstraintParseError;

    fn try_from(constraint: &Option<String>) -> Result<Self, Self::Error> {
        match constraint {
            Some(constraint) => {
                let package_version_req = constraint.parse()?;
                Ok(LockConstraint::Constrained(package_version_req))
            }
            None => Ok(LockConstraint::Unconstrained),
        }
    }
}

pub trait LockfilePermissions {}
#[derive(Clone)]
pub struct ReadOnly;
#[derive(Clone)]
pub struct ReadWrite;

impl LockfilePermissions for ReadOnly {}
impl LockfilePermissions for ReadWrite {}

#[derive(Clone, Debug, Serialize, Deserialize, Default)]
pub(crate) struct PackageLock {
    // NOTE: We cannot directly serialize to a `Sha256` object as they don't implement serde traits.
    // NOTE: We want to retain ordering of rocks when de/serializing.
    rocks: BTreeMap<LockedPackageId, LockedPackage>,

    /// The entrypoints of the tree. Enforces one entrypoint per package name.
    entrypoints: BTreeMap<PackageName, LockedPackageId>,
}

impl PackageLock {
    pub(crate) fn from_packages(
        packages: impl Iterator<Item = (LockedPackage, EntryType)>,
    ) -> Self {
        let mut lock = Self::default();

        for (package, entry_type) in packages {
            if matches!(entry_type, EntryType::Entrypoint) {
                lock.entrypoints
                    .insert(package.name().clone(), package.id());
            }
            lock.rocks.insert(package.id(), package);
        }

        lock
    }

    pub(crate) fn merge(&mut self, other: &PackageLock) {
        self.entrypoints.extend(
            other
                .entrypoints
                .iter()
                .map(|(name, id)| (name.clone(), id.clone())),
        );
        self.rocks.extend(
            other
                .rocks
                .iter()
                .map(|(id, package)| (id.clone(), package.clone())),
        );
    }

    fn get(&self, id: &LockedPackageId) -> Option<&LockedPackage> {
        self.rocks.get(id)
    }

    fn is_empty(&self) -> bool {
        self.rocks.is_empty()
    }

    pub(crate) fn rocks(&self) -> &BTreeMap<LockedPackageId, LockedPackage> {
        &self.rocks
    }

    fn is_entrypoint(&self, package: &LockedPackageId) -> bool {
        self.entrypoints.values().any(|id| id == package)
    }

    /// Returns the entrypoint with the given package name, if any.
    pub(crate) fn entrypoint(&self, name: &PackageName) -> Option<&LockedPackage> {
        self.entrypoints.get(name).and_then(|id| self.get(id))
    }

    fn is_dependency(&self, package: &LockedPackageId) -> bool {
        self.rocks
            .values()
            .flat_map(|rock| rock.dependencies())
            .any(|dep_id| dep_id == package)
    }

    fn list(&self) -> HashMap<PackageName, Vec<LockedPackage>> {
        self.rocks()
            .values()
            .cloned()
            .map(|locked_rock| (locked_rock.name().clone(), locked_rock))
            .into_group_map()
    }

    fn remove(&mut self, target: &LockedPackage) {
        self.remove_by_id(&target.id())
    }

    fn remove_by_id(&mut self, target: &LockedPackageId) {
        self.rocks.remove(target);
        self.entrypoints.retain(|_, id| id != target);
    }

    pub(crate) fn has_rock(
        &self,
        req: &PackageReq,
        filter: Option<RemotePackageTypeFilterSpec>,
    ) -> Option<LockedPackage> {
        self.list()
            .get(req.name())
            .map(|packages| {
                packages
                    .iter()
                    .filter(|package| match &filter {
                        Some(filter_spec) => match &package.source {
                            RemotePackageSource::LuarocksRockspec(_) => filter_spec.rockspec,
                            RemotePackageSource::LuarocksSrcRock(_) => filter_spec.src,
                            RemotePackageSource::LuarocksBinaryRock(_) => filter_spec.binary,
                            RemotePackageSource::Wally(_) => true,
                            RemotePackageSource::RockspecContent(_) => true,
                            RemotePackageSource::Local => true,
                            #[cfg(test)]
                            RemotePackageSource::Test => unimplemented!(),
                        },
                        None => true,
                    })
                    .rev()
                    .find(|package| req.version_req().matches(package.version()))
            })?
            .cloned()
    }

    fn has_rock_with_equal_constraint(&self, req: &LuaDependencySpec) -> Option<LockedPackage> {
        self.list()
            .get(req.name())
            .map(|packages| {
                packages
                    .iter()
                    .rev()
                    .find(|package| package.constraint().matches_version_req(req.version_req()))
            })?
            .cloned()
    }

    /// Synchronise a list of packages with this lock,
    /// producing a report of packages to add and packages to remove,
    /// based on the version constraint and the given [`SyncStrategy`].
    ///
    /// NOTE: The reason we produce a report and don't add/remove packages
    /// here is because packages need to be installed in order to be added.
    pub(crate) fn package_sync_spec(
        &self,
        packages: &[LuaDependencySpec],
        strategy: &SyncStrategy<'_>,
    ) -> PackageSyncSpec {
        let pkg_dir_exists = |pkg: &LockedPackage| match strategy {
            SyncStrategy::LockfileOnly => true,
            SyncStrategy::EnsureInstalled(tree) => tree.layout_for(pkg.spec()).root.is_dir(),
        };

        let entrypoints_to_keep: HashSet<LockedPackage> = self
            .entrypoints
            .values()
            .filter_map(|id| self.get(id))
            .filter(|local_pkg| {
                packages.iter().any(|req| {
                    local_pkg
                        .constraint()
                        .matches_version_req(req.version_req())
                })
            })
            .cloned()
            .collect();

        let packages_to_keep: HashSet<&LockedPackage> = entrypoints_to_keep
            .iter()
            .flat_map(|local_pkg| self.get_all_dependencies(&local_pkg.id()))
            .collect();

        let to_add = packages
            .iter()
            .filter(|pkg| {
                self.has_rock_with_equal_constraint(pkg)
                    .map(|local_pkg| !pkg_dir_exists(&local_pkg))
                    .unwrap_or(true)
            })
            .cloned()
            .collect_vec();

        let to_remove = self
            .rocks()
            .values()
            .filter(|pkg| !packages_to_keep.contains(*pkg))
            .cloned()
            .collect_vec();

        PackageSyncSpec { to_add, to_remove }
    }

    /// Return all dependencies of a package, including itself
    fn get_all_dependencies(&self, id: &LockedPackageId) -> HashSet<&LockedPackage> {
        let mut packages = HashSet::new();
        if let Some(local_pkg) = self.get(id) {
            packages.insert(local_pkg);
            packages.extend(
                local_pkg
                    .dependencies()
                    .iter()
                    .flat_map(|id| self.get_all_dependencies(id)),
            );
        }
        packages
    }

    /// The set of all packages reachable from the entrypoints via the
    /// dependency graph.
    fn reachable(&self) -> HashSet<&LockedPackage> {
        self.entrypoints
            .values()
            .flat_map(|id| self.get_all_dependencies(id))
            .collect()
    }
}

/// A lockfile for an install tree
#[derive(Clone, Debug, Serialize, Deserialize)]
pub struct Lockfile<P: LockfilePermissions> {
    #[serde(skip)]
    filepath: PathBuf,
    #[serde(skip)]
    _marker: PhantomData<P>,
    // TODO: Serialize this directly into a `Version`
    version: String,
    #[serde(default, skip_serializing_if = "BTreeMap::is_empty")]
    binaries: BTreeMap<LockedPackageId, RockBinaries>,
    #[serde(flatten)]
    lock: PackageLock,
}

#[derive(EnumIter, Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub enum LockedPackageLockType {
    Regular,
    Test,
    Build,
}

/// A lockfile for a Lua project
#[derive(Clone, Debug, Serialize, Deserialize)]
pub struct WorkspaceLockfile<P: LockfilePermissions> {
    #[serde(skip)]
    filepath: PathBuf,
    #[serde(skip)]
    _marker: PhantomData<P>,
    version: String,
    #[serde(default, skip_serializing_if = "PackageLock::is_empty")]
    dependencies: PackageLock,
    #[serde(default, skip_serializing_if = "PackageLock::is_empty")]
    test_dependencies: PackageLock,
    #[serde(default, skip_serializing_if = "PackageLock::is_empty")]
    build_dependencies: PackageLock,
}

#[derive(Error, Debug, Diagnostic)]
#[non_exhaustive]
pub enum Error {
    #[error(transparent)]
    #[diagnostic(transparent)]
    Fs(#[from] fs::Error),
    #[error("error parsing lockfile from JSON")]
    ParseJson(#[source] serde_json::Error),
    #[error("error writing lockfile to JSON")]
    WriteJson(#[source] serde_json::Error),
    #[error("invalid version string in lockfile: {version}")]
    #[diagnostic(help("did you or a tool modify the lockfile manually?"))]
    InvalidVersion {
        version: String,
        source: semver::Error,
    },
    #[error("lockfile '{}' has no `version` field", lockfile.display())]
    #[diagnostic(
        code(lux_lib::lockfile::missing_version),
        help("the lockfile is corrupt or was hand-edited")
    )]
    MissingVersion { lockfile: PathBuf },
    #[error(
        "incompatible lockfile version in '{}': expected {expected}, found {found}",
        lockfile.display()
    )]
    #[diagnostic(code(lux_lib::lockfile::incompatible_version), help("{help}"))]
    IncompatibleVersion {
        lockfile: PathBuf,
        found: String,
        expected: String,
        help: String,
    },
    #[error("cannot migrate lockfile '{}'", lockfile.display())]
    #[diagnostic(code(lux_lib::lockfile::conflicting_entrypoints), help("{help}"))]
    ConflictingEntrypoints { lockfile: PathBuf, help: String },
}

#[derive(Error, Debug, Diagnostic)]
#[non_exhaustive]
pub enum LockfileIntegrityError {
    #[error(
        r#"rockspec integrity mismatch.
- source: {src}
- expected: {expected}
- got: {got}
"#
    )]
    #[diagnostic(help(
        r#"the maintainer may have force-uploaded a different rockspec.
check the rockspec source, then rerun the command with `--no-lock` to update the hash.
"#
    ))]
    RockspecIntegrityMismatch {
        src: String,
        expected: Integrity,
        got: Integrity,
    },
    #[error(
        r#"source integrity mismatch.
- source: {src}
- expected: {expected}
- got: {got}"#
    )]
    #[diagnostic(help(
        r#"the maintainer may have force-uploaded a different source or moved a tag.
check the source URL, then rerun the command with `lx --no-lock` to update the hash.
"#
    ))]
    SourceIntegrityMismatch {
        src: String,
        expected: Integrity,
        got: Integrity,
    },
    #[error(
        "package {0} version {1} with pinned state {2} and constraint {3} not found in the lockfile."
    )]
    PackageNotFound(PackageName, Box<PackageVersion>, PinnedState, String),
}

#[derive(Error, Debug, Diagnostic)]
#[non_exhaustive]
#[error("error flushing the lockfile ({})", .filepath.display())]
#[diagnostic(help("make sure the parent directory is writeable"))]
pub struct FlushLockfileError {
    filepath: PathBuf,
    source: io::Error,
}

/// A specification for syncing a list of packages with a lockfile
#[derive(Debug, Default)]
pub(crate) struct PackageSyncSpec {
    pub to_add: Vec<LuaDependencySpec>,
    pub to_remove: Vec<LockedPackage>,
}

/// Controls how `package_sync_spec` determines whether a package exists.
pub(crate) enum SyncStrategy<'a> {
    /// Only check the lockfile for constraint matches.
    LockfileOnly,
    /// In addition to checking lockfile constraints, verify that each
    /// package's installation directory exists in the given tree.
    EnsureInstalled(&'a Tree),
}

impl<P: LockfilePermissions> Lockfile<P> {
    pub fn version(&self) -> &String {
        &self.version
    }

    pub fn rocks(&self) -> &BTreeMap<LockedPackageId, LockedPackage> {
        self.lock.rocks()
    }

    pub fn is_dependency(&self, package: &LockedPackageId) -> bool {
        self.lock.is_dependency(package)
    }

    pub fn is_entrypoint(&self, package: &LockedPackageId) -> bool {
        self.lock.is_entrypoint(package)
    }

    /// Returns the entrypoint with the given package name, if any.
    pub fn entrypoint(&self, name: &PackageName) -> Option<&LockedPackage> {
        self.lock.entrypoint(name)
    }

    pub fn entrypoints(&self) -> Vec<(&PackageName, &LockedPackage)> {
        self.lock
            .entrypoints
            .iter()
            .filter_map(|(name, id)| self.lock.get(id).map(|package| (name, package)))
            .collect()
    }

    /// Returns all rocks in the lockfile that are reachable from an entrypoint
    /// via the dependency graph.
    pub fn reachable_rocks(&self) -> Vec<LockedPackage> {
        self.lock.reachable().into_iter().cloned().collect()
    }

    pub fn entry_type(&self, package: &LockedPackageId) -> EntryType {
        if self.lock.is_entrypoint(package) {
            EntryType::Entrypoint
        } else {
            EntryType::DependencyOnly
        }
    }

    /// The binaries installed in the tree for a given package.
    pub(crate) fn binaries(&self, id: &LockedPackageId) -> Option<&RockBinaries> {
        self.binaries.get(id)
    }

    pub(crate) fn local_pkg_lock(&self) -> &PackageLock {
        &self.lock
    }

    pub fn get(&self, id: &LockedPackageId) -> Option<&LockedPackage> {
        self.lock.get(id)
    }

    /// Unsafe because this assumes a prior check if the package is present
    ///
    /// # Safety
    ///
    /// Ensure that the package is present in the lockfile before calling this function.
    pub unsafe fn get_unchecked(&self, id: &LockedPackageId) -> &LockedPackage {
        self.lock.get(id).unwrap_unchecked()
    }

    pub(crate) fn list(&self) -> HashMap<PackageName, Vec<LockedPackage>> {
        self.lock.list()
    }

    pub(crate) fn has_rock(
        &self,
        req: &PackageReq,
        filter: Option<RemotePackageTypeFilterSpec>,
    ) -> Option<LockedPackage> {
        self.lock.has_rock(req, filter)
    }

    /// Find all rocks that match the requirement
    pub(crate) fn find_rocks(&self, req: &PackageReq) -> Vec<LockedPackageId> {
        match self.list().get(req.name()) {
            Some(packages) => packages
                .iter()
                .rev()
                .filter(|package| req.version_req().matches(package.version()))
                .map(|package| package.id())
                .collect_vec(),
            None => Vec::default(),
        }
    }

    /// Validate the integrity of an installed package with the entry in this lockfile.
    pub(crate) fn validate_integrity(
        &self,
        expected_package: &LockedPackage,
    ) -> Result<(), LockfileIntegrityError> {
        // NOTE: We can't query by ID, because when installing from a lockfile (e.g. during sync),
        // the constraint is always `==`.
        match self.list().get(expected_package.name()) {
            None => Err(integrity_err_not_found(expected_package)),
            Some(rocks) => match rocks
                .iter()
                .find(|rock| rock.version() == expected_package.version())
            {
                None => Err(integrity_err_not_found(expected_package)),
                Some(installed_package) => {
                    if expected_package
                        .hashes
                        .rockspec
                        .matches(&installed_package.hashes.rockspec)
                        .is_none()
                    {
                        return Err(LockfileIntegrityError::RockspecIntegrityMismatch {
                            src: expected_package.source.to_string(),
                            expected: expected_package.hashes.rockspec.clone(),
                            got: installed_package.hashes.rockspec.clone(),
                        });
                    }
                    if expected_package
                        .hashes
                        .source
                        .matches(&installed_package.hashes.source)
                        .is_none()
                    {
                        return Err(LockfileIntegrityError::SourceIntegrityMismatch {
                            src: expected_package
                                .source_url
                                .as_ref()
                                .map(|url| url.to_string())
                                .unwrap_or(expected_package.source.to_string()),
                            expected: expected_package.hashes.source.clone(),
                            got: installed_package.hashes.source.clone(),
                        });
                    }
                    Ok(())
                }
            },
        }
    }

    fn flush(&self) -> Result<(), FlushLockfileError> {
        let content = serde_json::to_string_pretty(&self).map_err(|err| FlushLockfileError {
            filepath: self.filepath.to_path_buf(),
            source: io::Error::other(err),
        })?;

        fs::sync::write_atomic(&self.filepath, content).map_err(|err| FlushLockfileError {
            filepath: self.filepath.to_path_buf(),
            source: io::Error::other(err),
        })
    }
}

impl<P: LockfilePermissions> WorkspaceLockfile<P> {
    pub(crate) fn rocks(
        &self,
        deps: &LockedPackageLockType,
    ) -> &BTreeMap<LockedPackageId, LockedPackage> {
        match deps {
            LockedPackageLockType::Regular => self.dependencies.rocks(),
            LockedPackageLockType::Test => self.test_dependencies.rocks(),
            LockedPackageLockType::Build => self.build_dependencies.rocks(),
        }
    }

    pub(crate) fn get(
        &self,
        id: &LockedPackageId,
        deps: &LockedPackageLockType,
    ) -> Option<&LockedPackage> {
        match deps {
            LockedPackageLockType::Regular => self.dependencies.get(id),
            LockedPackageLockType::Test => self.test_dependencies.get(id),
            LockedPackageLockType::Build => self.build_dependencies.get(id),
        }
    }

    pub(crate) fn is_entrypoint(
        &self,
        package: &LockedPackageId,
        deps: &LockedPackageLockType,
    ) -> bool {
        match deps {
            LockedPackageLockType::Regular => self.dependencies.is_entrypoint(package),
            LockedPackageLockType::Test => self.test_dependencies.is_entrypoint(package),
            LockedPackageLockType::Build => self.build_dependencies.is_entrypoint(package),
        }
    }

    pub(crate) fn package_sync_spec(
        &self,
        packages: &[LuaDependencySpec],
        deps: &LockedPackageLockType,
        strategy: &SyncStrategy<'_>,
    ) -> PackageSyncSpec {
        match deps {
            LockedPackageLockType::Regular => {
                self.dependencies.package_sync_spec(packages, strategy)
            }
            LockedPackageLockType::Test => {
                self.test_dependencies.package_sync_spec(packages, strategy)
            }
            LockedPackageLockType::Build => self
                .build_dependencies
                .package_sync_spec(packages, strategy),
        }
    }

    pub(crate) fn local_pkg_lock(&self, deps: &LockedPackageLockType) -> &PackageLock {
        match deps {
            LockedPackageLockType::Regular => &self.dependencies,
            LockedPackageLockType::Test => &self.test_dependencies,
            LockedPackageLockType::Build => &self.build_dependencies,
        }
    }

    pub(crate) fn local_pkg_locks(&self) -> Vec<PackageLock> {
        vec![
            self.dependencies.clone(),
            self.test_dependencies.clone(),
            self.build_dependencies.clone(),
        ]
    }

    fn flush(&self) -> io::Result<()> {
        let content = serde_json::to_string_pretty(&self)?;

        fs::sync::write_atomic(&self.filepath, content).map_err(io::Error::other)?;

        Ok(())
    }
}

impl Lockfile<ReadOnly> {
    /// Create a new `Lockfile`, writing an empty file if none exists.
    #[tracing::instrument(level = "trace")]
    pub(crate) fn new(filepath: PathBuf) -> Result<Lockfile<ReadOnly>, Error> {
        // Ensure that the lockfile exists
        match File::options().create_new(true).write(true).open(&filepath) {
            Ok(mut file) => {
                let empty_lockfile: Lockfile<ReadOnly> = Lockfile {
                    filepath: filepath.clone(),
                    _marker: PhantomData,
                    version: LOCKFILE_VERSION_STR.into(),
                    binaries: BTreeMap::new(),
                    lock: PackageLock::default(),
                };
                let json_str = serde_json::to_string(&empty_lockfile).map_err(Error::WriteJson)?;
                write!(file, "{json_str}").map_err(|source| fs::Error::Write {
                    path: filepath.to_path_buf(),
                    source,
                })?;
            }
            Err(err) if err.kind() == ErrorKind::AlreadyExists => {}
            Err(source) => {
                return Err(Error::Fs(fs::Error::FileOpen {
                    path: filepath.to_path_buf(),
                    source,
                }));
            }
        }

        Self::load(filepath)
    }

    /// Load a `Lockfile`, failing if none exists.
    #[tracing::instrument(level = "trace")]
    pub fn load(filepath: PathBuf) -> Result<Lockfile<ReadOnly>, Error> {
        let content = fs::sync::read_to_string(&filepath)?;
        let tree_dir = filepath.parent().unwrap_or(filepath.as_path());
        let help = format!(
            "this install tree was created by an incompatible version of Lux.
remove the tree at '{}' and reinstall all packages it contained.",
            tree_dir.display()
        );
        let mut lockfile: Lockfile<ReadOnly> = load_lockfile(&content, &filepath, &help)?;
        lockfile.filepath = filepath;
        Ok(lockfile)
    }

    /// Creates a temporary, writeable lockfile which can never flush.
    pub(crate) fn into_temporary(self) -> Lockfile<ReadWrite> {
        Lockfile::<ReadWrite> {
            _marker: PhantomData,
            filepath: self.filepath,
            version: self.version,
            binaries: self.binaries,
            lock: self.lock,
        }
    }

    /// Creates a lockfile guard, flushing the lockfile automatically
    /// once the guard goes out of scope.
    pub fn write_guard(self) -> LockfileGuard {
        LockfileGuard(self.into_temporary())
    }

    /// Converts the current lockfile into a writeable one, executes `cb` and flushes
    /// the lockfile.
    #[tracing::instrument(level = "trace", skip_all)]
    pub fn map_then_flush<T, F, E>(self, cb: F) -> Result<T, FlushLockfileError>
    where
        F: FnOnce(&mut Lockfile<ReadWrite>) -> Result<T, E>,
        E: std::error::Error,
        E: From<io::Error>,
        E: Into<Box<dyn std::error::Error + Send + Sync>>,
    {
        let mut writeable_lockfile = self.into_temporary();

        let result = cb(&mut writeable_lockfile).map_err(|err| FlushLockfileError {
            filepath: writeable_lockfile.filepath.to_path_buf(),
            source: io::Error::other(err),
        })?;

        writeable_lockfile.flush()?;

        Ok(result)
    }

    // TODO: Add this once async closures are stabilized
    // Converts the current lockfile into a writeable one, executes `cb` asynchronously and flushes
    // the lockfile.
    //pub async fn map_then_flush_async<T, F, E, Fut>(self, cb: F) -> Result<T, E>
    //where
    //    F: AsyncFnOnce(&mut Lockfile<ReadWrite>) -> Result<T, E>,
    //    E: Error,
    //    E: From<io::Error>,
    //{
    //    let mut writeable_lockfile = self.into_temporary();
    //
    //    let result = cb(&mut writeable_lockfile).await?;
    //
    //    writeable_lockfile.flush()?;
    //
    //    Ok(result)
    //}
}

impl WorkspaceLockfile<ReadOnly> {
    /// Create a new `ProjectLockfile`, writing an empty file if none exists.
    #[tracing::instrument(level = "trace")]
    pub fn new(filepath: PathBuf) -> Result<WorkspaceLockfile<ReadOnly>, Error> {
        // Ensure that the lockfile exists
        match File::options().create_new(true).write(true).open(&filepath) {
            Ok(mut file) => {
                let empty_lockfile: WorkspaceLockfile<ReadOnly> = WorkspaceLockfile {
                    filepath: filepath.clone(),
                    _marker: PhantomData,
                    version: LOCKFILE_VERSION_STR.into(),
                    dependencies: PackageLock::default(),
                    test_dependencies: PackageLock::default(),
                    build_dependencies: PackageLock::default(),
                };
                let json_str = serde_json::to_string(&empty_lockfile).map_err(Error::WriteJson)?;
                write!(file, "{json_str}").map_err(|source| fs::Error::Write {
                    path: filepath.to_path_buf(),
                    source,
                })?;
            }
            Err(err) if err.kind() == ErrorKind::AlreadyExists => {}
            Err(source) => {
                return Err(Error::Fs(fs::Error::FileOpen {
                    path: filepath.to_path_buf(),
                    source,
                }));
            }
        }

        Self::load(filepath)
    }

    /// Load a `ProjectLockfile`, failing if none exists.
    #[tracing::instrument(level = "trace")]
    pub fn load(filepath: PathBuf) -> Result<WorkspaceLockfile<ReadOnly>, Error> {
        let content = fs::sync::read_to_string(&filepath)?;
        let mut lockfile: WorkspaceLockfile<ReadOnly> = load_lockfile(
            &content,
            &filepath,
            "this lockfile was created by an incompatible version of Lux.
remove the `lux.lock` and run `lx sync` to regenerate it.",
        )?;

        lockfile.filepath = filepath;

        Ok(lockfile)
    }

    /// Creates a temporary, writeable project lockfile which can never flush.
    fn into_temporary(self) -> WorkspaceLockfile<ReadWrite> {
        WorkspaceLockfile::<ReadWrite> {
            _marker: PhantomData,
            filepath: self.filepath,
            version: self.version,
            dependencies: self.dependencies,
            test_dependencies: self.test_dependencies,
            build_dependencies: self.build_dependencies,
        }
    }

    /// Creates a project lockfile guard, flushing the lockfile automatically
    /// once the guard goes out of scope.
    pub fn write_guard(self) -> ProjectLockfileGuard {
        ProjectLockfileGuard(self.into_temporary())
    }
}

impl Lockfile<ReadWrite> {
    pub(crate) fn add_entrypoint(&mut self, rock: &LockedPackage) {
        self.add(rock);
        self.lock.entrypoints.insert(rock.name().clone(), rock.id());
    }

    pub(crate) fn add(&mut self, rock: &LockedPackage) {
        // Since rocks entries are mutable, we only add the dependency if it
        // has not already been added.
        self.lock
            .rocks
            .entry(rock.id())
            .or_insert_with(|| rock.clone());
    }

    /// Add a dependency for a package.
    pub(crate) fn add_dependency(&mut self, target: &LockedPackage, dependency: &LockedPackage) {
        self.lock
            .rocks
            .entry(target.id())
            .and_modify(|rock| {
                let id = dependency.id();
                if !rock.spec.dependencies.contains(&id) {
                    rock.spec.dependencies.push(id);
                }
            })
            .or_insert_with(|| {
                let id = dependency.id();
                let mut target = target.clone();
                if !target.spec.dependencies.contains(&id) {
                    target.spec.dependencies.push(id);
                }
                target
            });
        self.add(dependency);
    }

    /// Add a build dependency for a package.
    pub(crate) fn add_build_dependency(
        &mut self,
        target: &LockedPackage,
        dependency: &LockedPackage,
    ) {
        self.lock
            .rocks
            .entry(target.id())
            .and_modify(|rock| {
                let id = dependency.id();
                if !rock.spec.build_dependencies.contains(&id) {
                    rock.spec.build_dependencies.push(id);
                }
            })
            .or_insert_with(|| {
                let id = dependency.id();
                let mut target = target.clone();
                if !target.spec.build_dependencies.contains(&id) {
                    target.spec.build_dependencies.push(id);
                }
                target
            });
    }

    /// Record the binaries installed for `package` into the tree lockfile.
    pub(crate) fn set_binaries(&mut self, package: &LockedPackage, binaries: RockBinaries) {
        self.binaries.insert(package.id(), binaries);
    }

    pub(crate) fn remove(&mut self, target: &LockedPackage) {
        self.binaries.remove(&target.id());
        self.lock.remove(target)
    }

    pub(crate) fn remove_by_id(&mut self, target: &LockedPackageId) {
        self.binaries.remove(target);
        self.lock.remove_by_id(target)
    }

    pub(crate) fn sync(&mut self, lock: &PackageLock) {
        self.lock = lock.clone();
    }

    // TODO: `fn entrypoints() -> Vec<LockedRock>`
}

impl WorkspaceLockfile<ReadWrite> {
    pub(crate) fn remove(&mut self, target: &LockedPackage, deps: &LockedPackageLockType) {
        match deps {
            LockedPackageLockType::Regular => self.dependencies.remove(target),
            LockedPackageLockType::Test => self.test_dependencies.remove(target),
            LockedPackageLockType::Build => self.build_dependencies.remove(target),
        }
    }

    pub(crate) fn sync(&mut self, lock: &PackageLock, deps: &LockedPackageLockType) {
        match deps {
            LockedPackageLockType::Regular => {
                self.dependencies = lock.clone();
            }
            LockedPackageLockType::Test => {
                self.test_dependencies = lock.clone();
            }
            LockedPackageLockType::Build => {
                self.build_dependencies = lock.clone();
            }
        }
    }

    pub(crate) fn merge(&mut self, lock: &PackageLock, deps: &LockedPackageLockType) {
        match deps {
            LockedPackageLockType::Regular => self.dependencies.merge(lock),
            LockedPackageLockType::Test => self.test_dependencies.merge(lock),
            LockedPackageLockType::Build => self.build_dependencies.merge(lock),
        }
    }
}

/// Flushes a lockfile automatically when it goes out of scope
pub struct LockfileGuard(Lockfile<ReadWrite>);

pub struct ProjectLockfileGuard(WorkspaceLockfile<ReadWrite>);

impl Serialize for LockfileGuard {
    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
    where
        S: serde::Serializer,
    {
        self.0.serialize(serializer)
    }
}

impl Serialize for ProjectLockfileGuard {
    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
    where
        S: serde::Serializer,
    {
        self.0.serialize(serializer)
    }
}

impl<'de> Deserialize<'de> for LockfileGuard {
    fn deserialize<D>(deserializer: D) -> std::result::Result<Self, D::Error>
    where
        D: serde::Deserializer<'de>,
    {
        Ok(LockfileGuard(Lockfile::<ReadWrite>::deserialize(
            deserializer,
        )?))
    }
}

impl<'de> Deserialize<'de> for ProjectLockfileGuard {
    fn deserialize<D>(deserializer: D) -> std::result::Result<Self, D::Error>
    where
        D: serde::Deserializer<'de>,
    {
        Ok(ProjectLockfileGuard(
            WorkspaceLockfile::<ReadWrite>::deserialize(deserializer)?,
        ))
    }
}

impl Deref for LockfileGuard {
    type Target = Lockfile<ReadWrite>;

    fn deref(&self) -> &Self::Target {
        &self.0
    }
}

impl Deref for ProjectLockfileGuard {
    type Target = WorkspaceLockfile<ReadWrite>;

    fn deref(&self) -> &Self::Target {
        &self.0
    }
}

impl DerefMut for LockfileGuard {
    fn deref_mut(&mut self) -> &mut Self::Target {
        &mut self.0
    }
}

impl DerefMut for ProjectLockfileGuard {
    fn deref_mut(&mut self) -> &mut Self::Target {
        &mut self.0
    }
}

impl Drop for LockfileGuard {
    fn drop(&mut self) {
        let _ = self.flush();
    }
}

impl Drop for ProjectLockfileGuard {
    fn drop(&mut self) {
        let _ = self.flush();
    }
}

fn load_lockfile<T>(content: &str, lockfile: &Path, help: &str) -> Result<T, Error>
where
    T: serde::de::DeserializeOwned,
{
    let mut value = parse_lockfile_json(content)?;
    if lockfile_version(&value, lockfile)?.major == 1 {
        migrate_v1_lockfile(&mut value, lockfile)?;
    }
    deserialize_lockfile(value, lockfile, help)
}

fn parse_lockfile_json(content: &str) -> Result<serde_json::Map<String, serde_json::Value>, Error> {
    serde_json::from_str(content).map_err(Error::ParseJson)
}

fn lockfile_version(
    value: &serde_json::Map<String, serde_json::Value>,
    lockfile: &Path,
) -> Result<semver::Version, Error> {
    let version = value
        .get("version")
        .and_then(|version| version.as_str())
        .ok_or(Error::MissingVersion {
            lockfile: lockfile.to_path_buf(),
        })?;
    semver::Version::parse(version).map_err(|source| Error::InvalidVersion {
        version: version.to_string(),
        source,
    })
}

fn deserialize_lockfile<T>(
    value: serde_json::Map<String, serde_json::Value>,
    lockfile: &Path,
    help: &str,
) -> Result<T, Error>
where
    T: serde::de::DeserializeOwned,
{
    let found = lockfile_version(&value, lockfile)?;
    if found.major != LOCKFILE_VERSION {
        return Err(Error::IncompatibleVersion {
            lockfile: lockfile.to_path_buf(),
            found: found.to_string(),
            expected: LOCKFILE_VERSION_STR.into(),
            help: help.into(),
        });
    }
    serde_json::from_value(serde_json::Value::Object(value)).map_err(Error::ParseJson)
}

fn migrate_v1_lockfile(
    value: &mut serde_json::Map<String, serde_json::Value>,
    lockfile: &Path,
) -> Result<(), Error> {
    let tree_dir = lockfile.parent().unwrap_or(lockfile);
    let found = lockfile_version(value, lockfile)?;
    if value.contains_key("entrypoint_layout") {
        return Err(Error::IncompatibleVersion {
            lockfile: lockfile.to_path_buf(),
            found: found.to_string(),
            expected: LOCKFILE_VERSION_STR.into(),
            help: format!(
                "this install tree uses a custom entrypoint layout that cannot be migrated automatically.
remove the tree at '{}' and reinstall all packages it contained.",
                tree_dir.display()
            ),
        });
    }

    let conflict_help = format!(
        "this lockfile contains entrypoints that cannot be represented in the current format.
remove the tree at '{}' and reinstall all packages it contained.",
        tree_dir.display()
    );

    if value.contains_key("entrypoints") {
        migrate_entrypoints(value, lockfile, &conflict_help)?;
    }
    for section in ["dependencies", "test_dependencies", "build_dependencies"] {
        if let Some(section) = value
            .get_mut(section)
            .and_then(|section| section.as_object_mut())
        {
            migrate_entrypoints(section, lockfile, &conflict_help)?;
        }
    }
    value.insert(
        "version".into(),
        serde_json::Value::String(LOCKFILE_VERSION_STR.into()),
    );
    Ok(())
}

fn migrate_entrypoints(
    section: &mut serde_json::Map<String, serde_json::Value>,
    lockfile: &Path,
    help: &str,
) -> Result<(), Error> {
    let conflicting_entrypoints = || Error::ConflictingEntrypoints {
        lockfile: lockfile.to_path_buf(),
        help: help.to_string(),
    };

    let rocks = section.get("rocks").and_then(|rocks| rocks.as_object());
    let mut entrypoints = serde_json::Map::new();
    if let Some(old_entrypoints) = section.get("entrypoints") {
        for id in old_entrypoints
            .as_array()
            .ok_or_else(conflicting_entrypoints)?
        {
            let id = id.as_str().ok_or_else(conflicting_entrypoints)?;
            let name = rocks
                .and_then(|rocks| rocks.get(id))
                .and_then(|package| package.get("name"))
                .and_then(|name| name.as_str())
                .ok_or_else(conflicting_entrypoints)?;
            if entrypoints.contains_key(name) {
                return Err(conflicting_entrypoints());
            }
            entrypoints.insert(name.to_string(), serde_json::Value::String(id.to_string()));
        }
    }
    section.insert("entrypoints".into(), serde_json::Value::Object(entrypoints));
    Ok(())
}

fn integrity_err_not_found(package: &LockedPackage) -> LockfileIntegrityError {
    LockfileIntegrityError::PackageNotFound(
        package.name().clone(),
        Box::new(package.version().clone()),
        package.spec.pinned,
        package
            .spec
            .constraint
            .clone()
            .unwrap_or("UNCONSTRAINED".into()),
    )
}

fn deserialize_url<'de, D>(deserializer: D) -> Result<Url, D::Error>
where
    D: serde::Deserializer<'de>,
{
    let s = String::deserialize(deserializer)?;
    Url::parse(&s).map_err(serde::de::Error::custom)
}

fn serialize_url<S>(url: &Url, serializer: S) -> Result<S::Ok, S::Error>
where
    S: Serializer,
{
    url.as_str().serialize(serializer)
}

#[cfg(test)]
mod tests {
    use super::*;
    use std::{fs::remove_file, path::PathBuf};

    use assert_fs::fixture::PathCopy;
    use insta::{assert_json_snapshot, sorted_redaction};

    use crate::{config::ConfigBuilder, lua_version::LuaVersion, package::PackageSpec};

    #[test]
    fn parse_lockfile() {
        let temp = assert_fs::TempDir::new().unwrap();
        temp.copy_from(
            PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("resources/test/sample-tree"),
            &["**"],
        )
        .unwrap();

        let config = ConfigBuilder::new()
            .unwrap()
            .user_tree(Some(temp.to_path_buf()))
            .build()
            .unwrap();
        let tree = config.user_tree(LuaVersion::Lua51).unwrap();
        let lockfile = tree.lockfile().unwrap();

        assert_json_snapshot!(lockfile, { ".**" => sorted_redaction() });
    }

    #[test]
    fn add_rocks() {
        let temp = assert_fs::TempDir::new().unwrap();
        temp.copy_from(
            PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("resources/test/sample-tree"),
            &["**"],
        )
        .unwrap();

        let mock_hashes = LockedPackageHashes {
            rockspec: "sha256-uU0nuZNNPgilLlLX2n2r+sSE7+N6U4DukIj3rOLvzek="
                .parse()
                .unwrap(),
            source: "sha256-uU0nuZNNPgilLlLX2n2r+sSE7+N6U4DukIj3rOLvzek="
                .parse()
                .unwrap(),
        };

        let config = ConfigBuilder::new()
            .unwrap()
            .user_tree(Some(temp.to_path_buf()))
            .build()
            .unwrap();
        let tree = config.user_tree(LuaVersion::Lua51).unwrap();
        let mut lockfile = tree.lockfile().unwrap().write_guard();

        let test_package = PackageSpec::parse("test1".to_string(), "0.1.0".to_string()).unwrap();
        let test_local_package = LockedPackage::from(
            &test_package,
            crate::lockfile::LockConstraint::Unconstrained,
            RemotePackageSource::Test,
            None,
            mock_hashes.clone(),
        );
        lockfile.add_entrypoint(&test_local_package);

        let test_dep_package =
            PackageSpec::parse("test2".to_string(), "0.1.0".to_string()).unwrap();
        let mut test_local_dep_package = LockedPackage::from(
            &test_dep_package,
            crate::lockfile::LockConstraint::Constrained(">= 1.0.0".parse().unwrap()),
            RemotePackageSource::Test,
            None,
            mock_hashes.clone(),
        );
        test_local_dep_package.spec.pinned = PinnedState::Pinned;
        lockfile.add_dependency(&test_local_package, &test_local_dep_package);

        assert_json_snapshot!(lockfile, { ".**" => sorted_redaction() });
    }

    #[test]
    fn parse_nonexistent_lockfile() {
        let tree_path =
            PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("resources/test/sample-tree");

        let temp = assert_fs::TempDir::new().unwrap();
        temp.copy_from(&tree_path, &["**"]).unwrap();

        remove_file(temp.join("5.1/lux.lock")).unwrap();

        let config = ConfigBuilder::new()
            .unwrap()
            .user_tree(Some(temp.to_path_buf()))
            .build()
            .unwrap();
        let tree = config.user_tree(LuaVersion::Lua51).unwrap();

        let _ = tree.lockfile().unwrap().write_guard(); // Try to create the lockfile but don't actually do anything with it.
    }

    fn get_test_lockfile() -> Lockfile<ReadOnly> {
        let sample_tree = PathBuf::from(env!("CARGO_MANIFEST_DIR"))
            .join("resources/test/sample-tree/5.1/lux.lock");
        Lockfile::new(sample_tree).unwrap()
    }

    #[test]
    fn test_sync_spec() {
        let lockfile = get_test_lockfile();
        let packages = vec![
            PackageReq::parse("neorg@8.8.1-1").unwrap().into(),
            PackageReq::parse("lua-cjson@2.1.0").unwrap().into(),
            PackageReq::parse("nonexistent").unwrap().into(),
        ];

        let sync_spec = lockfile
            .lock
            .package_sync_spec(&packages, &SyncStrategy::LockfileOnly);

        assert_eq!(sync_spec.to_add.len(), 1);

        // Should keep dependencies of neorg 8.8.1-1
        assert!(!sync_spec
            .to_remove
            .iter()
            .any(|pkg| pkg.name().to_string() == "nvim-nio"
                && pkg.constraint()
                    == LockConstraint::Constrained(">=1.7.0, <1.8.0".parse().unwrap())));
        assert!(!sync_spec
            .to_remove
            .iter()
            .any(|pkg| pkg.name().to_string() == "lua-utils.nvim"
                && pkg.constraint() == LockConstraint::Constrained("=1.0.2".parse().unwrap())));
        assert!(!sync_spec
            .to_remove
            .iter()
            .any(|pkg| pkg.name().to_string() == "plenary.nvim"
                && pkg.constraint() == LockConstraint::Constrained("=0.1.4".parse().unwrap())));
        assert!(!sync_spec
            .to_remove
            .iter()
            .any(|pkg| pkg.name().to_string() == "nui.nvim"
                && pkg.constraint() == LockConstraint::Constrained("=0.3.0".parse().unwrap())));
        assert!(!sync_spec
            .to_remove
            .iter()
            .any(|pkg| pkg.name().to_string() == "pathlib.nvim"
                && pkg.constraint()
                    == LockConstraint::Constrained(">=2.2.0, <2.3.0".parse().unwrap())));
    }

    #[test]
    fn test_sync_spec_remove() {
        let lockfile = get_test_lockfile();
        let packages = vec![
            PackageReq::parse("lua-cjson@2.1.0").unwrap().into(),
            PackageReq::parse("nonexistent").unwrap().into(),
        ];

        let sync_spec = lockfile
            .lock
            .package_sync_spec(&packages, &SyncStrategy::LockfileOnly);

        assert_eq!(sync_spec.to_add.len(), 1);

        // Should remove:
        // - neorg
        // - dependencies unique to neorg
        assert!(sync_spec
            .to_remove
            .iter()
            .any(|pkg| pkg.name().to_string() == "neorg"
                && pkg.version() == &"8.8.1-1".parse().unwrap()));
        assert!(sync_spec
            .to_remove
            .iter()
            .any(|pkg| pkg.name().to_string() == "nvim-nio"
                && pkg.constraint()
                    == LockConstraint::Constrained(">=1.7.0, <1.8.0".parse().unwrap())));
        assert!(sync_spec
            .to_remove
            .iter()
            .any(|pkg| pkg.name().to_string() == "lua-utils.nvim"
                && pkg.constraint() == LockConstraint::Constrained("=1.0.2".parse().unwrap())));
        assert!(sync_spec
            .to_remove
            .iter()
            .any(|pkg| pkg.name().to_string() == "plenary.nvim"
                && pkg.constraint() == LockConstraint::Constrained("=0.1.4".parse().unwrap())));
        assert!(sync_spec
            .to_remove
            .iter()
            .any(|pkg| pkg.name().to_string() == "nui.nvim"
                && pkg.constraint() == LockConstraint::Constrained("=0.3.0".parse().unwrap())));
        assert!(sync_spec
            .to_remove
            .iter()
            .any(|pkg| pkg.name().to_string() == "pathlib.nvim"
                && pkg.constraint()
                    == LockConstraint::Constrained(">=2.2.0, <2.3.0".parse().unwrap())));
    }

    #[test]
    fn test_sync_spec_empty() {
        let lockfile = get_test_lockfile();
        let packages = vec![];
        let sync_spec = lockfile
            .lock
            .package_sync_spec(&packages, &SyncStrategy::LockfileOnly);

        // Should remove all packages
        assert!(sync_spec.to_add.is_empty());
        assert_eq!(sync_spec.to_remove.len(), lockfile.rocks().len());
    }

    #[test]
    fn test_sync_spec_different_constraints() {
        let lockfile = get_test_lockfile();
        let packages = vec![PackageReq::parse("nvim-nio>=2.0.0").unwrap().into()];
        let sync_spec = lockfile
            .lock
            .package_sync_spec(&packages, &SyncStrategy::LockfileOnly);

        let expected: PackageVersionReq = ">=2.0.0".parse().unwrap();
        assert!(sync_spec
            .to_add
            .iter()
            .any(|req| req.name().to_string() == "nvim-nio" && req.version_req() == &expected));

        assert!(sync_spec
            .to_remove
            .iter()
            .any(|pkg| pkg.name().to_string() == "nvim-nio"));
    }

    #[test]
    fn test_sync_spec_ensure_installed() {
        let temp = assert_fs::TempDir::new().unwrap();
        temp.copy_from(
            PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("resources/test/sample-tree"),
            &["**"],
        )
        .unwrap();

        let config = ConfigBuilder::new()
            .unwrap()
            .user_tree(Some(temp.to_path_buf()))
            .build()
            .unwrap();
        let tree = config.user_tree(LuaVersion::Lua51).unwrap();
        let lockfile = tree.lockfile().unwrap();

        let packages: Vec<LuaDependencySpec> = vec![
            PackageReq::parse("neorg@8.8.1-1").unwrap().into(),
            // This package isn't installed in the tree
            PackageReq::parse("lua-cjson@2.1.0").unwrap().into(),
            // And neither is this
            PackageReq::parse("nonexistent").unwrap().into(),
        ];

        // Since lua-cjson is not present in the tree, it should get put into `to_add`
        let sync_spec = lockfile
            .lock
            .package_sync_spec(&packages, &SyncStrategy::EnsureInstalled(&tree));

        assert!(!sync_spec
            .to_add
            .iter()
            .any(|req| req.name().to_string() == "neorg"));

        assert!(sync_spec
            .to_add
            .iter()
            .any(|req| req.name().to_string() == "lua-cjson"));

        assert!(sync_spec
            .to_add
            .iter()
            .any(|req| req.name().to_string() == "nonexistent"));
    }

    #[test]
    fn verify_lockfile_version() {
        let deserialize = |version: &str| {
            let content = format!(r#"{{"version": "{version}"}}"#);
            parse_lockfile_json(&content).and_then(|value| {
                deserialize_lockfile::<Lockfile<ReadOnly>>(value, Path::new("lux.lock"), "")
            })
        };

        assert!(matches!(
            deserialize("3.0.0"),
            Err(Error::IncompatibleVersion { .. })
        ));
        assert!(matches!(
            deserialize("not-a-version"),
            Err(Error::InvalidVersion { .. })
        ));
    }

    #[test]
    fn migrate_v1_tree_lockfile() {
        let sample =
            PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("resources/test/lockfile-v1/tree.lock");

        let temp = assert_fs::TempDir::new().unwrap();
        let lockfile_path = temp.path().join("lux.lock");
        std::fs::copy(&sample, &lockfile_path).unwrap();

        let lockfile = Lockfile::load(lockfile_path.clone()).unwrap();

        let on_disk: serde_json::Value =
            serde_json::from_str(&std::fs::read_to_string(&lockfile_path).unwrap()).unwrap();
        assert_eq!(on_disk["version"], "1.0.0");
        assert!(on_disk["entrypoints"].is_array());

        assert_eq!(lockfile.version, "2.0.0");
        assert_eq!(lockfile.lock.entrypoints.len(), 3);
    }

    #[test]
    fn refuse_migration_for_custom_layout() {
        let lockfile = r#"{"version":"1.0.0","entrypoint_layout":{"root":"site/pack/lux"}}"#;
        assert!(matches!(
            load_lockfile::<Lockfile<ReadOnly>>(lockfile, Path::new("lux.lock"), ""),
            Err(Error::IncompatibleVersion { .. })
        ));
    }

    #[test]
    fn refuse_migration_for_conflicting_entrypoints() {
        let lockfile = r#"{
            "version": "1.0.0",
            "rocks": {
                "a": { "name": "foo" },
                "b": { "name": "foo" }
            },
            "entrypoints": ["a", "b"]
        }"#;
        assert!(matches!(
            load_lockfile::<Lockfile<ReadOnly>>(lockfile, Path::new("lux.lock"), ""),
            Err(Error::ConflictingEntrypoints { .. })
        ));
    }

    #[test]
    fn migrate_v1_workspace_lockfile() {
        let sample = PathBuf::from(env!("CARGO_MANIFEST_DIR"))
            .join("resources/test/lockfile-v1/workspace.lock");

        let temp = assert_fs::TempDir::new().unwrap();
        let lockfile_path = temp.path().join("lux.lock");
        std::fs::copy(&sample, &lockfile_path).unwrap();

        let lockfile = WorkspaceLockfile::load(lockfile_path.clone()).unwrap();

        let on_disk: serde_json::Value =
            serde_json::from_str(&std::fs::read_to_string(&lockfile_path).unwrap()).unwrap();
        assert_eq!(on_disk["version"], "1.0.0");
        assert!(on_disk["dependencies"]["entrypoints"].is_array());

        assert_eq!(lockfile.version, "2.0.0");
        assert_eq!(lockfile.dependencies.entrypoints.len(), 3);
    }
}