o_/

pm.rs

use crate::lock::{LockCollector, write_lockfile};
use crate::report::Report;
use flate2::read::GzDecoder;
use home::home_dir;
use nodejs_semver::{Range, Version};
use reqwest::blocking::Client;
use serde::Deserialize;
use serde_json::Value;
use ssri::Integrity;
use std::collections::{HashMap, HashSet};
use std::fs;
use std::io;
use std::path::{Path, PathBuf};
use tar::Archive;
use tempfile::tempdir;

#[derive(Debug, Clone, Deserialize)]
pub struct Manifest {
    pub name: String,
    pub version: String,
    #[serde(default)]
    pub dependencies: Option<HashMap<String, String>>,
    #[serde(default, rename = "devDependencies")]
    pub dev_dependencies: Option<HashMap<String, String>>,
    #[serde(default, rename = "optionalDependencies")]
    pub optional_dependencies: Option<HashMap<String, String>>,
    #[serde(default, rename = "peerDependencies")]
    pub peer_dependencies: Option<HashMap<String, String>>,
}

pub fn read_manifest(path: &str) -> io::Result<Manifest> {
    let manifest_path = find_manifest_path(Path::new(path))?;
    read_manifest_from_path(&manifest_path)
}

fn read_manifest_from_path(path: &Path) -> io::Result<Manifest> {
    let source = fs::read_to_string(path)?;
    serde_json::from_str(&source).map_err(|source| {
        io::Error::new(
            io::ErrorKind::InvalidData,
            format!("failed to parse {}: {source}", path.display()),
        )
    })
}

fn find_manifest_path(start: &Path) -> io::Result<PathBuf> {
    let mut current = if start.is_dir() {
        start.to_path_buf()
    } else {
        start
            .parent()
            .map(Path::to_path_buf)
            .unwrap_or_else(|| PathBuf::from("."))
    };

    loop {
        let candidate = current.join("package.json");
        if candidate.is_file() {
            return Ok(candidate);
        }

        if !current.pop() {
            return Err(io::Error::new(
                io::ErrorKind::NotFound,
                format!("failed to find package.json from {}", start.display()),
            ));
        }
    }
}

#[derive(Debug, Deserialize)]
struct Packument {
    versions: HashMap<String, RegistryVersion>,
}

#[derive(Debug, Deserialize)]
struct RegistryVersion {
    #[serde(default)]
    dependencies: HashMap<String, String>,
    #[serde(default)]
    optional_dependencies: HashMap<String, String>,
    #[serde(default)]
    peer_dependencies: HashMap<String, String>,
    dist: RegistryDist,
}

#[derive(Debug, Deserialize)]
struct RegistryDist {
    tarball: String,
    integrity: Option<String>,
}

#[derive(Debug)]
struct ResolvedPackage {
    name: String,
    version: String,
    dependencies: HashMap<String, String>,
    optional_dependencies: HashMap<String, String>,
    peer_dependencies: HashMap<String, String>,
    tarball_url: String,
    integrity: Option<String>,
}

#[derive(Debug, Clone, Copy)]
enum DependencyKind {
    Prod,
    Dev,
    Optional,
    Peer,
}

#[derive(Debug, Default)]
struct InstallSummary {
    prod_installed: usize,
    dev_installed: usize,
    optional_installed: usize,
    peer_installed: usize,
    warnings: Vec<String>,
}

impl InstallSummary {
    fn record_install(&mut self, kind: DependencyKind) {
        match kind {
            DependencyKind::Prod => self.prod_installed += 1,
            DependencyKind::Dev => self.dev_installed += 1,
            DependencyKind::Optional => self.optional_installed += 1,
            DependencyKind::Peer => self.peer_installed += 1,
        }
    }

    fn warn(&mut self, warning: impl Into<String>) {
        self.warnings.push(warning.into());
    }
}

#[derive(Debug)]
pub enum PmError {
    HomeDirUnavailable,
    MissingGlobalPackageSpec,
    InvalidPackageSpec {
        spec: String,
    },
    PackageNotInstalled {
        name: String,
        path: PathBuf,
    },
    FindManifest {
        start: PathBuf,
        source: io::Error,
    },
    ReadManifest {
        path: PathBuf,
        source: io::Error,
    },
    ParseManifest {
        path: PathBuf,
        source: serde_json::Error,
    },
    ProjectRootMissing {
        path: PathBuf,
    },
    CreateDir {
        path: PathBuf,
        source: io::Error,
    },
    FetchMetadata {
        package: String,
        source: reqwest::Error,
    },
    MetadataStatus {
        package: String,
        source: reqwest::Error,
    },
    ReadMetadataBody {
        package: String,
        source: reqwest::Error,
    },
    ParseMetadata {
        package: String,
        source: serde_json::Error,
    },
    InvalidRange {
        package: String,
        range: String,
        source: String,
    },
    VersionNotFound {
        package: String,
        range: String,
    },
    MissingResolvedVersion {
        package: String,
        version: String,
    },
    DownloadTarball {
        package: String,
        source: reqwest::Error,
    },
    TarballStatus {
        package: String,
        source: reqwest::Error,
    },
    ReadTarballBody {
        package: String,
        source: reqwest::Error,
    },
    InvalidIntegrity {
        package: String,
        version: String,
        source: String,
    },
    IntegrityMismatch {
        package: String,
        version: String,
        source: String,
    },
    ExtractTarball {
        package: String,
        source: io::Error,
    },
    MissingPackageDir {
        package: String,
        path: PathBuf,
    },
    RemoveExistingInstall {
        path: PathBuf,
        source: io::Error,
    },
    CopyInstall {
        from: PathBuf,
        to: PathBuf,
        source: io::Error,
    },
    ReadInstalledManifest {
        path: PathBuf,
        source: io::Error,
    },
    MissingInstalledName {
        path: PathBuf,
    },
    InvalidBinField {
        path: PathBuf,
    },
    InvalidBinEntry {
        path: PathBuf,
        entry: String,
    },
    AmbiguousBinEntry {
        package: String,
        path: PathBuf,
        available: Vec<String>,
    },
    MissingBinTarget {
        package_dir: PathBuf,
        target: PathBuf,
    },
    CreateTempDir {
        source: io::Error,
    },
    CurrentDir {
        source: io::Error,
    },
    WriteGeneratedManifest {
        path: PathBuf,
        source: io::Error,
    },
    WriteProcessOutput {
        source: io::Error,
    },
    MissingPackageBinary {
        package: String,
        command: String,
        path: PathBuf,
    },
    SpawnPackageBinary {
        package: String,
        command: PathBuf,
        source: io::Error,
    },
    PackageBinaryFailed {
        package: String,
        command: PathBuf,
        status: String,
        stderr: Option<String>,
    },
    CreateBinLink {
        command: String,
        path: PathBuf,
        source: io::Error,
    },
    RemoveBinLink {
        command: String,
        path: PathBuf,
        source: io::Error,
    },
    RemoveInstalledPackage {
        path: PathBuf,
        source: io::Error,
    },
    ReadLockfile {
        path: PathBuf,
        source: io::Error,
    },
    ParseLockfile {
        path: PathBuf,
        source: serde_json::Error,
    },
    WriteLockfile {
        path: PathBuf,
        source: io::Error,
    },
    InvalidTempPath {
        path: PathBuf,
    },
}

