mise 2026.8.11

use std::path::PathBuf;

use eyre::{Result, bail};
use toml_edit::{DocumentMut, InlineTable, Item, Table, Value};

use crate::config::config_file::ConfigFile;
use crate::config::config_file::mise_toml::MiseToml;
use crate::config::{Config, ConfigPathOptions, resolve_target_config_path};
use crate::dirs;
use crate::file;
use crate::path::PathExt;
use crate::system;
use crate::system::files::{FileMode, FileRequest};
use crate::ui::prompt;

/// Add or update dotfiles in `[dotfiles]`
///
/// If the target is already managed, this updates its source from the live
/// target. Otherwise it creates a `[dotfiles]` entry and seeds the source
/// under `dotfiles.root` unless `--source` is provided.
#[derive(Debug, usage_rs::Args)]
#[usage(verbatim_doc_comment, after_long_help = AFTER_LONG_HELP)]
pub(crate) struct DotfilesAdd {
    /// Targets to add or update
    #[usage(value_name = "TARGET", required = true)]
    pub(super) targets: Vec<String>,

    /// Overwrite existing sources without prompting
    #[usage(long, short)]
    pub(super) force: bool,

    /// Write to the global config
    #[usage(long, short, conflicts = ["local", "path"])]
    pub(super) global: bool,

    /// Write to the local config instead of the global config
    #[usage(long, short, conflicts = ["global", "path"])]
    pub(super) local: bool,

    /// Dotfile mode to write
    #[usage(long, short)]
    pub(super) mode: Option<String>,

    /// Print the config/source updates without writing anything
    #[usage(long, short = 'n')]
    pub(super) dry_run: bool,

    /// Add the entry without applying it
    #[usage(long)]
    pub(super) no_apply: bool,

    /// Write to this config file or directory
    // No `--file` alias here: `-f` on this command is `--force`, and `targets` accepts any
    // string, so `-f <path>` silently adds the config file as a dotfile instead of writing
    // to it. See `mise unset --path` for the commands where the short form is free.
    #[usage(long, short, value_name = "PATH", conflicts = ["global", "local"])]
    pub(super) path: Option<PathBuf>,

    /// Source path to use for a single target
    #[usage(long, short, value_name = "PATH")]
    pub(super) source: Option<PathBuf>,

    /// Skip the confirmation prompt
    #[usage(long, short)]
    pub(super) yes: bool,
}

