dot_agent_core/

installer.rs

use std::fs;
use std::path::{Path, PathBuf};

use crate::error::{DotAgentError, Result};
use crate::json_merge::{is_mergeable_json, merge_json_file, unmerge_json_file};
use crate::metadata::{compute_file_hash, compute_hash, Metadata};
use crate::profile::{IgnoreConfig, Profile};

const CLAUDE_MD: &str = "CLAUDE.md";
const CLAUDE_DIR: &str = ".claude";

/// Callback type for file operation progress reporting
pub type FileCallback<'a> = Option<&'a dyn Fn(&str, &str)>;

// Directories where files should be prefixed with profile name
const PREFIXED_DIRS: &[&str] = &["agents", "commands", "rules"];
// Directories where subdirectories should be prefixed (skills has SKILL.md inside)
const PREFIXED_SUBDIRS: &[&str] = &["skills"];

/// Generate metadata key with profile prefix.
/// Format: "{profile}:{relative_path}"
fn make_meta_key(profile_name: &str, relative_path: &str) -> String {
    format!("{}:{}", profile_name, relative_path)
}

#[derive(Debug, Clone, Copy, PartialEq)]
pub enum FileStatus {
    Unchanged,
    Modified,
    Added,
    Missing,
}

#[derive(Debug)]
pub struct FileInfo {
    pub relative_path: PathBuf,
    pub status: FileStatus,
}

#[derive(Debug, Default)]
pub struct InstallResult {
    pub installed: usize,
    pub skipped: usize,
    pub conflicts: usize,
    pub merged: usize,
}

#[derive(Debug, Default)]
pub struct DiffResult {
    pub unchanged: usize,
    pub modified: usize,
    pub added: usize,
    pub missing: usize,
    pub files: Vec<FileInfo>,
}

/// Options for install/upgrade/remove operations
#[derive(Default)]
pub struct InstallOptions<'a> {
    /// Force overwrite of existing files
    pub force: bool,
    /// Don't actually make changes, just show what would happen
    pub dry_run: bool,
    /// Don't add profile prefix to file names (install/upgrade only)
    pub no_prefix: bool,
    /// Don't merge JSON files (hooks.json, settings.json, etc.)
    pub no_merge: bool,
    /// File ignore configuration
    pub ignore_config: IgnoreConfig,
    /// Callback for file operation progress
    pub on_file: FileCallback<'a>,
}

impl std::fmt::Debug for InstallOptions<'_> {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_struct("InstallOptions")
            .field("force", &self.force)
            .field("dry_run", &self.dry_run)
            .field("no_prefix", &self.no_prefix)
            .field("no_merge", &self.no_merge)
            .field("ignore_config", &self.ignore_config)
            .field("on_file", &self.on_file.is_some())
            .finish()
    }
}

impl<'a> InstallOptions<'a> {
    /// Create new options with defaults
    pub fn new() -> Self {
        Self {
            ignore_config: IgnoreConfig::with_defaults(),
            ..Default::default()
        }
    }

    /// Set force flag
    pub fn force(mut self, force: bool) -> Self {
        self.force = force;
        self
    }

    /// Set dry_run flag
    pub fn dry_run(mut self, dry_run: bool) -> Self {
        self.dry_run = dry_run;
        self
    }

    /// Set no_prefix flag
    pub fn no_prefix(mut self, no_prefix: bool) -> Self {
        self.no_prefix = no_prefix;
        self
    }

    /// Set no_merge flag
    pub fn no_merge(mut self, no_merge: bool) -> Self {
        self.no_merge = no_merge;
        self
    }

    /// Set ignore configuration
    pub fn ignore_config(mut self, config: IgnoreConfig) -> Self {
        self.ignore_config = config;
        self
    }

    /// Set file callback
    pub fn on_file(mut self, callback: FileCallback<'a>) -> Self {
        self.on_file = callback;
        self
    }
}

pub struct Installer {
    base_dir: PathBuf,
}

impl Installer {
    pub fn new(base_dir: PathBuf) -> Self {
        Self { base_dir }
    }