impl PmError {
    pub fn report(&self) -> Report {
        match self {
            Self::HomeDirUnavailable => Report::new("could not resolve home directory")
                .detail("`$HOME` is unavailable in the current environment"),
            Self::MissingGlobalPackageSpec => Report::new("global install requires a package name")
                .detail("example: `o- install --global cowsay`"),
            Self::InvalidPackageSpec { spec } => Report::new("failed to parse package spec")
                .detail(format!("spec: {spec}"))
                .detail("expected `name`, `name@version`, `@scope/name`, or `@scope/name@version`"),
            Self::PackageNotInstalled { name, path } => {
                Report::new(format!("package `{name}` is not installed"))
                    .detail(format!("path: {}", path.display()))
            }
            Self::FindManifest { start, source } => Report::new("failed to find package.json")
                .detail(format!("start: {}", start.display()))
                .detail(format!("cause: {source}")),
            Self::ReadManifest { path, source } => Report::new("failed to read package.json")
                .detail(format!("path: {}", path.display()))
                .detail(format!("cause: {source}")),
            Self::ParseManifest { path, source } => Report::new("failed to parse package.json")
                .detail(format!("path: {}", path.display()))
                .detail(format!("cause: {source}")),
            Self::ProjectRootMissing { path } => Report::new("failed to resolve project root")
                .detail(format!("path: {}", path.display())),
            Self::CreateDir { path, source } => Report::new("failed to create directory")
                .detail(format!("path: {}", path.display()))
                .detail(format!("cause: {source}")),
            Self::FetchMetadata { package, source } => {
                Report::new(format!("failed to fetch package metadata for `{package}`"))
                    .detail(format!("cause: {source}"))
            }
            Self::MetadataStatus { package, source } => {
                Report::new(format!("registry returned an error for `{package}`"))
                    .detail(format!("cause: {source}"))
            }
            Self::ReadMetadataBody { package, source } => Report::new(format!(
                "failed to read package metadata body for `{package}`"
            ))
            .detail(format!("cause: {source}")),
            Self::ParseMetadata { package, source } => {
                Report::new(format!("failed to decode package metadata for `{package}`"))
                    .detail(format!("cause: {source}"))
            }
            Self::InvalidRange {
                package,
                range,
                source,
            } => Report::new(format!("invalid semver range for `{package}`"))
                .detail(format!("range: {range}"))
                .detail(format!("cause: {source}")),
            Self::VersionNotFound { package, range } => Report::new(format!(
                "no version of `{package}` satisfies the requested range"
            ))
            .detail(format!("range: {range}")),
            Self::MissingResolvedVersion { package, version } => {
                Report::new(format!("registry metadata is incomplete for `{package}`"))
                    .detail(format!("version: {version}"))
            }
            Self::DownloadTarball { package, source } => {
                Report::new(format!("failed to download tarball for `{package}`"))
                    .detail(format!("cause: {source}"))
            }
            Self::TarballStatus { package, source } => {
                Report::new(format!("tarball request failed for `{package}`"))
                    .detail(format!("cause: {source}"))
            }
            Self::ReadTarballBody { package, source } => {
                Report::new(format!("failed to read tarball body for `{package}`"))
                    .detail(format!("cause: {source}"))
            }
            Self::InvalidIntegrity {
                package,
                version,
                source,
            } => Report::new(format!("registry integrity is invalid for `{package}`"))
                .detail(format!("version: {version}"))
                .detail(format!("cause: {source}")),
            Self::IntegrityMismatch {
                package,
                version,
                source,
            } => Report::new(format!("integrity check failed for `{package}`"))
                .detail(format!("version: {version}"))
                .detail(format!("cause: {source}")),
            Self::ExtractTarball { package, source } => {
                Report::new(format!("failed to extract tarball for `{package}`"))
                    .detail(format!("cause: {source}"))
            }
            Self::MissingPackageDir { package, path } => {
                Report::new(format!("downloaded tarball for `{package}` is malformed"))
                    .detail(format!("missing: {}", path.display()))
            }
            Self::RemoveExistingInstall { path, source } => {
                Report::new("failed to remove existing package installation")
                    .detail(format!("path: {}", path.display()))
                    .detail(format!("cause: {source}"))
            }
            Self::CopyInstall { from, to, source } => Report::new("failed to copy package files")
                .detail(format!("from: {}", from.display()))
                .detail(format!("to: {}", to.display()))
                .detail(format!("cause: {source}")),
            Self::ReadInstalledManifest { path, source } => {
                Report::new("failed to read installed package manifest")
                    .detail(format!("path: {}", path.display()))
                    .detail(format!("cause: {source}"))
            }
            Self::MissingInstalledName { path } => {
                Report::new("installed package manifest is missing `name`")
                    .detail(format!("path: {}", path.display()))
            }
            Self::InvalidBinField { path } => {
                Report::new("installed package has an invalid `bin` field")
                    .detail(format!("path: {}", path.display()))
            }
            Self::InvalidBinEntry { path, entry } => {
                Report::new("installed package has an invalid `bin` entry")
                    .detail(format!("path: {}", path.display()))
                    .detail(format!("entry: {entry}"))
            }
            Self::AmbiguousBinEntry {
                package,
                path,
                available,
            } => Report::new(format!("package `{package}` exposes multiple bin commands"))
                .detail(format!("path: {}", path.display()))
                .detail(format!("available: {}", available.join(", "))),
            Self::MissingBinTarget {
                package_dir,
                target,
            } => Report::new("installed package bin target does not exist")
                .detail(format!("package: {}", package_dir.display()))
                .detail(format!("target: {}", target.display())),
            Self::CreateTempDir { source } => Report::new("failed to create temporary directory")
                .detail(format!("cause: {source}")),
            Self::CurrentDir { source } => Report::new("failed to resolve current directory")
                .detail(format!("cause: {source}")),
            Self::WriteGeneratedManifest { path, source } => {
                Report::new("failed to write generated package.json")
                    .detail(format!("path: {}", path.display()))
                    .detail(format!("cause: {source}"))
            }
            Self::WriteProcessOutput { source } => {
                Report::new("failed to write package output").detail(format!("cause: {source}"))
            }
            Self::MissingPackageBinary {
                package,
                command,
                path,
            } => Report::new(format!(
                "package `{package}` does not expose a runnable binary"
            ))
            .detail(format!("command: {command}"))
            .detail(format!("expected shim: {}", path.display())),
            Self::SpawnPackageBinary {
                package,
                command,
                source,
            } => Report::new(format!("failed to execute package binary for `{package}`"))
                .detail(format!("command: {}", command.display()))
                .detail(format!("cause: {source}")),
            Self::PackageBinaryFailed {
                package,
                command,
                status,
                stderr,
            } => {
                let report = Report::new(format!("package binary `{package}` failed"))
                    .detail(format!("command: {}", command.display()))
                    .detail(format!("status: {status}"));
                if let Some(stderr) = stderr {
                    report.detail(format!("stderr: {stderr}"))
                } else {
                    report
                }
            }
            Self::CreateBinLink {
                command,
                path,
                source,
            } => Report::new(format!("failed to create bin link `{command}`"))
                .detail(format!("path: {}", path.display()))
                .detail(format!("cause: {source}")),
            Self::RemoveBinLink {
                command,
                path,
                source,
            } => Report::new(format!("failed to remove bin link `{command}`"))
                .detail(format!("path: {}", path.display()))
                .detail(format!("cause: {source}")),
            Self::RemoveInstalledPackage { path, source } => {
                Report::new("failed to remove installed package")
                    .detail(format!("path: {}", path.display()))
                    .detail(format!("cause: {source}"))
            }
            Self::ReadLockfile { path, source } => Report::new("failed to read package-lock.json")
                .detail(format!("path: {}", path.display()))
                .detail(format!("cause: {source}")),
            Self::ParseLockfile { path, source } => {
                Report::new("failed to parse package-lock.json")
                    .detail(format!("path: {}", path.display()))
                    .detail(format!("cause: {source}"))
            }
            Self::WriteLockfile { path, source } => {
                Report::new("failed to write package-lock.json")
                    .detail(format!("path: {}", path.display()))
                    .detail(format!("cause: {source}"))
            }
            Self::InvalidTempPath { path } => Report::new("failed to run x")
                .detail("note: tempdir failed".to_string())
                .detail(format!("path: {}", path.display())),
        }
    }

