use std::collections::HashSet;

use turbopath::{AbsoluteSystemPath, AnchoredSystemPathBuf};
use turborepo_repository::{
    change_mapper::{ChangeMapError, ChangeMapper, LockfileChange, PackageChanges},
    package_graph::{PackageGraph, WorkspaceName},
};
use turborepo_scm::SCM;

/// Given two git refs, determine which packages have changed between them.
pub trait GitChangeDetector {
    fn changed_packages(
        &self,
        from_ref: &str,
        to_ref: &str,
    ) -> Result<HashSet<WorkspaceName>, ChangeMapError>;
}

pub struct ScopeChangeDetector<'a> {
    turbo_root: &'a AbsoluteSystemPath,
    change_mapper: ChangeMapper<'a>,
    scm: &'a SCM,
    pkg_graph: &'a PackageGraph,
}

impl<'a> ScopeChangeDetector<'a> {
    pub fn new(
        turbo_root: &'a AbsoluteSystemPath,
        scm: &'a SCM,
        pkg_graph: &'a PackageGraph,
        global_deps: Vec<String>,
        ignore_patterns: Vec<String>,
    ) -> Self {
        let change_mapper = ChangeMapper::new(pkg_graph, global_deps, ignore_patterns);

        Self {
            turbo_root,
            change_mapper,
            scm,
            pkg_graph,
        }
    }

    /// Gets the lockfile content from SCM if it has changed.
    /// Does *not* error if cannot get content, instead just
    /// returns an empty lockfile change
    fn get_lockfile_contents(
        &self,
        from_ref: &str,
        changed_files: &HashSet<AnchoredSystemPathBuf>,
    ) -> Option<LockfileChange> {
        let lockfile_path = self
            .pkg_graph
            .package_manager()
            .lockfile_path(self.turbo_root);

        if !ChangeMapper::lockfile_changed(self.turbo_root, changed_files, &lockfile_path) {
            return None;
        }

        let lockfile_path = self
            .pkg_graph
            .package_manager()
            .lockfile_path(self.turbo_root);

        let Ok(content) = self.scm.previous_content(from_ref, &lockfile_path) else {
            return Some(LockfileChange::Empty);
        };

        Some(LockfileChange::WithContent(content))
    }
}

impl<'a> GitChangeDetector for ScopeChangeDetector<'a> {
    fn changed_packages(
        &self,
        from_ref: &str,
        to_ref: &str,
    ) -> Result<HashSet<WorkspaceName>, ChangeMapError> {
        let mut changed_files = HashSet::new();
        if !from_ref.is_empty() {
            changed_files = self
                .scm
                .changed_files(self.turbo_root, Some(from_ref), to_ref)?;
        }

        let lockfile_contents = self.get_lockfile_contents(from_ref, &changed_files);

        match self
            .change_mapper
            .changed_packages(changed_files, lockfile_contents)?
        {
            PackageChanges::All => Ok(self
                .pkg_graph
                .workspaces()
                .map(|(name, _)| name.to_owned())
                .collect()),
            PackageChanges::Some(packages) => Ok(packages
                .iter()
                .map(|package| package.name.to_owned())
                .collect()),
        }
    }
}