impl DotfilesAdd {
    /// Validate and capture the requested targets as one transactional update.
    pub(crate) async fn run(self) -> Result<()> {
        if self.source.is_some() && self.targets.len() != 1 {
            bail!("--source can only be used with one target");
        }
        let mode = match self.mode.as_deref() {
            Some(mode) => {
                FileMode::parse(mode).ok_or_else(|| eyre::eyre!("unknown dotfile mode: {mode}"))?
            }
            None => system::files::default_mode(),
        };
        let config = Config::get().await?;
        let managed = system::files::files_from_config(&config)?;
        let config_path = resolve_target_config_path(ConfigPathOptions {
            global: self.global || !self.local,
            path: self.path.clone(),
            env: None,
            cwd: None,
            prefer_toml: true,
            prevent_home_local: true,
        })?;

        let mut planned = vec![];
        let managed_edits = system::edits::edits_from_config(&config)?;
        for target_raw in &self.targets {
            let target = system::files::resolve_target_arg(target_raw)
                .components()
                .collect::<PathBuf>();
            if target.is_relative() {
                bail!("{target_raw}: target must be absolute or start with ~/");
            }
            if managed_edits.iter().any(|req| {
                system::files::matches_target(
                    &req.path,
                    &req.path_raw,
                    std::slice::from_ref(target_raw),
                )
            }) {
                bail!(
                    "{target_raw}: target is already managed by [dotfiles] edits; remove or rename those entries before adding a whole-file dotfile"
                );
            }
            let existing = managed.iter().find(|req| {
                system::files::matches_target(
                    &req.target,
                    &req.target_raw,
                    std::slice::from_ref(target_raw),
                )
            });
            let source = if let Some(req) = existing {
                req.source.clone()
            } else if let Some(source) = &self.source {
                file::replace_path(source)
            } else {
                system::files::implied_source(&target)?
            };
            let write_mode = existing.map(|req| req.mode).unwrap_or(mode);
            if let Some(req) = existing
                && self.mode.is_some()
                && req.mode != mode
            {
                warn!(
                    "dotfiles: {} is already managed with mode {}; --mode {} was ignored",
                    target_raw,
                    req.mode.name(),
                    mode.name()
                );
            }
            let item = PlannedAdd {
                target_raw: normalized_target_raw(&target),
                target,
                source,
                mode: write_mode,
                implied_source: self.source.is_none(),
                explicit_mode: self.mode.is_some(),
                already_managed: existing.cloned(),
            };
            // Wildcard expansion and equivalent spellings can resolve more
            // than one argument to the same config key. The eventual config
            // write is unique by target, so plan its filesystem work once too.
            if !planned
                .iter()
                .any(|planned: &PlannedAdd| planned.target == item.target)
            {
                planned.push(item);
            }
        }

        let mut prospective = managed.clone();
        for item in &planned {
            if let Some(existing) = &item.already_managed
                && let Some(index) = prospective.iter().position(|request| {
                    request.target == existing.target
                        && request.source == existing.source
                        && request.mode == existing.mode
                        && request.origin.config == existing.origin.config
                })
            {
                prospective.remove(index);
            }
            prospective.push(item.validation_request(&config_path));
        }
        system::files::validate_composed_file_footprints(&prospective)?;

        if self.dry_run {
            for item in &planned {
                if item.already_managed.is_none() {
                    miseprintln!(
                        "{}: \"{}\" = {}",
                        config_path.display_user(),
                        item.target_raw,
                        inline_entry(item)
                    );
                }
                if item.target.exists() {
                    miseprintln!(
                        "cp {} {}",
                        item.target.display_user(),
                        item.source.display_user()
                    );
                }
                if !self.no_apply {
                    miseprintln!("{}", describe_apply(item));
                }
            }
            return Ok(());
        }

        let backup_dir = tempfile::tempdir()?;
        let original_config = if config_path.exists() {
            Some(file::read(&config_path)?)
        } else {
            None
        };
        let mut source_backups = vec![];
        let mut target_backups = vec![];
        let mut moved_targets = vec![];
        let mut apply_started = false;
        let result = (|| -> Result<()> {
            let mut accepted = vec![];
            let mut updated_targets = vec![];
            let mut apply_requests = vec![];
            for (index, item) in planned.iter().enumerate() {
                if item.target.exists() && !same_file(&item.target, &item.source) {
                    if item.source.exists()
                        && !self.force
                        && !self.yes
                        && console::user_attended_stderr()
                    {
                        let ok = prompt::confirm(format!(
                            "dotfiles: overwrite source {} from {}?",
                            item.source.display_user(),
                            item.target.display_user()
                        ))?
                        .is_yes();
                        if !ok {
                            info!("dotfiles: skipped {}", item.target_raw);
                            continue;
                        }
                    }
                    source_backups.push((
                        item.source.clone(),
                        backup_path(
                            &item.source,
                            &backup_dir.path().join("sources").join(index.to_string()),
                        )?,
                    ));
                    let move_before_apply = item.mode == FileMode::Symlink
                        || (item.mode == FileMode::SymlinkEach && item.already_managed.is_none());
                    if !self.no_apply && move_before_apply && !item.target.is_symlink() {
                        remove_path(&item.source)?;
                        if let Some(parent) = item.source.parent() {
                            file::create_dir_all(parent)?;
                        }
                        match file::try_rename(&item.target, &item.source) {
                            Ok(()) => {
                                moved_targets.push(MovedTarget {
                                    target: item.target.clone(),
                                    source: item.source.clone(),
                                    recovery: None,
                                });
                            }
                            Err(err) if err.kind() == std::io::ErrorKind::CrossesDevices => {
                                let parent = item.target.parent().ok_or_else(|| {
                                    eyre::eyre!(
                                        "cannot stage target without a parent: {}",
                                        item.target.display_user()
                                    )
                                })?;
                                let recovery = tempfile::Builder::new()
                                    .prefix(".mise-dotfiles-rollback-")
                                    .tempdir_in(parent)?;
                                let recovery_path = recovery.path().join("target");
                                // Keep the original on its own filesystem until
                                // the entire transaction succeeds. This makes
                                // rollback an atomic rename instead of another
                                // potentially failing recursive removal.
                                file::rename(&item.target, &recovery_path)?;
                                moved_targets.push(MovedTarget {
                                    target: item.target.clone(),
                                    source: item.source.clone(),
                                    recovery: Some(recovery),
                                });
                                system::files::copy_path(&recovery_path, &item.source)?;
                            }
                            Err(err) => bail!(
                                "failed rename: {} -> {}: {err}",
                                item.target.display_user(),
                                item.source.display_user()
                            ),
                        }
                        info!(
                            "dotfiles: moved {} to {}",
                            item.target.display_user(),
                            item.source.display_user()
                        );
                    } else {
                        target_backups.push((
                            item.target.clone(),
                            backup_path(
                                &item.target,
                                &backup_dir.path().join("targets").join(index.to_string()),
                            )?,
                        ));
                        system::files::copy_path(&item.target, &item.source)?;
                        info!(
                            "dotfiles: copied {} to {}",
                            item.target.display_user(),
                            item.source.display_user()
                        );
                    }
                } else if !item.source.exists() {
                    target_backups.push((
                        item.target.clone(),
                        backup_path(
                            &item.target,
                            &backup_dir.path().join("targets").join(index.to_string()),
                        )?,
                    ));
                    source_backups.push((item.source.clone(), PathBackup::Missing));
                    if let Some(parent) = item.source.parent() {
                        file::create_dir_all(parent)?;
                    }
                    file::write(&item.source, "")?;
                    info!("dotfiles: created {}", item.source.display_user());
                } else if !self.no_apply {
                    target_backups.push((
                        item.target.clone(),
                        backup_path(
                            &item.target,
                            &backup_dir.path().join("targets").join(index.to_string()),
                        )?,
                    ));
                }
                if item.already_managed.is_some() {
                    updated_targets.push(item.target_raw.as_str());
                }
                accepted.push(item);
                apply_requests.push(item.managed_request(&config_path));
            }

            let apply_opts = system::files::ApplyOpts {
                dry_run: false,
                verbose: false,
                force: false,
                force_hint: "run `mise bootstrap dotfiles apply --force`",
                yes: true,
            };
            let apply_plan = if !self.no_apply && !apply_requests.is_empty() {
                Some(system::files::plan_apply(
                    &config,
                    &apply_requests,
                    &apply_opts,
                )?)
            } else {
                None
            };

            let added_targets = accepted
                .iter()
                .filter(|item| item.already_managed.is_none())
                .collect::<Vec<_>>();
            if !added_targets.is_empty() {
                if !config_path.exists() {
                    let cf = MiseToml::init(&config_path);
                    cf.save()?;
                }
                let raw = file::read_to_string(&config_path)?;
                let mut doc: DocumentMut = raw.parse()?;
                ensure_dotfiles_table(&mut doc);
                for item in &added_targets {
                    write_entry(&mut doc, item);
                }
                sort_dotfiles_table(&mut doc);
                file::write(&config_path, doc.to_string())?;
                for item in &added_targets {
                    info!(
                        "dotfiles: added {} to {}",
                        item.target_raw,
                        config_path.display_user()
                    );
                }
            }
            if !updated_targets.is_empty() {
                info!("dotfiles: updated {}", updated_targets.join(", "));
            }
            if let Some(plan) = apply_plan {
                apply_started = true;
                system::files::execute_apply(plan, &apply_opts)?;
            }
            Ok(())
        })();
        if let Err(err) = result {
            if let Err(rollback_err) = rollback_add(
                &source_backups,
                &target_backups,
                &mut moved_targets,
                apply_started,
                &config_path,
                original_config.as_deref(),
            ) {
                bail!("{err}\ndotfiles: rollback failed: {rollback_err}");
            }
            return Err(err);
        }
        Ok(())
    }
}