    fn warning_summary(&self) -> String {
        self.report().summary().to_string()
    }
}

pub fn install() -> Result<Report, PmError> {
    install_from(".")
}

pub fn global_install(package_spec: Option<&str>) -> Result<Report, PmError> {
    let package_spec = package_spec.ok_or(PmError::MissingGlobalPackageSpec)?;
    if package_spec.trim().is_empty() {
        return Err(PmError::MissingGlobalPackageSpec);
    }
    let (package_name, package_range) = parse_package_spec(package_spec);
    let global_root = global_packages_root()?;
    let node_modules = global_node_modules_dir(&global_root);
    fs::create_dir_all(&node_modules).map_err(|source| PmError::CreateDir {
        path: node_modules.clone(),
        source,
    })?;

    let mut installed = HashSet::new();
    let mut lock = LockCollector::new();
    let mut root_dependencies = HashMap::new();
    root_dependencies.insert(package_name.clone(), package_range.clone());
    let empty_dependencies = HashMap::new();
    lock.insert_root_fields(
        "o--global",
        "0.0.0",
        &root_dependencies,
        &empty_dependencies,
        &empty_dependencies,
        &empty_dependencies,
    );

    let mut summary = InstallSummary::default();
    let client = Client::new();
    install_dependency(
        &client,
        &global_root,
        &package_name,
        &package_range,
        &node_modules,
        &mut installed,
        &mut lock,
        &mut summary,
        DependencyKind::Prod,
    )?;

    let installed_manifest_path = install_dir(&node_modules, &package_name).join("package.json");
    let installed_manifest = read_manifest_from_path(&installed_manifest_path)
        .map_err(|source| map_manifest_error(&installed_manifest_path, source))?;
    let lockfile = lock.into_lockfile_fields("o--global", "0.0.0");
    let lockfile_path =
        write_lockfile(&global_root, &lockfile).map_err(|source| PmError::WriteLockfile {
            path: global_root.join("package-lock.json"),
            source,
        })?;

    Ok(Report::new(format!(
        "installed global package `{}`",
        installed_manifest.name
    ))
    .detail(format!("requested: {package_spec}"))
    .detail(format!("resolved version: {}", installed_manifest.version))
    .detail(format!("root: {}", global_root.display()))
    .detail(format!(
        "bin dir: {}",
        global_bin_dir(&node_modules).display()
    ))
    .detail(format!("lockfile: {}", lockfile_path.display()))
    .detail(format!("dependencies: {}", summary.prod_installed))
    .detail(format!(
        "optionalDependencies: {}",
        summary.optional_installed
    ))
    .detail(format!("peerDependencies: {}", summary.peer_installed))
    .detail(format!("peer warnings: {}", summary.warnings.len()))
    .detail(if summary.warnings.is_empty() {
        "peer/optional warnings: none".to_string()
    } else {
        format!("peer/optional warnings: {}", summary.warnings.join(" | "))
    }))
}