    /// Get target directory (either project/.claude or global ~/.claude)
    pub fn resolve_target(&self, target: Option<&Path>, global: bool) -> Result<PathBuf> {
        if global {
            let home = dirs::home_dir().ok_or(DotAgentError::HomeNotFound)?;
            Ok(home.join(".claude"))
        } else {
            let base = target
                .map(|p| p.to_path_buf())
                .unwrap_or_else(|| std::env::current_dir().unwrap());

            if !base.exists() {
                return Err(DotAgentError::TargetNotFound { path: base });
            }

            Ok(base.join(CLAUDE_DIR))
        }
    }

    /// Install a profile to target
    pub fn install(
        &self,
        profile: &Profile,
        target: &Path,
        opts: &InstallOptions<'_>,
    ) -> Result<InstallResult> {
        let mut result = InstallResult::default();
        let mut metadata = Metadata::load(target)?.unwrap_or_else(|| Metadata::new(&self.base_dir));

        // Ensure target directory exists
        if !opts.dry_run && !target.exists() {
            fs::create_dir_all(target)?;
        }

        let files = profile.list_files_with_config(&opts.ignore_config)?;

        for relative_path in files {
            let src = profile.path.join(&relative_path);
            let prefixed_path = if opts.no_prefix {
                relative_path.clone()
            } else {
                prefix_path(&relative_path, &profile.name)
            };
            let dst = target.join(&prefixed_path);
            let relative_str = prefixed_path.to_string_lossy().to_string();

            let is_claude_md = relative_path.to_string_lossy() == CLAUDE_MD;
            let is_mergeable = is_mergeable_json(&relative_path);

            // Handle mergeable JSON files
            if is_mergeable && !opts.no_merge && dst.exists() {
                let merge_result = merge_json_file(&dst, &src, &profile.name)?;

                if !merge_result.changed {
                    if let Some(f) = opts.on_file {
                        f("SKIP", &relative_str);
                    }
                    result.skipped += 1;
                    continue;
                }

                if !opts.dry_run {
                    if let Some(parent) = dst.parent() {
                        fs::create_dir_all(parent)?;
                    }
                    fs::write(&dst, &merge_result.content)?;
                    metadata.add_merged(
                        &profile.name,
                        &relative_str,
                        merge_result.record.added_paths,
                    );
                }

                if let Some(f) = opts.on_file {
                    f("MERGE", &relative_str);
                }
                result.merged += 1;
                continue;
            }

            let src_content = fs::read(&src)?;
            let src_hash = compute_hash(&src_content);

            if dst.exists() {
                let dst_hash = compute_file_hash(&dst)?;

                if src_hash == dst_hash {
                    // Same content - skip
                    if let Some(f) = opts.on_file {
                        f("SKIP", &relative_str);
                    }
                    result.skipped += 1;
                    continue;
                }

                // CLAUDE.md is never overwritten
                if is_claude_md {
                    if let Some(f) = opts.on_file {
                        f("WARN", &relative_str);
                    }
                    result.skipped += 1;
                    continue;
                }

                // Different content - for JSON files with no_merge, this is a conflict
                if !opts.force {
                    if let Some(f) = opts.on_file {
                        f("CONFLICT", &relative_str);
                    }
                    result.conflicts += 1;
                    continue;
                }
            }

            // Copy file (or write merged JSON for new files)
            if !opts.dry_run {
                if let Some(parent) = dst.parent() {
                    fs::create_dir_all(parent)?;
                }

                // For new mergeable JSON files, add profile marker
                if is_mergeable && !opts.no_merge {
                    let merge_result = merge_json_file(&dst, &src, &profile.name)?;
                    fs::write(&dst, &merge_result.content)?;
                    metadata.add_merged(
                        &profile.name,
                        &relative_str,
                        merge_result.record.added_paths,
                    );
                } else {
                    fs::write(&dst, &src_content)?;
                    let meta_key = make_meta_key(&profile.name, &relative_str);
                    metadata.add_file(&meta_key, &src_hash);
                }
            }

            if let Some(f) = opts.on_file {
                f("OK", &relative_str);
            }
            result.installed += 1;
        }

        if !opts.dry_run && result.conflicts == 0 {
            metadata.add_profile(&profile.name);
            metadata.save(target)?;
        }

        Ok(result)
    }