#[derive(Debug)]
struct PlannedAdd {
    target_raw: String,
    target: PathBuf,
    source: PathBuf,
    mode: FileMode,
    implied_source: bool,
    explicit_mode: bool,
    already_managed: Option<FileRequest>,
}

impl PlannedAdd {
    /// Build the request that will exist after capture, using the live target
    /// as the future source footprint when add is about to copy or move it.
    fn validation_request(&self, config_path: &std::path::Path) -> FileRequest {
        let mut request = self.managed_request(config_path);
        if self.target.exists() && !same_file(&self.target, &self.source) {
            request.source = self.target.clone();
        } else if !self.source.exists() && self.mode != FileMode::SymlinkEach {
            // Add creates an empty source file in this case. Content mode gives
            // the validator that known leaf shape without touching the disk.
            request.mode = FileMode::Content;
            request.content = Some(String::new());
        }
        request
    }

    /// Preserve an existing declaration's options when an add refreshes its
    /// source; newly added declarations use the command-derived request.
    fn managed_request(&self, config_path: &std::path::Path) -> FileRequest {
        self.already_managed
            .clone()
            .unwrap_or_else(|| self.as_request(config_path))
    }

    /// Convert a newly added target into its configured file request.
    fn as_request(&self, config_path: &std::path::Path) -> FileRequest {
        FileRequest {
            target_raw: self.target_raw.clone(),
            target: self.target.clone(),
            source: self.source.clone(),
            content: None,
            mode: self.mode,
            exclude: vec![],
            base: config_path
                .parent()
                .unwrap_or(std::path::Path::new("."))
                .to_path_buf(),
            origin: crate::system::resources::ResourceOrigin {
                config: config_path.to_path_buf(),
                config_root: crate::config::config_file::config_root::config_root(config_path),
                environment: crate::config::environments_for_config_path(config_path),
                source: Some(self.source.clone()),
            },
        }
    }
}