pub fn install_from(path: &str) -> Result<Report, PmError> {
    let manifest_path =
        find_manifest_path(Path::new(path)).map_err(|source| PmError::FindManifest {
            start: PathBuf::from(path),
            source,
        })?;
    let project_root = manifest_path
        .parent()
        .ok_or_else(|| PmError::ProjectRootMissing {
            path: manifest_path.clone(),
        })?;
    let manifest = read_manifest_from_path(&manifest_path)
        .map_err(|source| map_manifest_error(&manifest_path, source))?;
    let node_modules = project_root.join("node_modules");
    fs::create_dir_all(&node_modules).map_err(|source| PmError::CreateDir {
        path: node_modules.clone(),
        source,
    })?;

    let mut installed = HashSet::new();
    let mut lock = LockCollector::new();
    lock.insert_root_fields(
        &manifest.name,
        &manifest.version,
        &manifest.dependencies.clone().unwrap_or_default(),
        &manifest.dev_dependencies.clone().unwrap_or_default(),
        &manifest.optional_dependencies.clone().unwrap_or_default(),
        &manifest.peer_dependencies.clone().unwrap_or_default(),
    );
    let mut summary = InstallSummary::default();
    let client = Client::new();

    let root_dependencies = manifest.dependencies.clone().unwrap_or_default();
    let root_dev_dependencies = manifest.dev_dependencies.clone().unwrap_or_default();
    let root_optional_dependencies = manifest.optional_dependencies.clone().unwrap_or_default();
    let root_peer_dependencies = manifest.peer_dependencies.clone().unwrap_or_default();

    install_dependency_set(
        &client,
        project_root,
        &root_dependencies,
        &node_modules,
        &mut installed,
        &mut lock,
        &mut summary,
        DependencyKind::Prod,
    )?;
    install_dependency_set(
        &client,
        project_root,
        &root_dev_dependencies,
        &node_modules,
        &mut installed,
        &mut lock,
        &mut summary,
        DependencyKind::Dev,
    )?;
    install_optional_dependency_set(
        &client,
        project_root,
        &root_optional_dependencies,
        &node_modules,
        &mut installed,
        &mut lock,
        &mut summary,
        "root package",
    );
    reconcile_peer_dependencies(
        "root package",
        &root_peer_dependencies,
        &client,
        project_root,
        &node_modules,
        &mut installed,
        &mut lock,
        &mut summary,
    );

    let lockfile = lock.into_lockfile_fields(&manifest.name, &manifest.version);
    let lockfile_path =
        write_lockfile(project_root, &lockfile).map_err(|source| PmError::WriteLockfile {
            path: project_root.join("package-lock.json"),
            source,
        })?;

    Ok(Report::new("installed project dependencies")
        .detail(format!("root: {}", project_root.display()))
        .detail(format!("dependencies: {}", summary.prod_installed))
        .detail(format!("devDependencies: {}", summary.dev_installed))
        .detail(format!(
            "optionalDependencies: {}",
            summary.optional_installed
        ))
        .detail(format!("peerDependencies: {}", summary.peer_installed))
        .detail(format!("peer warnings: {}", summary.warnings.len()))
        .detail(format!("lockfile: {}", lockfile_path.display()))
        .detail(format!(
            "declared dependencies: {}",
            root_dependencies.len()
        ))
        .detail(format!(
            "declared devDependencies: {}",
            root_dev_dependencies.len()
        ))
        .detail(format!(
            "declared optionalDependencies: {}",
            root_optional_dependencies.len()
        ))
        .detail(format!(
            "declared peerDependencies: {}",
            root_peer_dependencies.len()
        ))
        .detail(if summary.warnings.is_empty() {
            "peer/optional warnings: none".to_string()
        } else {
            format!("peer/optional warnings: {}", summary.warnings.join(" | "))
        }))
}

fn install_dependency_set(
    client: &Client,
    project_root: &Path,
    dependencies: &HashMap<String, String>,
    node_modules_dir: &Path,
    installed: &mut HashSet<String>,
    lock: &mut LockCollector,
    summary: &mut InstallSummary,
    kind: DependencyKind,
) -> Result<(), PmError> {
    for (name, range) in dependencies {
        install_dependency(
            client,
            project_root,
            name,
            range,
            node_modules_dir,
            installed,
            lock,
            summary,
            kind,
        )?;
    }

    Ok(())
}

fn install_optional_dependency_set(
    client: &Client,
    project_root: &Path,
    dependencies: &HashMap<String, String>,
    node_modules_dir: &Path,
    installed: &mut HashSet<String>,
    lock: &mut LockCollector,
    summary: &mut InstallSummary,
    owner: &str,
) {
    for (name, range) in dependencies {
        if let Err(error) = install_dependency(
            client,
            project_root,
            name,
            range,
            node_modules_dir,
            installed,
            lock,
            summary,
            DependencyKind::Optional,
        ) {
            summary.warn(format!(
                "optional dependency `{name}` for `{owner}` was skipped: {}",
                error.warning_summary()
            ));
        }
    }
}