    /// Compare profile with installed files
    pub fn diff(
        &self,
        profile: &Profile,
        target: &Path,
        ignore_config: &IgnoreConfig,
    ) -> Result<DiffResult> {
        let mut result = DiffResult::default();

        if !target.exists() {
            // All files are missing
            for relative_path in profile.list_files_with_config(ignore_config)? {
                let prefixed_path = prefix_path(&relative_path, &profile.name);
                result.files.push(FileInfo {
                    relative_path: prefixed_path,
                    status: FileStatus::Missing,
                });
                result.missing += 1;
            }
            return Ok(result);
        }

        let metadata = Metadata::load(target)?;
        let profile_files = profile.list_files_with_config(ignore_config)?;

        // Build set of prefixed paths for comparison
        let prefixed_files: Vec<_> = profile_files
            .iter()
            .map(|p| prefix_path(p, &profile.name))
            .collect();

        // Check profile files against target
        for (idx, relative_path) in profile_files.iter().enumerate() {
            let src = profile.path.join(relative_path);
            let prefixed_path = &prefixed_files[idx];
            let dst = target.join(prefixed_path);

            if !dst.exists() {
                result.files.push(FileInfo {
                    relative_path: prefixed_path.clone(),
                    status: FileStatus::Missing,
                });
                result.missing += 1;
                continue;
            }

            let src_hash = compute_file_hash(&src)?;
            let dst_hash = compute_file_hash(&dst)?;

            if src_hash == dst_hash {
                result.files.push(FileInfo {
                    relative_path: prefixed_path.clone(),
                    status: FileStatus::Unchanged,
                });
                result.unchanged += 1;
            } else {
                result.files.push(FileInfo {
                    relative_path: prefixed_path.clone(),
                    status: FileStatus::Modified,
                });
                result.modified += 1;
            }
        }

        // Check for files in metadata that aren't in profile (user added)
        if let Some(meta) = &metadata {
            for file_path in meta.files.keys() {
                let path = PathBuf::from(file_path);
                if !prefixed_files.contains(&path) {
                    let full_path = target.join(&path);
                    if full_path.exists() {
                        result.files.push(FileInfo {
                            relative_path: path,
                            status: FileStatus::Added,
                        });
                        result.added += 1;
                    }
                }
            }
        }

        result
            .files
            .sort_by(|a, b| a.relative_path.cmp(&b.relative_path));
        Ok(result)
    }