fn ensure_dotfiles_table(doc: &mut DocumentMut) {
    if !doc.as_table().contains_key("dotfiles") {
        doc["dotfiles"] = Item::Table(Table::new());
    }
}

fn sort_dotfiles_table(doc: &mut DocumentMut) {
    if let Some(table) = doc["dotfiles"].as_table_mut() {
        table.sort_values();
    }
}

fn write_entry(doc: &mut DocumentMut, item: &PlannedAdd) {
    doc["dotfiles"][&item.target_raw] = Item::Value(inline_entry(item));
}

fn inline_entry(item: &PlannedAdd) -> Value {
    let mut table = InlineTable::new();
    if !item.implied_source {
        table.insert(
            "source",
            Value::String(toml_edit::Formatted::new(
                item.source.display_user().to_string(),
            )),
        );
    } else if let Some(req) = &item.already_managed
        && !system::files::source_is_implied(req)
    {
        table.insert(
            "source",
            Value::String(toml_edit::Formatted::new(
                item.source.display_user().to_string(),
            )),
        );
    }
    if item.explicit_mode || item.mode != FileMode::Symlink {
        table.insert(
            "mode",
            Value::String(toml_edit::Formatted::new(item.mode.name().to_string())),
        );
    }
    Value::InlineTable(table)
}

fn normalized_target_raw(target: &std::path::Path) -> String {
    let normalized = target.components().collect::<PathBuf>();
    match normalized.strip_prefix(*dirs::HOME) {
        Ok(rel) if !rel.as_os_str().is_empty() => format!("~/{}", rel.display()),
        Ok(_) => "~".to_string(),
        Err(_) => normalized.to_string_lossy().to_string(),
    }
}

fn describe_apply(item: &PlannedAdd) -> String {
    let source = item.source.display_user();
    let target = item.target.display_user();
    match item.mode {
        FileMode::Symlink => format!("ln -sf {source} {target}"),
        FileMode::SymlinkEach => format!("ln -sf {source}/* into {target}/"),
        FileMode::Copy if item.source.is_dir() => format!("cp -r {source} {target}"),
        FileMode::Copy => format!("cp {source} {target}"),
        FileMode::Template => format!("render {source} -> {target}"),
        FileMode::Content => unreachable!("dotfiles add always captures a source file"),
    }
}

fn same_file(a: &std::path::Path, b: &std::path::Path) -> bool {
    match (a.canonicalize(), b.canonicalize()) {
        (Ok(a), Ok(b)) => a == b,
        _ => false,
    }
}

enum PathBackup {
    Missing,
    Copied(PathBuf),
    Symlink(PathBuf),
}

fn backup_path(path: &std::path::Path, backup: &std::path::Path) -> Result<PathBackup> {
    if path.is_symlink() {
        Ok(PathBackup::Symlink(std::fs::read_link(path)?))
    } else if path.exists() {
        system::files::copy_path(path, backup)?;
        Ok(PathBackup::Copied(backup.to_path_buf()))
    } else {
        Ok(PathBackup::Missing)
    }
}

fn restore_paths(backups: &[(PathBuf, PathBackup)]) -> Result<()> {
    let mut errors = vec![];
    for (path, backup) in backups.iter().rev() {
        let result = (|| -> Result<()> {
            remove_path(path)?;
            match backup {
                PathBackup::Missing => {}
                PathBackup::Copied(backup) => system::files::copy_path(backup, path)?,
                PathBackup::Symlink(target) => {
                    if let Some(parent) = path.parent() {
                        file::create_dir_all(parent)?;
                    }
                    file::make_symlink(target, path)?;
                }
            }
            Ok(())
        })();
        if let Err(err) = result {
            errors.push(format!("{}: {err}", path.display_user()));
        }
    }
    if !errors.is_empty() {
        bail!("{}", errors.join("\n"));
    }
    Ok(())
}