fn install_dependency(
    client: &Client,
    project_root: &Path,
    name: &str,
    range: &str,
    node_modules_dir: &Path,
    installed: &mut HashSet<String>,
    lock: &mut LockCollector,
    summary: &mut InstallSummary,
    kind: DependencyKind,
) -> Result<(), PmError> {
    let resolved = resolve_package(client, name, range)?;
    let install_key = format!(
        "{}@{}::{}",
        resolved.name,
        resolved.version,
        node_modules_dir.display()
    );

    if !installed.insert(install_key) {
        return Ok(());
    }

    let target_dir = install_dir(node_modules_dir, &resolved.name);
    if is_matching_install(&target_dir, &resolved.version)? {
        lock.insert_package(
            project_root,
            &target_dir,
            &resolved.name,
            &resolved.version,
            &resolved.tarball_url,
            resolved.integrity.as_deref(),
            &resolved.dependencies,
            &resolved.optional_dependencies,
            &resolved.peer_dependencies,
        )
        .map_err(|source| PmError::WriteLockfile {
            path: project_root.join("package-lock.json"),
            source,
        })?;
        install_dependency_set(
            client,
            project_root,
            &resolved.dependencies,
            &target_dir.join("node_modules"),
            installed,
            lock,
            summary,
            DependencyKind::Prod,
        )?;
        install_optional_dependency_set(
            client,
            project_root,
            &resolved.optional_dependencies,
            &target_dir.join("node_modules"),
            installed,
            lock,
            summary,
            &resolved.name,
        );
        reconcile_peer_dependencies(
            &resolved.name,
            &resolved.peer_dependencies,
            client,
            project_root,
            node_modules_dir,
            installed,
            lock,
            summary,
        );
        return Ok(());
    }

    if let Some(parent) = target_dir.parent() {
        fs::create_dir_all(parent).map_err(|source| PmError::CreateDir {
            path: parent.to_path_buf(),
            source,
        })?;
    }

    let package_root = download_and_extract_package(client, &resolved)?;

    if target_dir.exists() {
        fs::remove_dir_all(&target_dir).map_err(|source| PmError::RemoveExistingInstall {
            path: target_dir.clone(),
            source,
        })?;
    }
    copy_dir_all(&package_root, &target_dir).map_err(|source| PmError::CopyInstall {
        from: package_root.clone(),
        to: target_dir.clone(),
        source,
    })?;
    create_bin_links(node_modules_dir, &target_dir)?;
    summary.record_install(kind);
    lock.insert_package(
        project_root,
        &target_dir,
        &resolved.name,
        &resolved.version,
        &resolved.tarball_url,
        resolved.integrity.as_deref(),
        &resolved.dependencies,
        &resolved.optional_dependencies,
        &resolved.peer_dependencies,
    )
    .map_err(|source| PmError::WriteLockfile {
        path: project_root.join("package-lock.json"),
        source,
    })?;

    let nested_node_modules = target_dir.join("node_modules");
    fs::create_dir_all(&nested_node_modules).map_err(|source| PmError::CreateDir {
        path: nested_node_modules.clone(),
        source,
    })?;
    install_dependency_set(
        client,
        project_root,
        &resolved.dependencies,
        &nested_node_modules,
        installed,
        lock,
        summary,
        DependencyKind::Prod,
    )?;
    install_optional_dependency_set(
        client,
        project_root,
        &resolved.optional_dependencies,
        &nested_node_modules,
        installed,
        lock,
        summary,
        &resolved.name,
    );
    reconcile_peer_dependencies(
        &resolved.name,
        &resolved.peer_dependencies,
        client,
        project_root,
        node_modules_dir,
        installed,
        lock,
        summary,
    );

    Ok(())
}

fn resolve_package(client: &Client, name: &str, range: &str) -> Result<ResolvedPackage, PmError> {
    let url = resolve_npm_url(name);
    let response = client
        .get(url)
        .send()
        .map_err(|source| PmError::FetchMetadata {
            package: name.to_string(),
            source,
        })?;

    let response = response
        .error_for_status()
        .map_err(|source| PmError::MetadataStatus {
            package: name.to_string(),
            source,
        })?;

    let body = response
        .text()
        .map_err(|source| PmError::ReadMetadataBody {
            package: name.to_string(),
            source,
        })?;

    let packument: Packument =
        serde_json::from_str(&body).map_err(|source| PmError::ParseMetadata {
            package: name.to_string(),
            source,
        })?;

    let range: Range = range
        .parse()
        .map_err(|source: nodejs_semver::SemverError| PmError::InvalidRange {
            package: name.to_string(),
            range: range.to_string(),
            source: source.to_string(),
        })?;

    let version = packument
        .versions
        .keys()
        .filter_map(|raw_version| {
            Version::parse(raw_version)
                .ok()
                .map(|parsed| (raw_version, parsed))
        })
        .filter(|(_, parsed)| parsed.satisfies(&range))
        .map(|(_, parsed)| parsed)
        .max()
        .ok_or_else(|| PmError::VersionNotFound {
            package: name.to_string(),
            range: range.to_string(),
        })?;

    let version_string = version.to_string();
    let metadata =
        packument
            .versions
            .get(&version_string)
            .ok_or_else(|| PmError::MissingResolvedVersion {
                package: name.to_string(),
                version: version_string.clone(),
            })?;

    Ok(ResolvedPackage {
        name: name.to_string(),
        version: version_string,
        dependencies: metadata.dependencies.clone(),
        optional_dependencies: metadata.optional_dependencies.clone(),
        peer_dependencies: metadata.peer_dependencies.clone(),
        tarball_url: metadata.dist.tarball.clone(),
        integrity: metadata.dist.integrity.clone(),
    })
}

fn download_and_extract_package(
    client: &Client,
    package: &ResolvedPackage,
) -> Result<PathBuf, PmError> {
    let response =
        client
            .get(&package.tarball_url)
            .send()
            .map_err(|source| PmError::DownloadTarball {
                package: package.name.clone(),
                source,
            })?;
    let response = response
        .error_for_status()
        .map_err(|source| PmError::TarballStatus {
            package: package.name.clone(),
            source,
        })?;

    let bytes = response
        .bytes()
        .map_err(|source| PmError::ReadTarballBody {
            package: package.name.clone(),
            source,
        })?;
    verify_integrity(package, bytes.as_ref())?;

    let temp = tempdir().map_err(|source| PmError::ExtractTarball {
        package: package.name.clone(),
        source,
    })?;
    let temp_path = temp.keep();

    let tar = GzDecoder::new(bytes.as_ref());
    let mut archive = Archive::new(tar);
    archive
        .unpack(&temp_path)
        .map_err(|source| PmError::ExtractTarball {
            package: package.name.clone(),
            source,
        })?;

    locate_extracted_package_root(&temp_path).ok_or_else(|| PmError::MissingPackageDir {
        package: package.name.clone(),
        path: temp_path.join("package"),
    })
}