    /// Remove installed profile files
    pub fn remove(
        &self,
        profile: &Profile,
        target: &Path,
        opts: &InstallOptions<'_>,
    ) -> Result<(usize, usize, usize)> {
        if !target.exists() {
            return Ok((0, 0, 0));
        }

        let mut metadata = Metadata::load(target)?.unwrap_or_else(|| Metadata::new(&self.base_dir));
        let diff = self.diff(profile, target, &opts.ignore_config)?;

        // Check for local modifications
        // Skip mergeable JSON files - they are expected to differ due to profile markers
        if !opts.force {
            let modified: Vec<_> = diff
                .files
                .iter()
                .filter(|f| {
                    f.status == FileStatus::Modified
                        && (opts.no_merge || !is_mergeable_json(&f.relative_path))
                })
                .map(|f| f.relative_path.clone())
                .collect();

            if !modified.is_empty() {
                return Err(DotAgentError::LocalModifications { paths: modified });
            }
        }

        let mut removed = 0;
        let mut kept = 0;
        let mut unmerged = 0;

        // First, handle unmerging from JSON files
        if !opts.no_merge {
            if let Some(merged_files) = metadata.get_merged_files(&profile.name).cloned() {
                for (file_path, _json_paths) in merged_files {
                    let dst = target.join(&file_path);
                    if !dst.exists() {
                        continue;
                    }

                    if let Some(result) = unmerge_json_file(&dst, &profile.name)? {
                        if result.changed {
                            if !opts.dry_run {
                                fs::write(&dst, &result.content)?;
                            }
                            if let Some(f) = opts.on_file {
                                f("UNMERGE", &file_path);
                            }
                            unmerged += 1;
                        }
                    }
                }
            }
        }

        for file_info in &diff.files {
            let dst = target.join(&file_info.relative_path);
            let relative_str = file_info.relative_path.to_string_lossy().to_string();

            // Never remove CLAUDE.md
            if relative_str == CLAUDE_MD {
                if let Some(f) = opts.on_file {
                    f("KEEP", &relative_str);
                }
                kept += 1;
                continue;
            }

            // Skip user-added files
            if file_info.status == FileStatus::Added {
                if let Some(f) = opts.on_file {
                    f("KEEP", &relative_str);
                }
                kept += 1;
                continue;
            }

            // Skip missing files
            if file_info.status == FileStatus::Missing {
                continue;
            }

            // Skip merged JSON files (already handled above)
            if !opts.no_merge && is_mergeable_json(&file_info.relative_path) {
                // Only delete if we own this file entirely (not merged)
                if metadata.get_merged(&profile.name, &relative_str).is_some() {
                    continue;
                }
            }

            // Remove file
            if !opts.dry_run && dst.exists() {
                fs::remove_file(&dst)?;
                let meta_key = make_meta_key(&profile.name, &relative_str);
                metadata.remove_file(&meta_key);

                // Remove empty parent directories
                if let Some(parent) = dst.parent() {
                    let _ = remove_empty_dirs(parent, target);
                }
            }

            if let Some(f) = opts.on_file {
                f("DEL", &relative_str);
            }
            removed += 1;
        }

        if !opts.dry_run {
            metadata.remove_profile(&profile.name);
            metadata.remove_merged(&profile.name);
            if metadata.installed.profiles.is_empty()
                && metadata.files.is_empty()
                && metadata.merged.is_empty()
            {
                // Remove metadata file if no profiles left
                let meta_path = target.join(".dot-agent-meta.toml");
                let _ = fs::remove_file(meta_path);
            } else {
                metadata.save(target)?;
            }
        }

        Ok((removed, kept, unmerged))
    }

    /// Upgrade profile files
    pub fn upgrade(
        &self,
        profile: &Profile,
        target: &Path,
        opts: &InstallOptions<'_>,
    ) -> Result<(usize, usize, usize, usize)> {
        // updated, new, skipped, unchanged
        if !target.exists() {
            // Just install everything
            let result = self.install(profile, target, opts)?;
            return Ok((0, result.installed, 0, 0));
        }

        let mut metadata = Metadata::load(target)?.unwrap_or_else(|| Metadata::new(&self.base_dir));
        let mut updated = 0;
        let mut new = 0;
        let mut skipped = 0;
        let mut unchanged = 0;

        let files = profile.list_files_with_config(&opts.ignore_config)?;

        for relative_path in files {
            let src = profile.path.join(&relative_path);
            let prefixed_path = if opts.no_prefix {
                relative_path.clone()
            } else {
                prefix_path(&relative_path, &profile.name)
            };
            let dst = target.join(&prefixed_path);
            let relative_str = prefixed_path.to_string_lossy().to_string();
            let is_claude_md = relative_path.to_string_lossy() == CLAUDE_MD;

            let src_content = fs::read(&src)?;
            let src_hash = compute_hash(&src_content);

            let meta_key = make_meta_key(&profile.name, &relative_str);

            if !dst.exists() {
                // New file
                if !opts.dry_run {
                    if let Some(parent) = dst.parent() {
                        fs::create_dir_all(parent)?;
                    }
                    fs::write(&dst, &src_content)?;
                    metadata.add_file(&meta_key, &src_hash);
                }
                if let Some(f) = opts.on_file {
                    f("NEW", &relative_str);
                }
                new += 1;
                continue;
            }

            let dst_hash = compute_file_hash(&dst)?;

            if src_hash == dst_hash {
                if let Some(f) = opts.on_file {
                    f("OK", &relative_str);
                }
                unchanged += 1;
                continue;
            }

            // CLAUDE.md is never overwritten
            if is_claude_md {
                if let Some(f) = opts.on_file {
                    f("WARN", &relative_str);
                }
                skipped += 1;
                continue;
            }

            // Check if file was modified locally
            let original_hash = metadata.get_file_hash(&meta_key);
            let locally_modified = original_hash.map(|h| h != &dst_hash).unwrap_or(false);

            if locally_modified && !opts.force {
                if let Some(f) = opts.on_file {
                    f("SKIP", &relative_str);
                }
                skipped += 1;
                continue;
            }

            // Update file
            if !opts.dry_run {
                fs::write(&dst, &src_content)?;
                metadata.add_file(&meta_key, &src_hash);
            }
            if let Some(f) = opts.on_file {
                f("UPDATE", &relative_str);
            }
            updated += 1;
        }

        if !opts.dry_run {
            metadata.add_profile(&profile.name);
            metadata.save(target)?;
        }

        Ok((updated, new, skipped, unchanged))
    }
}