fn remove_path(path: &std::path::Path) -> Result<()> {
    if path.is_symlink() || path.is_file() {
        file::remove_file(path)?;
    } else if path.is_dir() {
        file::remove_all(path)?;
    }
    Ok(())
}

struct MovedTarget {
    target: PathBuf,
    source: PathBuf,
    /// A same-filesystem staging directory used by cross-device captures.
    /// Keeping it alive preserves the original until the transaction commits.
    recovery: Option<tempfile::TempDir>,
}

impl MovedTarget {
    fn keep_recovery(&mut self) -> Option<PathBuf> {
        self.recovery
            .take()
            .map(|recovery| recovery.keep().join("target"))
    }
}

fn rollback_add(
    source_backups: &[(PathBuf, PathBackup)],
    target_backups: &[(PathBuf, PathBackup)],
    moved_targets: &mut [MovedTarget],
    restore_targets: bool,
    config_path: &std::path::Path,
    original_config: Option<&[u8]>,
) -> Result<()> {
    let mut errors = vec![];
    for moved in moved_targets.iter_mut().rev() {
        let result = (|| -> Result<()> {
            remove_path(&moved.target)?;
            if let Some(parent) = moved.target.parent() {
                file::create_dir_all(parent)?;
            }
            if let Some(recovery) = &moved.recovery {
                file::rename(recovery.path().join("target"), &moved.target)
            } else {
                file::move_file(&moved.source, &moved.target)
            }
        })();
        if let Err(err) = result {
            let mut message = format!(
                "moved target {} from {}: {err}",
                moved.target.display_user(),
                moved.source.display_user()
            );
            if let Some(recovery) = moved.keep_recovery() {
                message.push_str(&format!(
                    "; original preserved at {}",
                    recovery.display_user()
                ));
            }
            errors.push(message);
        }
    }
    if restore_targets && let Err(err) = restore_paths(target_backups) {
        errors.push(format!("targets: {err}"));
    }
    if let Err(err) = restore_paths(source_backups) {
        errors.push(format!("sources: {err}"));
    }
    let config_result = match original_config {
        Some(contents) => file::write(config_path, contents),
        None => remove_path(config_path),
    };
    if let Err(err) = config_result {
        errors.push(format!("config: {err}"));
    }
    if !errors.is_empty() {
        bail!("{}", errors.join("\n"));
    }
    Ok(())
}

static AFTER_LONG_HELP: &str = color_print::cstr!(
    r#"<bold><underline>Examples:</underline></bold>

    $ <bold>mise bootstrap dotfiles add ~/.zshrc</bold>
    $ <bold>mise bootstrap dotfiles add --mode copy ~/.config/starship.toml</bold>
    $ <bold>mise bootstrap dotfiles add --source dotfiles/gitconfig ~/.gitconfig</bold>
"#
);

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn rollback_uses_staged_cross_device_original() {
        let root = tempfile::tempdir().unwrap();
        let target = root.path().join("target");
        let source = root.path().join("source");
        let source_backup = root.path().join("source-backup");
        let recovery = tempfile::tempdir_in(root.path()).unwrap();
        let recovery_target = recovery.path().join("target");

        file::create_dir_all(&recovery_target).unwrap();
        file::write(recovery_target.join("file"), "live target").unwrap();
        file::create_dir_all(&source).unwrap();
        file::write(source.join("file"), "captured copy").unwrap();
        file::create_dir_all(&source_backup).unwrap();
        file::write(source_backup.join("file"), "old source").unwrap();

        let mut moved_targets = vec![MovedTarget {
            target: target.clone(),
            source: source.clone(),
            recovery: Some(recovery),
        }];
        rollback_add(
            &[(source.clone(), PathBackup::Copied(source_backup))],
            &[],
            &mut moved_targets,
            false,
            &root.path().join("config.toml"),
            None,
        )
        .unwrap();

        assert_eq!(
            file::read_to_string(target.join("file")).unwrap(),
            "live target"
        );
        assert_eq!(
            file::read_to_string(source.join("file")).unwrap(),
            "old source"
        );
    }

    #[test]
    fn failed_rollback_can_keep_staged_original() {
        let root = tempfile::tempdir().unwrap();
        let recovery = tempfile::tempdir_in(root.path()).unwrap();
        let recovery_target = recovery.path().join("target");
        file::write(&recovery_target, "original").unwrap();
        let mut moved = MovedTarget {
            target: root.path().join("target"),
            source: root.path().join("source"),
            recovery: Some(recovery),
        };

        let kept = moved.keep_recovery().unwrap();
        drop(moved);

        assert_eq!(file::read_to_string(kept).unwrap(), "original");
    }
}