fn locate_extracted_package_root(temp_path: &Path) -> Option<PathBuf> {
    let package_json = temp_path.join("package.json");
    if package_json.is_file() {
        return Some(temp_path.to_path_buf());
    }

    let package_dir = temp_path.join("package");
    if package_json.is_file() || package_dir.join("package.json").is_file() {
        return Some(package_dir);
    }

    let entries = fs::read_dir(temp_path).ok()?;
    let mut candidate_dir: Option<PathBuf> = None;

    for entry in entries.flatten() {
        let path = entry.path();
        if path.is_file() {
            continue;
        }

        if path.join("package.json").is_file() {
            if candidate_dir.is_some() {
                return None;
            }
            candidate_dir = Some(path);
        }
    }

    candidate_dir
}

fn verify_integrity(package: &ResolvedPackage, bytes: &[u8]) -> Result<(), PmError> {
    let Some(integrity) = &package.integrity else {
        return Ok(());
    };

    let parsed: Integrity =
        integrity
            .parse()
            .map_err(|source: ssri::Error| PmError::InvalidIntegrity {
                package: package.name.clone(),
                version: package.version.clone(),
                source: source.to_string(),
            })?;

    parsed
        .check(bytes)
        .map_err(|source: ssri::Error| PmError::IntegrityMismatch {
            package: package.name.clone(),
            version: package.version.clone(),
            source: source.to_string(),
        })?;

    Ok(())
}

fn install_dir(node_modules: &Path, package_name: &str) -> PathBuf {
    if let Some((scope, name)) = package_name.split_once('/') {
        node_modules.join(scope).join(name)
    } else {
        node_modules.join(package_name)
    }
}

fn is_matching_install(path: &Path, version: &str) -> Result<bool, PmError> {
    let manifest_path = path.join("package.json");
    if !manifest_path.is_file() {
        return Ok(false);
    }

    let manifest = read_manifest_from_path(&manifest_path)
        .map_err(|source| map_manifest_error(&manifest_path, source))?;
    Ok(manifest.version == version)
}

fn copy_dir_all(src: &Path, dst: &Path) -> io::Result<()> {
    fs::create_dir_all(dst)?;

    for entry in fs::read_dir(src)? {
        let entry = entry?;
        let file_type = entry.file_type()?;
        let from = entry.path();
        let to = dst.join(entry.file_name());

        if file_type.is_dir() {
            copy_dir_all(&from, &to)?;
        } else {
            fs::copy(&from, &to)?;
        }
    }

    Ok(())
}

fn create_bin_links(node_modules_dir: &Path, package_dir: &Path) -> Result<(), PmError> {
    let bin_entries = read_bin_entries(package_dir)?;
    if bin_entries.is_empty() {
        return Ok(());
    }

    let bin_dir = node_modules_dir.join(".bin");
    fs::create_dir_all(&bin_dir).map_err(|source| PmError::CreateDir {
        path: bin_dir.clone(),
        source,
    })?;

    for (command_name, relative_target) in bin_entries {
        let target = package_dir.join(normalize_package_relative_path(&relative_target));
        if !target.is_file() {
            return Err(PmError::MissingBinTarget {
                package_dir: package_dir.to_path_buf(),
                target,
            });
        }

        create_bin_link(&bin_dir, &command_name, &target)?;
    }

    Ok(())
}

fn read_bin_entries(package_dir: &Path) -> Result<Vec<(String, String)>, PmError> {
    let package_json_path = package_dir.join("package.json");
    let source = fs::read_to_string(&package_json_path).map_err(|source| {
        PmError::ReadInstalledManifest {
            path: package_json_path.clone(),
            source,
        }
    })?;
    let value: Value = serde_json::from_str(&source).map_err(|source| PmError::ParseManifest {
        path: package_json_path.clone(),
        source,
    })?;

    let package_name = value
        .get("name")
        .and_then(Value::as_str)
        .map(default_bin_name)
        .ok_or_else(|| PmError::MissingInstalledName {
            path: package_json_path.clone(),
        })?;

    let Some(bin_value) = value.get("bin") else {
        return Ok(Vec::new());
    };

    match bin_value {
        Value::String(path) => Ok(vec![(package_name, path.clone())]),
        Value::Object(entries) => {
            let mut bins = Vec::with_capacity(entries.len());
            for (command_name, target) in entries {
                let target = target.as_str().ok_or_else(|| PmError::InvalidBinEntry {
                    path: package_json_path.clone(),
                    entry: command_name.clone(),
                })?;
                bins.push((command_name.clone(), target.to_string()));
            }
            Ok(bins)
        }
        Value::Null => Ok(Vec::new()),
        _ => Err(PmError::InvalidBinField {
            path: package_json_path,
        }),
    }
}

fn default_bin_name(package_name: &str) -> String {
    package_name
        .rsplit_once('/')
        .map(|(_, name)| name.to_string())
        .unwrap_or_else(|| package_name.to_string())
}

fn normalize_package_relative_path(path: &str) -> PathBuf {
    let trimmed = path.strip_prefix("./").unwrap_or(path);
    PathBuf::from(trimmed)
}

fn reconcile_peer_dependencies(
    owner: &str,
    peer_dependencies: &HashMap<String, String>,
    client: &Client,
    project_root: &Path,
    node_modules_dir: &Path,
    installed: &mut HashSet<String>,
    lock: &mut LockCollector,
    summary: &mut InstallSummary,
) {
    for (name, range) in peer_dependencies {
        if peer_dependency_warning(name, range, node_modules_dir).is_some() {
            if let Err(error) = install_dependency(
                client,
                project_root,
                name,
                range,
                node_modules_dir,
                installed,
                lock,
                summary,
                DependencyKind::Peer,
            ) {
                summary.warn(format!(
                    "peer dependency `{name}` for `{owner}` could not be installed: {}",
                    error.warning_summary()
                ));
            }
        }
    }

    validate_peer_dependencies(owner, peer_dependencies, node_modules_dir, summary);
}