fn remove_empty_dirs(dir: &Path, root: &Path) -> std::io::Result<()> {
    if dir == root {
        return Ok(());
    }

    if dir.is_dir() && fs::read_dir(dir)?.next().is_none() {
        fs::remove_dir(dir)?;
        if let Some(parent) = dir.parent() {
            remove_empty_dirs(parent, root)?;
        }
    }

    Ok(())
}

/// Transform relative path to add profile prefix where needed
/// Examples:
///   agents/code-reviewer.md → agents/{profile}-code-reviewer.md
///   skills/my-skill/SKILL.md → skills/{profile}-my-skill/SKILL.md
///   rules/testing.md → rules/{profile}-testing.md
///   commands/profile:cmd.md → commands/profile:cmd.md (already prefixed)
///   CLAUDE.md → CLAUDE.md (no change)
fn prefix_path(relative_path: &Path, profile_name: &str) -> PathBuf {
    let components: Vec<_> = relative_path.components().collect();

    if components.is_empty() {
        return relative_path.to_path_buf();
    }

    // Get first component (top-level directory)
    let first = components[0].as_os_str().to_string_lossy();

    // Check if this is a directory where we prefix files directly
    if PREFIXED_DIRS.contains(&first.as_ref()) && components.len() >= 2 {
        let filename = components[1].as_os_str().to_string_lossy();

        // Skip if already prefixed (contains ':' or starts with profile name)
        if filename.contains(':') || filename.starts_with(&format!("{}-", profile_name)) {
            return relative_path.to_path_buf();
        }

        // agents/code-reviewer.md → agents/{profile}-code-reviewer.md
        let mut result = PathBuf::from(components[0].as_os_str());
        result.push(format!("{}-{}", profile_name, filename));

        // Add remaining components if any
        for comp in &components[2..] {
            result.push(comp.as_os_str());
        }
        return result;
    }

    // Check if this is a directory where we prefix subdirectories
    if PREFIXED_SUBDIRS.contains(&first.as_ref()) && components.len() >= 2 {
        let subdir = components[1].as_os_str().to_string_lossy();

        // Skip if already prefixed
        if subdir.contains(':') || subdir.starts_with(&format!("{}-", profile_name)) {
            return relative_path.to_path_buf();
        }

        // skills/my-skill/SKILL.md → skills/{profile}-my-skill/SKILL.md
        let mut result = PathBuf::from(components[0].as_os_str());
        result.push(format!("{}-{}", profile_name, subdir));

        // Add remaining components
        for comp in &components[2..] {
            result.push(comp.as_os_str());
        }
        return result;
    }

    // No transformation needed
    relative_path.to_path_buf()
}