Skip to main content

yui/
cmd.rs

1//! Command implementations.
2//!
3//! Each `Command` variant in `cli.rs` calls one of these. Currently
4//! implemented: `apply`, `init`, `doctor`. The rest are `todo!()`.
5
6use std::fmt::Write as _;
7
8use anyhow::{Context as _, Result};
9use camino::{Utf8Path, Utf8PathBuf};
10use tera::Context as TeraContext;
11use tracing::{info, warn};
12
13use crate::config::{self, Config, HookPhase, IconsMode, MountStrategy};
14use crate::hook::{self, HookOutcome};
15use crate::icons::Icons;
16use crate::link::{self, EffectiveDirMode, EffectiveFileMode, resolve_dir_mode, resolve_file_mode};
17use crate::marker::{self, MarkerSpec};
18use crate::mount::{self, ResolvedMount};
19use crate::render::{self, RenderReport};
20use crate::template;
21use crate::vars::YuiVars;
22use crate::{absorb, backup, paths};
23
24// NOTE: `owo_colors::OwoColorize` is intentionally NOT imported at module
25// scope — its blanket impl shadows inherent methods of unrelated types
26// (e.g. `ignore::WalkBuilder::hidden(bool)` collides with
27// `OwoColorize::hidden(&self)`). Each print function imports the trait
28// locally with `use owo_colors::OwoColorize as _;`.
29
30pub fn init(source: Option<Utf8PathBuf>, _git_hooks: bool) -> Result<()> {
31    let dir = match source {
32        Some(s) => absolutize(&s)?,
33        None => current_dir_utf8()?,
34    };
35    std::fs::create_dir_all(&dir)?;
36    let config_path = dir.join("config.toml");
37    if config_path.exists() {
38        anyhow::bail!("config.toml already exists at {config_path}");
39    }
40    std::fs::write(&config_path, SKELETON_CONFIG)?;
41    let gitignore_path = dir.join(".gitignore");
42    if !gitignore_path.exists() {
43        std::fs::write(&gitignore_path, SKELETON_GITIGNORE)?;
44    }
45    info!("initialized yui source repo at {dir}");
46    info!("created: {config_path}");
47    info!("next: edit config.toml, then run `yui apply`");
48    Ok(())
49}
50
51pub fn apply(source: Option<Utf8PathBuf>, dry_run: bool) -> Result<()> {
52    let source = resolve_source(source)?;
53    let yui = YuiVars::detect(&source);
54    let config = config::load(&source, &yui)?;
55    // Load `.yuiignore` once and thread through render + walk so the
56    // matcher isn't re-built per-flow.
57    let yuiignore = paths::load_yuiignore(&source)?;
58
59    let mut engine = template::Engine::new();
60    let tera_ctx = template::template_context(&yui, &config.vars);
61
62    // 0. Pre-apply hooks (before render / link). Bail on hook failure so
63    //    apply doesn't proceed past a broken bootstrap.
64    hook::run_phase(
65        &config,
66        &source,
67        &yui,
68        &mut engine,
69        &tera_ctx,
70        HookPhase::Pre,
71        dry_run,
72    )?;
73
74    // 1. Render templates first so the link walk picks up rendered files.
75    let render_report = render::render_all(&source, &config, &yui, &yuiignore, dry_run)?;
76    log_render_report(&render_report);
77    if render_report.has_drift() {
78        anyhow::bail!(
79            "render drift detected ({} file(s)); reflect target edits back into the .tera before re-running apply",
80            render_report.diverged.len()
81        );
82    }
83
84    // 2. Resolve mounts and link.
85    let mounts = mount::resolve(
86        &config.mount.entry,
87        config.mount.default_strategy,
88        &mut engine,
89        &tera_ctx,
90    )?;
91
92    let backup_root = source.join(&config.backup.dir);
93    let ctx = ApplyCtx {
94        config: &config,
95        source: &source,
96        yuiignore: &yuiignore,
97        file_mode: resolve_file_mode(config.link.file_mode),
98        dir_mode: resolve_dir_mode(config.link.dir_mode),
99        backup_root: &backup_root,
100        dry_run,
101    };
102
103    info!("source: {source}");
104    info!("modes: file={:?} dir={:?}", ctx.file_mode, ctx.dir_mode);
105    if dry_run {
106        info!("dry-run: nothing will be written");
107    }
108
109    for m in &mounts {
110        info!("mount: {} → {}", m.src, m.dst);
111        process_mount(&source, m, &ctx, &mut engine, &tera_ctx)?;
112    }
113
114    // 3. Post-apply hooks (after every link is in place).
115    hook::run_phase(
116        &config,
117        &source,
118        &yui,
119        &mut engine,
120        &tera_ctx,
121        HookPhase::Post,
122        dry_run,
123    )?;
124    Ok(())
125}
126
127fn log_render_report(r: &RenderReport) {
128    if !r.written.is_empty() {
129        info!("rendered {} new file(s)", r.written.len());
130    }
131    if !r.unchanged.is_empty() {
132        info!("rendered {} file(s) unchanged", r.unchanged.len());
133    }
134    if !r.skipped_when_false.is_empty() {
135        info!(
136            "skipped {} template(s) (when=false)",
137            r.skipped_when_false.len()
138        );
139    }
140    for d in &r.diverged {
141        warn!("rendered file diverged from template: {d}");
142    }
143}
144
145/// Bundle of immutable settings threaded through the apply walk.
146struct ApplyCtx<'a> {
147    config: &'a Config,
148    /// Source repo root — needed for git-clean checks during absorb and
149    /// for resolving paths inside `is_ignored` against `.yuiignore`.
150    source: &'a Utf8Path,
151    /// Patterns from `$source/.yuiignore`. Empty matcher when the file
152    /// is absent.
153    yuiignore: &'a ignore::gitignore::Gitignore,
154    file_mode: EffectiveFileMode,
155    dir_mode: EffectiveDirMode,
156    backup_root: &'a Utf8Path,
157    dry_run: bool,
158}
159
160/// Show the resolved src→dst mappings for the current source repo.
161///
162/// By default only entries whose `when` matches the current host are shown
163/// (`active`). With `--all`, inactive entries are included with a dim row
164/// and the `when` condition that excluded them.
165pub fn list(
166    source: Option<Utf8PathBuf>,
167    all: bool,
168    icons_override: Option<IconsMode>,
169    no_color: bool,
170) -> Result<()> {
171    let source = resolve_source(source)?;
172    let yui = YuiVars::detect(&source);
173    let config = config::load(&source, &yui)?;
174
175    let icons_mode = icons_override.unwrap_or(config.ui.icons);
176    let icons = Icons::for_mode(icons_mode);
177    let color = !no_color && supports_color_stdout();
178
179    let items = collect_list_items(&source, &config, &yui)?;
180    let displayed: Vec<&ListItem> = if all {
181        items.iter().collect()
182    } else {
183        items.iter().filter(|i| i.active).collect()
184    };
185
186    print_list_table(&displayed, icons, color);
187
188    let total = items.len();
189    let active = items.iter().filter(|i| i.active).count();
190    let inactive = total - active;
191    println!();
192    if all {
193        println!("  {total} entries · {active} active · {inactive} inactive");
194    } else {
195        println!(
196            "  {} of {} entries shown ({} inactive hidden — use --all)",
197            active, total, inactive
198        );
199    }
200    Ok(())
201}
202
203#[derive(Debug)]
204struct ListItem {
205    src: Utf8PathBuf,
206    dst: String,
207    when: Option<String>,
208    active: bool,
209}
210
211fn collect_list_items(source: &Utf8Path, config: &Config, yui: &YuiVars) -> Result<Vec<ListItem>> {
212    let mut engine = template::Engine::new();
213    let tera_ctx = template::template_context(yui, &config.vars);
214    let yuiignore = paths::load_yuiignore(source)?;
215    let mut items = Vec::new();
216
217    // 1. config.toml [[mount.entry]] entries
218    for entry in &config.mount.entry {
219        let active = match &entry.when {
220            None => true,
221            Some(w) => template::eval_truthy(w, &mut engine, &tera_ctx)?,
222        };
223        let dst = engine
224            .render(&entry.dst, &tera_ctx)
225            .map(|s| paths::expand_tilde(s.trim()).to_string())
226            .unwrap_or_else(|_| entry.dst.clone());
227        items.push(ListItem {
228            src: entry.src.clone(),
229            dst,
230            when: entry.when.clone(),
231            active,
232        });
233    }
234
235    // 2. .yuilink overrides under source
236    let walker = paths::source_walker(source).build();
237    let marker_filename = &config.mount.marker_filename;
238    for entry in walker {
239        let entry = match entry {
240            Ok(e) => e,
241            Err(_) => continue,
242        };
243        if !entry.file_type().map(|t| t.is_file()).unwrap_or(false) {
244            continue;
245        }
246        if entry.path().file_name().and_then(|n| n.to_str()) != Some(marker_filename.as_str()) {
247            continue;
248        }
249        let dir = match entry.path().parent() {
250            Some(d) => d,
251            None => continue,
252        };
253        let dir_utf8 = match Utf8PathBuf::from_path_buf(dir.to_path_buf()) {
254            Ok(p) => p,
255            Err(_) => continue,
256        };
257        // .yuiignore filter — markers inside ignored subtrees are skipped.
258        if paths::is_ignored(&yuiignore, source, &dir_utf8, true) {
259            continue;
260        }
261        let spec = match marker::read_spec(&dir_utf8, marker_filename)? {
262            Some(s) => s,
263            None => continue,
264        };
265        let MarkerSpec::Explicit { links } = spec else {
266            continue; // PassThrough markers are already implied by mount entry
267        };
268        let rel = dir_utf8
269            .strip_prefix(source)
270            .map(Utf8PathBuf::from)
271            .unwrap_or(dir_utf8);
272        for link in &links {
273            let active = match &link.when {
274                None => true,
275                Some(w) => template::eval_truthy(w, &mut engine, &tera_ctx)?,
276            };
277            let dst = engine
278                .render(&link.dst, &tera_ctx)
279                .map(|s| paths::expand_tilde(s.trim()).to_string())
280                .unwrap_or_else(|_| link.dst.clone());
281            // File-level entry (`[[link]] src = "<filename>"`) targets a
282            // single file inside the marker dir; show that file path
283            // instead of the bare dir so `yui list` makes the scope
284            // obvious at a glance.
285            let src_display = match &link.src {
286                Some(filename) => rel.join(filename),
287                None => rel.clone(),
288            };
289            items.push(ListItem {
290                src: src_display,
291                dst,
292                when: link.when.clone(),
293                active,
294            });
295        }
296    }
297
298    items.sort_by(|a, b| a.src.cmp(&b.src).then_with(|| a.dst.cmp(&b.dst)));
299    Ok(items)
300}
301
302fn supports_color_stdout() -> bool {
303    use std::io::IsTerminal;
304    std::io::stdout().is_terminal() && std::env::var_os("NO_COLOR").is_none()
305}
306
307fn print_list_table(items: &[&ListItem], icons: Icons, color: bool) {
308    let src_w = items
309        .iter()
310        .map(|i| i.src.as_str().chars().count())
311        .max()
312        .unwrap_or(0)
313        .max("SRC".len());
314    let dst_w = items
315        .iter()
316        .map(|i| i.dst.chars().count())
317        .max()
318        .unwrap_or(0)
319        .max("DST".len());
320
321    let status_w = "STATUS".len();
322    let arrow_w = icons.arrow.chars().count();
323
324    // Header
325    print_header(status_w, src_w, arrow_w, dst_w, color);
326
327    // Separator
328    let sep = render_separator(icons.sep, status_w, src_w, arrow_w, dst_w);
329    if color {
330        use owo_colors::OwoColorize as _;
331        println!("{}", sep.dimmed());
332    } else {
333        println!("{sep}");
334    }
335
336    // Rows
337    for item in items {
338        print_row(item, icons, status_w, src_w, arrow_w, dst_w, color);
339    }
340}
341
342fn print_header(status_w: usize, src_w: usize, arrow_w: usize, dst_w: usize, color: bool) {
343    use owo_colors::OwoColorize as _;
344    let mut line = String::new();
345    let _ = write!(
346        &mut line,
347        "  {:<status_w$}  {:<src_w$}  {:<arrow_w$}  {:<dst_w$}  WHEN",
348        "STATUS", "SRC", "", "DST"
349    );
350    if color {
351        println!("{}", line.bold());
352    } else {
353        println!("{line}");
354    }
355}
356
357fn render_separator(
358    sep_ch: char,
359    status_w: usize,
360    src_w: usize,
361    arrow_w: usize,
362    dst_w: usize,
363) -> String {
364    let bar = |n: usize| sep_ch.to_string().repeat(n);
365    format!(
366        "  {}  {}  {}  {}  {}",
367        bar(status_w),
368        bar(src_w),
369        bar(arrow_w),
370        bar(dst_w),
371        bar("WHEN".len())
372    )
373}
374
375fn print_row(
376    item: &ListItem,
377    icons: Icons,
378    status_w: usize,
379    src_w: usize,
380    arrow_w: usize,
381    dst_w: usize,
382    color: bool,
383) {
384    use owo_colors::OwoColorize as _;
385    let status = if item.active {
386        icons.active
387    } else {
388        icons.inactive
389    };
390    let when_str = item
391        .when
392        .as_deref()
393        .map(strip_braces)
394        .unwrap_or_else(|| "(always)".to_string());
395
396    // Normalize backslashes to forward slashes for cross-platform display.
397    let src_display = item.src.as_str().replace('\\', "/");
398    let src = src_display.as_str();
399    let dst = &item.dst;
400    let arrow = icons.arrow;
401
402    // Pad each cell to its column width FIRST, then apply color. Doing it
403    // the other way round lets ANSI escape codes count as printable chars
404    // in `format!("{:<w$}")`, which silently breaks alignment when colors
405    // are enabled (caught in PR #11 review).
406    let cell_status = format!("{:<status_w$}", status);
407    let cell_src = format!("{:<src_w$}", src);
408    let cell_arrow = format!("{:<arrow_w$}", arrow);
409    let cell_dst = format!("{:<dst_w$}", dst);
410
411    if !color {
412        println!("  {cell_status}  {cell_src}  {cell_arrow}  {cell_dst}  {when_str}");
413        return;
414    }
415
416    if item.active {
417        println!(
418            "  {}  {}  {}  {}  {}",
419            cell_status.green(),
420            cell_src.cyan(),
421            cell_arrow.dimmed(),
422            cell_dst.green(),
423            when_str.dimmed()
424        );
425    } else {
426        println!(
427            "  {}  {}  {}  {}  {}",
428            cell_status.red().dimmed(),
429            cell_src.dimmed(),
430            cell_arrow.dimmed(),
431            cell_dst.dimmed(),
432            when_str.dimmed()
433        );
434    }
435}
436
437/// Strip the outer `{{ ... }}` Tera braces from a `when` expression for
438/// display purposes (shorter line, easier to read at a glance).
439fn strip_braces(expr: &str) -> String {
440    let trimmed = expr.trim();
441    if let Some(inner) = trimmed
442        .strip_prefix("{{")
443        .and_then(|s| s.strip_suffix("}}"))
444    {
445        inner.trim().to_string()
446    } else {
447        trimmed.to_string()
448    }
449}
450
451pub fn render(source: Option<Utf8PathBuf>, check: bool, dry_run: bool) -> Result<()> {
452    let source = resolve_source(source)?;
453    let yui = YuiVars::detect(&source);
454    let config = config::load(&source, &yui)?;
455    let yuiignore = paths::load_yuiignore(&source)?;
456    // --check is a stricter dry-run: never writes, exits non-zero on drift.
457    let report = render::render_all(&source, &config, &yui, &yuiignore, dry_run || check)?;
458    log_render_report(&report);
459    if check && report.has_drift() {
460        anyhow::bail!("render drift detected ({} file(s))", report.diverged.len());
461    }
462    Ok(())
463}
464
465pub fn link(source: Option<Utf8PathBuf>, dry_run: bool) -> Result<()> {
466    // For now `link` and `apply` do the same thing (no render/absorb yet).
467    apply(source, dry_run)
468}
469
470pub fn unlink(source: Option<Utf8PathBuf>, paths_arg: Vec<Utf8PathBuf>) -> Result<()> {
471    let _source = resolve_source(source)?;
472    if paths_arg.is_empty() {
473        anyhow::bail!("yui unlink: provide at least one target path");
474    }
475    for p in paths_arg {
476        let abs = absolutize(&p)?;
477        info!("unlink: {abs}");
478        link::unlink(&abs)?;
479    }
480    Ok(())
481}
482
483/// Show every src→dst pair's drift state against the current host.
484///
485/// Walks each `[[mount.entry]]`'s source tree, honoring `.yuilink`
486/// markers (PassThrough = single dir-level link, Override = one or more
487/// custom dsts), classifies each pair via [`crate::absorb::classify`],
488/// and additionally surfaces any **render drift** — rendered files
489/// whose content has diverged from what the matching `.tera` template
490/// would produce now (i.e. the user edited the rendered file in place
491/// without reflecting the change back into the template).
492///
493/// Exits non-zero (via `anyhow::bail!`) when anything diverges, so
494/// `yui status && …` can gate workflows on a clean tree.
495pub fn status(
496    source: Option<Utf8PathBuf>,
497    icons_override: Option<IconsMode>,
498    no_color: bool,
499) -> Result<()> {
500    let source = resolve_source(source)?;
501    let yui = YuiVars::detect(&source);
502    let config = config::load(&source, &yui)?;
503
504    let mut engine = template::Engine::new();
505    let tera_ctx = template::template_context(&yui, &config.vars);
506    let mounts = mount::resolve(
507        &config.mount.entry,
508        config.mount.default_strategy,
509        &mut engine,
510        &tera_ctx,
511    )?;
512
513    let icons_mode = icons_override.unwrap_or(config.ui.icons);
514    let icons = Icons::for_mode(icons_mode);
515    let color = !no_color && supports_color_stdout();
516
517    let mut report: Vec<StatusItem> = Vec::new();
518    // Load `.yuiignore` once and reuse for both render-drift detection
519    // and the link-drift walk below.
520    let yuiignore = paths::load_yuiignore(&source)?;
521
522    // 1. Template drift — render in dry-run mode and surface anything
523    //    whose rendered counterpart on disk no longer matches.
524    let render_report =
525        render::render_all(&source, &config, &yui, &yuiignore, /* dry_run */ true)?;
526    for rendered in &render_report.diverged {
527        // `diverged` holds the rendered path; the template lives at
528        // `<rendered>.tera`. Show the .tera as src so it's clear which
529        // file the user needs to update.
530        let tera_path = Utf8PathBuf::from(format!("{rendered}.tera"));
531        report.push(StatusItem {
532            src: relative_for_display(&source, &tera_path),
533            dst: rendered.clone(),
534            state: StatusState::RenderDrift,
535        });
536    }
537
538    // 2. Link drift — classify each src→dst pair under every mount.
539    for m in &mounts {
540        let src_root = source.join(&m.src);
541        if !src_root.is_dir() {
542            warn!("mount src missing: {src_root}");
543            continue;
544        }
545        classify_walk(
546            &src_root,
547            &m.dst,
548            &config,
549            m.strategy,
550            &mut engine,
551            &tera_ctx,
552            &source,
553            &yuiignore,
554            &mut report,
555        )?;
556    }
557
558    report.sort_by(|a, b| a.src.cmp(&b.src).then_with(|| a.dst.cmp(&b.dst)));
559
560    print_status_table(&report, icons, color);
561
562    let drift = report.iter().filter(|r| !r.state.is_in_sync()).count();
563
564    println!();
565    let total = report.len();
566    let in_sync = total - drift;
567    if drift == 0 {
568        println!("  {total} entries · all in sync");
569        Ok(())
570    } else {
571        println!("  {total} entries · {in_sync} in sync · {drift} diverged");
572        anyhow::bail!("status: {drift} entries diverged from source")
573    }
574}
575
576#[derive(Debug)]
577struct StatusItem {
578    /// Path under the source tree (display only).
579    src: Utf8PathBuf,
580    /// Resolved target path (or rendered output path for `RenderDrift`).
581    dst: Utf8PathBuf,
582    state: StatusState,
583}
584
585#[derive(Debug, Clone, Copy)]
586enum StatusState {
587    Link(absorb::AbsorbDecision),
588    /// Rendered output diverges from current `.tera` template — user
589    /// edited the rendered file directly without updating the template.
590    RenderDrift,
591}
592
593impl StatusState {
594    fn is_in_sync(self) -> bool {
595        matches!(self, Self::Link(absorb::AbsorbDecision::InSync))
596    }
597}
598
599#[allow(clippy::too_many_arguments)]
600fn classify_walk(
601    src_dir: &Utf8Path,
602    dst_dir: &Utf8Path,
603    config: &Config,
604    strategy: MountStrategy,
605    engine: &mut template::Engine,
606    tera_ctx: &TeraContext,
607    source_root: &Utf8Path,
608    yuiignore: &ignore::gitignore::Gitignore,
609    report: &mut Vec<StatusItem>,
610) -> Result<()> {
611    classify_walk_inner(
612        src_dir,
613        dst_dir,
614        config,
615        strategy,
616        engine,
617        tera_ctx,
618        source_root,
619        yuiignore,
620        report,
621        false,
622    )
623}
624
625#[allow(clippy::too_many_arguments)]
626fn classify_walk_inner(
627    src_dir: &Utf8Path,
628    dst_dir: &Utf8Path,
629    config: &Config,
630    strategy: MountStrategy,
631    engine: &mut template::Engine,
632    tera_ctx: &TeraContext,
633    source_root: &Utf8Path,
634    yuiignore: &ignore::gitignore::Gitignore,
635    report: &mut Vec<StatusItem>,
636    parent_covered: bool,
637) -> Result<()> {
638    if paths::is_ignored(yuiignore, source_root, src_dir, /* is_dir */ true) {
639        return Ok(());
640    }
641
642    let marker_filename = &config.mount.marker_filename;
643    let mut covered = parent_covered;
644
645    if strategy == MountStrategy::Marker {
646        match marker::read_spec(src_dir, marker_filename)? {
647            None => {}
648            Some(MarkerSpec::PassThrough) => {
649                let decision = absorb::classify(src_dir, dst_dir)?;
650                report.push(StatusItem {
651                    src: relative_for_display(source_root, src_dir),
652                    dst: dst_dir.to_path_buf(),
653                    state: StatusState::Link(decision),
654                });
655                covered = true;
656            }
657            Some(MarkerSpec::Explicit { links }) => {
658                let mut emitted_dir_link = false;
659                for link in &links {
660                    if let Some(when) = &link.when {
661                        if !template::eval_truthy(when, engine, tera_ctx)? {
662                            continue;
663                        }
664                    }
665                    let dst_str = engine.render(&link.dst, tera_ctx)?;
666                    let dst = paths::expand_tilde(dst_str.trim());
667                    if let Some(filename) = &link.src {
668                        let file_src = src_dir.join(filename);
669                        if !file_src.is_file() {
670                            anyhow::bail!(
671                                "marker at {src_dir}: [[link]] src={filename:?} \
672                                 not found"
673                            );
674                        }
675                        let decision = absorb::classify(&file_src, &dst)?;
676                        report.push(StatusItem {
677                            src: relative_for_display(source_root, &file_src),
678                            dst,
679                            state: StatusState::Link(decision),
680                        });
681                    } else {
682                        let decision = absorb::classify(src_dir, &dst)?;
683                        report.push(StatusItem {
684                            src: relative_for_display(source_root, src_dir),
685                            dst,
686                            state: StatusState::Link(decision),
687                        });
688                        emitted_dir_link = true;
689                    }
690                }
691                if emitted_dir_link {
692                    covered = true;
693                }
694            }
695        }
696    }
697
698    for entry in std::fs::read_dir(src_dir)? {
699        let entry = entry?;
700        let name_os = entry.file_name();
701        let Some(name) = name_os.to_str() else {
702            continue;
703        };
704        if name == marker_filename || name.ends_with(".tera") {
705            continue;
706        }
707        let src_path = src_dir.join(name);
708        let dst_path = dst_dir.join(name);
709        let ft = entry.file_type()?;
710        if paths::is_ignored(yuiignore, source_root, &src_path, ft.is_dir()) {
711            continue;
712        }
713        if ft.is_dir() {
714            classify_walk_inner(
715                &src_path,
716                &dst_path,
717                config,
718                strategy,
719                engine,
720                tera_ctx,
721                source_root,
722                yuiignore,
723                report,
724                covered,
725            )?;
726        } else if ft.is_file() && !covered {
727            let decision = absorb::classify(&src_path, &dst_path)?;
728            report.push(StatusItem {
729                src: relative_for_display(source_root, &src_path),
730                dst: dst_path,
731                state: StatusState::Link(decision),
732            });
733        }
734    }
735    Ok(())
736}
737
738fn relative_for_display(source_root: &Utf8Path, p: &Utf8Path) -> Utf8PathBuf {
739    p.strip_prefix(source_root)
740        .map(Utf8PathBuf::from)
741        .unwrap_or_else(|_| p.to_path_buf())
742}
743
744fn print_status_table(items: &[StatusItem], icons: Icons, color: bool) {
745    let src_w = items
746        .iter()
747        .map(|i| i.src.as_str().chars().count())
748        .max()
749        .unwrap_or(0)
750        .max("SRC".len());
751    let dst_w = items
752        .iter()
753        .map(|i| i.dst.as_str().chars().count())
754        .max()
755        .unwrap_or(0)
756        .max("DST".len());
757    // STATE column = icon (1ch) + space + longest label
758    let state_label_w = items
759        .iter()
760        .map(|i| state_label(i.state).len())
761        .max()
762        .unwrap_or(0)
763        .max("STATE".len() - 2); // "STATE" header takes 5 chars; the icon prefix accounts for 2
764    let state_w = state_label_w + 2; // " " + label
765
766    print_status_header(state_w, src_w, dst_w, color);
767    let sep = render_status_separator(icons.sep, state_w, src_w, dst_w, icons.arrow);
768    if color {
769        use owo_colors::OwoColorize as _;
770        println!("{}", sep.dimmed());
771    } else {
772        println!("{sep}");
773    }
774    for item in items {
775        print_status_row(item, icons, state_w, src_w, dst_w, color);
776    }
777}
778
779fn state_label(s: StatusState) -> &'static str {
780    use absorb::AbsorbDecision::*;
781    match s {
782        StatusState::Link(InSync) => "in-sync",
783        StatusState::Link(RelinkOnly) => "relink",
784        StatusState::Link(AutoAbsorb) => "drift (auto)",
785        StatusState::Link(NeedsConfirm) => "drift (anomaly)",
786        StatusState::Link(Restore) => "missing",
787        StatusState::RenderDrift => "render drift",
788    }
789}
790
791fn state_icon(s: StatusState, icons: Icons) -> &'static str {
792    use absorb::AbsorbDecision::*;
793    match s {
794        StatusState::Link(InSync) => icons.ok,
795        StatusState::Link(RelinkOnly) => icons.warn,
796        StatusState::Link(AutoAbsorb) => icons.warn,
797        StatusState::Link(NeedsConfirm) => icons.error,
798        StatusState::Link(Restore) => icons.info,
799        StatusState::RenderDrift => icons.error,
800    }
801}
802
803fn print_status_header(state_w: usize, src_w: usize, dst_w: usize, color: bool) {
804    use owo_colors::OwoColorize as _;
805    // STATE is the only column with data above; "WHEN" intentionally omitted
806    // since status only shows mounts that are already active on this host.
807    let line = format!(
808        "  {:<state_w$}  {:<src_w$}     {:<dst_w$}",
809        "STATE", "SRC", "DST"
810    );
811    if color {
812        println!("{}", line.bold());
813    } else {
814        println!("{line}");
815    }
816}
817
818fn render_status_separator(
819    sep_ch: char,
820    state_w: usize,
821    src_w: usize,
822    dst_w: usize,
823    arrow: &str,
824) -> String {
825    let bar = |n: usize| sep_ch.to_string().repeat(n);
826    format!(
827        "  {}  {}  {}  {}",
828        bar(state_w),
829        bar(src_w),
830        bar(arrow.chars().count()),
831        bar(dst_w)
832    )
833}
834
835fn print_status_row(
836    item: &StatusItem,
837    icons: Icons,
838    state_w: usize,
839    src_w: usize,
840    dst_w: usize,
841    color: bool,
842) {
843    use owo_colors::OwoColorize as _;
844    let icon = state_icon(item.state, icons);
845    let label = state_label(item.state);
846    let state_text = format!("{icon} {label}");
847    let src_display = item.src.as_str().replace('\\', "/");
848    let dst_display = item.dst.as_str().replace('\\', "/");
849    let arrow = icons.arrow;
850
851    let cell_state = format!("{:<state_w$}", state_text);
852    let cell_src = format!("{:<src_w$}", src_display);
853    let cell_dst = format!("{:<dst_w$}", dst_display);
854
855    if !color {
856        println!("  {cell_state}  {cell_src}  {arrow}  {cell_dst}");
857        return;
858    }
859
860    use absorb::AbsorbDecision::*;
861    let state_colored = match item.state {
862        StatusState::Link(InSync) => cell_state.green().to_string(),
863        StatusState::Link(RelinkOnly) | StatusState::Link(AutoAbsorb) => {
864            cell_state.yellow().to_string()
865        }
866        StatusState::Link(NeedsConfirm) => cell_state.red().to_string(),
867        StatusState::Link(Restore) => cell_state.cyan().to_string(),
868        StatusState::RenderDrift => cell_state.red().to_string(),
869    };
870    let src_colored = cell_src.cyan().to_string();
871    let arrow_colored = arrow.dimmed().to_string();
872    let dst_colored = cell_dst.dimmed().to_string();
873    println!("  {state_colored}  {src_colored}  {arrow_colored}  {dst_colored}");
874}
875
876/// Manually absorb a single target file back into source.
877///
878/// Used when `apply` has skipped an anomaly (`[absorb] on_anomaly = "skip"`
879/// or non-TTY ask) but the user has decided that target is right. Bypasses
880/// policy + git-clean checks: this is an explicit user request.
881///
882/// Walks `[[mount.entry]]` and `.yuilink` overrides to find which source
883/// path "owns" the given target. Errors loudly if no mount claims it.
884pub fn absorb(source: Option<Utf8PathBuf>, target: Utf8PathBuf, dry_run: bool) -> Result<()> {
885    let source = resolve_source(source)?;
886    let target = absolutize(&target)?;
887    let yui = YuiVars::detect(&source);
888    let config = config::load(&source, &yui)?;
889
890    let mut engine = template::Engine::new();
891    let tera_ctx = template::template_context(&yui, &config.vars);
892    // Single load — the matcher is shared with both find_source_for_target
893    // and the eventual ApplyCtx below.
894    let yuiignore = paths::load_yuiignore(&source)?;
895
896    let src_path = match find_source_for_target(
897        &source,
898        &config,
899        &target,
900        &mut engine,
901        &tera_ctx,
902        &yuiignore,
903    )? {
904        Some(s) => s,
905        None => anyhow::bail!(
906            "no mount entry / .yuilink override claims target {target}; \
907                 pass a path inside a known dst"
908        ),
909    };
910
911    info!("source for {target}: {src_path}");
912
913    if dry_run {
914        info!("[dry-run] would absorb {target} → {src_path}");
915        return Ok(());
916    }
917
918    let backup_root = source.join(&config.backup.dir);
919    let ctx = ApplyCtx {
920        config: &config,
921        source: &source,
922        yuiignore: &yuiignore,
923        file_mode: resolve_file_mode(config.link.file_mode),
924        dir_mode: resolve_dir_mode(config.link.dir_mode),
925        backup_root: &backup_root,
926        dry_run: false,
927    };
928
929    // Manual absorb is an explicit user request — bypass `auto`,
930    // `require_clean_git`, and `on_anomaly` policy entirely.
931    absorb_target_into_source(&src_path, &target, &ctx)
932}
933
934/// Walk mount entries + `.yuilink` Override markers to find the source
935/// file/dir that the given target maps back to. Returns `None` when no
936/// mount or marker claims the path.
937fn find_source_for_target(
938    source: &Utf8Path,
939    config: &Config,
940    target: &Utf8Path,
941    engine: &mut template::Engine,
942    tera_ctx: &TeraContext,
943    yuiignore: &ignore::gitignore::Gitignore,
944) -> Result<Option<Utf8PathBuf>> {
945    // 1. Mount entries — render dst, see if target is inside it.
946    for entry in &config.mount.entry {
947        if let Some(when) = &entry.when {
948            if !template::eval_truthy(when, engine, tera_ctx)? {
949                continue;
950            }
951        }
952        let dst_str = engine.render(&entry.dst, tera_ctx)?;
953        let dst_root = paths::expand_tilde(dst_str.trim());
954        if let Ok(rel) = target.strip_prefix(&dst_root) {
955            let candidate = source.join(&entry.src).join(rel);
956            // Honor `.yuiignore` even on manual absorb — if you've
957            // ignored a path, you've explicitly opted out of yui's
958            // managing it.
959            if paths::is_ignored(yuiignore, source, &candidate, candidate.is_dir()) {
960                continue;
961            }
962            return Ok(Some(candidate));
963        }
964    }
965
966    // 2. `.yuilink` Override markers — walk source, parse, render each
967    //    `[[link]] dst`, see if target is the rendered dst (or nested
968    //    inside a junction'd dir). Skips `.yui/` (backup mirrors etc.).
969    let walker = paths::source_walker(source).build();
970    let marker_filename = &config.mount.marker_filename;
971    for ent in walker {
972        let ent = match ent {
973            Ok(e) => e,
974            Err(_) => continue,
975        };
976        if !ent.file_type().map(|t| t.is_file()).unwrap_or(false) {
977            continue;
978        }
979        if ent.path().file_name().and_then(|n| n.to_str()) != Some(marker_filename.as_str()) {
980            continue;
981        }
982        let dir = match ent.path().parent() {
983            Some(d) => d,
984            None => continue,
985        };
986        let dir_utf8 = match Utf8PathBuf::from_path_buf(dir.to_path_buf()) {
987            Ok(p) => p,
988            Err(_) => continue,
989        };
990        if paths::is_ignored(yuiignore, source, &dir_utf8, true) {
991            continue;
992        }
993        let spec = match marker::read_spec(&dir_utf8, marker_filename)? {
994            Some(s) => s,
995            None => continue,
996        };
997        let MarkerSpec::Explicit { links } = spec else {
998            continue;
999        };
1000        for link in &links {
1001            if let Some(when) = &link.when {
1002                if !template::eval_truthy(when, engine, tera_ctx)? {
1003                    continue;
1004                }
1005            }
1006            let dst_str = engine.render(&link.dst, tera_ctx)?;
1007            let dst = paths::expand_tilde(dst_str.trim());
1008            // File-level entry: dst points at a single file, so a match
1009            // resolves directly to `<marker-dir>/<src filename>`. Mirror
1010            // the existence check that apply / status do so a missing
1011            // sibling produces the same clear message regardless of
1012            // entry point — consistent with the `marker at … src=… not
1013            // found` shape users already see from those flows.
1014            if let Some(filename) = &link.src {
1015                let file_src = dir_utf8.join(filename);
1016                if !file_src.is_file() {
1017                    anyhow::bail!(
1018                        "marker at {dir_utf8}: [[link]] src={filename:?} \
1019                         not found"
1020                    );
1021                }
1022                if target == dst {
1023                    return Ok(Some(file_src));
1024                }
1025                continue;
1026            }
1027            if target == dst {
1028                return Ok(Some(dir_utf8));
1029            }
1030            if let Ok(rel) = target.strip_prefix(&dst) {
1031                return Ok(Some(dir_utf8.join(rel)));
1032            }
1033        }
1034    }
1035
1036    Ok(None)
1037}
1038
1039pub fn doctor(source: Option<Utf8PathBuf>) -> Result<()> {
1040    let yui = YuiVars::detect(Utf8Path::new("."));
1041    println!("yui doctor");
1042    println!("==========");
1043    println!("os:    {}", yui.os);
1044    println!("arch:  {}", yui.arch);
1045    println!("user:  {}", yui.user);
1046    println!("host:  {}", yui.host);
1047    match resolve_source(source) {
1048        Ok(s) => {
1049            println!("source: {s}");
1050            // Probe: try loading config
1051            match config::load(&s, &yui) {
1052                Ok(cfg) => println!(
1053                    "config: ok ({} mount entries, {} render rules)",
1054                    cfg.mount.entry.len(),
1055                    cfg.render.rule.len()
1056                ),
1057                Err(e) => println!("config: ERROR — {e}"),
1058            }
1059        }
1060        Err(e) => println!("source: NOT FOUND — {e}"),
1061    }
1062    println!();
1063    println!("link mode (auto resolves to):");
1064    if cfg!(windows) {
1065        println!("  files: hardlink");
1066        println!("  dirs:  junction");
1067    } else {
1068        println!("  files: symlink");
1069        println!("  dirs:  symlink");
1070    }
1071    Ok(())
1072}
1073
1074pub fn gc_backup(_source: Option<Utf8PathBuf>, _older_than: Option<String>) -> Result<()> {
1075    todo!("yui gc-backup — clean up old backups")
1076}
1077
1078/// `yui hooks list` — show every configured hook + its last-run state.
1079pub fn hooks_list(source: Option<Utf8PathBuf>) -> Result<()> {
1080    let source = resolve_source(source)?;
1081    let yui = YuiVars::detect(&source);
1082    let config = config::load(&source, &yui)?;
1083    let state = hook::State::load(&source)?;
1084
1085    if config.hook.is_empty() {
1086        println!("(no [[hook]] entries in config)");
1087        return Ok(());
1088    }
1089
1090    for h in &config.hook {
1091        let phase = match h.phase {
1092            HookPhase::Pre => "pre",
1093            HookPhase::Post => "post",
1094        };
1095        let when_run = match h.when_run {
1096            config::WhenRun::Once => "once",
1097            config::WhenRun::Onchange => "onchange",
1098            config::WhenRun::Every => "every",
1099        };
1100        let last = state
1101            .hooks
1102            .get(&h.name)
1103            .and_then(|s| s.last_run_at.as_deref())
1104            .unwrap_or("(never)");
1105        println!(
1106            "{name:<20}  phase={phase:<4}  when_run={when_run:<8}  last_run_at={last}",
1107            name = h.name,
1108        );
1109        if let Some(when) = &h.when {
1110            println!("                       when = {when}");
1111        }
1112        println!("                       script = {}", h.script);
1113        println!(
1114            "                       command = {} {}",
1115            h.command,
1116            h.args.join(" ")
1117        );
1118    }
1119    Ok(())
1120}
1121
1122/// `yui hooks run [<name>] [--force]` — run a single hook (or every
1123/// hook) on demand. `--force` bypasses the `when_run` state check;
1124/// the `when` filter (`yui.os == 'macos'` etc.) is always honored.
1125pub fn hooks_run(source: Option<Utf8PathBuf>, name: Option<String>, force: bool) -> Result<()> {
1126    let source = resolve_source(source)?;
1127    let yui = YuiVars::detect(&source);
1128    let config = config::load(&source, &yui)?;
1129    let mut engine = template::Engine::new();
1130    let tera_ctx = template::template_context(&yui, &config.vars);
1131
1132    let targets: Vec<&config::HookConfig> = match &name {
1133        Some(want) => {
1134            let m = config
1135                .hook
1136                .iter()
1137                .find(|h| &h.name == want)
1138                .ok_or_else(|| {
1139                    anyhow::anyhow!(
1140                        "no [[hook]] named {want:?}; run `yui hooks list` to see available names"
1141                    )
1142                })?;
1143            vec![m]
1144        }
1145        None => config.hook.iter().collect(),
1146    };
1147
1148    let mut state = hook::State::load(&source)?;
1149    for h in targets {
1150        let outcome = hook::run_hook(
1151            h,
1152            &source,
1153            &yui,
1154            &config.vars,
1155            &mut engine,
1156            &tera_ctx,
1157            &mut state,
1158            /* dry_run */ false,
1159            force,
1160        )?;
1161        let label = match outcome {
1162            HookOutcome::Ran => "ran",
1163            HookOutcome::SkippedOnce => "skipped (once: already ran)",
1164            HookOutcome::SkippedUnchanged => "skipped (onchange: hash matches)",
1165            HookOutcome::SkippedWhenFalse => "skipped (when=false)",
1166            HookOutcome::DryRun => "would run (dry-run)",
1167        };
1168        info!("hook[{}]: {label}", h.name);
1169        if outcome == HookOutcome::Ran {
1170            state.save(&source)?;
1171        }
1172    }
1173    Ok(())
1174}
1175
1176// ---------------------------------------------------------------------------
1177// internals
1178// ---------------------------------------------------------------------------
1179
1180fn process_mount(
1181    source: &Utf8Path,
1182    m: &ResolvedMount,
1183    ctx: &ApplyCtx<'_>,
1184    engine: &mut template::Engine,
1185    tera_ctx: &TeraContext,
1186) -> Result<()> {
1187    let src_root = source.join(&m.src);
1188    if !src_root.is_dir() {
1189        warn!("mount src missing: {src_root}");
1190        return Ok(());
1191    }
1192    walk_and_link(&src_root, &m.dst, ctx, m.strategy, engine, tera_ctx, false)
1193}
1194
1195#[allow(clippy::too_many_arguments)]
1196fn walk_and_link(
1197    src_dir: &Utf8Path,
1198    dst_dir: &Utf8Path,
1199    ctx: &ApplyCtx<'_>,
1200    strategy: MountStrategy,
1201    engine: &mut template::Engine,
1202    tera_ctx: &TeraContext,
1203    parent_covered: bool,
1204) -> Result<()> {
1205    // `.yuiignore` short-circuit — entire subtrees that match are skipped
1206    // without even reading their marker / iterating their children.
1207    if paths::is_ignored(ctx.yuiignore, ctx.source, src_dir, /* is_dir */ true) {
1208        return Ok(());
1209    }
1210
1211    let marker_filename = &ctx.config.mount.marker_filename;
1212    let mut covered = parent_covered;
1213
1214    if strategy == MountStrategy::Marker {
1215        match marker::read_spec(src_dir, marker_filename)? {
1216            None => {} // no marker — fall through to recursive walk
1217            Some(MarkerSpec::PassThrough) => {
1218                // Empty marker = junction this dir at the natural
1219                // mount-derived dst. Subsequent recursion keeps going so
1220                // descendant markers can layer on extra dsts.
1221                link_dir_with_backup(src_dir, dst_dir, ctx)?;
1222                covered = true;
1223            }
1224            Some(MarkerSpec::Explicit { links }) => {
1225                let mut emitted_dir_link = false;
1226                let mut emitted_any = false;
1227                for link in &links {
1228                    // Nested ifs (not let-chains) so the crate's MSRV
1229                    // (rust-version = "1.85") stays buildable.
1230                    if let Some(when) = &link.when {
1231                        if !template::eval_truthy(when, engine, tera_ctx)? {
1232                            continue;
1233                        }
1234                    }
1235                    let dst_str = engine.render(&link.dst, tera_ctx)?;
1236                    let dst = paths::expand_tilde(dst_str.trim());
1237                    if let Some(filename) = &link.src {
1238                        let file_src = src_dir.join(filename);
1239                        if !file_src.is_file() {
1240                            anyhow::bail!(
1241                                "marker at {src_dir}: [[link]] src={filename:?} \
1242                                 not found"
1243                            );
1244                        }
1245                        link_file_with_backup(&file_src, &dst, ctx)?;
1246                    } else {
1247                        link_dir_with_backup(src_dir, &dst, ctx)?;
1248                        emitted_dir_link = true;
1249                    }
1250                    emitted_any = true;
1251                }
1252                if !emitted_any {
1253                    // v0.6+ semantics: with no active links, the walker
1254                    // still descends and per-file defaults still apply.
1255                    // Phrase it so users don't read "skipping" as
1256                    // "subtree blocked" (the v0.5 behaviour).
1257                    info!(
1258                        "marker at {src_dir} had no active links \
1259                         — falling back to defaults"
1260                    );
1261                }
1262                if emitted_dir_link {
1263                    covered = true;
1264                }
1265            }
1266        }
1267    }
1268
1269    for entry in std::fs::read_dir(src_dir)? {
1270        let entry = entry?;
1271        let name_os = entry.file_name();
1272        let Some(name) = name_os.to_str() else {
1273            continue;
1274        };
1275        if name == marker_filename {
1276            continue;
1277        }
1278        if name.ends_with(".tera") {
1279            // Templates are handled by the render flow before linking.
1280            continue;
1281        }
1282        let src_path = src_dir.join(name);
1283        let dst_path = dst_dir.join(name);
1284        let ft = entry.file_type()?;
1285
1286        if paths::is_ignored(ctx.yuiignore, ctx.source, &src_path, ft.is_dir()) {
1287            continue;
1288        }
1289
1290        if ft.is_dir() {
1291            walk_and_link(
1292                &src_path, &dst_path, ctx, strategy, engine, tera_ctx, covered,
1293            )?;
1294        } else if ft.is_file() {
1295            // If an ancestor (or this dir itself) created a dir-level
1296            // junction, the file is already accessible via that junction
1297            // — emitting another per-file link would just duplicate work
1298            // (and on Windows might land at a path that's already
1299            // hard-linked through the parent).
1300            if !covered {
1301                link_file_with_backup(&src_path, &dst_path, ctx)?;
1302            }
1303        }
1304    }
1305    Ok(())
1306}
1307
1308fn link_file_with_backup(src: &Utf8Path, dst: &Utf8Path, ctx: &ApplyCtx<'_>) -> Result<()> {
1309    use absorb::AbsorbDecision::*;
1310
1311    let decision = absorb::classify(src, dst)?;
1312
1313    if ctx.dry_run {
1314        info!("[dry-run] {decision:?}: {src} → {dst}");
1315        return Ok(());
1316    }
1317
1318    match decision {
1319        InSync => {
1320            // Link is intact (same inode/file-id). Nothing to do.
1321            Ok(())
1322        }
1323        Restore => {
1324            info!("link: {src} → {dst}");
1325            link::link_file(src, dst, ctx.file_mode)?;
1326            Ok(())
1327        }
1328        RelinkOnly => {
1329            // Same content, different inode (e.g. hardlink broken by an
1330            // editor's atomic save). Re-link without touching source.
1331            info!("relink: {src} → {dst}");
1332            link::unlink(dst)?;
1333            link::link_file(src, dst, ctx.file_mode)?;
1334            Ok(())
1335        }
1336        AutoAbsorb => {
1337            // Target newer + content differs: target wins, source updated.
1338            // Honor `[absorb] auto` (kill-switch) and `require_clean_git`.
1339            if !ctx.config.absorb.auto {
1340                return handle_anomaly(
1341                    src,
1342                    dst,
1343                    ctx,
1344                    "absorb.auto = false; treating divergence as anomaly",
1345                );
1346            }
1347            if ctx.config.absorb.require_clean_git && !source_repo_is_clean(ctx.source) {
1348                return handle_anomaly(
1349                    src,
1350                    dst,
1351                    ctx,
1352                    "source repo is dirty; deferring auto-absorb",
1353                );
1354            }
1355            absorb_target_into_source(src, dst, ctx)
1356        }
1357        NeedsConfirm => handle_anomaly(
1358            src,
1359            dst,
1360            ctx,
1361            "anomaly: source equals/newer than target but content differs",
1362        ),
1363    }
1364}
1365
1366/// Back up the source-side file, copy the target's content into source,
1367/// then re-link so the freshly-updated source is what target points at.
1368/// "Target wins" — yui's core philosophy.
1369fn absorb_target_into_source(src: &Utf8Path, dst: &Utf8Path, ctx: &ApplyCtx<'_>) -> Result<()> {
1370    info!("absorb: {dst} → {src}");
1371    backup_existing(src, ctx.backup_root, /* is_dir */ false)?;
1372    std::fs::copy(dst, src)?;
1373    link::unlink(dst)?;
1374    link::link_file(src, dst, ctx.file_mode)?;
1375    Ok(())
1376}
1377
1378/// Decide what to do for an anomaly (NeedsConfirm or AutoAbsorb that was
1379/// escalated by `auto = false` / dirty git). Per `[absorb] on_anomaly`:
1380///   - `skip`  → log warning, leave target alone
1381///   - `force` → behave like AutoAbsorb (target wins)
1382///   - `ask`   → on a TTY, show diff + prompt. Off-TTY, downgrade to skip.
1383fn handle_anomaly(src: &Utf8Path, dst: &Utf8Path, ctx: &ApplyCtx<'_>, reason: &str) -> Result<()> {
1384    use crate::config::AnomalyAction::*;
1385    match ctx.config.absorb.on_anomaly {
1386        Skip => {
1387            warn!("anomaly skip: {dst} ({reason})");
1388            Ok(())
1389        }
1390        Force => {
1391            warn!("anomaly force: {dst} ({reason}) — absorbing target into source");
1392            absorb_target_into_source(src, dst, ctx)
1393        }
1394        Ask => {
1395            use std::io::IsTerminal;
1396            if std::io::stdin().is_terminal() && std::io::stdout().is_terminal() {
1397                if prompt_absorb_with_diff(src, dst, reason)? {
1398                    absorb_target_into_source(src, dst, ctx)
1399                } else {
1400                    warn!("anomaly skipped by user: {dst}");
1401                    Ok(())
1402                }
1403            } else {
1404                warn!("anomaly skip (non-TTY ask mode): {dst} ({reason})");
1405                Ok(())
1406            }
1407        }
1408    }
1409}
1410
1411fn prompt_absorb_with_diff(src: &Utf8Path, dst: &Utf8Path, reason: &str) -> Result<bool> {
1412    use std::io::Write as _;
1413    let src_content = std::fs::read_to_string(src).unwrap_or_default();
1414    let dst_content = std::fs::read_to_string(dst).unwrap_or_default();
1415    eprintln!();
1416    eprintln!("anomaly: {reason}");
1417    eprintln!("  src: {src}");
1418    eprintln!("  dst: {dst}");
1419    eprintln!();
1420    eprintln!("--- diff (- source, + target) ---");
1421    let diff = similar::TextDiff::from_lines(&src_content, &dst_content);
1422    for change in diff.iter_all_changes() {
1423        let sign = match change.tag() {
1424            similar::ChangeTag::Delete => "-",
1425            similar::ChangeTag::Insert => "+",
1426            similar::ChangeTag::Equal => " ",
1427        };
1428        eprint!("{sign}{change}");
1429    }
1430    eprintln!();
1431    eprint!("absorb target into source? [y/N]: ");
1432    // Flush stderr (where the prompt was written) — flushing stdout was a
1433    // bug; on a buffered stderr (rare but possible) the prompt would be
1434    // hidden until after the user typed something. Caught in PR #15
1435    // review (gemini-code-assist).
1436    std::io::stderr().flush().ok();
1437    let mut input = String::new();
1438    std::io::stdin().read_line(&mut input)?;
1439    let answer = input.trim();
1440    Ok(answer.eq_ignore_ascii_case("y") || answer.eq_ignore_ascii_case("yes"))
1441}
1442
1443/// Resilient git-clean check: if `git` isn't available or `source` isn't
1444/// a repo, log a warning and proceed as if clean. We don't want a missing
1445/// `git` to block apply — the require_clean_git knob is a *safety net*,
1446/// not a hard prerequisite.
1447fn source_repo_is_clean(source: &Utf8Path) -> bool {
1448    match crate::git::is_clean(source) {
1449        Ok(b) => b,
1450        Err(e) => {
1451            warn!("git clean check failed at {source}: {e} — treating as clean");
1452            true
1453        }
1454    }
1455}
1456
1457fn link_dir_with_backup(src: &Utf8Path, dst: &Utf8Path, ctx: &ApplyCtx<'_>) -> Result<()> {
1458    use absorb::AbsorbDecision::*;
1459    let decision = absorb::classify(src, dst)?;
1460
1461    if ctx.dry_run {
1462        info!("[dry-run] dir {decision:?}: {src} → {dst}");
1463        return Ok(());
1464    }
1465
1466    match decision {
1467        InSync => Ok(()),
1468        Restore => {
1469            info!("link dir: {src} → {dst}");
1470            link::link_dir(src, dst, ctx.dir_mode)?;
1471            Ok(())
1472        }
1473        RelinkOnly => {
1474            // For dirs the classifier doesn't currently produce
1475            // `RelinkOnly` (only InSync / NeedsConfirm), but handle it
1476            // for symmetry with the file path: contents already match,
1477            // so just swap the target for a junction to source.
1478            info!("relink dir: {src} → {dst}");
1479            remove_dir_link_or_real(dst)?;
1480            link::link_dir(src, dst, ctx.dir_mode)?;
1481            Ok(())
1482        }
1483        AutoAbsorb | NeedsConfirm => {
1484            // Reaching `link_dir_with_backup` means we're acting on a
1485            // `.yuilink` marker (or a `[[mount.entry]]` whose `src` is a
1486            // directory) — the user has explicitly opted into
1487            // "this whole subtree is target-as-truth". A dir-level
1488            // NeedsConfirm here is therefore *not* the same kind of
1489            // anomaly that file-level NeedsConfirm represents (a single
1490            // file the user edited and source got newer); it's just
1491            // "source and target dirs are different inodes" — the
1492            // marker already authorised us to merge.
1493            //
1494            // Per-file content conflicts *inside* the merge are still
1495            // a real concern (target has X, source has X with
1496            // different content). Those are surfaced from inside the
1497            // merge itself — see `merge_dir_target_into_source`'s
1498            // file-level dispatch — so the outer-dir decision falls
1499            // straight through to absorb.
1500            //
1501            // The `auto` / `require_clean_git` knobs still gate, so
1502            // turning them off restores the prompt before any
1503            // whole-dir absorb.
1504            if !ctx.config.absorb.auto {
1505                return handle_anomaly_dir(
1506                    src,
1507                    dst,
1508                    ctx,
1509                    "absorb.auto = false; treating divergence as anomaly",
1510                );
1511            }
1512            if ctx.config.absorb.require_clean_git && !source_repo_is_clean(ctx.source) {
1513                return handle_anomaly_dir(
1514                    src,
1515                    dst,
1516                    ctx,
1517                    "source repo is dirty; deferring auto-absorb",
1518                );
1519            }
1520            absorb_target_dir_into_source(src, dst, ctx)
1521        }
1522    }
1523}
1524
1525/// `link::unlink` with a documented fallback for the chezmoi-migration
1526/// shape: target is a real (non-link) directory packed with files. The
1527/// caller is responsible for ensuring the target's prior content is
1528/// preserved (in `.yui/backup/...` or because we just merged it into
1529/// source) before reaching here.
1530///
1531/// Anything other than the "non-empty regular dir" case — permission
1532/// denied, target gone, target now a junction or symlink — propagates
1533/// rather than being silently coerced into `remove_dir_all`.
1534fn remove_dir_link_or_real(dst: &Utf8Path) -> Result<()> {
1535    if let Err(unlink_err) = link::unlink(dst) {
1536        let meta = std::fs::symlink_metadata(dst)
1537            .with_context(|| format!("stat {dst} after link::unlink failed: {unlink_err}"))?;
1538        let ft = meta.file_type();
1539        if ft.is_dir() && !ft.is_symlink() {
1540            std::fs::remove_dir_all(dst).with_context(|| {
1541                format!(
1542                    "remove_dir_all({dst}) after link::unlink failed: \
1543                     {unlink_err}"
1544                )
1545            })?;
1546        } else {
1547            return Err(unlink_err).with_context(|| format!("unlink({dst}) before relink"));
1548        }
1549    }
1550    Ok(())
1551}
1552
1553/// Recursively merge target's files into source: target wins on file
1554/// conflicts, source-only files are preserved, sub-dirs are created
1555/// in source as needed. Non-regular entries (symlinks / junctions /
1556/// device files) are skipped with a warning — copying their content
1557/// is ill-defined and following them risks looping into target via
1558/// some chain back to source.
1559///
1560/// Mirrors the file-level "AutoAbsorb backs up source, copies target's
1561/// content into source before relinking" semantic for whole dirs.
1562fn merge_dir_target_into_source(
1563    target: &Utf8Path,
1564    source: &Utf8Path,
1565    ctx: &ApplyCtx<'_>,
1566) -> Result<()> {
1567    for entry in std::fs::read_dir(target)? {
1568        let entry = entry?;
1569        let name_os = entry.file_name();
1570        let Some(name) = name_os.to_str() else {
1571            continue;
1572        };
1573        let target_path = target.join(name);
1574        let source_path = source.join(name);
1575        let ft = entry.file_type()?;
1576
1577        if ft.is_dir() && !ft.is_symlink() {
1578            // Target is a real dir. If source has a non-dir entry at
1579            // the same name (regular file, symlink, junction), it
1580            // would block `create_dir_all` and the recursive merge.
1581            // Honor target-wins by clearing the conflicting source
1582            // entry first.
1583            if let Ok(src_meta) = std::fs::symlink_metadata(&source_path) {
1584                let sft = src_meta.file_type();
1585                if !sft.is_dir() || sft.is_symlink() {
1586                    link::unlink(&source_path).with_context(|| {
1587                        format!("remove conflicting source entry before dir merge: {source_path}")
1588                    })?;
1589                }
1590            }
1591            if !source_path.exists() {
1592                std::fs::create_dir_all(&source_path).with_context(|| {
1593                    format!("create_dir_all({source_path}) during target→source merge")
1594                })?;
1595            }
1596            merge_dir_target_into_source(&target_path, &source_path, ctx)?;
1597        } else if ft.is_file() {
1598            // Target is a regular file. Symmetrical handling: if
1599            // source has a directory or symlink at the same name,
1600            // tear it down first so the file copy can land.
1601            if let Ok(src_meta) = std::fs::symlink_metadata(&source_path) {
1602                let sft = src_meta.file_type();
1603                if sft.is_dir() && !sft.is_symlink() {
1604                    remove_dir_link_or_real(&source_path).with_context(|| {
1605                        format!("remove conflicting source dir before file merge: {source_path}")
1606                    })?;
1607                } else if sft.is_symlink() {
1608                    link::unlink(&source_path).with_context(|| {
1609                        format!(
1610                            "remove conflicting source symlink before file merge: {source_path}"
1611                        )
1612                    })?;
1613                }
1614            }
1615            if let Some(parent) = source_path.parent() {
1616                if !parent.exists() {
1617                    std::fs::create_dir_all(parent)?;
1618                }
1619            }
1620            // If both sides are now regular files at the same path, run
1621            // the file-level absorb classifier so this single overlap
1622            // is resolved against `[absorb]` policy (auto / skip /
1623            // force / ask) instead of being silently overwritten. The
1624            // dir-level marker provides consent for the *whole-tree*
1625            // merge, but a per-file content collision where the
1626            // source side is *newer* is still a legitimate anomaly
1627            // worth surfacing.
1628            //
1629            // Source-only files were already preserved by virtue of
1630            // the merge not visiting them. Target-only files (where
1631            // `source_path` doesn't exist) skip the classifier and go
1632            // straight to copy below.
1633            if source_path.is_file() {
1634                merge_resolve_file_conflict(&target_path, &source_path, ctx)?;
1635            } else {
1636                std::fs::copy(&target_path, &source_path)
1637                    .with_context(|| format!("copy({target_path} → {source_path}) during merge"))?;
1638            }
1639        } else {
1640            warn!(
1641                "merge: skipping non-regular entry {target_path} \
1642                 (symlink / junction / special — content not copied)"
1643            );
1644        }
1645    }
1646    Ok(())
1647}
1648
1649/// Per-file conflict resolution inside the dir merge. Both
1650/// `target_path` and `source_path` exist as regular files — run the
1651/// absorb classifier on the pair and route to the matching policy:
1652///
1653/// - `InSync` / `RelinkOnly` → no-op (contents already match)
1654/// - `AutoAbsorb` (target newer + diff) → copy target → source,
1655///   target-wins per the AutoAbsorb contract.
1656/// - `NeedsConfirm` (source newer + diff, the genuine anomaly) →
1657///   `[absorb] on_anomaly` dispatch:
1658///     - `skip` → leave source alone, target's version is dropped
1659///       (after the outer junction, target ends up with source's content)
1660///     - `force` → copy target → source (target wins anyway)
1661///     - `ask` → TTY prompt with diff; downgrade to skip off-TTY
1662fn merge_resolve_file_conflict(
1663    target_path: &Utf8Path,
1664    source_path: &Utf8Path,
1665    ctx: &ApplyCtx<'_>,
1666) -> Result<()> {
1667    use absorb::AbsorbDecision::*;
1668    let decision = absorb::classify(source_path, target_path)?;
1669    match decision {
1670        InSync | RelinkOnly => Ok(()),
1671        AutoAbsorb => {
1672            std::fs::copy(target_path, source_path).with_context(|| {
1673                format!("copy({target_path} → {source_path}) during merge AutoAbsorb")
1674            })?;
1675            Ok(())
1676        }
1677        Restore => {
1678            // `Restore` is the classifier's "target is missing" arm.
1679            // We only enter this function after the merge loop saw
1680            // `target_path` as a regular file in the read_dir
1681            // iteration, and the caller guards on `source_path.is_file()`
1682            // — both exist by construction, so this branch is
1683            // unreachable.
1684            unreachable!(
1685                "merge_resolve_file_conflict reached with both files present, \
1686                 but classify returned Restore (target {target_path} / source {source_path})"
1687            )
1688        }
1689        NeedsConfirm => {
1690            use crate::config::AnomalyAction::*;
1691            match ctx.config.absorb.on_anomaly {
1692                Skip => {
1693                    warn!(
1694                        "merge anomaly skip: {target_path} (source-newer / content drift) \
1695                         — keeping source version, target version dropped"
1696                    );
1697                    Ok(())
1698                }
1699                Force => {
1700                    warn!(
1701                        "merge anomaly force: {target_path} \
1702                         (source-newer / content drift) — overwriting source"
1703                    );
1704                    std::fs::copy(target_path, source_path)?;
1705                    Ok(())
1706                }
1707                Ask => {
1708                    use std::io::IsTerminal;
1709                    if std::io::stdin().is_terminal() && std::io::stdout().is_terminal() {
1710                        if prompt_absorb_with_diff(
1711                            source_path,
1712                            target_path,
1713                            "merge: file content differs and source is newer",
1714                        )? {
1715                            std::fs::copy(target_path, source_path)?;
1716                        } else {
1717                            warn!("merge: kept source version by user choice: {source_path}");
1718                        }
1719                        Ok(())
1720                    } else {
1721                        warn!(
1722                            "merge anomaly skip (non-TTY ask mode): {target_path} \
1723                             — keeping source version"
1724                        );
1725                        Ok(())
1726                    }
1727                }
1728            }
1729        }
1730    }
1731}
1732
1733/// Back up source-side, merge target's content into source (target
1734/// wins on conflict), then replace target with a junction to source.
1735/// "Target wins" — yui's core philosophy, generalised from the file
1736/// path to whole directories so a chezmoi-style migrated `~/.config/`
1737/// keeps every file the user actually had instead of stranding most
1738/// of them in `.yui/backup/...`.
1739fn absorb_target_dir_into_source(src: &Utf8Path, dst: &Utf8Path, ctx: &ApplyCtx<'_>) -> Result<()> {
1740    info!("absorb dir: {dst} → {src}");
1741    backup_existing(src, ctx.backup_root, /* is_dir */ true)?;
1742    merge_dir_target_into_source(dst, src, ctx)?;
1743    // Source now carries every regular file from target. Tear down the
1744    // original target dir and re-expose source via a junction.
1745    remove_dir_link_or_real(dst)?;
1746    link::link_dir(src, dst, ctx.dir_mode)?;
1747    Ok(())
1748}
1749
1750/// Dir-level counterpart to `handle_anomaly`. Same `[absorb] on_anomaly`
1751/// dispatch — `skip` warns and walks away, `force` absorbs anyway,
1752/// `ask` prompts on a TTY (downgraded to skip off-TTY).
1753fn handle_anomaly_dir(
1754    src: &Utf8Path,
1755    dst: &Utf8Path,
1756    ctx: &ApplyCtx<'_>,
1757    reason: &str,
1758) -> Result<()> {
1759    use crate::config::AnomalyAction::*;
1760    match ctx.config.absorb.on_anomaly {
1761        Skip => {
1762            warn!("anomaly skip dir: {dst} ({reason})");
1763            Ok(())
1764        }
1765        Force => {
1766            warn!(
1767                "anomaly force dir: {dst} ({reason}) \
1768                 — absorbing target into source"
1769            );
1770            absorb_target_dir_into_source(src, dst, ctx)
1771        }
1772        Ask => {
1773            use std::io::IsTerminal;
1774            if std::io::stdin().is_terminal() && std::io::stdout().is_terminal() {
1775                eprintln!();
1776                eprintln!("anomaly: {dst}");
1777                eprintln!("  {reason}");
1778                eprintln!("  source: {src}");
1779                eprint!("  absorb target dir into source? (y/N) ");
1780                use std::io::{BufRead as _, Write as _};
1781                std::io::stderr().flush().ok();
1782                let mut buf = String::new();
1783                std::io::stdin().lock().read_line(&mut buf)?;
1784                let answer = buf.trim();
1785                if answer.eq_ignore_ascii_case("y") || answer.eq_ignore_ascii_case("yes") {
1786                    absorb_target_dir_into_source(src, dst, ctx)
1787                } else {
1788                    warn!("anomaly skipped by user: {dst}");
1789                    Ok(())
1790                }
1791            } else {
1792                warn!("anomaly skip (non-TTY ask mode): {dst} ({reason})");
1793                Ok(())
1794            }
1795        }
1796    }
1797}
1798
1799fn backup_existing(target: &Utf8Path, backup_root: &Utf8Path, is_dir: bool) -> Result<()> {
1800    let abs_target = absolutize(target)?;
1801    let ts = backup::current_timestamp("%Y%m%d_%H%M%S%3f")?;
1802    let bp = paths::append_timestamp(&paths::mirror_into_backup(backup_root, &abs_target), &ts);
1803    info!("backup → {bp}");
1804    if is_dir {
1805        backup::backup_dir(target, &bp)?;
1806    } else {
1807        backup::backup_file(target, &bp)?;
1808    }
1809    Ok(())
1810}
1811
1812fn resolve_source(source: Option<Utf8PathBuf>) -> Result<Utf8PathBuf> {
1813    if let Some(s) = source {
1814        return absolutize(&s);
1815    }
1816    if let Ok(s) = std::env::var("YUI_SOURCE") {
1817        return absolutize(Utf8Path::new(&s));
1818    }
1819    let cwd = current_dir_utf8()?;
1820    for ancestor in cwd.ancestors() {
1821        if ancestor.join("config.toml").is_file() {
1822            return Ok(ancestor.to_path_buf());
1823        }
1824    }
1825    if let Some(home) = paths::home_dir() {
1826        for c in ["dotfiles", ".dotfiles", "src/dotfiles"] {
1827            let p = home.join(c);
1828            if p.join("config.toml").is_file() {
1829                return Ok(p);
1830            }
1831        }
1832    }
1833    anyhow::bail!("source repo not found (set --source / $YUI_SOURCE)")
1834}
1835
1836fn absolutize(p: &Utf8Path) -> Result<Utf8PathBuf> {
1837    // Expand `~` first so callers can pass `--source ~/dotfiles` directly.
1838    let expanded = paths::expand_tilde(p.as_str());
1839    if expanded.is_absolute() {
1840        return Ok(expanded);
1841    }
1842    let cwd = current_dir_utf8()?;
1843    Ok(cwd.join(expanded))
1844}
1845
1846fn current_dir_utf8() -> Result<Utf8PathBuf> {
1847    let cwd = std::env::current_dir().context("getting cwd")?;
1848    Utf8PathBuf::from_path_buf(cwd).map_err(|p| anyhow::anyhow!("non-UTF8 cwd: {}", p.display()))
1849}
1850
1851// Note: `home_dir()` lives in `paths.rs` so the tilde-expansion helper and
1852// `resolve_source` share one HOME/USERPROFILE lookup.
1853
1854const SKELETON_CONFIG: &str = r#"# yui config — see https://github.com/yukimemi/yui
1855
1856[vars]
1857# user-defined values; templates can reference these as {{ vars.foo }}
1858
1859# [link]
1860# file_mode = "auto"   # auto | symlink | hardlink
1861# dir_mode  = "auto"   # auto | symlink | junction
1862
1863[mount]
1864default_strategy = "marker"
1865
1866[[mount.entry]]
1867src = "home"
1868# `~` expands to $HOME / $USERPROFILE per OS at apply time, no Tera needed.
1869dst = "~"
1870
1871# [[mount.entry]]
1872# src  = "appdata"
1873# dst  = "{{ env(name='APPDATA') }}"
1874# # NOTE: write `when` as a *bare* expression (no `{{ … }}`) so it survives
1875# # config.toml's whole-file Tera render and shows up cleanly in `yui list`.
1876# when = "yui.os == 'windows'"
1877"#;
1878
1879const SKELETON_GITIGNORE: &str = r#"# yui per-machine state and backups (regenerable, do not commit).
1880# .yui/bin/ is intentionally tracked — it holds your hook scripts.
1881/.yui/state.json
1882/.yui/state.json.tmp
1883/.yui/backup/
1884
1885# >>> yui rendered (auto-managed, do not edit) >>>
1886# <<< yui rendered (auto-managed) <<<
1887
1888# config.local.toml is per-machine; commit a config.local.example.toml instead.
1889config.local.toml
1890"#;
1891
1892#[cfg(test)]
1893mod tests {
1894    use super::*;
1895    use tempfile::TempDir;
1896
1897    fn utf8(p: std::path::PathBuf) -> Utf8PathBuf {
1898        Utf8PathBuf::from_path_buf(p).unwrap()
1899    }
1900
1901    /// Convert a path to a TOML-string-safe form (forward slashes).
1902    fn toml_path(p: &Utf8Path) -> String {
1903        p.as_str().replace('\\', "/")
1904    }
1905
1906    #[test]
1907    fn apply_links_a_raw_file() {
1908        let tmp = TempDir::new().unwrap();
1909        let source = utf8(tmp.path().join("dotfiles"));
1910        let target = utf8(tmp.path().join("target"));
1911        std::fs::create_dir_all(source.join("home")).unwrap();
1912        std::fs::create_dir_all(&target).unwrap();
1913        std::fs::write(source.join("home/.bashrc"), "echo hi\n").unwrap();
1914
1915        let cfg = format!(
1916            r#"
1917[[mount.entry]]
1918src = "home"
1919dst = "{}"
1920"#,
1921            toml_path(&target)
1922        );
1923        std::fs::write(source.join("config.toml"), cfg).unwrap();
1924
1925        apply(Some(source), false).unwrap();
1926
1927        let linked = target.join(".bashrc");
1928        assert!(linked.exists(), "expected {linked} to exist");
1929        assert_eq!(std::fs::read_to_string(&linked).unwrap(), "echo hi\n");
1930    }
1931
1932    #[test]
1933    fn apply_with_marker_links_whole_directory() {
1934        let tmp = TempDir::new().unwrap();
1935        let source = utf8(tmp.path().join("dotfiles"));
1936        let target = utf8(tmp.path().join("target"));
1937        let nvim_src = source.join("home/nvim");
1938        std::fs::create_dir_all(&nvim_src).unwrap();
1939        std::fs::create_dir_all(&target).unwrap();
1940        std::fs::write(nvim_src.join(".yuilink"), "").unwrap();
1941        std::fs::write(nvim_src.join("init.lua"), "-- hi\n").unwrap();
1942        std::fs::write(nvim_src.join("plugins.lua"), "-- plugins\n").unwrap();
1943
1944        let cfg = format!(
1945            r#"
1946[[mount.entry]]
1947src = "home"
1948dst = "{}"
1949"#,
1950            toml_path(&target)
1951        );
1952        std::fs::write(source.join("config.toml"), cfg).unwrap();
1953
1954        apply(Some(source.clone()), false).unwrap();
1955
1956        let nvim_dst = target.join("nvim");
1957        assert!(nvim_dst.exists());
1958        assert_eq!(
1959            std::fs::read_to_string(nvim_dst.join("init.lua")).unwrap(),
1960            "-- hi\n"
1961        );
1962        // Marker file itself shouldn't be visible as a separate link in target;
1963        // however with junction/symlink the whole dir shows up so the marker
1964        // file IS visible inside. That's fine — the marker is informational.
1965    }
1966
1967    #[test]
1968    fn apply_dry_run_does_not_write() {
1969        let tmp = TempDir::new().unwrap();
1970        let source = utf8(tmp.path().join("dotfiles"));
1971        let target = utf8(tmp.path().join("target"));
1972        std::fs::create_dir_all(source.join("home")).unwrap();
1973        std::fs::create_dir_all(&target).unwrap();
1974        std::fs::write(source.join("home/.bashrc"), "echo hi").unwrap();
1975
1976        let cfg = format!(
1977            r#"
1978[[mount.entry]]
1979src = "home"
1980dst = "{}"
1981"#,
1982            toml_path(&target)
1983        );
1984        std::fs::write(source.join("config.toml"), cfg).unwrap();
1985
1986        apply(Some(source), true).unwrap();
1987
1988        assert!(!target.join(".bashrc").exists());
1989    }
1990
1991    #[test]
1992    fn apply_renders_templates_then_links_rendered_outputs() {
1993        let tmp = TempDir::new().unwrap();
1994        let source = utf8(tmp.path().join("dotfiles"));
1995        let target = utf8(tmp.path().join("target"));
1996        std::fs::create_dir_all(source.join("home")).unwrap();
1997        std::fs::create_dir_all(&target).unwrap();
1998        std::fs::write(
1999            source.join("home/.gitconfig.tera"),
2000            "[user]\n  os = {{ yui.os }}\n",
2001        )
2002        .unwrap();
2003        std::fs::write(source.join("home/.bashrc"), "raw").unwrap();
2004
2005        let cfg = format!(
2006            r#"
2007[[mount.entry]]
2008src = "home"
2009dst = "{}"
2010"#,
2011            toml_path(&target)
2012        );
2013        std::fs::write(source.join("config.toml"), cfg).unwrap();
2014
2015        apply(Some(source.clone()), false).unwrap();
2016
2017        // Raw file: linked.
2018        assert!(target.join(".bashrc").exists());
2019        // Template's rendered output: written to source then linked.
2020        assert!(source.join("home/.gitconfig").exists());
2021        assert!(target.join(".gitconfig").exists());
2022        // The .tera file itself is never linked into target.
2023        assert!(!target.join(".gitconfig.tera").exists());
2024        // Rendered file content carries the yui.os substitution.
2025        let linked = std::fs::read_to_string(target.join(".gitconfig")).unwrap();
2026        assert!(linked.contains("os = "));
2027    }
2028
2029    #[test]
2030    fn apply_marker_override_links_to_custom_dst() {
2031        let tmp = TempDir::new().unwrap();
2032        let source = utf8(tmp.path().join("dotfiles"));
2033        let target_a = utf8(tmp.path().join("target_a"));
2034        let target_b = utf8(tmp.path().join("target_b"));
2035        std::fs::create_dir_all(source.join("home/.config/nvim")).unwrap();
2036        std::fs::create_dir_all(&target_a).unwrap();
2037        std::fs::create_dir_all(&target_b).unwrap();
2038        std::fs::write(
2039            source.join("home/.config/nvim/init.lua"),
2040            "-- nvim config\n",
2041        )
2042        .unwrap();
2043
2044        // Marker tells yui to ignore the parent mount's dst for this dir
2045        // and link it to two custom places (the second only if condition matches).
2046        std::fs::write(
2047            source.join("home/.config/nvim/.yuilink"),
2048            format!(
2049                r#"
2050[[link]]
2051dst = "{}/nvim"
2052
2053[[link]]
2054dst = "{}/nvim"
2055when = "{{{{ yui.os == '{}' }}}}"
2056"#,
2057                toml_path(&target_a),
2058                toml_path(&target_b),
2059                std::env::consts::OS
2060            ),
2061        )
2062        .unwrap();
2063
2064        let parent_target = utf8(tmp.path().join("parent_target"));
2065        std::fs::create_dir_all(&parent_target).unwrap();
2066        let cfg = format!(
2067            r#"
2068[[mount.entry]]
2069src = "home"
2070dst = "{}"
2071"#,
2072            toml_path(&parent_target)
2073        );
2074        std::fs::write(source.join("config.toml"), cfg).unwrap();
2075
2076        apply(Some(source.clone()), false).unwrap();
2077
2078        // Both override targets received the link (the second's when matches OS).
2079        assert!(
2080            target_a.join("nvim/init.lua").exists(),
2081            "target_a/nvim/init.lua should be reachable through the link"
2082        );
2083        assert!(
2084            target_b.join("nvim/init.lua").exists(),
2085            "target_b/nvim/init.lua should be reachable through the link"
2086        );
2087        // Parent mount did NOT also link this dir (it would have appeared at
2088        // parent_target/.config/nvim — the marker claims the dir).
2089        assert!(
2090            !parent_target.join(".config/nvim").exists(),
2091            "parent mount should have skipped the marker-claimed sub-dir"
2092        );
2093    }
2094
2095    #[test]
2096    fn apply_marker_inactive_link_falls_through_to_default() {
2097        // v0.6+ semantics: a marker that has only inactive links no
2098        // longer suppresses the parent mount's natural placement. The
2099        // walker keeps descending so per-file defaults still apply.
2100        // (Use `.yuiignore` to actually exclude a subtree.)
2101        let tmp = TempDir::new().unwrap();
2102        let source = utf8(tmp.path().join("dotfiles"));
2103        let target_inactive = utf8(tmp.path().join("inactive"));
2104        let parent_target = utf8(tmp.path().join("parent"));
2105        std::fs::create_dir_all(source.join("home/.config/nvim")).unwrap();
2106        std::fs::create_dir_all(&parent_target).unwrap();
2107        std::fs::write(source.join("home/.config/nvim/init.lua"), "x").unwrap();
2108
2109        // when=false on every link → marker has no active links.
2110        std::fs::write(
2111            source.join("home/.config/nvim/.yuilink"),
2112            format!(
2113                r#"
2114[[link]]
2115dst = "{}/nvim"
2116when = "{{{{ yui.os == 'no-such-os' }}}}"
2117"#,
2118                toml_path(&target_inactive)
2119            ),
2120        )
2121        .unwrap();
2122
2123        let cfg = format!(
2124            r#"
2125[[mount.entry]]
2126src = "home"
2127dst = "{}"
2128"#,
2129            toml_path(&parent_target)
2130        );
2131        std::fs::write(source.join("config.toml"), cfg).unwrap();
2132
2133        apply(Some(source.clone()), false).unwrap();
2134
2135        // Inactive marker target untouched.
2136        assert!(!target_inactive.join("nvim").exists());
2137        // Parent mount's natural placement IS produced — the marker had
2138        // no active dir-level link to claim coverage with.
2139        assert!(parent_target.join(".config/nvim/init.lua").exists());
2140    }
2141
2142    #[test]
2143    fn list_shows_mount_entries_and_marker_overrides() {
2144        let tmp = TempDir::new().unwrap();
2145        let source = utf8(tmp.path().join("dotfiles"));
2146        std::fs::create_dir_all(source.join("home/.config/nvim")).unwrap();
2147        std::fs::write(source.join("home/.config/nvim/init.lua"), "x").unwrap();
2148        std::fs::write(
2149            source.join("home/.config/nvim/.yuilink"),
2150            r#"
2151[[link]]
2152dst = "/custom/nvim"
2153"#,
2154        )
2155        .unwrap();
2156        std::fs::write(
2157            source.join("config.toml"),
2158            r#"
2159[[mount.entry]]
2160src = "home"
2161dst = "/h"
2162"#,
2163        )
2164        .unwrap();
2165
2166        // Just verify it runs without error — output format is covered by
2167        // unit-level helpers below.
2168        list(Some(source), false, None, true).unwrap();
2169    }
2170
2171    #[test]
2172    fn status_reports_in_sync_after_apply() {
2173        let tmp = TempDir::new().unwrap();
2174        let source = utf8(tmp.path().join("dotfiles"));
2175        let target = utf8(tmp.path().join("target"));
2176        std::fs::create_dir_all(source.join("home")).unwrap();
2177        std::fs::create_dir_all(&target).unwrap();
2178        std::fs::write(source.join("home/.bashrc"), "echo hi\n").unwrap();
2179        let cfg = format!(
2180            r#"
2181[[mount.entry]]
2182src = "home"
2183dst = "{}"
2184"#,
2185            toml_path(&target)
2186        );
2187        std::fs::write(source.join("config.toml"), cfg).unwrap();
2188        // First link the target so the link is intact.
2189        apply(Some(source.clone()), false).unwrap();
2190        // status should succeed (everything in-sync).
2191        status(Some(source), None, true).unwrap();
2192    }
2193
2194    #[test]
2195    fn status_reports_template_drift() {
2196        let tmp = TempDir::new().unwrap();
2197        let source = utf8(tmp.path().join("dotfiles"));
2198        let target = utf8(tmp.path().join("target"));
2199        std::fs::create_dir_all(source.join("home")).unwrap();
2200        std::fs::create_dir_all(&target).unwrap();
2201        // Template would render to "fresh" but the rendered file on disk
2202        // says "stale" — simulating a manual edit not reflected back.
2203        std::fs::write(source.join("home/.gitconfig.tera"), "fresh").unwrap();
2204        std::fs::write(source.join("home/.gitconfig"), "stale").unwrap();
2205
2206        let cfg = format!(
2207            r#"
2208[[mount.entry]]
2209src = "home"
2210dst = "{}"
2211"#,
2212            toml_path(&target)
2213        );
2214        std::fs::write(source.join("config.toml"), cfg).unwrap();
2215
2216        let err = status(Some(source), None, true).unwrap_err();
2217        assert!(format!("{err}").contains("diverged"));
2218    }
2219
2220    #[test]
2221    fn status_fails_when_target_missing() {
2222        let tmp = TempDir::new().unwrap();
2223        let source = utf8(tmp.path().join("dotfiles"));
2224        let target = utf8(tmp.path().join("target"));
2225        std::fs::create_dir_all(source.join("home")).unwrap();
2226        std::fs::create_dir_all(&target).unwrap();
2227        std::fs::write(source.join("home/.bashrc"), "echo hi\n").unwrap();
2228        let cfg = format!(
2229            r#"
2230[[mount.entry]]
2231src = "home"
2232dst = "{}"
2233"#,
2234            toml_path(&target)
2235        );
2236        std::fs::write(source.join("config.toml"), cfg).unwrap();
2237        // No apply yet — target/.bashrc doesn't exist.
2238        let err = status(Some(source), None, true).unwrap_err();
2239        assert!(format!("{err}").contains("diverged"));
2240    }
2241
2242    #[test]
2243    fn strip_braces_removes_outer_template_braces() {
2244        assert_eq!(strip_braces("{{ yui.os == 'linux' }}"), "yui.os == 'linux'");
2245        assert_eq!(strip_braces("yui.os == 'linux'"), "yui.os == 'linux'");
2246        assert_eq!(strip_braces("  {{x}}  "), "x");
2247    }
2248
2249    #[test]
2250    fn apply_aborts_on_render_drift() {
2251        let tmp = TempDir::new().unwrap();
2252        let source = utf8(tmp.path().join("dotfiles"));
2253        let target = utf8(tmp.path().join("target"));
2254        std::fs::create_dir_all(source.join("home")).unwrap();
2255        std::fs::create_dir_all(&target).unwrap();
2256        std::fs::write(source.join("home/foo.tera"), "fresh body").unwrap();
2257        std::fs::write(source.join("home/foo"), "manually edited").unwrap();
2258
2259        let cfg = format!(
2260            r#"
2261[[mount.entry]]
2262src = "home"
2263dst = "{}"
2264"#,
2265            toml_path(&target)
2266        );
2267        std::fs::write(source.join("config.toml"), cfg).unwrap();
2268
2269        let err = apply(Some(source.clone()), false).unwrap_err();
2270        assert!(format!("{err}").contains("drift"));
2271        // Existing rendered file untouched.
2272        assert_eq!(
2273            std::fs::read_to_string(source.join("home/foo")).unwrap(),
2274            "manually edited"
2275        );
2276        // Linking aborted — target empty.
2277        assert!(!target.join("foo").exists());
2278    }
2279
2280    #[test]
2281    fn init_creates_skeleton_when_dir_empty() {
2282        let tmp = TempDir::new().unwrap();
2283        let dir = utf8(tmp.path().join("new_dotfiles"));
2284        init(Some(dir.clone()), false).unwrap();
2285        assert!(dir.join("config.toml").is_file());
2286        assert!(dir.join(".gitignore").is_file());
2287    }
2288
2289    #[test]
2290    fn init_refuses_to_overwrite_existing_config() {
2291        let tmp = TempDir::new().unwrap();
2292        let dir = utf8(tmp.path().join("dotfiles"));
2293        std::fs::create_dir_all(&dir).unwrap();
2294        std::fs::write(dir.join("config.toml"), "preexisting").unwrap();
2295        let err = init(Some(dir), false).unwrap_err();
2296        assert!(format!("{err}").contains("already exists"));
2297    }
2298
2299    /// Build a minimal `apply`-able dotfiles tree for absorb tests.
2300    /// Returns (source_dir, target_dir).
2301    fn setup_minimal_dotfiles(tmp: &TempDir) -> (Utf8PathBuf, Utf8PathBuf) {
2302        let source = utf8(tmp.path().join("dotfiles"));
2303        let target = utf8(tmp.path().join("target"));
2304        std::fs::create_dir_all(source.join("home")).unwrap();
2305        std::fs::create_dir_all(&target).unwrap();
2306        let cfg = format!(
2307            r#"
2308[[mount.entry]]
2309src = "home"
2310dst = "{}"
2311"#,
2312            toml_path(&target)
2313        );
2314        std::fs::write(source.join("config.toml"), cfg).unwrap();
2315        (source, target)
2316    }
2317
2318    fn write_with_mtime(path: &Utf8Path, body: &str, when: std::time::SystemTime) {
2319        std::fs::write(path, body).unwrap();
2320        let f = std::fs::OpenOptions::new()
2321            .write(true)
2322            .open(path)
2323            .expect("open writable");
2324        f.set_modified(when).expect("set_modified");
2325    }
2326
2327    #[test]
2328    fn apply_target_newer_absorbs_target_into_source() {
2329        // Target has the user's edit and is mtime-newer than source —
2330        // classifier returns `AutoAbsorb`. yui's "target-as-truth"
2331        // philosophy: target wins, source is updated and backed up.
2332        let tmp = TempDir::new().unwrap();
2333        let (source, target) = setup_minimal_dotfiles(&tmp);
2334
2335        let now = std::time::SystemTime::now();
2336        let past = now - std::time::Duration::from_secs(120);
2337        write_with_mtime(&source.join("home/.bashrc"), "default from repo", past);
2338        // Pre-existing target with user's edit, NEWER mtime.
2339        write_with_mtime(&target.join(".bashrc"), "user's edit", now);
2340
2341        apply(Some(source.clone()), false).unwrap();
2342
2343        // Target's content survives — that's the whole point.
2344        assert_eq!(
2345            std::fs::read_to_string(target.join(".bashrc")).unwrap(),
2346            "user's edit"
2347        );
2348        // Source has been updated to match target.
2349        assert_eq!(
2350            std::fs::read_to_string(source.join("home/.bashrc")).unwrap(),
2351            "user's edit"
2352        );
2353        // Source's previous content lives under .yui/backup.
2354        let backup_root = source.join(".yui/backup");
2355        let mut found_old = false;
2356        for entry in walkdir(&backup_root) {
2357            if let Ok(s) = std::fs::read_to_string(&entry) {
2358                if s == "default from repo" {
2359                    found_old = true;
2360                    break;
2361                }
2362            }
2363        }
2364        assert!(found_old, "expected backup containing 'default from repo'");
2365    }
2366
2367    #[test]
2368    fn apply_in_sync_target_is_a_no_op() {
2369        // After an initial `apply`, running `apply` again classifies as
2370        // `InSync` and does nothing.
2371        let tmp = TempDir::new().unwrap();
2372        let (source, target) = setup_minimal_dotfiles(&tmp);
2373        std::fs::write(source.join("home/.bashrc"), "echo hi\n").unwrap();
2374        apply(Some(source.clone()), false).unwrap();
2375        let backup_root = source.join(".yui/backup");
2376        let backup_count_after_first = walkdir(&backup_root).len();
2377
2378        // Second apply — nothing should change.
2379        apply(Some(source.clone()), false).unwrap();
2380        assert_eq!(
2381            std::fs::read_to_string(target.join(".bashrc")).unwrap(),
2382            "echo hi\n"
2383        );
2384        let backup_count_after_second = walkdir(&backup_root).len();
2385        assert_eq!(
2386            backup_count_after_first, backup_count_after_second,
2387            "second apply on an in-sync tree should not produce backups"
2388        );
2389    }
2390
2391    #[test]
2392    fn apply_skip_policy_leaves_anomaly_alone() {
2393        // Source newer than target + content differs = NeedsConfirm.
2394        // With on_anomaly = "skip", target stays untouched.
2395        let tmp = TempDir::new().unwrap();
2396        let source = utf8(tmp.path().join("dotfiles"));
2397        let target = utf8(tmp.path().join("target"));
2398        std::fs::create_dir_all(source.join("home")).unwrap();
2399        std::fs::create_dir_all(&target).unwrap();
2400        let cfg = format!(
2401            r#"
2402[absorb]
2403on_anomaly = "skip"
2404
2405[[mount.entry]]
2406src = "home"
2407dst = "{}"
2408"#,
2409            toml_path(&target)
2410        );
2411        std::fs::write(source.join("config.toml"), cfg).unwrap();
2412
2413        let now = std::time::SystemTime::now();
2414        let past = now - std::time::Duration::from_secs(120);
2415        write_with_mtime(&target.join(".bashrc"), "user's edit (older)", past);
2416        write_with_mtime(&source.join("home/.bashrc"), "fresh from upstream", now);
2417
2418        apply(Some(source.clone()), false).unwrap();
2419
2420        // Target untouched (skip policy honored).
2421        assert_eq!(
2422            std::fs::read_to_string(target.join(".bashrc")).unwrap(),
2423            "user's edit (older)"
2424        );
2425        // Source untouched too.
2426        assert_eq!(
2427            std::fs::read_to_string(source.join("home/.bashrc")).unwrap(),
2428            "fresh from upstream"
2429        );
2430    }
2431
2432    #[test]
2433    fn apply_force_policy_absorbs_anomaly_anyway() {
2434        // Same anomaly setup, but on_anomaly = "force" → target wins.
2435        let tmp = TempDir::new().unwrap();
2436        let source = utf8(tmp.path().join("dotfiles"));
2437        let target = utf8(tmp.path().join("target"));
2438        std::fs::create_dir_all(source.join("home")).unwrap();
2439        std::fs::create_dir_all(&target).unwrap();
2440        let cfg = format!(
2441            r#"
2442[absorb]
2443on_anomaly = "force"
2444
2445[[mount.entry]]
2446src = "home"
2447dst = "{}"
2448"#,
2449            toml_path(&target)
2450        );
2451        std::fs::write(source.join("config.toml"), cfg).unwrap();
2452
2453        let now = std::time::SystemTime::now();
2454        let past = now - std::time::Duration::from_secs(120);
2455        write_with_mtime(&target.join(".bashrc"), "user's edit (older)", past);
2456        write_with_mtime(&source.join("home/.bashrc"), "fresh from upstream", now);
2457
2458        apply(Some(source.clone()), false).unwrap();
2459
2460        // Target wins despite being mtime-older — force policy.
2461        assert_eq!(
2462            std::fs::read_to_string(target.join(".bashrc")).unwrap(),
2463            "user's edit (older)"
2464        );
2465        assert_eq!(
2466            std::fs::read_to_string(source.join("home/.bashrc")).unwrap(),
2467            "user's edit (older)"
2468        );
2469    }
2470
2471    /// Regression for the Windows-error-145 bug: a `home/.config/.yuilink`
2472    /// (PassThrough) marker pointing at a non-empty regular `~/.config`
2473    /// directory (the typical chezmoi-migrated state, where every file
2474    /// inside is an individual hardlink) used to fail the absorb with
2475    /// `Directory not empty` because `link::unlink` refuses to recurse.
2476    /// After backup we now `remove_dir_all` as a fallback.
2477    ///
2478    /// v0.7+: also exercises the target-wins merge — target's
2479    /// `config.toml` overwrites source's, target's `state.json` lands
2480    /// in source (target was the source of truth), and source-only
2481    /// scaffolding (`.yuilink`) survives the absorb.
2482    #[test]
2483    fn apply_absorbs_non_empty_target_dir_target_wins() {
2484        let tmp = TempDir::new().unwrap();
2485        let source = utf8(tmp.path().join("dotfiles"));
2486        let target = utf8(tmp.path().join("target"));
2487        std::fs::create_dir_all(source.join("home/.config/app")).unwrap();
2488        std::fs::create_dir_all(target.join(".config/app")).unwrap();
2489        // Marker that says "junction this dir at the parent mount's dst"
2490        // — same shape as a typical home/.config/.yuilink.
2491        std::fs::write(source.join("home/.config/.yuilink"), "").unwrap();
2492        std::fs::write(source.join("home/.config/app/config.toml"), "src side").unwrap();
2493        // Source-only scaffolding that the absorb must preserve.
2494        std::fs::write(source.join("home/.config/app/source-only.toml"), "src").unwrap();
2495        // Pre-existing non-empty regular dir at the target — chezmoi /
2496        // any per-file dotfiles flow leaves things in this shape.
2497        std::fs::write(target.join(".config/app/config.toml"), "target side").unwrap();
2498        std::fs::write(target.join(".config/app/state.json"), "{}").unwrap();
2499
2500        let cfg = format!(
2501            r#"
2502[absorb]
2503on_anomaly = "force"
2504
2505[[mount.entry]]
2506src = "home"
2507dst = "{}"
2508"#,
2509            toml_path(&target)
2510        );
2511        std::fs::write(source.join("config.toml"), cfg).unwrap();
2512
2513        // Used to bail with `unlink: ... Directory not empty` here.
2514        apply(Some(source.clone()), false).unwrap();
2515
2516        // Target wins on the conflicting file.
2517        assert_eq!(
2518            std::fs::read_to_string(target.join(".config/app/config.toml")).unwrap(),
2519            "target side"
2520        );
2521        // Target-only file is now reachable via the junction.
2522        assert_eq!(
2523            std::fs::read_to_string(target.join(".config/app/state.json")).unwrap(),
2524            "{}"
2525        );
2526        // Source's pre-merge state was backed up before being overwritten,
2527        // so the original "src side" / `.yuilink` survive in `.yui/backup/`.
2528        let backup_root = source.join(".yui/backup");
2529        let mut backup_files: Vec<String> = Vec::new();
2530        for entry in walkdir(&backup_root) {
2531            if let Some(n) = entry.file_name() {
2532                backup_files.push(n.to_string());
2533            }
2534        }
2535        assert!(
2536            backup_files.iter().any(|f| f == "config.toml"),
2537            "expected source's config.toml to land in the backup tree, got {backup_files:?}"
2538        );
2539        // Source-only scaffolding survives the merge.
2540        assert!(
2541            source.join("home/.config/app/source-only.toml").exists(),
2542            "source-only file should survive a target-wins merge"
2543        );
2544        // Source picked up target-only state.json via the merge.
2545        assert!(
2546            source.join("home/.config/app/state.json").exists(),
2547            "target-only state.json should be merged into source"
2548        );
2549    }
2550
2551    /// v0.7+: `home/.config/.yuilink` is the user's explicit
2552    /// "this whole subtree is target-as-truth" declaration. A
2553    /// dir-level NeedsConfirm at the marker root is therefore not a
2554    /// real anomaly — the marker is consent. Default `[absorb]` (ask
2555    /// + require_clean_git) should still absorb, no prompt.
2556    #[test]
2557    fn marker_dir_absorbs_with_default_ask_policy() {
2558        let tmp = TempDir::new().unwrap();
2559        let source = utf8(tmp.path().join("dotfiles"));
2560        let target = utf8(tmp.path().join("target"));
2561        std::fs::create_dir_all(source.join("home/.config")).unwrap();
2562        std::fs::create_dir_all(target.join(".config/gh")).unwrap();
2563        // Marker — user opts the whole .config dir into target-as-truth.
2564        std::fs::write(source.join("home/.config/.yuilink"), "").unwrap();
2565        // gh exists only on the target side (no entry in source).
2566        std::fs::write(target.join(".config/gh/hosts.yml"), "oauth_token: x\n").unwrap();
2567
2568        // Default [absorb] (no override) — `on_anomaly = "ask"`,
2569        // `auto = true`, `require_clean_git = true`. Pre-v0.7 this
2570        // would have been routed through the ask prompt at dir level.
2571        let cfg = format!(
2572            r#"
2573[[mount.entry]]
2574src = "home"
2575dst = "{}"
2576"#,
2577            toml_path(&target)
2578        );
2579        std::fs::write(source.join("config.toml"), cfg).unwrap();
2580
2581        // Even with default `ask`, the marker-rooted absorb proceeds.
2582        // Test would hang on a stdin prompt if dir-level still treated
2583        // this as an anomaly.
2584        apply(Some(source.clone()), false).unwrap();
2585
2586        // Target-only file is now reachable through the junction and
2587        // recorded in source.
2588        assert!(target.join(".config/gh/hosts.yml").exists());
2589        assert!(source.join("home/.config/gh/hosts.yml").exists());
2590    }
2591
2592    /// File↔dir collisions during merge. Honor target-wins: if source
2593    /// has a regular file at a path where target has a dir, the file
2594    /// gets removed and the dir is created. Symmetrical for the
2595    /// inverse case. Without the conflict-clearing the merge would
2596    /// fail with `not a directory` / `path exists` deep in the recursion.
2597    #[test]
2598    fn merge_handles_file_vs_dir_collisions_target_wins() {
2599        let tmp = TempDir::new().unwrap();
2600        let source = utf8(tmp.path().join("dotfiles"));
2601        let target = utf8(tmp.path().join("target"));
2602        std::fs::create_dir_all(source.join("home/.config/foo")).unwrap();
2603        std::fs::create_dir_all(target.join(".config")).unwrap();
2604        std::fs::write(source.join("home/.config/.yuilink"), "").unwrap();
2605
2606        // Conflict A: source has `foo` as dir, target has `foo` as file.
2607        std::fs::write(source.join("home/.config/foo/leaf.txt"), "src").unwrap();
2608        std::fs::write(target.join(".config/foo"), "target file body").unwrap();
2609        // Conflict B: source has `bar` as file, target has `bar` as dir.
2610        std::fs::write(source.join("home/.config/bar"), "src file body").unwrap();
2611        std::fs::create_dir_all(target.join(".config/bar")).unwrap();
2612        std::fs::write(target.join(".config/bar/inside.txt"), "target nested").unwrap();
2613
2614        let cfg = format!(
2615            r#"
2616[absorb]
2617on_anomaly = "force"
2618
2619[[mount.entry]]
2620src = "home"
2621dst = "{}"
2622"#,
2623            toml_path(&target)
2624        );
2625        std::fs::write(source.join("config.toml"), cfg).unwrap();
2626        apply(Some(source.clone()), false).unwrap();
2627
2628        // After absorb the target's view (which equals source via
2629        // junction) carries target's shapes:
2630        // `foo` is a regular file
2631        let foo_meta = std::fs::symlink_metadata(target.join(".config/foo")).unwrap();
2632        assert!(foo_meta.file_type().is_file(), "foo should be a file");
2633        assert_eq!(
2634            std::fs::read_to_string(target.join(".config/foo")).unwrap(),
2635            "target file body"
2636        );
2637        // `bar` is a directory with the nested file
2638        let bar_meta = std::fs::symlink_metadata(target.join(".config/bar")).unwrap();
2639        assert!(bar_meta.file_type().is_dir(), "bar should be a dir");
2640        assert_eq!(
2641            std::fs::read_to_string(target.join(".config/bar/inside.txt")).unwrap(),
2642            "target nested"
2643        );
2644    }
2645
2646    /// Per-file conflict in dir merge — target newer + content
2647    /// differs → AutoAbsorb. Target wins automatically without
2648    /// touching `[absorb] on_anomaly`.
2649    #[test]
2650    fn merge_per_file_target_newer_auto_absorbs() {
2651        let tmp = TempDir::new().unwrap();
2652        let source = utf8(tmp.path().join("dotfiles"));
2653        let target = utf8(tmp.path().join("target"));
2654        std::fs::create_dir_all(source.join("home/.config")).unwrap();
2655        std::fs::create_dir_all(target.join(".config")).unwrap();
2656        std::fs::write(source.join("home/.config/.yuilink"), "").unwrap();
2657
2658        // Source has the older copy, target has the newer edit.
2659        let past = std::time::SystemTime::now() - std::time::Duration::from_secs(120);
2660        write_with_mtime(&source.join("home/.config/app.toml"), "old src", past);
2661        std::fs::write(target.join(".config/app.toml"), "user's live edit").unwrap();
2662
2663        // Default `ask` policy — should NOT prompt because the
2664        // classifier returns AutoAbsorb (target newer + diff), which
2665        // bypasses `on_anomaly` entirely.
2666        let cfg = format!(
2667            r#"
2668[[mount.entry]]
2669src = "home"
2670dst = "{}"
2671"#,
2672            toml_path(&target)
2673        );
2674        std::fs::write(source.join("config.toml"), cfg).unwrap();
2675        apply(Some(source.clone()), false).unwrap();
2676
2677        // Target wins.
2678        assert_eq!(
2679            std::fs::read_to_string(target.join(".config/app.toml")).unwrap(),
2680            "user's live edit"
2681        );
2682    }
2683
2684    /// Per-file conflict — source newer + content differs +
2685    /// `on_anomaly = "skip"` → keep source's version. After the outer
2686    /// junction, target ends up with source's content (so target's
2687    /// file is effectively dropped, matching the file-level `skip`
2688    /// semantic).
2689    #[test]
2690    fn merge_per_file_source_newer_skip_keeps_source() {
2691        let tmp = TempDir::new().unwrap();
2692        let source = utf8(tmp.path().join("dotfiles"));
2693        let target = utf8(tmp.path().join("target"));
2694        std::fs::create_dir_all(source.join("home/.config")).unwrap();
2695        std::fs::create_dir_all(target.join(".config")).unwrap();
2696        std::fs::write(source.join("home/.config/.yuilink"), "").unwrap();
2697
2698        // Target has the older copy, source has the newer edit.
2699        let past = std::time::SystemTime::now() - std::time::Duration::from_secs(120);
2700        write_with_mtime(&target.join(".config/app.toml"), "old target", past);
2701        std::fs::write(source.join("home/.config/app.toml"), "fresh source").unwrap();
2702
2703        let cfg = format!(
2704            r#"
2705[absorb]
2706on_anomaly = "skip"
2707
2708[[mount.entry]]
2709src = "home"
2710dst = "{}"
2711"#,
2712            toml_path(&target)
2713        );
2714        std::fs::write(source.join("config.toml"), cfg).unwrap();
2715        apply(Some(source.clone()), false).unwrap();
2716
2717        // Source kept — target now reads source's version through the
2718        // junction (so target's old text is dropped).
2719        assert_eq!(
2720            std::fs::read_to_string(target.join(".config/app.toml")).unwrap(),
2721            "fresh source"
2722        );
2723    }
2724
2725    /// Per-file conflict — source newer + content differs +
2726    /// `on_anomaly = "force"` → target wins anyway.
2727    #[test]
2728    fn merge_per_file_source_newer_force_overwrites_source() {
2729        let tmp = TempDir::new().unwrap();
2730        let source = utf8(tmp.path().join("dotfiles"));
2731        let target = utf8(tmp.path().join("target"));
2732        std::fs::create_dir_all(source.join("home/.config")).unwrap();
2733        std::fs::create_dir_all(target.join(".config")).unwrap();
2734        std::fs::write(source.join("home/.config/.yuilink"), "").unwrap();
2735
2736        let past = std::time::SystemTime::now() - std::time::Duration::from_secs(120);
2737        write_with_mtime(&target.join(".config/app.toml"), "old target", past);
2738        std::fs::write(source.join("home/.config/app.toml"), "fresh source").unwrap();
2739
2740        let cfg = format!(
2741            r#"
2742[absorb]
2743on_anomaly = "force"
2744
2745[[mount.entry]]
2746src = "home"
2747dst = "{}"
2748"#,
2749            toml_path(&target)
2750        );
2751        std::fs::write(source.join("config.toml"), cfg).unwrap();
2752        apply(Some(source.clone()), false).unwrap();
2753
2754        // Target overrides source despite being mtime-older.
2755        assert_eq!(
2756            std::fs::read_to_string(target.join(".config/app.toml")).unwrap(),
2757            "old target"
2758        );
2759    }
2760
2761    /// Per-file conflict — bytes match → no-op. The merge classifies
2762    /// this as RelinkOnly and skips the copy entirely (saves a lot of
2763    /// I/O when migrating big chezmoi repos where source and target
2764    /// have already shared inodes).
2765    #[test]
2766    fn merge_per_file_identical_content_is_noop() {
2767        let tmp = TempDir::new().unwrap();
2768        let source = utf8(tmp.path().join("dotfiles"));
2769        let target = utf8(tmp.path().join("target"));
2770        std::fs::create_dir_all(source.join("home/.config")).unwrap();
2771        std::fs::create_dir_all(target.join(".config")).unwrap();
2772        std::fs::write(source.join("home/.config/.yuilink"), "").unwrap();
2773        std::fs::write(source.join("home/.config/app.toml"), "same").unwrap();
2774        std::fs::write(target.join(".config/app.toml"), "same").unwrap();
2775
2776        // Default policy — bytes match, classifier returns RelinkOnly,
2777        // merge skips the copy. Apply must succeed without prompting.
2778        let cfg = format!(
2779            r#"
2780[[mount.entry]]
2781src = "home"
2782dst = "{}"
2783"#,
2784            toml_path(&target)
2785        );
2786        std::fs::write(source.join("config.toml"), cfg).unwrap();
2787        apply(Some(source.clone()), false).unwrap();
2788
2789        assert_eq!(
2790            std::fs::read_to_string(target.join(".config/app.toml")).unwrap(),
2791            "same"
2792        );
2793    }
2794
2795    #[test]
2796    fn manual_absorb_command_pulls_target_into_source() {
2797        // Manual `yui absorb <target>` bypasses policy + git checks.
2798        let tmp = TempDir::new().unwrap();
2799        let source = utf8(tmp.path().join("dotfiles"));
2800        let target = utf8(tmp.path().join("target"));
2801        std::fs::create_dir_all(source.join("home")).unwrap();
2802        std::fs::create_dir_all(&target).unwrap();
2803        // on_anomaly = "skip" so passive `apply` would NOT touch this.
2804        let cfg = format!(
2805            r#"
2806[absorb]
2807on_anomaly = "skip"
2808
2809[[mount.entry]]
2810src = "home"
2811dst = "{}"
2812"#,
2813            toml_path(&target)
2814        );
2815        std::fs::write(source.join("config.toml"), cfg).unwrap();
2816        std::fs::write(target.join(".bashrc"), "user picked this").unwrap();
2817        std::fs::write(source.join("home/.bashrc"), "default").unwrap();
2818
2819        // Run absorb directly on the target.
2820        absorb(
2821            Some(source.clone()),
2822            target.join(".bashrc"),
2823            /* dry_run */ false,
2824        )
2825        .unwrap();
2826
2827        // Source picked up target's content (manual absorb is forceful).
2828        assert_eq!(
2829            std::fs::read_to_string(source.join("home/.bashrc")).unwrap(),
2830            "user picked this"
2831        );
2832    }
2833
2834    #[test]
2835    fn manual_absorb_errors_when_target_outside_known_mounts() {
2836        let tmp = TempDir::new().unwrap();
2837        let (source, _target) = setup_minimal_dotfiles(&tmp);
2838        std::fs::write(source.join("home/.bashrc"), "x").unwrap();
2839        let stranger = utf8(tmp.path().join("not-managed/foo"));
2840        std::fs::create_dir_all(stranger.parent().unwrap()).unwrap();
2841        std::fs::write(&stranger, "not yui's").unwrap();
2842        let err = absorb(Some(source), stranger, false).unwrap_err();
2843        assert!(format!("{err}").contains("no mount entry"));
2844    }
2845
2846    #[test]
2847    fn yuiignore_excludes_file_from_linking() {
2848        let tmp = TempDir::new().unwrap();
2849        let (source, target) = setup_minimal_dotfiles(&tmp);
2850        std::fs::write(source.join("home/.bashrc"), "kept").unwrap();
2851        std::fs::write(source.join("home/lock.json"), "ignored").unwrap();
2852        // Exclude `lock.json` files anywhere under source.
2853        std::fs::write(source.join(".yuiignore"), "**/lock.json\n").unwrap();
2854        apply(Some(source.clone()), false).unwrap();
2855        assert!(target.join(".bashrc").exists());
2856        assert!(
2857            !target.join("lock.json").exists(),
2858            "yuiignore should keep lock.json out of target"
2859        );
2860    }
2861
2862    #[test]
2863    fn yuiignore_excludes_directory_subtree() {
2864        let tmp = TempDir::new().unwrap();
2865        let (source, target) = setup_minimal_dotfiles(&tmp);
2866        std::fs::create_dir_all(source.join("home/cache")).unwrap();
2867        std::fs::write(source.join("home/.bashrc"), "kept").unwrap();
2868        std::fs::write(source.join("home/cache/a"), "ignored").unwrap();
2869        std::fs::write(source.join("home/cache/b"), "also ignored").unwrap();
2870        // Trailing slash → match dirs only; entire subtree skipped.
2871        std::fs::write(source.join(".yuiignore"), "home/cache/\n").unwrap();
2872        apply(Some(source.clone()), false).unwrap();
2873        assert!(target.join(".bashrc").exists());
2874        assert!(
2875            !target.join("cache").exists(),
2876            "yuiignore'd subtree should not appear in target"
2877        );
2878    }
2879
2880    #[test]
2881    fn yuiignore_negation_re_includes_file() {
2882        let tmp = TempDir::new().unwrap();
2883        let (source, target) = setup_minimal_dotfiles(&tmp);
2884        std::fs::write(source.join("home/keep.cache"), "kept by negation").unwrap();
2885        std::fs::write(source.join("home/drop.cache"), "ignored").unwrap();
2886        // Ignore all .cache files except keep.cache.
2887        std::fs::write(source.join(".yuiignore"), "*.cache\n!keep.cache\n").unwrap();
2888        apply(Some(source.clone()), false).unwrap();
2889        assert!(target.join("keep.cache").exists());
2890        assert!(!target.join("drop.cache").exists());
2891    }
2892
2893    #[test]
2894    fn yuiignore_skips_template_in_render() {
2895        let tmp = TempDir::new().unwrap();
2896        let source = utf8(tmp.path().join("dotfiles"));
2897        let target = utf8(tmp.path().join("target"));
2898        std::fs::create_dir_all(source.join("home")).unwrap();
2899        std::fs::create_dir_all(&target).unwrap();
2900        std::fs::write(source.join("home/note.tera"), "{{ yui.os }}").unwrap();
2901        std::fs::write(source.join(".yuiignore"), "home/note*\n").unwrap();
2902        let cfg = format!(
2903            r#"
2904[[mount.entry]]
2905src = "home"
2906dst = "{}"
2907"#,
2908            toml_path(&target)
2909        );
2910        std::fs::write(source.join("config.toml"), cfg).unwrap();
2911        apply(Some(source.clone()), false).unwrap();
2912        // Neither the template nor the rendered output linked.
2913        assert!(!source.join("home/note").exists());
2914        assert!(!target.join("note").exists());
2915        assert!(!target.join("note.tera").exists());
2916    }
2917
2918    /// v0.6+: parent `.yuilink` doesn't stop the walker. A parent
2919    /// marker can junction the whole dir, AND a child marker can layer
2920    /// on extra dsts (e.g. an OS-specific alternate location).
2921    #[test]
2922    fn nested_marker_accumulates_extra_dst() {
2923        let tmp = TempDir::new().unwrap();
2924        let source = utf8(tmp.path().join("dotfiles"));
2925        let parent_target = utf8(tmp.path().join("home"));
2926        let extra_target = utf8(tmp.path().join("extra"));
2927        std::fs::create_dir_all(source.join("home/.config/nvim")).unwrap();
2928        std::fs::create_dir_all(&parent_target).unwrap();
2929        std::fs::create_dir_all(&extra_target).unwrap();
2930        std::fs::write(source.join("home/.config/nvim/init.lua"), "-- nvim\n").unwrap();
2931
2932        // Parent: junction the whole .config dir to <home>/.config.
2933        std::fs::write(
2934            source.join("home/.config/.yuilink"),
2935            format!(
2936                r#"
2937[[link]]
2938dst = "{}/.config"
2939"#,
2940                toml_path(&parent_target)
2941            ),
2942        )
2943        .unwrap();
2944        // Child: ALSO junction nvim/ to an extra path, but only on the
2945        // running OS (so the test exercises an active link).
2946        std::fs::write(
2947            source.join("home/.config/nvim/.yuilink"),
2948            format!(
2949                r#"
2950[[link]]
2951dst = "{}/nvim"
2952when = "{{{{ yui.os == '{}' }}}}"
2953"#,
2954                toml_path(&extra_target),
2955                std::env::consts::OS
2956            ),
2957        )
2958        .unwrap();
2959
2960        let cfg = format!(
2961            r#"
2962[[mount.entry]]
2963src = "home"
2964dst = "{}"
2965"#,
2966            toml_path(&parent_target)
2967        );
2968        std::fs::write(source.join("config.toml"), cfg).unwrap();
2969
2970        apply(Some(source.clone()), false).unwrap();
2971
2972        // Both links are present: parent's whole-.config junction reaches
2973        // init.lua, and the child marker added an additional path.
2974        assert!(parent_target.join(".config/nvim/init.lua").exists());
2975        assert!(extra_target.join("nvim/init.lua").exists());
2976    }
2977
2978    /// v0.6+: `[[link]] src = "<filename>"` links a single sibling file
2979    /// to a custom dst, leaving the rest of the dir to default
2980    /// behaviour. Useful for paths like the PowerShell profile that
2981    /// have to live in a non-`~/.config` location on Windows.
2982    #[test]
2983    fn marker_file_link_targets_specific_file() {
2984        let tmp = TempDir::new().unwrap();
2985        let source = utf8(tmp.path().join("dotfiles"));
2986        let parent_target = utf8(tmp.path().join("home"));
2987        let docs_target = utf8(tmp.path().join("docs"));
2988        std::fs::create_dir_all(source.join("home/.config/powershell")).unwrap();
2989        std::fs::create_dir_all(&parent_target).unwrap();
2990        std::fs::create_dir_all(&docs_target).unwrap();
2991        std::fs::write(
2992            source.join("home/.config/powershell/profile.ps1"),
2993            "# profile\n",
2994        )
2995        .unwrap();
2996        std::fs::write(source.join("home/.config/powershell/extra.txt"), "extra\n").unwrap();
2997
2998        // File-level entry only — no dir-level [[link]], so the dir
2999        // itself still falls through to the default mount placement.
3000        std::fs::write(
3001            source.join("home/.config/powershell/.yuilink"),
3002            format!(
3003                r#"
3004[[link]]
3005src = "profile.ps1"
3006dst = "{}/Microsoft.PowerShell_profile.ps1"
3007"#,
3008                toml_path(&docs_target)
3009            ),
3010        )
3011        .unwrap();
3012
3013        let cfg = format!(
3014            r#"
3015[[mount.entry]]
3016src = "home"
3017dst = "{}"
3018"#,
3019            toml_path(&parent_target)
3020        );
3021        std::fs::write(source.join("config.toml"), cfg).unwrap();
3022
3023        apply(Some(source.clone()), false).unwrap();
3024
3025        // File-level target gets the link.
3026        assert!(
3027            docs_target
3028                .join("Microsoft.PowerShell_profile.ps1")
3029                .exists()
3030        );
3031        // Default per-file placement still happens for ALL files in the
3032        // dir (the marker had no dir-level [[link]] to claim coverage).
3033        assert!(
3034            parent_target
3035                .join(".config/powershell/profile.ps1")
3036                .exists()
3037        );
3038        assert!(parent_target.join(".config/powershell/extra.txt").exists());
3039    }
3040
3041    /// File-level [[link]] errors clearly when src points at a missing
3042    /// file — config bug, not a silent skip.
3043    #[test]
3044    fn marker_file_link_missing_src_errors() {
3045        let tmp = TempDir::new().unwrap();
3046        let source = utf8(tmp.path().join("dotfiles"));
3047        let parent_target = utf8(tmp.path().join("home"));
3048        let docs_target = utf8(tmp.path().join("docs"));
3049        std::fs::create_dir_all(source.join("home/.config/powershell")).unwrap();
3050        std::fs::create_dir_all(&parent_target).unwrap();
3051        std::fs::create_dir_all(&docs_target).unwrap();
3052
3053        std::fs::write(
3054            source.join("home/.config/powershell/.yuilink"),
3055            format!(
3056                r#"
3057[[link]]
3058src = "missing.ps1"
3059dst = "{}/profile.ps1"
3060"#,
3061                toml_path(&docs_target)
3062            ),
3063        )
3064        .unwrap();
3065
3066        let cfg = format!(
3067            r#"
3068[[mount.entry]]
3069src = "home"
3070dst = "{}"
3071"#,
3072            toml_path(&parent_target)
3073        );
3074        std::fs::write(source.join("config.toml"), cfg).unwrap();
3075
3076        let err = apply(Some(source.clone()), false).unwrap_err();
3077        assert!(format!("{err:#}").contains("missing.ps1"));
3078    }
3079
3080    fn walkdir(root: &Utf8Path) -> Vec<Utf8PathBuf> {
3081        let mut out = Vec::new();
3082        let mut stack = vec![root.to_path_buf()];
3083        while let Some(dir) = stack.pop() {
3084            let Ok(entries) = std::fs::read_dir(&dir) else {
3085                continue;
3086            };
3087            for e in entries.flatten() {
3088                let p = utf8(e.path());
3089                if e.file_type().map(|t| t.is_dir()).unwrap_or(false) {
3090                    stack.push(p);
3091                } else {
3092                    out.push(p);
3093                }
3094            }
3095        }
3096        out
3097    }
3098}