fn validate_peer_dependencies(
    owner: &str,
    peer_dependencies: &HashMap<String, String>,
    node_modules_dir: &Path,
    summary: &mut InstallSummary,
) {
    for (name, range) in peer_dependencies {
        if let Some(warning) = peer_dependency_warning(name, range, node_modules_dir) {
            summary.warn(format!("peer dependency for `{owner}`: {warning}"));
        }
    }
}

fn peer_dependency_warning(name: &str, range: &str, node_modules_dir: &Path) -> Option<String> {
    let package_dir = install_dir(node_modules_dir, name);
    let manifest_path = package_dir.join("package.json");
    if !manifest_path.is_file() {
        return Some(format!("missing `{name}` required by range `{range}`"));
    }

    let manifest = match read_manifest_from_path(&manifest_path) {
        Ok(manifest) => manifest,
        Err(error) => {
            return Some(format!(
                "failed to read installed `{name}` manifest: {error}"
            ));
        }
    };
    let installed_version = match Version::parse(&manifest.version) {
        Ok(version) => version,
        Err(error) => {
            return Some(format!(
                "`{name}` is installed with invalid version `{}`: {error}",
                manifest.version
            ));
        }
    };
    let expected_range: Range = match range.parse() {
        Ok(parsed) => parsed,
        Err(error) => {
            return Some(format!(
                "`{name}` requires invalid peer range `{range}`: {error}"
            ));
        }
    };
    if installed_version.satisfies(&expected_range) {
        None
    } else {
        Some(format!(
            "`{name}` is installed as `{}` but `{range}` is required",
            manifest.version
        ))
    }
}

#[cfg(unix)]
fn create_bin_link(bin_dir: &Path, command_name: &str, target: &Path) -> Result<(), PmError> {
    use std::os::unix::fs::symlink;

    let link_path = bin_dir.join(command_name);
    remove_existing_link_path(&link_path).map_err(|source| PmError::CreateBinLink {
        command: command_name.to_string(),
        path: link_path.clone(),
        source,
    })?;
    symlink(target, &link_path).map_err(|source| PmError::CreateBinLink {
        command: command_name.to_string(),
        path: link_path,
        source,
    })
}

#[cfg(windows)]
fn create_bin_link(bin_dir: &Path, command_name: &str, target: &Path) -> Result<(), PmError> {
    let link_path = bin_dir.join(format!("{command_name}.cmd"));
    remove_existing_link_path(&link_path).map_err(|source| PmError::CreateBinLink {
        command: command_name.to_string(),
        path: link_path.clone(),
        source,
    })?;
    let script = format!("@ECHO off\r\nnode \"{}\" %*\r\n", target.display());
    fs::write(&link_path, script).map_err(|source| PmError::CreateBinLink {
        command: command_name.to_string(),
        path: link_path,
        source,
    })
}

fn remove_existing_link_path(path: &Path) -> io::Result<()> {
    if !path.exists() {
        return Ok(());
    }

    let metadata = fs::symlink_metadata(path)?;
    if metadata.file_type().is_dir() && !metadata.file_type().is_symlink() {
        fs::remove_dir_all(path)
    } else {
        fs::remove_file(path)
    }
}

fn resolve_npm_url(package: &str) -> String {
    let encoded = package.replace('@', "%40").replace('/', "%2F");
    format!("https://registry.npmjs.org/{encoded}")
}

fn global_packages_root() -> Result<PathBuf, PmError> {
    let mut path = home_dir().ok_or(PmError::HomeDirUnavailable)?;
    path.push(".config");
    path.push("o-");
    path.push("packages");
    Ok(path)
}

fn global_node_modules_dir(global_root: &Path) -> PathBuf {
    global_root.join("node_modules")
}

fn global_bin_dir(node_modules_dir: &Path) -> PathBuf {
    node_modules_dir.join(".bin")
}

fn parse_package_spec(spec: &str) -> (String, String) {
    let trimmed = spec.trim();
    if trimmed.is_empty() {
        return (String::new(), "*".to_string());
    }

    if let Some((name, range)) = split_package_spec(trimmed) {
        return (name.to_string(), normalize_package_range(range).to_string());
    }

    (trimmed.to_string(), "*".to_string())
}

fn split_package_spec(spec: &str) -> Option<(&str, &str)> {
    if spec.starts_with('@') {
        let slash = spec.find('/')?;
        let tail = &spec[slash + 1..];
        let at = tail.rfind('@')?;
        let split_index = slash + 1 + at;
        let name = &spec[..split_index];
        let range = &spec[split_index + 1..];
        if range.is_empty() {
            None
        } else {
            Some((name, range))
        }
    } else if let Some((name, range)) = spec.rsplit_once('@') {
        if name.is_empty() || range.is_empty() {
            None
        } else {
            Some((name, range))
        }
    } else {
        None
    }
}

fn normalize_package_range(range: &str) -> &str {
    if range == "latest" { "*" } else { range }
}

pub fn remove_shim(node_modules_dir: &Path, command: &str) -> Result<bool, PmError> {
    let shim_path = shim_path_for_command(node_modules_dir, command);
    if !shim_path.exists() {
        return Ok(false);
    }

    remove_existing_link_path(&shim_path).map_err(|source| PmError::RemoveBinLink {
        command: command.to_string(),
        path: shim_path,
        source,
    })?;
    Ok(true)
}

pub fn uninstall(name: &str) -> Result<Report, PmError> {
    let global_root = global_packages_root()?;
    let node_modules_dir = global_node_modules_dir(&global_root);
    let package_dir = install_dir(&node_modules_dir, name);
    if !package_dir.is_dir() {
        return Err(PmError::PackageNotInstalled {
            name: name.to_string(),
            path: package_dir,
        });
    }

    let manifest_path = package_dir.join("package.json");
    let manifest = read_manifest_from_path(&manifest_path)
        .map_err(|source| map_manifest_error(&manifest_path, source))?;
    let bin_entries = read_bin_entries(&package_dir)?;

    fs::remove_dir_all(&package_dir).map_err(|source| PmError::RemoveInstalledPackage {
        path: package_dir.clone(),
        source,
    })?;
    remove_empty_scope_dir(&package_dir)?;

    let mut removed_shims = Vec::new();
    for (command, _) in bin_entries {
        if remove_shim(&node_modules_dir, &command)? {
            removed_shims.push(command);
        }
    }

    let lockfile_path = remove_global_lockfile_entry(&global_root, &package_dir, &manifest.name)?;

    Ok(
        Report::new(format!("uninstalled package `{}`", manifest.name))
            .detail(format!("version: {}", manifest.version))
            .detail(format!("package: {}", package_dir.display()))
            .detail(format!(
                "bin dir: {}",
                global_bin_dir(&node_modules_dir).display()
            ))
            .detail(match lockfile_path {
                Some(path) => format!("lockfile: {}", path.display()),
                None => "lockfile: none".to_string(),
            })
            .detail(if removed_shims.is_empty() {
                "removed shims: none".to_string()
            } else {
                format!("removed shims: {}", removed_shims.join(", "))
            }),
    )
}

#[cfg(unix)]
fn shim_path_for_command(node_modules_dir: &Path, command: &str) -> PathBuf {
    global_bin_dir(node_modules_dir).join(command)
}

#[cfg(windows)]
fn shim_path_for_command(node_modules_dir: &Path, command: &str) -> PathBuf {
    global_bin_dir(node_modules_dir).join(format!("{command}.cmd"))
}

fn remove_empty_scope_dir(package_dir: &Path) -> Result<(), PmError> {
    let Some(parent) = package_dir.parent() else {
        return Ok(());
    };

    let Some(scope_name) = parent.file_name().and_then(|name| name.to_str()) else {
        return Ok(());
    };

    if !scope_name.starts_with('@') {
        return Ok(());
    }

    if parent
        .read_dir()
        .map_err(|source| PmError::RemoveInstalledPackage {
            path: parent.to_path_buf(),
            source,
        })?
        .next()
        .is_none()
    {
        fs::remove_dir(parent).map_err(|source| PmError::RemoveInstalledPackage {
            path: parent.to_path_buf(),
            source,
        })?;
    }

    Ok(())
}

fn remove_global_lockfile_entry(
    global_root: &Path,
    package_dir: &Path,
    package_name: &str,
) -> Result<Option<PathBuf>, PmError> {
    let lockfile_path = global_root.join("package-lock.json");
    if !lockfile_path.is_file() {
        return Ok(None);
    }

    let source = fs::read_to_string(&lockfile_path).map_err(|source| PmError::ReadLockfile {
        path: lockfile_path.clone(),
        source,
    })?;
    let mut lockfile: crate::lock::LockFile =
        serde_json::from_str(&source).map_err(|source| PmError::ParseLockfile {
            path: lockfile_path.clone(),
            source,
        })?;

    let key = package_dir
        .strip_prefix(global_root)
        .map(|path| path.to_string_lossy().replace('\\', "/"))
        .unwrap_or_else(|_| package_dir.to_string_lossy().replace('\\', "/"));
    lockfile.packages.remove(&key);

    if let Some(root) = lockfile.packages.get_mut("") {
        remove_dependency_entry(&mut root.dependencies, package_name);
        remove_dependency_entry(&mut root.dev_dependencies, package_name);
        remove_dependency_entry(&mut root.optional_dependencies, package_name);
        remove_dependency_entry(&mut root.peer_dependencies, package_name);
    }

    let rewritten =
        write_lockfile(global_root, &lockfile).map_err(|source| PmError::WriteLockfile {
            path: lockfile_path,
            source,
        })?;
    Ok(Some(rewritten))
}

fn remove_dependency_entry(
    dependencies: &mut Option<std::collections::BTreeMap<String, String>>,
    name: &str,
) {
    let should_clear = if let Some(entries) = dependencies.as_mut() {
        entries.remove(name);
        entries.is_empty()
    } else {
        false
    };

    if should_clear {
        *dependencies = None;
    }
}

fn map_manifest_error(path: &Path, source: io::Error) -> PmError {
    match source.kind() {
        io::ErrorKind::InvalidData => {
            let parse_source =
                serde_json::Error::io(io::Error::new(source.kind(), source.to_string()));
            PmError::ParseManifest {
                path: path.to_path_buf(),
                source: parse_source,
            }
        }
        _ => PmError::ReadManifest {
            path: path.to_path_buf(),
            source,
        },
    }
}

#[cfg(test)]
mod tests {
    use super::locate_extracted_package_root;
    use std::fs;
    use tempfile::tempdir;

    #[test]
    fn uses_temp_root_when_package_json_is_at_root() {
        let temp = tempdir().unwrap();
        fs::write(temp.path().join("package.json"), "{}").unwrap();

        let root = locate_extracted_package_root(temp.path()).unwrap();
        assert_eq!(root, temp.path());
    }

    #[test]
    fn uses_package_directory_when_present() {
        let temp = tempdir().unwrap();
        let package_dir = temp.path().join("package");
        fs::create_dir(&package_dir).unwrap();
        fs::write(package_dir.join("package.json"), "{}").unwrap();

        let root = locate_extracted_package_root(temp.path()).unwrap();
        assert_eq!(root, package_dir);
    }

    #[test]
    fn uses_single_top_level_directory_as_fallback() {
        let temp = tempdir().unwrap();
        let package_dir = temp.path().join("nested");
        fs::create_dir(&package_dir).unwrap();
        fs::write(package_dir.join("package.json"), "{}").unwrap();

        let root = locate_extracted_package_root(temp.path()).unwrap();
        assert_eq!(root, package_dir);
    }

    #[test]
    fn rejects_ambiguous_top_level_layouts() {
        let temp = tempdir().unwrap();
        let left = temp.path().join("left");
        let right = temp.path().join("right");
        fs::create_dir(&left).unwrap();
        fs::create_dir(&right).unwrap();
        fs::write(left.join("package.json"), "{}").unwrap();
        fs::write(right.join("package.json"), "{}").unwrap();

        assert!(locate_extracted_package_root(temp.path()).is_none());
    }
}