OXYGENHUB
# c8d9059e4e70 oxygen / src / cli / workspace.rs
95.6 KiB · Rust · last changed oxy resolve PATH settles a replace conflict with the working copy's content (closes #140) 10 days ago source blame history raw
fce44ae9 Andy McCall ↶ 1
//! Commands that act on the local repository and working copy.
2
75167672 Andy McCall 3
use super::format::{
5cde7882 Andy McCall ↶ 4
    author_name, deps_line, deps_of, plural, short_id, signature_line, signature_mark_of,
5
    subject_of, via, when,
75167672 Andy McCall 6
};
c6aa5bd9 Andy McCall ↶ 7
use super::json;
fce44ae9 Andy McCall ↶ 8
use super::open;
135212de Andy McCall ↶ 9
use super::ui;
fce44ae9 Andy McCall ↶ 10
use anyhow::Result;
94084087 Andy McCall ↶ 11
use oxy_core::objects::ChangeKind;
75167672 Andy McCall 12
use oxy_core::repo::SignatureStatus;
fce44ae9 Andy McCall ↶ 13
use oxy_core::repo::{render_change, ChangeStatus, Repo};
fddf64ba Andy McCall ↶ 14
use std::io::Write;
fce44ae9 Andy McCall ↶ 15
use std::path::PathBuf;
16
d069701a Andy McCall ↶ 17
pub fn init(path: Option<PathBuf>, bare: bool, store: Option<&str>) -> Result<i32> {
fce44ae9 Andy McCall ↶ 18
    let path = path.unwrap_or(std::env::current_dir()?);
d069701a Andy McCall ↶ 19
    let repo = Repo::init_new(&path, bare, store)?;
75167672 Andy McCall 20
    println!(
21
        "{} {}oxy repository in {}",
22
        ui::done("Initialized"),
23
        if bare { "bare " } else { "" },
24
        ui::path(repo.root().display())
25
    );
26
    println!(
27
        "{}",
28
        ui::kv("bookmark", ui::bookmark(&repo.config.bookmark))
29
    );
d069701a Andy McCall ↶ 30
    if let Some(url) = &repo.config.storage_url {
31
        println!(
32
            "{}",
33
            ui::kv(
34
                "store",
35
                format!(
36
                    "{url} (cache in {}, up to {} MiB)",
37
                    ui::path(repo.root().join(".oxy/cache").display()),
38
                    repo.config.storage_cache >> 20
39
                )
40
            )
41
        );
42
    }
75167672 Andy McCall 43
    println!(
44
        "{}",
45
        ui::kv(
46
            "author",
47
            format!(
48
                "{} <{}>",
49
                ui::author(&repo.config.author_name),
50
                repo.config.author_email
51
            )
52
        )
53
    );
fce44ae9 Andy McCall ↶ 54
    Ok(0)
55
}
56
57
fn status_tag(s: ChangeStatus) -> &'static str {
58
    match s {
59
        ChangeStatus::Added => "A",
60
        ChangeStatus::Modified => "M",
61
        ChangeStatus::Removed => "D",
6e723b81 Andy McCall ↶ 62
        ChangeStatus::Renamed => "R",
fce44ae9 Andy McCall ↶ 63
        ChangeStatus::ModeChanged => "X",
64
        ChangeStatus::ConflictEdited => "C!",
6e723b81 Andy McCall ↶ 65
        ChangeStatus::Resolved => "V",
fce44ae9 Andy McCall ↶ 66
    }
67
}
68
b689779e Andy McCall ↶ 69
/// New untracked files left out of the snapshot for their size (oxygen#98):
70
/// named, with how to keep them out quietly or take them in.
71
fn print_skipped_large(repo: &Repo) {
72
    let skipped = repo.skipped_large();
73
    if skipped.is_empty() {
74
        return;
75
    }
76
    let size = |n: u64| {
77
        if n >= 1 << 30 {
78
            format!("{:.1} GiB", n as f64 / (1u64 << 30) as f64)
79
        } else {
80
            format!("{:.0} MiB", n as f64 / (1u64 << 20) as f64)
81
        }
82
    };
83
    println!();
84
    println!(
85
        "{} {}",
86
        ui::section(ui::warn("Not tracked")),
87
        ui::dim(format!("new files over {}", size(repo.config.max_new_file)))
88
    );
89
    for (p, n) in &skipped {
90
        println!("  {} {}  {}", ui::caution(), ui::path(p), ui::dim(size(*n)));
91
    }
92
    println!(
93
        "  {} {}",
94
        ui::note(),
95
        ui::dim(
96
            "to leave them out quietly, list them in .oxyignore; to track one, \
97
             raise snapshot.max-new-file-size (oxy config snapshot.max-new-file-size 2G)"
98
        )
99
    );
100
}
101
9b723f69 Andy McCall ↶ 102
/// Name what a checkout removed from the working copy (oxygen#102): a
103
/// switch to a bookmark lacking files takes them off disk, which an
104
/// editor open on the tree notices before a person reading a count does.
105
fn print_removed(r: &oxy_core::repo::CheckoutReport) {
106
    if r.removed == 0 {
107
        return;
108
    }
109
    let more = r.removed.saturating_sub(r.removed_paths.len());
110
    let names = r.removed_paths.join(", ");
111
    let tail = if more > 0 {
112
        format!(" and {more} more")
113
    } else {
114
        String::new()
115
    };
116
    let lead = if r.removed >= 20 {
117
        ui::caution()
118
    } else {
119
        ui::note()
120
    };
121
    println!(
122
        "  {} {}",
123
        lead,
124
        ui::dim(format!(
125
            "removed from disk (the bookmark does not have them): {names}{tail}"
126
        ))
127
    );
128
}
129
d4dbab97 Andy McCall ↶ 130
/// Folders holding a repository of their own, which oxy leaves out (oxygen#43):
131
/// named, with how to track them.
132
fn print_untracked_nested(repo: &Repo, nested: &[String]) {
133
    if nested.is_empty() {
134
        return;
135
    }
136
    println!();
137
    println!(
138
        "{} {}",
139
        ui::section(ui::warn("Not tracked")),
140
        ui::dim("folders holding a repository of their own")
141
    );
142
    for p in nested {
143
        let kind = if repo.root().join(p).join(".git").exists() {
144
            "Git"
145
        } else {
146
            "Oxygen"
147
        };
148
        println!(
149
            "  {} {}  {}",
150
            ui::caution(),
151
            ui::path(format!("{p}/")),
152
            ui::dim(format!("a nested {kind} repository"))
153
        );
154
    }
155
    println!(
156
        "  {} {}",
157
        ui::note(),
158
        ui::dim(
159
            "to track one's files, remove its .git (or .oxy); to keep it a repository of its own, \
69692f14 Andy McCall ↶ 160
             oxy submodule add URL PATH; to leave it out quietly, list it in .oxyignore"
d4dbab97 Andy McCall ↶ 161
        )
162
    );
163
}
164
36041796 Andy McCall ↶ 165
/// The working copy's changed paths, or that it is clean.
166
fn print_statuses(statuses: &[oxy_core::repo::PathStatus]) {
167
    if statuses.is_empty() {
168
        println!(
169
            "  {} {}",
170
            ui::ok(ui::glyph("✔", "*")),
171
            ui::dim("nothing to commit, working copy clean")
172
        );
173
        return;
174
    }
175
    println!();
176
    for s in statuses {
177
        println!(
178
            "  {} {}",
179
            ui::status_mark(status_tag(s.status)),
180
            ui::path(&s.path)
181
        );
182
    }
183
}
184
f6dccc45 Andy McCall ↶ 185
/// Say when the working copy has fallen behind its own bookmark: a
186
/// "clean" status there hid a pull that left it on old code (oxygen#89).
187
fn print_behind(repo: &Repo) {
188
    if let Ok(Some(n)) = repo.working_copy_behind() {
189
        println!(
190
            "  {} {}",
191
            ui::caution(),
192
            ui::warn(format!(
193
                "the working copy is behind {}: it lacks {} the bookmark has; oxy checkout brings it up",
194
                repo.config.bookmark,
195
                plural(n, "change")
196
            ))
197
        );
198
    }
199
}
200
50a046e4 Andy McCall ↶ 201
pub fn status(json: bool) -> Result<i32> {
fce44ae9 Andy McCall ↶ 202
    let repo = open()?;
36041796 Andy McCall ↶ 203
    // Answered by the manifest alone, with no history to load (oxygen#35,
204
    // #41): a clean working copy, or one as the last snapshot left it.
205
    let known = if json {
206
        None
207
    } else {
208
        repo.status_by_manifest()?
209
    };
210
    if let Some(statuses) = known {
da868d5d Andy McCall ↶ 211
        let (_, wc) = repo.working_copy()?;
212
        let heads = repo.bookmark_heads()?;
213
        println!(
214
            "{} {}{}working copy {}{}{}",
215
            ui::dim("On"),
216
            ui::bookmark(&repo.config.bookmark),
217
            ui::sep(),
218
            ui::id(wc.change_id.short(8)),
219
            ui::sep(),
220
            ui::dim(plural(heads.len(), "head"))
221
        );
f6dccc45 Andy McCall ↶ 222
        print_behind(&repo);
36041796 Andy McCall ↶ 223
        print_statuses(&statuses);
d4dbab97 Andy McCall ↶ 224
        print_untracked_nested(&repo, &repo.untracked_nested());
b689779e Andy McCall ↶ 225
        print_skipped_large(&repo);
da868d5d Andy McCall ↶ 226
        return Ok(0);
227
    }
135212de Andy McCall ↶ 228
    let sp = ui::spinner("Snapshotting the working copy…");
cc7a0056 Andy McCall ↶ 229
    let snap = repo.snapshot_status("oxy status")?;
4cce912a Andy McCall ↶ 230
    let heads = {
231
        let _t = oxy_core::timing::phase("status: heads");
232
        repo.bookmark_heads()?
233
    };
234
    let _drop = oxy_core::timing::phase("status: print and drop");
135212de Andy McCall ↶ 235
    sp.clear();
50a046e4 Andy McCall ↶ 236
    if json {
237
        println!("{}", repo.status_json(&snap)?);
75167672 Andy McCall 238
        return Ok(if snap.base_snapshot.is_conflicted() {
239
            2
240
        } else {
241
            0
242
        });
50a046e4 Andy McCall ↶ 243
    }
135212de Andy McCall ↶ 244
    println!(
94084087 Andy McCall ↶ 245
        "{} {}{}working copy {}{}{}",
246
        ui::dim("On"),
247
        ui::bookmark(&repo.config.bookmark),
248
        ui::sep(),
135212de Andy McCall ↶ 249
        ui::id(snap.wc.change_id.short(8)),
94084087 Andy McCall ↶ 250
        ui::sep(),
251
        ui::dim(plural(heads.len(), "head"))
135212de Andy McCall ↶ 252
    );
f6dccc45 Andy McCall ↶ 253
    print_behind(&repo);
36041796 Andy McCall ↶ 254
    print_statuses(&snap.statuses);
d4dbab97 Andy McCall ↶ 255
    print_untracked_nested(&repo, &snap.untracked_nested);
b689779e Andy McCall ↶ 256
    print_skipped_large(&repo);
fce44ae9 Andy McCall ↶ 257
    if !snap.base_snapshot.is_conflicted() {
258
        return Ok(0);
259
    }
135212de Andy McCall ↶ 260
    println!();
75167672 Andy McCall 261
    println!(
262
        "{} {}",
263
        ui::section(ui::err("Conflicts")),
264
        ui::dim("edit the marked files, or the .oxy-conflict-N candidates, then commit")
265
    );
fce44ae9 Andy McCall ↶ 266
    for c in &snap.base_snapshot.tree.conflicts {
94084087 Andy McCall ↶ 267
        println!("  {} tree: {c:?}", ui::err(ui::glyph("✱", "!")));
fce44ae9 Andy McCall ↶ 268
    }
269
    for (p, f) in &snap.base_snapshot.files {
270
        if f.is_conflicted() {
75167672 Andy McCall 271
            let state = match snap
272
                .statuses
273
                .iter()
274
                .find(|s| s.path == *p)
275
                .map(|s| s.status)
276
            {
277
                Some(ChangeStatus::Resolved) => {
278
                    ui::ok(format!("{} resolved in working copy", ui::glyph("✔", "*")))
279
                }
280
                Some(ChangeStatus::ConflictEdited) => {
281
                    ui::warn(format!("{} markers remain", ui::glyph("‼", "!")))
282
                }
94084087 Andy McCall ↶ 283
                _ => ui::err(format!("{} unresolved", ui::glyph("✱", "!"))),
fce44ae9 Andy McCall ↶ 284
            };
135212de Andy McCall ↶ 285
            println!("  {}  {state}", ui::path(p));
fce44ae9 Andy McCall ↶ 286
        }
287
    }
288
    Ok(2)
289
}
290
75167672 Andy McCall 291
pub fn commit(
292
    message: &str,
293
    depends_on: &[String],
294
    paths: &[String],
295
    interactive: bool,
296
    hunks: &[usize],
297
) -> Result<i32> {
fce44ae9 Andy McCall ↶ 298
    let repo = open()?;
299
    let mut deps = Vec::new();
300
    for d in depends_on {
301
        deps.push(repo.resolve_change(d)?);
302
    }
c6aa5bd9 Andy McCall ↶ 303
    let selection = if !hunks.is_empty() {
304
        oxy_core::repo::Selection::Hunks(hunks.to_vec())
305
    } else if interactive {
957fee30 Andy McCall ↶ 306
        let (_, hunks) = repo.hunks()?;
307
        if hunks.is_empty() {
308
            anyhow::bail!("nothing to commit: the working copy matches its base");
309
        }
75167672 Andy McCall 310
        let hunks: Vec<_> = hunks
311
            .into_iter()
312
            .filter(|h| {
313
                paths.is_empty()
314
                    || paths.iter().any(|p| {
315
                        h.path == *p || h.path.starts_with(&format!("{}/", p.trim_end_matches('/')))
316
                    })
317
            })
318
            .collect();
957fee30 Andy McCall ↶ 319
        let mut chosen = Vec::new();
320
        let stdin = std::io::stdin();
321
        let mut take_rest = false;
322
        for (n, h) in hunks.iter().enumerate() {
323
            if take_rest {
324
                chosen.push(h.index);
325
                continue;
326
            }
75167672 Andy McCall 327
            println!(
328
                "{} {} {}",
329
                ui::section(format!("hunk {} of {}", n + 1, hunks.len())),
330
                ui::path(&h.path),
331
                ui::dim(format!("(#{})", h.index))
332
            );
957fee30 Andy McCall ↶ 333
            for line in h.text.lines() {
75167672 Andy McCall 334
                let styled = if line.starts_with('+') {
335
                    ui::ok(line)
336
                } else if line.starts_with('-') {
337
                    ui::err(line)
338
                } else {
339
                    ui::dim(line)
340
                };
957fee30 Andy McCall ↶ 341
                println!("  {styled}");
342
            }
343
            loop {
75167672 Andy McCall 344
                print!(
345
                    "{} ",
346
                    ui::warn("Commit this hunk? [y]es [n]o [a]ll remaining [q]uit:")
347
                );
957fee30 Andy McCall ↶ 348
                use std::io::Write as _;
349
                std::io::stdout().flush()?;
350
                let mut answer = String::new();
351
                if stdin.read_line(&mut answer)? == 0 {
352
                    anyhow::bail!("aborted");
353
                }
354
                match answer.trim() {
355
                    "y" | "Y" => {
356
                        chosen.push(h.index);
357
                        break;
358
                    }
359
                    "n" | "N" | "" => break,
360
                    "a" | "A" => {
361
                        chosen.push(h.index);
362
                        take_rest = true;
363
                        break;
364
                    }
365
                    "q" | "Q" => anyhow::bail!("aborted; nothing committed"),
366
                    _ => continue,
367
                }
368
            }
369
        }
370
        if chosen.is_empty() {
371
            anyhow::bail!("no hunks chosen; nothing committed");
372
        }
373
        oxy_core::repo::Selection::Hunks(chosen)
374
    } else if paths.is_empty() {
375
        oxy_core::repo::Selection::All
376
    } else {
377
        oxy_core::repo::Selection::Paths(paths.to_vec())
378
    };
135212de Andy McCall ↶ 379
    let sp = ui::spinner("Committing…");
75051941 Andy McCall ↶ 380
    let (h, partial) = repo.commit_selection_partial(message, &deps, &selection)?;
5b4c5130 Andy McCall ↶ 381
    // The metadata only: a first commit's change holds the whole tree.
382
    let meta = repo.store().get_change_meta(&h)?;
75167672 Andy McCall 383
    sp.done(&format!(
384
        "{} {} {}  {}",
385
        ui::done("Committed"),
5b4c5130 Andy McCall ↶ 386
        ui::id(meta.change_id.short(8)),
75167672 Andy McCall 387
        ui::hash(format!("#{}", h.short(12))),
388
        ui::subject(subject_of(message))
389
    ));
d4dbab97 Andy McCall ↶ 390
    print_untracked_nested(&repo, &repo.untracked_nested());
b689779e Andy McCall ↶ 391
    print_skipped_large(&repo);
75051941 Andy McCall ↶ 392
    // Only a commit that took part of the working copy leaves anything.
393
    if partial {
8ab265dc Andy McCall ↶ 394
        let rest = repo.snapshot_status("oxy commit")?;
957fee30 Andy McCall ↶ 395
        if !rest.statuses.is_empty() {
75167672 Andy McCall 396
            println!(
397
                "  {} {}",
398
                ui::note(),
399
                ui::dim(format!(
400
                    "{} left uncommitted",
401
                    plural(rest.statuses.len(), "path")
402
                ))
403
            );
957fee30 Andy McCall ↶ 404
        }
405
    }
98a1620a Andy McCall ↶ 406
    where_now(&repo);
957fee30 Andy McCall ↶ 407
    Ok(0)
408
}
409
410
pub fn blame(path: &str, json: bool) -> Result<i32> {
411
    let repo = open()?;
412
    let lines = repo.blame(path)?;
413
    if json {
414
        let items = lines.iter().map(|l| {
415
            format!(
416
                "{{\"line\":{},\"change_id\":{},\"hash\":{},\"index\":{},\"author\":{},\"when_ms\":{}}}",
417
                l.line,
418
                oxy_core::repo::json_str(&l.change_id.to_hex()),
419
                oxy_core::repo::json_str(&l.hash.to_hex()),
420
                l.index,
421
                oxy_core::repo::json_str(&l.author),
422
                l.when_ms
423
            )
424
        });
425
        println!("{}", oxy_core::repo::json_array(items));
426
        return Ok(0);
427
    }
75167672 Andy McCall 428
    let width = lines
429
        .iter()
430
        .map(|l| l.author.chars().count())
431
        .max()
432
        .unwrap_or(0)
433
        .min(20);
957fee30 Andy McCall ↶ 434
    let digits = lines.len().to_string().len();
fddf64ba Andy McCall ↶ 435
    let _t = oxy_core::timing::phase("blame: print");
436
    // Most lines share their change with many others: each change's id,
437
    // author and age are formatted once, and the lines go out through one
438
    // buffered lock on stdout.
439
    let mut prefix: std::collections::HashMap<oxy_core::hash::ChangeHash, String> =
440
        std::collections::HashMap::new();
441
    let bar = ui::dim(ui::glyph("│", "|"));
442
    let mut out = std::io::BufWriter::with_capacity(1 << 16, std::io::stdout().lock());
957fee30 Andy McCall ↶ 443
    for l in &lines {
fddf64ba Andy McCall ↶ 444
        let p = prefix.entry(l.hash).or_insert_with(|| {
445
            let author: String = l.author.chars().take(20).collect();
446
            format!(
447
                "{} {} {}",
448
                ui::id(l.change_id.short(8)),
449
                ui::author(format!("{author:<width$}")),
450
                ui::dim(ui::ago(l.when_ms))
451
            )
452
        });
957fee30 Andy McCall ↶ 453
        let text = String::from_utf8_lossy(&l.text);
fddf64ba Andy McCall ↶ 454
        if writeln!(
455
            out,
456
            "{p} {:>digits$}{bar} {}",
957fee30 Andy McCall ↶ 457
            l.line,
458
            text.trim_end_matches('\n')
fddf64ba Andy McCall ↶ 459
        )
460
        .is_err()
461
        {
462
            // A closed pipe (blame | head): stop quietly.
463
            return Ok(0);
464
        }
957fee30 Andy McCall ↶ 465
    }
fddf64ba Andy McCall ↶ 466
    let _ = out.flush();
957fee30 Andy McCall ↶ 467
    Ok(0)
468
}
469
470
fn bisect_report(repo: &Repo, step: &oxy_core::repo::BisectStep) -> Result<i32> {
471
    use oxy_core::repo::BisectStep::*;
472
    match step {
473
        NeedMarks { good, bad } => {
75167672 Andy McCall 474
            println!(
475
                "{} {} {}",
476
                ui::note(),
477
                ui::dim(format!("{good} good, {bad} bad;")),
478
                ui::dim(if *good == 0 {
479
                    "mark a known-good change or bookmark: oxy bisect good CHANGE"
480
                } else {
481
                    "mark a known-bad change or bookmark: oxy bisect bad CHANGE"
482
                })
483
            );
957fee30 Andy McCall ↶ 484
            Ok(0)
485
        }
75167672 Andy McCall 486
        Testing {
487
            heads,
488
            remaining,
489
            steps_left,
490
        } => {
491
            println!(
492
                "{} {} {}",
493
                ui::done("Testing"),
494
                ui::dim(format!(
495
                    "{} in the state under test;",
496
                    plural(heads.len(), "head")
497
                )),
498
                ui::dim(format!(
499
                    "{remaining} candidates left, about {steps_left} more steps"
500
                ))
501
            );
502
            println!(
503
                "  {} {}",
504
                ui::note(),
505
                ui::dim("run your test, then oxy bisect good | bad | skip")
506
            );
957fee30 Andy McCall ↶ 507
            Ok(0)
508
        }
509
        Found { culprit } => {
510
            let c = repo.store().get_change(culprit)?;
75167672 Andy McCall 511
            println!(
512
                "{} {} {}  {}",
513
                ui::done("Found"),
514
                ui::id(short_id(&c)),
515
                ui::hash(format!("#{}", culprit.short(12))),
516
                ui::subject(subject_of(&c.message))
517
            );
957fee30 Andy McCall ↶ 518
            println!("  {} {}", ui::note(), ui::dim("the first change whose inclusion makes the state bad; oxy bisect reset to return"));
519
            Ok(0)
520
        }
521
        Inconclusive { candidates } => {
75167672 Andy McCall 522
            println!(
523
                "{} {}",
524
                ui::caution(),
525
                ui::warn(format!(
526
                    "every remaining candidate was skipped ({}); oxy bisect reset to return",
527
                    plural(candidates.len(), "change")
528
                ))
529
            );
957fee30 Andy McCall ↶ 530
            Ok(1)
531
        }
532
    }
533
}
534
535
pub fn bisect_start() -> Result<i32> {
536
    let repo = open()?;
537
    repo.bisect_start()?;
75167672 Andy McCall 538
    println!(
539
        "{} bisect; {}",
540
        ui::done("Started"),
541
        ui::dim("now: oxy bisect good CHANGE and oxy bisect bad CHANGE")
542
    );
957fee30 Andy McCall ↶ 543
    Ok(0)
544
}
545
546
pub fn bisect_mark(good: Option<bool>, changes: &[String]) -> Result<i32> {
547
    let repo = open()?;
548
    let mut heads = Vec::new();
549
    for c in changes {
550
        heads.extend(repo.resolve_heads(c)?);
551
    }
552
    let step = repo.bisect_mark(good, &heads)?;
553
    bisect_report(&repo, &step)
554
}
555
556
pub fn bisect_run(command: &[String]) -> Result<i32> {
557
    let repo = open()?;
558
    let step = repo.bisect_run(command)?;
559
    bisect_report(&repo, &step)
560
}
561
c6aa5bd9 Andy McCall ↶ 562
pub fn bisect_status(as_json: bool) -> Result<i32> {
957fee30 Andy McCall ↶ 563
    let repo = open()?;
c6aa5bd9 Andy McCall ↶ 564
    if as_json {
565
        use oxy_core::repo::BisectStep::*;
566
        let v = match repo.bisect_status()? {
567
            None => "null".to_string(),
75167672 Andy McCall 568
            Some(NeedMarks { good, bad }) => json::obj([
569
                ("state", json::s("need-marks")),
570
                ("good", json::n(good)),
571
                ("bad", json::n(bad)),
572
            ]),
573
            Some(Testing {
574
                heads,
575
                remaining,
576
                steps_left,
577
            }) => json::obj([
578
                ("state", json::s("testing")),
579
                ("heads", json::hashes(&heads)),
580
                ("remaining", json::n(remaining)),
581
                ("steps_left", json::n(steps_left)),
582
            ]),
583
            Some(Found { culprit }) => json::obj([
584
                ("state", json::s("found")),
585
                ("culprit", json::s(&culprit.to_hex())),
586
            ]),
587
            Some(Inconclusive { candidates }) => json::obj([
588
                ("state", json::s("inconclusive")),
589
                ("candidates", json::hashes(&candidates)),
590
            ]),
c6aa5bd9 Andy McCall ↶ 591
        };
592
        println!("{v}");
593
        return Ok(0);
594
    }
957fee30 Andy McCall ↶ 595
    match repo.bisect_status()? {
596
        None => {
597
            println!("{}", ui::dim("No bisect in progress."));
598
            Ok(0)
599
        }
600
        Some(step) => bisect_report(&repo, &step),
601
    }
602
}
603
604
pub fn bisect_reset() -> Result<i32> {
605
    let repo = open()?;
606
    repo.bisect_reset()?;
607
    println!("{} bisect; working copy restored", ui::done("Reset"));
fce44ae9 Andy McCall ↶ 608
    Ok(0)
609
}
610
94084087 Andy McCall ↶ 611
/// The log: one entry per change, newest first, joined by a rail. The
612
/// default view is the subject and a metadata line; `verbose` adds the full
613
/// message, dependencies, and the authenticating key.
c6aa5bd9 Andy McCall ↶ 614
fn signature_json(repo: &Repo, h: &oxy_core::hash::ChangeHash) -> Result<String> {
615
    Ok(match repo.authenticate_change(h)? {
616
        SignatureStatus::Unsigned => json::obj([("status", json::s("unsigned"))]),
75167672 Andy McCall 617
        SignatureStatus::Valid {
618
            key,
619
            name,
620
            email,
621
            sealed,
622
            revoked,
623
        } => json::obj([
c6aa5bd9 Andy McCall ↶ 624
            ("status", json::s(if sealed { "sealed" } else { "signed" })),
625
            ("key", json::s(&key.to_hex())),
626
            ("name", json::s(&name)),
627
            ("email", json::s(&email)),
628
            ("revoked", json::b(revoked)),
629
        ]),
75167672 Andy McCall 630
        SignatureStatus::Invalid(why) => {
631
            json::obj([("status", json::s("invalid")), ("reason", json::s(&why))])
632
        }
c6aa5bd9 Andy McCall ↶ 633
    })
634
}
635
636
fn log_entry_json(repo: &Repo, e: &oxy_core::repo::LogEntry) -> Result<String> {
637
    let c = &e.change;
638
    Ok(json::obj([
639
        ("hash", json::s(&e.hash.to_hex())),
640
        ("change_id", json::s(&c.change_id.to_hex())),
641
        ("author", json::s(&author_name(repo, c))),
642
        ("author_key", json::s(&c.author.to_hex())),
5cde7882 Andy McCall ↶ 643
        (
644
            "via",
645
            match via(
646
                repo,
647
                &repo.authenticate_change_known(&e.hash, Some(c))?,
648
                &c.author,
649
            ) {
650
                Some(v) => json::obj([
651
                    ("name", json::s(&v.name)),
652
                    ("email", json::s(&v.email)),
653
                    ("key", json::s(&v.key.to_hex())),
654
                    ("scope", json::opt(v.scope.as_deref(), json::s)),
655
                    ("expires_ms", json::opt(v.expires, json::n)),
656
                    ("revoked", json::b(v.revoked)),
657
                ]),
658
                None => "null".to_string(),
659
            },
660
        ),
c6aa5bd9 Andy McCall ↶ 661
        ("when_ms", json::n(c.hlc.wall_ms)),
662
        ("message", json::s(&c.message)),
663
        ("kind", json::s(&format!("{:?}", c.kind).to_lowercase())),
664
        ("deps", json::hashes(&c.deps)),
665
        ("head", json::b(e.is_head)),
666
        ("working_copy", json::b(e.is_working_copy)),
75167672 Andy McCall 667
        (
668
            "signature",
669
            if e.is_working_copy {
670
                "null".into()
671
            } else {
672
                signature_json(repo, &e.hash)?
673
            },
674
        ),
c6aa5bd9 Andy McCall ↶ 675
    ]))
676
}
677
83d61e80 Andy McCall ↶ 678
/// `START,END:PATH` for `oxy log -L`.
679
fn parse_line_range(spec: &str) -> Result<(usize, usize, &str)> {
680
    let bad = || anyhow::anyhow!("-L {spec}: expected START,END:PATH, as 10,24:src/lib.rs");
681
    let (range, path) = spec.split_once(':').ok_or_else(bad)?;
682
    let (a, b) = range.split_once(',').ok_or_else(bad)?;
683
    Ok((
684
        a.trim().parse().map_err(|_| bad())?,
685
        b.trim().parse().map_err(|_| bad())?,
686
        path,
687
    ))
688
}
689
690
pub fn log(
47ce8b8e Andy McCall ↶ 691
    revset: Option<&str>,
83d61e80 Andy McCall ↶ 692
    limit: Option<usize>,
693
    verbose: bool,
694
    path: Option<&str>,
695
    lines: Option<&str>,
696
    as_json: bool,
697
) -> Result<i32> {
fce44ae9 Andy McCall ↶ 698
    let repo = open()?;
69816d68 Andy McCall ↶ 699
    // Whether the working copy is clean and conflicted: from the manifest
700
    // when it proves the working copy clean, else from a snapshot.
33472ccc Andy McCall ↶ 701
    // The working copy as last recorded, without looking at the files (as
702
    // Git's log does not): whether its change holds edits, and whether its
703
    // base was conflicted. `oxy status` is what checks the disk.
704
    let conflicted = !repo.is_at_op() && repo.recorded_conflicted()?;
4cce912a Andy McCall ↶ 705
    let listing = oxy_core::timing::phase("log: entries");
83d61e80 Andy McCall ↶ 706
    let entries = match (path, lines) {
47ce8b8e Andy McCall ↶ 707
        _ if revset.is_some() => repo.log_revset(revset.unwrap_or_default())?,
83d61e80 Andy McCall ↶ 708
        (_, Some(spec)) => {
709
            let (first, last, p) = parse_line_range(spec)?;
69816d68 Andy McCall ↶ 710
            let (_, wc) = repo.working_copy()?;
711
            let base = repo.load_state_history(&wc.deps)?.maximal(&wc.deps);
712
            repo.line_history(&base, p, first, last)?
83d61e80 Andy McCall ↶ 713
        }
714
        (Some(p), None) => repo.log_for_path(p)?,
715
        (None, None) => repo.log()?,
c6aa5bd9 Andy McCall ↶ 716
    };
4cce912a Andy McCall ↶ 717
    drop(listing);
718
    let _print = oxy_core::timing::phase("log: print");
c6aa5bd9 Andy McCall ↶ 719
    if as_json {
720
        let shown = limit.unwrap_or(entries.len()).min(entries.len());
721
        let mut items = Vec::new();
722
        for e in entries.iter().take(shown) {
723
            items.push(log_entry_json(&repo, e)?);
724
        }
75167672 Andy McCall 725
        println!(
726
            "{}",
727
            json::obj([
728
                ("bookmark", json::s(&repo.config.bookmark)),
729
                ("total", json::n(entries.len())),
730
                ("changes", json::arr(items))
731
            ])
732
        );
c6aa5bd9 Andy McCall ↶ 733
        return Ok(0);
734
    }
94084087 Andy McCall ↶ 735
    let total = entries.len();
736
    let shown = limit.unwrap_or(total).min(total);
2bf22305 Andy McCall ↶ 737
    // Lay out a little past what is shown, so edges into the cut are drawn.
738
    let window: Vec<(ChangeHash, Vec<ChangeHash>)> = entries
739
        .iter()
740
        .take((shown + MAX_EDGE).min(total))
741
        .map(|e| (e.hash, e.change.deps.clone()))
742
        .collect();
743
    let rows = oxy_core::lanes::graph(&window, MAX_EDGE);
744
    // Highlight the shortest prefix that names each change: against the
745
    // whole log, which is what `resolve_change` searches.
746
    let everything;
47ce8b8e Andy McCall ↶ 747
    let all = if path.is_some() || lines.is_some() || revset.is_some() {
2bf22305 Andy McCall ↶ 748
        everything = repo.log()?;
749
        &everything
750
    } else {
751
        &entries
752
    };
79b5a90f Andy McCall ↶ 753
    let prefixes = oxy_core::timing::phase("log: id prefixes");
754
    let id_prefix = id_prefixes(
755
        all.iter().map(|e| e.change.change_id.0).collect(),
756
        entries.iter().take(shown).map(|e| e.change.change_id.0),
757
    );
758
    drop(prefixes);
2bf22305 Andy McCall ↶ 759
    let labels = reference_labels(&repo)?;
760
    for (e, row) in entries.iter().take(shown).zip(&rows) {
fce44ae9 Andy McCall ↶ 761
        let c = &e.change;
75167672 Andy McCall 762
        let bullet = ui::log_bullet(
763
            e.is_working_copy,
764
            e.is_head,
765
            e.is_working_copy && conflicted,
766
        );
2bf22305 Andy McCall ↶ 767
        let id = c.change_id.to_hex();
79b5a90f Andy McCall ↶ 768
        let n = id_prefix.get(&c.change_id.0).copied().unwrap_or(8).min(8);
2bf22305 Andy McCall ↶ 769
        let mut head = vec![format!("{}{}", ui::id(&id[..n]), ui::dim(&id[n..8]))];
770
        head.push(ui::author(author_name(&repo, c)));
771
        head.push(ui::dim(ui::ago(c.hlc.wall_ms)));
772
        if let Some(names) = labels.get(&e.hash) {
773
            head.push(names.join(" "));
fce44ae9 Andy McCall ↶ 774
        }
94084087 Andy McCall ↶ 775
        if c.kind == ChangeKind::Resolution {
2bf22305 Andy McCall ↶ 776
            head.push(ui::warn("resolution"));
fce44ae9 Andy McCall ↶ 777
        }
2bf22305 Andy McCall ↶ 778
        if !e.is_working_copy {
779
            head.push(ui::hash(format!("#{}", e.hash.short(8))));
5cde7882 Andy McCall ↶ 780
            let status = repo.authenticate_change_known(&e.hash, Some(c))?;
781
            if let Some(agent) = via(&repo, &status, &c.author) {
782
                head.push(ui::dim(format!("via {}", agent.name)));
783
            }
784
            head.push(signature_mark_of(&status));
2bf22305 Andy McCall ↶ 785
        }
786
        println!("{}{}", node_line(row, &bullet), head.join(" "));
94084087 Andy McCall ↶ 787
        let subject = if e.is_working_copy && c.message.is_empty() {
33472ccc Andy McCall ↶ 788
            let edited = repo
789
                .store()
790
                .get_change(&e.hash)
791
                .is_ok_and(|w| !w.tree_ops.is_empty() || !w.edits.is_empty());
792
            ui::dim(if edited {
2bf22305 Andy McCall ↶ 793
                "(working copy, uncommitted changes)"
33472ccc Andy McCall ↶ 794
            } else {
795
                "(working copy)"
75167672 Andy McCall 796
            })
2bf22305 Andy McCall ↶ 797
        } else if c.message.trim().is_empty() {
798
            ui::dim("(no description)")
94084087 Andy McCall ↶ 799
        } else {
2bf22305 Andy McCall ↶ 800
            subject_of(&c.message).to_string()
94084087 Andy McCall ↶ 801
        };
2bf22305 Andy McCall ↶ 802
        println!("{}{subject}", below_line(row));
94084087 Andy McCall ↶ 803
        if verbose {
2bf22305 Andy McCall ↶ 804
            let rail = continuation(row);
94084087 Andy McCall ↶ 805
            if !e.is_working_copy {
75167672 Andy McCall 806
                println!(
2bf22305 Andy McCall ↶ 807
                    "{rail}{}",
75167672 Andy McCall 808
                    ui::dim(format!(
809
                        "{} {}",
810
                        when(c.hlc.wall_ms),
811
                        signature_line(&repo, &e.hash)?
812
                    ))
813
                );
94084087 Andy McCall ↶ 814
            }
815
            if !c.deps.is_empty() {
75167672 Andy McCall 816
                println!(
2bf22305 Andy McCall ↶ 817
                    "{rail}{} {}",
75167672 Andy McCall 818
                    ui::dim(ui::glyph("↳", "->")),
819
                    ui::dim(deps_of(&c.deps))
820
                );
94084087 Andy McCall ↶ 821
            }
75167672 Andy McCall 822
            let body: Vec<&str> = c
823
                .message
824
                .lines()
825
                .skip(1)
826
                .skip_while(|l| l.trim().is_empty())
827
                .collect();
2bf22305 Andy McCall ↶ 828
            for line in body {
829
                println!("{rail}  {line}");
94084087 Andy McCall ↶ 830
            }
fce44ae9 Andy McCall ↶ 831
        }
832
    }
94084087 Andy McCall ↶ 833
    if shown < total {
2bf22305 Andy McCall ↶ 834
        if let Some(last) = rows.get(shown.saturating_sub(1)) {
835
            let cut: String = last
836
                .below
837
                .iter()
838
                .map(|c| match c {
839
                    Cell::Blank | Cell::Elided | Cell::JoinRight => "  ",
840
                    _ => "~ ",
841
                })
842
                .collect();
843
            if cut.trim().is_empty() {
844
                println!("{}", ui::dim("~"));
845
            } else {
846
                println!("{}", ui::dim(cut.trim_end()));
847
            }
848
        }
75167672 Andy McCall 849
        println!(
850
            "{}",
851
            ui::dim(format!(
852
                "{} {} more; pass -n to see them",
853
                ui::glyph("…", "..."),
854
                total - shown
855
            ))
856
        );
94084087 Andy McCall ↶ 857
    }
fce44ae9 Andy McCall ↶ 858
    Ok(0)
2bf22305 Andy McCall ↶ 859
}
860
861
/// Rows a drawn edge may span in `oxy log`; a longer one is elided (`~`).
862
const MAX_EDGE: usize = 12;
863
864
use oxy_core::hash::ChangeHash;
865
use oxy_core::lanes::{Cell, GraphRow};
866
867
/// A graph cell, two columns wide.
868
fn cell(c: Cell) -> &'static str {
869
    match c {
870
        Cell::Blank => "  ",
871
        Cell::Line => ui::glyph("│ ", "| "),
872
        Cell::Elided => "~ ",
873
        Cell::JoinLeft => ui::glyph("├─", "|-"),
874
        Cell::JoinPass => ui::glyph("──", "--"),
875
        Cell::JoinCross => ui::glyph("┼─", "+-"),
876
        Cell::JoinRight => ui::glyph("╯ ", "' "),
877
    }
878
}
879
880
/// The graph part of a change's own line, its bullet in its column.
881
fn node_line(row: &GraphRow, bullet: &str) -> String {
882
    let width = row
883
        .passing
884
        .iter()
885
        .copied()
886
        .max()
887
        .unwrap_or(0)
888
        .max(row.column)
889
        + 1;
890
    let mut out = String::new();
891
    for c in 0..width {
892
        if c == row.column {
893
            out.push_str(bullet);
894
            out.push(' ');
895
        } else if row.passing.contains(&c) {
896
            out.push_str(&ui::dim(cell(Cell::Line)));
897
        } else {
898
            out.push_str("  ");
899
        }
900
    }
901
    out.push(' ');
902
    out
903
}
904
905
/// The graph part of the line under a change: lanes carrying on, a join,
906
/// or `~` for an elided edge.
907
fn below_line(row: &GraphRow) -> String {
908
    let g: String = row.below.iter().map(|c| cell(*c)).collect();
909
    format!("{} ", ui::dim(g))
910
}
911
912
/// The graph part of any further line of a change (`-v`).
913
fn continuation(row: &GraphRow) -> String {
914
    let g: String = row
915
        .below
916
        .iter()
917
        .map(|c| match c {
918
            Cell::Line | Cell::JoinLeft | Cell::JoinCross => cell(Cell::Line),
919
            _ => "  ",
920
        })
921
        .collect();
922
    format!("{} ", ui::dim(g))
923
}
924
925
/// For each hex ID, the length of its shortest prefix no other ID shares.
79b5a90f Andy McCall ↶ 926
/// The shortest prefix, in hex digits, that names each of `wanted` among
927
/// `all`: worked out for the few shown rather than for the whole history.
928
fn id_prefixes(
929
    mut all: Vec<[u8; 16]>,
930
    wanted: impl Iterator<Item = [u8; 16]>,
931
) -> std::collections::HashMap<[u8; 16], usize> {
932
    all.sort_unstable();
933
    all.dedup();
934
    // Hex digits two ids share at the start.
935
    let common = |a: &[u8; 16], b: &[u8; 16]| {
936
        let bytes = a.iter().zip(b).take_while(|(x, y)| x == y).count();
937
        match (a.get(bytes), b.get(bytes)) {
938
            (Some(x), Some(y)) if x >> 4 == y >> 4 => 2 * bytes + 1,
939
            _ => 2 * bytes,
940
        }
941
    };
942
    let mut out = std::collections::HashMap::new();
943
    for id in wanted {
944
        let Ok(i) = all.binary_search(&id) else {
945
            continue;
946
        };
947
        let before = i.checked_sub(1).map_or(0, |j| common(&all[j], &id));
948
        let after = all.get(i + 1).map_or(0, |n| common(&id, n));
949
        out.insert(id, before.max(after) + 1);
950
    }
951
    out
952
}
953
2bf22305 Andy McCall ↶ 954
fn unique_prefixes(ids: impl Iterator<Item = String>) -> std::collections::HashMap<String, usize> {
955
    let mut v: Vec<String> = ids.collect();
956
    v.sort();
957
    v.dedup();
958
    let common = |a: &str, b: &str| a.bytes().zip(b.bytes()).take_while(|(x, y)| x == y).count();
959
    let mut out = std::collections::HashMap::with_capacity(v.len());
960
    for i in 0..v.len() {
961
        let before = if i > 0 { common(&v[i - 1], &v[i]) } else { 0 };
962
        let after = if i + 1 < v.len() {
963
            common(&v[i], &v[i + 1])
964
        } else {
965
            0
966
        };
967
        out.insert(v[i].clone(), before.max(after) + 1);
968
    }
969
    out
970
}
971
972
/// Bookmark, remote bookmark and tag names to show beside the changes they
973
/// point at. A name is shown on each head of its state; one with more than
974
/// three heads names too much of the log to help, so it is left out (oxy
975
/// bookmark list shows it). A remote bookmark at the same state as the
976
/// local one of that name is not repeated.
977
fn reference_labels(repo: &Repo) -> Result<std::collections::HashMap<ChangeHash, Vec<String>>> {
978
    const MAX_HEADS: usize = 3;
979
    let view = repo.view()?;
980
    let mut out: std::collections::HashMap<ChangeHash, Vec<String>> = Default::default();
981
    let mut add = |heads: Vec<ChangeHash>, label: String| {
982
        if heads.len() <= MAX_HEADS {
983
            for h in heads {
984
                out.entry(h).or_default().push(label.clone());
985
            }
986
        }
987
    };
988
    for (name, st) in &view.bookmarks {
989
        add(repo.store().get_state(st)?.heads, ui::bookmark(name));
990
    }
991
    for (remote, bookmarks) in &view.remotes {
992
        for (name, st) in bookmarks {
993
            if view.bookmarks.get(name) == Some(st) {
994
                continue;
995
            }
996
            add(
997
                repo.store().get_state(st)?.heads,
998
                ui::dim(format!("{remote}/{name}")),
999
            );
1000
        }
1001
    }
1002
    for t in repo.tags()? {
1003
        add(t.heads.clone(), ui::warn(&t.name));
1004
    }
1005
    Ok(out)
fce44ae9 Andy McCall ↶ 1006
}
1007
c6aa5bd9 Andy McCall ↶ 1008
pub fn diff(change: Option<&str>, as_json: bool) -> Result<i32> {
1009
    let repo = open()?;
1010
    let hash = change.map(|c| repo.resolve_change(c)).transpose()?;
1011
    if as_json {
1012
        let files = repo.diff_files(hash.as_ref())?;
1013
        let items = files.iter().map(|f| {
1014
            json::obj([
1015
                ("path", json::s(&f.path)),
1016
                ("kind", json::s(&f.kind)),
1017
                ("from", json::opt(f.from.as_deref(), json::s)),
1018
                (
1019
                    "hunks",
1020
                    json::arr(f.hunks.iter().map(|h| {
1021
                        json::obj([
1022
                            ("base_line", json::n(h.base_line)),
75167672 Andy McCall 1023
                            (
1024
                                "added",
1025
                                json::arr(
1026
                                    h.added.iter().map(|l| json::s(&String::from_utf8_lossy(l))),
1027
                                ),
1028
                            ),
1029
                            (
1030
                                "removed",
1031
                                json::arr(h.removed.iter().map(|(n, l)| {
1032
                                    json::obj([
1033
                                        ("line", json::n(*n)),
1034
                                        ("text", json::s(&String::from_utf8_lossy(l))),
1035
                                    ])
1036
                                })),
1037
                            ),
c6aa5bd9 Andy McCall ↶ 1038
                        ])
1039
                    })),
1040
                ),
1041
            ])
1042
        });
75167672 Andy McCall 1043
        println!(
1044
            "{}",
1045
            json::obj([
1046
                ("change", json::opt(hash, |h| json::s(&h.to_hex()))),
1047
                ("files", json::arr(items))
1048
            ])
1049
        );
c6aa5bd9 Andy McCall ↶ 1050
        return Ok(0);
1051
    }
1052
    match hash {
1053
        Some(h) => {
1054
            let c = repo.store().get_change(&h)?;
8ab265dc Andy McCall ↶ 1055
            let (base, snapshot) = repo.change_base(&c)?;
c6aa5bd9 Andy McCall ↶ 1056
            print!("{}", render_change(&c, &base, &snapshot));
1057
        }
1058
        None => {
77445673 Andy McCall ↶ 1059
            let (wc, base, snapshot) = repo.working_copy_with_base()?;
10b4fd43 Andy McCall ↶ 1060
            let text = {
1061
                let _t = oxy_core::timing::phase("diff: render the change");
1062
                render_change(&wc, &base, &snapshot)
1063
            };
1064
            print!("{text}");
1065
            // The process ends here: freeing the base's tree and graphs
1066
            // one by one would only cost time (about 27 ms on Linux).
1067
            std::mem::forget((wc, base, snapshot));
c6aa5bd9 Andy McCall ↶ 1068
        }
1069
    }
1070
    Ok(0)
1071
}
1072
1073
pub fn cat(path: &str, change: Option<&str>) -> Result<i32> {
1074
    let repo = open()?;
1075
    let heads = change.map(|c| repo.resolve_heads(c)).transpose()?;
1076
    let bytes = repo.file_at(path, heads.as_deref())?;
1077
    use std::io::Write as _;
1078
    std::io::stdout().write_all(&bytes)?;
1079
    Ok(0)
1080
}
1081
1082
pub fn root(as_json: bool) -> Result<i32> {
1083
    let Ok(repo) = Repo::open(&std::env::current_dir()?) else {
1084
        if as_json {
1085
            println!("null");
1086
        }
1087
        return Ok(1);
1088
    };
1089
    if as_json {
75167672 Andy McCall 1090
        println!(
1091
            "{}",
1092
            json::obj([
1093
                ("root", json::s(&repo.root().to_string_lossy())),
1094
                ("workspace", json::s(&repo.workspace().to_hex())),
1095
                ("bookmark", json::s(&repo.config.bookmark))
1096
            ])
1097
        );
c6aa5bd9 Andy McCall ↶ 1098
    } else {
1099
        println!("{}", repo.root().display());
1100
    }
1101
    Ok(0)
1102
}
1103
1104
pub fn version(as_json: bool) -> Result<i32> {
1105
    if as_json {
75167672 Andy McCall 1106
        println!(
1107
            "{}",
1108
            json::obj([
1109
                ("version", json::s(env!("CARGO_PKG_VERSION"))),
e293c90d Andy McCall ↶ 1110
                // What was built: a release's or install's `oxy describe`,
1111
                // a Flow build's heads, or "local build".
1112
                ("build", json::s(env!("OXY_BUILD_ID"))),
75167672 Andy McCall 1113
                ("protocol", json::n(oxy_core::protocol::PROTOCOL_VERSION)),
1114
                ("format", json::n(2))
1115
            ])
1116
        );
c6aa5bd9 Andy McCall ↶ 1117
    } else {
75167672 Andy McCall 1118
        println!(
1119
            "oxy {} (repository format 2, protocol {})",
e293c90d Andy McCall ↶ 1120
            env!("OXY_VERSION_LONG"),
75167672 Andy McCall 1121
            oxy_core::protocol::PROTOCOL_VERSION
1122
        );
c6aa5bd9 Andy McCall ↶ 1123
    }
1124
    Ok(0)
1125
}
1126
1127
pub fn discard(paths: &[String], hunks: &[usize]) -> Result<i32> {
1128
    let repo = open()?;
1129
    if paths.is_empty() && hunks.is_empty() {
1130
        anyhow::bail!("name paths to restore, or --hunks to undo chosen hunks");
1131
    }
1132
    let mut n = 0;
1133
    if !hunks.is_empty() {
1134
        n += repo.discard_hunks(hunks)?;
1135
    }
1136
    if !paths.is_empty() {
1137
        n += repo.discard(paths)?;
1138
    }
75167672 Andy McCall 1139
    println!(
1140
        "{} {} restored to the working copy's base",
1141
        ui::done("Discarded"),
1142
        plural(n, "path")
1143
    );
c6aa5bd9 Andy McCall ↶ 1144
    Ok(0)
1145
}
1146
1147
pub fn hunks(as_json: bool) -> Result<i32> {
fce44ae9 Andy McCall ↶ 1148
    let repo = open()?;
c6aa5bd9 Andy McCall ↶ 1149
    let (_, hunks) = repo.hunks()?;
1150
    if as_json {
75167672 Andy McCall 1151
        println!(
1152
            "{}",
1153
            json::arr(hunks.iter().map(|h| json::obj([
1154
                ("index", json::n(h.index)),
1155
                ("path", json::s(&h.path)),
1156
                ("text", json::s(&h.text))
1157
            ])))
1158
        );
c6aa5bd9 Andy McCall ↶ 1159
        return Ok(0);
1160
    }
1161
    if hunks.is_empty() {
75167672 Andy McCall 1162
        println!(
1163
            "{}",
1164
            ui::dim("No hunks: the working copy matches its base.")
1165
        );
c6aa5bd9 Andy McCall ↶ 1166
    }
1167
    for h in &hunks {
75167672 Andy McCall 1168
        println!(
1169
            "{} {}",
1170
            ui::section(format!("hunk {}", h.index)),
1171
            ui::path(&h.path)
1172
        );
c6aa5bd9 Andy McCall ↶ 1173
        for line in h.text.lines() {
75167672 Andy McCall 1174
            let styled = if line.starts_with('+') {
1175
                ui::ok(line)
1176
            } else if line.starts_with('-') {
1177
                ui::err(line)
1178
            } else {
1179
                ui::dim(line)
1180
            };
c6aa5bd9 Andy McCall ↶ 1181
            println!("  {styled}");
1182
        }
1183
    }
fce44ae9 Andy McCall ↶ 1184
    Ok(0)
1185
}
1186
0299076f Andy McCall ↶ 1187
pub fn show(change: Option<&str>, revset: Option<&str>, stat: bool, as_json: bool) -> Result<i32> {
fce44ae9 Andy McCall ↶ 1188
    let repo = open()?;
cc7a0056 Andy McCall ↶ 1189
    repo.snapshot_status("oxy show")?;
2c9ddac3 Andy McCall ↶ 1190
    let Some(r) = revset else {
1191
        let h = repo.resolve_change(change.unwrap_or("@"))?;
0299076f Andy McCall ↶ 1192
        return show_one(&repo, h, as_json, false, stat);
2c9ddac3 Andy McCall ↶ 1193
    };
1194
    // Several changes: a JSON array, or the changes one after another.
1195
    let hashes = repo.revset(r)?.hashes;
1196
    if as_json {
1197
        print!("[");
1198
        for (i, h) in hashes.iter().enumerate() {
1199
            if i > 0 {
1200
                print!(",");
1201
            }
0299076f Andy McCall ↶ 1202
            show_one(&repo, *h, true, true, stat)?;
2c9ddac3 Andy McCall ↶ 1203
        }
1204
        println!("]");
1205
        return Ok(0);
1206
    }
1207
    if hashes.is_empty() {
1208
        println!("{}", ui::dim(format!("No change matches {r:?}.")));
1209
        return Ok(0);
1210
    }
1211
    for (i, h) in hashes.iter().enumerate() {
1212
        if i > 0 {
1213
            println!();
1214
        }
0299076f Andy McCall ↶ 1215
        show_one(&repo, *h, false, false, stat)?;
2c9ddac3 Andy McCall ↶ 1216
    }
1217
    Ok(0)
1218
}
1219
0299076f Andy McCall ↶ 1220
/// Lines added and removed in one file's diff.
1221
fn counts(f: &oxy_core::repo::FileDiff) -> (usize, usize) {
1222
    f.hunks
1223
        .iter()
1224
        .fold((0, 0), |(a, r), h| (a + h.added.len(), r + h.removed.len()))
1225
}
1226
1227
/// What `--stat` prints (oxygen#119): each file the change touched, its
1228
/// lines added and removed, then the total.
1229
fn print_stat(files: &[oxy_core::repo::FileDiff]) {
1230
    let width = files
1231
        .iter()
1232
        .map(|f| f.path.chars().count())
1233
        .max()
1234
        .unwrap_or(0)
1235
        .min(60);
1236
    let (mut added, mut removed) = (0, 0);
1237
    for f in files {
1238
        let (a, r) = counts(f);
1239
        added += a;
1240
        removed += r;
1241
        let path = match &f.from {
1242
            Some(from) => format!("{from} → {}", f.path),
1243
            None => f.path.clone(),
1244
        };
1245
        println!(
1246
            " {path:<width$}  {} {}  {}",
1247
            ui::ok(format!("{:>6}", format!("+{a}"))),
1248
            ui::err(format!("{:>6}", format!("-{r}"))),
1249
            ui::dim(&f.kind)
1250
        );
1251
    }
1252
    println!(
1253
        " {}",
1254
        ui::dim(format!(
1255
            "{} changed, {added} added, {removed} removed",
1256
            plural(files.len(), "file")
1257
        ))
1258
    );
1259
}
1260
2c9ddac3 Andy McCall ↶ 1261
/// One change, as `oxy show` prints it; `inline` JSON goes without its
0299076f Andy McCall ↶ 1262
/// newline, for an array. `stat` lists the files instead of the diff.
1263
fn show_one(repo: &Repo, h: ChangeHash, as_json: bool, inline: bool, stat: bool) -> Result<i32> {
fce44ae9 Andy McCall ↶ 1264
    let c = repo.store().get_change(&h)?;
c6aa5bd9 Andy McCall ↶ 1265
    if as_json {
1266
        let files = repo.diff_files(Some(&h))?;
1267
        let heads = repo.bookmark_heads()?;
75167672 Andy McCall 1268
        let entry = oxy_core::repo::LogEntry {
1269
            hash: h,
1270
            change: oxy_core::state::ChangeMeta::of(&c),
1271
            is_working_copy: c.kind == oxy_core::objects::ChangeKind::WorkingCopy,
1272
            is_head: heads.contains(&h),
1273
        };
2c9ddac3 Andy McCall ↶ 1274
        let mut body = log_entry_json(repo, &entry)?;
c6aa5bd9 Andy McCall ↶ 1275
        body.pop();
75167672 Andy McCall 1276
        body.push_str(&format!(
1277
            ",{}:{}}}",
1278
            json::s("files"),
0299076f Andy McCall ↶ 1279
            json::arr(files.iter().map(|f| {
1280
                let (a, r) = counts(f);
1281
                json::obj([
1282
                    ("path", json::s(&f.path)),
1283
                    ("kind", json::s(&f.kind)),
1284
                    ("from", json::opt(f.from.as_deref(), json::s)),
1285
                    ("added", a.to_string()),
1286
                    ("removed", r.to_string()),
1287
                ])
1288
            }))
75167672 Andy McCall 1289
        ));
2c9ddac3 Andy McCall ↶ 1290
        if inline {
1291
            print!("{body}");
1292
        } else {
1293
            println!("{body}");
1294
        }
c6aa5bd9 Andy McCall ↶ 1295
        return Ok(0);
1296
    }
75167672 Andy McCall 1297
    println!(
1298
        "{} {} {}",
1299
        ui::section("change"),
1300
        ui::id(short_id(&c)),
1301
        ui::hash(format!("#{h}"))
1302
    );
94084087 Andy McCall ↶ 1303
    println!("{}", ui::kv("kind", format!("{:?}", c.kind).to_lowercase()));
b35df36b Andy McCall ↶ 1304
    // Fetched but not pulled (oxygen#56): say where it is.
1305
    if let Some(at) = repo.remote_only_location(&h)? {
1306
        println!(
1307
            "{}",
1308
            ui::kv(
1309
                "only in",
1310
                ui::warn(format!(
1311
                    "{at}: fetched, not yet in a local bookmark (oxy pull)"
1312
                ))
1313
            )
1314
        );
1315
    }
5cde7882 Andy McCall ↶ 1316
    let meta = oxy_core::state::ChangeMeta::of(&c);
1317
    let status = repo.authenticate_change_known(&h, Some(&meta))?;
1318
    println!("{}", ui::kv("author", ui::author(author_name(repo, &meta))));
1319
    if let Some(agent) = via(repo, &status, &meta.author) {
1320
        println!("{}", ui::kv("via", agent.line()));
1321
    }
75167672 Andy McCall 1322
    println!(
1323
        "{}",
1324
        ui::kv(
1325
            "date",
1326
            format!(
1327
                "{} {}",
1328
                when(c.hlc.wall_ms),
1329
                ui::dim(format!("({})", ui::ago(c.hlc.wall_ms)))
1330
            )
1331
        )
1332
    );
94084087 Andy McCall ↶ 1333
    println!("{}", ui::kv("deps", ui::dim(deps_line(&c))));
2c9ddac3 Andy McCall ↶ 1334
    println!("{}", ui::kv("sig", signature_line(repo, &h)?));
94084087 Andy McCall ↶ 1335
    println!();
75167672 Andy McCall 1336
    for line in (if c.message.is_empty() {
1337
        "(no description)"
1338
    } else {
1339
        &c.message
1340
    })
1341
    .lines()
1342
    {
94084087 Andy McCall ↶ 1343
        println!("    {line}");
1344
    }
1345
    println!();
0299076f Andy McCall ↶ 1346
    if stat {
1347
        print_stat(&repo.diff_files(Some(&h))?);
1348
    } else {
1349
        print!("{}", repo.describe_change(&h)?);
1350
    }
fce44ae9 Andy McCall ↶ 1351
    Ok(0)
1352
}
1353
1c2bd9ab Andy McCall ↶ 1354
pub fn checkout(changes: &[String], discard: bool, carry: bool) -> Result<i32> {
fce44ae9 Andy McCall ↶ 1355
    let repo = open()?;
a46f3e4f Andy McCall ↶ 1356
    // `oxy checkout BOOKMARK` is being on that bookmark, as `git checkout
1357
    // BRANCH` is: checking its files out under another bookmark's name
1358
    // made the next commit move the wrong bookmark (oxygen#60).
1359
    if let [name] = changes {
1360
        if *name != repo.config.bookmark && names_bookmark(&repo, name)? {
1361
            drop(repo);
1362
            return bookmark_switch(name, discard, carry);
1363
        }
1364
    }
fce44ae9 Andy McCall ↶ 1365
    let heads = if changes.is_empty() {
1366
        repo.bookmark_heads()?
1367
    } else {
132b884b Andy McCall ↶ 1368
        let mut heads = Vec::new();
1369
        for c in changes {
1370
            heads.extend(repo.resolve_heads(c)?);
1371
        }
1372
        heads
fce44ae9 Andy McCall ↶ 1373
    };
75167672 Andy McCall 1374
    let spec = if changes.is_empty() {
1375
        repo.config.bookmark.clone()
1376
    } else {
1377
        changes.join(" ")
1378
    };
135212de Andy McCall ↶ 1379
    let sp = ui::spinner(&format!("Checking out {spec}…"));
1c2bd9ab Andy McCall ↶ 1380
    let command = format!("oxy checkout {spec}");
1381
    let (r, carried) = if carry {
1382
        repo.checkout_carry(&heads, &spec, &command)?
1383
    } else {
1384
        (repo.checkout(&heads, discard, &command)?, vec![])
1385
    };
75167672 Andy McCall 1386
    sp.done(&format!(
1c2bd9ab Andy McCall ↶ 1387
        "{} on {}: {} written, {} removed{}",
75167672 Andy McCall 1388
        ui::done("Working copy"),
1389
        ui::bookmark(&spec),
1390
        plural(r.written, "file"),
1c2bd9ab Andy McCall ↶ 1391
        r.removed,
1392
        if carried.is_empty() {
1393
            String::new()
1394
        } else {
1395
            format!("; carried {}", plural(carried.len(), "changed path"))
1396
        }
75167672 Andy McCall 1397
    ));
9b723f69 Andy McCall ↶ 1398
    print_removed(&r);
4dee277c Andy McCall ↶ 1399
    for n in &r.notes {
1400
        println!("  {} {n}", ui::note());
1401
    }
fce44ae9 Andy McCall ↶ 1402
    if r.conflicted {
75167672 Andy McCall 1403
        println!(
1404
            "  {} {}",
1405
            ui::caution(),
1406
            ui::warn("this state has conflicts; see oxy status")
1407
        );
fce44ae9 Andy McCall ↶ 1408
    }
98a1620a Andy McCall ↶ 1409
    where_now(&repo);
fce44ae9 Andy McCall ↶ 1410
    Ok(0)
1411
}
1412
157d02e9 Andy McCall ↶ 1413
pub fn resolve(paths: &[String], as_json: bool) -> Result<i32> {
fce44ae9 Andy McCall ↶ 1414
    let repo = open()?;
157d02e9 Andy McCall ↶ 1415
    let mut snap = repo.snapshot("oxy resolve")?;
1416
    if !paths.is_empty() {
1417
        // A replace conflict has no markers to edit out: the user says which
1418
        // content wins (oxygen#140, spec §9.3), and the snapshot builds it.
1419
        let paths: Vec<String> = paths
1420
            .iter()
1421
            .map(|p| p.trim_start_matches("./").trim_end_matches('/').to_string())
1422
            .collect();
1423
        repo.choose_resolutions(&snap.base_snapshot, &paths)?;
1424
        snap = repo.snapshot("oxy resolve")?;
1425
    }
c6aa5bd9 Andy McCall ↶ 1426
    if as_json {
1427
        println!("{}", repo.status_json(&snap)?);
75167672 Andy McCall 1428
        return Ok(if snap.base_snapshot.is_conflicted() {
1429
            2
1430
        } else {
1431
            0
1432
        });
c6aa5bd9 Andy McCall ↶ 1433
    }
fce44ae9 Andy McCall ↶ 1434
    if !snap.base_snapshot.is_conflicted() {
1435
        println!("The working copy's base has no conflicts.");
1436
        return Ok(0);
1437
    }
1438
    let mut remaining = 0;
1439
    for (p, f) in &snap.base_snapshot.files {
1440
        if !f.is_conflicted() {
1441
            continue;
1442
        }
75167672 Andy McCall 1443
        match snap
1444
            .statuses
1445
            .iter()
1446
            .find(|s| s.path == *p)
1447
            .map(|s| s.status)
1448
        {
fce44ae9 Andy McCall ↶ 1449
            Some(ChangeStatus::Resolved) => println!("resolved   {p}"),
1450
            Some(ChangeStatus::ConflictEdited) => {
1451
                remaining += 1;
1452
                println!("markers    {p}");
1453
            }
1454
            _ => {
1455
                remaining += 1;
1456
                println!("unresolved {p}");
1457
            }
1458
        }
1459
    }
1460
    for c in &snap.base_snapshot.tree.conflicts {
1461
        remaining += 1;
1462
        println!("tree       {c:?}");
1463
    }
1464
    if remaining == 0 {
1465
        println!("All conflicts are resolved; `oxy commit` will record the resolution.");
1466
        Ok(0)
1467
    } else {
1468
        Ok(2)
1469
    }
1470
}
1471
1472
pub fn undo() -> Result<i32> {
1473
    let repo = open()?;
530a0fc6 Andy McCall ↶ 1474
    let u = repo.undo()?;
1475
    println!("{} {}", ui::done("Undone:"), u.command);
1476
    for c in &u.changes {
1477
        println!("  {} {}", ui::note(), change_words(&repo, c));
1478
    }
1479
    if u.others_kept > 0 {
1480
        let s = if u.others_kept == 1 { "" } else { "s" };
1481
        println!(
1482
            "  {} {}",
1483
            ui::note(),
1484
            ui::dim(format!(
1485
                "{} newer operation{s} from other workspaces left as they are",
1486
                u.others_kept
1487
            ))
1488
        );
1489
    }
1490
    for (bookmark, remote) in &u.pushed {
1491
        println!(
1492
            "  {} {}",
1493
            ui::caution(),
1494
            ui::warn(format!(
1495
                "{remote}/{bookmark} already has what this undid: undone here only; the next push of {bookmark} needs --force to take it back there"
1496
            ))
1497
        );
1498
    }
98a1620a Andy McCall ↶ 1499
    where_now(&repo);
fce44ae9 Andy McCall ↶ 1500
    Ok(0)
1501
}
1502
98a1620a Andy McCall ↶ 1503
pub fn op_log(limit: Option<usize>, as_json: bool) -> Result<i32> {
fce44ae9 Andy McCall ↶ 1504
    let repo = open()?;
98a1620a Andy McCall ↶ 1505
    let ops = repo.op_log()?;
1506
    // What each operation changed, against the one before it.
1507
    let changes_of = |i: usize| -> Result<Vec<oxy_core::repo::ViewChange>> {
1508
        match ops.get(i + 1) {
1509
            Some((_, before)) => Ok(repo.view_changes(&before.view, &ops[i].1.view)?),
1510
            None => Ok(vec![]),
1511
        }
1512
    };
c6aa5bd9 Andy McCall ↶ 1513
    if as_json {
98a1620a Andy McCall ↶ 1514
        let mut items = Vec::new();
1515
        for (i, (r, op)) in ops.iter().enumerate().take(limit.unwrap_or(usize::MAX)) {
1516
            let summary: Vec<String> = changes_of(i)?
1517
                .iter()
1518
                .map(|c| change_words(&repo, c))
1519
                .collect();
1520
            items.push(json::obj([
1521
                ("id", json::s(&r.to_hex())),
1522
                ("when_ms", json::n(op.hlc.wall_ms)),
1523
                ("command", json::s(&op.command)),
1524
                ("snapshot", json::b(op.is_snapshot())),
1525
                ("current", json::b(i == 0)),
1526
                ("changes", json::arr(summary.iter().map(|x| json::s(x)))),
1527
            ]));
1528
        }
1529
        println!("{}", json::arr(items));
c6aa5bd9 Andy McCall ↶ 1530
        return Ok(0);
1531
    }
98a1620a Andy McCall ↶ 1532
    let id_prefix = unique_prefixes(ops.iter().map(|(r, _)| r.to_hex()));
1533
    let id = |r: &oxy_core::hash::ObjectRef| {
1534
        let hex = r.to_hex();
1535
        let n = id_prefix.get(&hex).copied().unwrap_or(12).min(12);
1536
        format!("{}{}", ui::id(&hex[..n]), ui::dim(&hex[n..12]))
1537
    };
1538
    let rail = ui::dim(ui::glyph("│", "|"));
1539
    let (mut i, mut shown) = (0, 0);
1540
    while i < ops.len() && shown < limit.unwrap_or(usize::MAX) {
1541
        let (r, op) = &ops[i];
75167672 Andy McCall 1542
        let bullet = if i == 0 {
98a1620a Andy McCall ↶ 1543
            ui::ok(ui::glyph("@", "@"))
75167672 Andy McCall 1544
        } else {
1545
            ui::dim(ui::glyph("○", "o"))
1546
        };
98a1620a Andy McCall ↶ 1547
        // A run of snapshots folds into one entry.
1548
        let mut run = 1;
1549
        while op.is_snapshot() && i + run < ops.len() && ops[i + run].1.is_snapshot() {
1550
            run += 1;
1551
        }
1552
        let when = ui::dim(ui::ago(op.hlc.wall_ms));
1553
        if op.is_snapshot() {
1554
            let span = if run > 1 {
1555
                format!("{} {}", ui::dim("…"), id(&ops[i + run - 1].0))
1556
            } else {
1557
                String::new()
1558
            };
1559
            println!(
1560
                "{bullet}  {}{} {when}",
1561
                id(r),
1562
                if span.is_empty() {
1563
                    String::new()
1564
                } else {
1565
                    format!(" {span}")
1566
                }
1567
            );
1568
            println!(
1569
                "{rail}  {}",
1570
                ui::dim(if run == 1 {
1571
                    "working-copy snapshot".to_string()
1572
                } else {
1573
                    format!("{run} working-copy snapshots")
1574
                })
1575
            );
1576
        } else {
1577
            println!("{bullet}  {} {when}", id(r));
1578
            let words: Vec<String> = changes_of(i)?
1579
                .iter()
1580
                .map(|c| change_words(&repo, c))
1581
                .collect();
1582
            let summary = if words.is_empty() {
1583
                if i + 1 == ops.len() {
1584
                    "created the repository".to_string()
1585
                } else {
1586
                    "no bookmark or working-copy change".to_string()
1587
                }
1588
            } else {
1589
                words.join("; ")
1590
            };
1591
            println!(
1592
                "{rail}  {}",
1593
                if i == 0 {
1594
                    ui::subject(&summary)
1595
                } else {
1596
                    summary
1597
                }
1598
            );
1599
            let cmd = op.command.lines().next().unwrap_or("");
1600
            let cmd: String = cmd.chars().take(100).collect();
1601
            println!("{rail}  {}", ui::dim(cmd));
1602
        }
1603
        i += run;
1604
        shown += 1;
1605
    }
1606
    if i < ops.len() {
75167672 Andy McCall 1607
        println!(
98a1620a Andy McCall ↶ 1608
            "{}",
1609
            ui::dim(format!(
1610
                "{} {} older; pass -n to see more",
1611
                ui::glyph("…", "..."),
1612
                plural(ops.len() - i, "operation")
1613
            ))
1614
        );
1615
    }
1616
    Ok(0)
1617
}
1618
1619
/// One change an operation made, in words.
1620
fn change_words(repo: &Repo, c: &oxy_core::repo::ViewChange) -> String {
1621
    use oxy_core::repo::ViewChange::*;
1622
    match c {
1623
        BookmarkCreated { name, heads } => {
1624
            format!("created bookmark {name} ({})", plural(*heads, "head"))
1625
        }
1626
        BookmarkDeleted { name } => format!("deleted bookmark {name}"),
1627
        BookmarkMoved { name, from, to } if from == to => format!("moved {name}"),
1628
        BookmarkMoved { name, from, to } => format!(
1629
            "{name}: {from} {} {}",
1630
            ui::glyph("→", "->"),
1631
            plural(*to, "head")
1632
        ),
1633
        RemoteMoved { remote, name } => format!("{remote}/{name} updated"),
1634
        TagCreated { name } => format!("tagged {name}"),
1635
        TagDeleted { name } => format!("removed tag {name}"),
1636
        Policy => "trust policy changed".to_string(),
1637
        WorkingCopy { change } => match repo.store().get_change_meta(change) {
1638
            Ok(m) => format!("working copy now {}", m.change_id.short(8)),
1639
            Err(_) => "working copy moved".to_string(),
1640
        },
1641
    }
1642
}
1643
1644
pub fn op_restore(op: &str, discard: bool) -> Result<i32> {
1645
    let repo = open()?;
1646
    let sp = ui::spinner(&format!("Restoring to operation {op}…"));
1647
    let r = repo.op_restore(op, discard)?;
1648
    sp.done(&format!(
1649
        "{} to operation {} {}",
1650
        ui::done("Restored"),
1651
        ui::hash(r.op.short(12)),
1652
        ui::dim(r.command.lines().next().unwrap_or(""))
1653
    ));
1654
    println!(
1655
        "  {} {}",
1656
        ui::note(),
1657
        ui::dim(format!(
1658
            "{} written, {} removed; oxy undo goes back again",
1659
            plural(r.checkout.written, "file"),
1660
            r.checkout.removed
1661
        ))
1662
    );
1663
    where_now(&repo);
1664
    Ok(0)
1665
}
1666
1667
pub fn op_revert(op: &str) -> Result<i32> {
1668
    let repo = open()?;
1669
    let r = repo.op_revert(op)?;
1670
    let view = repo.view()?;
1671
    let names: Vec<String> = r
1672
        .reverted
1673
        .iter()
1674
        .map(|n| {
1675
            if view.bookmarks.contains_key(n) {
1676
                ui::bookmark(n)
75167672 Andy McCall 1677
            } else {
98a1620a Andy McCall ↶ 1678
                format!("{} (removed)", ui::bookmark(n))
75167672 Andy McCall 1679
            }
98a1620a Andy McCall ↶ 1680
        })
1681
        .collect();
1682
    println!(
1683
        "{} operation {} {}",
1684
        ui::done("Reverted"),
1685
        ui::hash(r.op.short(12)),
1686
        ui::dim(r.command.lines().next().unwrap_or(""))
1687
    );
1688
    println!(
1689
        "  {} {}",
1690
        ui::note(),
1691
        ui::dim(format!("as before it: {}", names.join(", ")))
1692
    );
1693
    if !r.kept.is_empty() {
1694
        println!(
1695
            "  {} {}",
1696
            ui::caution(),
1697
            ui::warn(format!(
1698
                "left as they are, having moved again since: {}",
1699
                r.kept.join(", ")
1700
            ))
75167672 Andy McCall 1701
        );
fce44ae9 Andy McCall ↶ 1702
    }
98a1620a Andy McCall ↶ 1703
    if !r.checked_out && r.reverted.contains(&repo.config.bookmark) {
1704
        println!(
1705
            "  {} {}",
1706
            ui::note(),
1707
            ui::dim("the working copy has work of its own: oxy checkout to move it")
1708
        );
1709
    }
1710
    where_now(&repo);
fce44ae9 Andy McCall ↶ 1711
    Ok(0)
98a1620a Andy McCall ↶ 1712
}
1713
1714
/// Where the working copy and bookmark are now, after a command moved
1715
/// them (as jj says "Working copy now at").
1716
pub fn where_now(repo: &Repo) {
1717
    let Ok((_, wc)) = repo.working_copy() else {
1718
        return;
1719
    };
1720
    let heads = repo.bookmark_heads().unwrap_or_default();
1721
    let newest = heads
1722
        .iter()
1723
        .filter_map(|h| repo.store().get_change_meta(h).ok().map(|m| (m, *h)))
1724
        .max_by_key(|(m, _)| m.hlc);
1725
    println!(
1726
        "  {} {}",
1727
        ui::dim(format!("{:<13}", "Working copy")),
1728
        ui::id(wc.change_id.short(8))
1729
    );
1730
    let tail = match newest {
1731
        Some((m, _)) => format!(
1732
            ", newest {} {}",
1733
            ui::id(m.change_id.short(8)),
1734
            subject_of(&m.message)
1735
        ),
1736
        None => String::new(),
1737
    };
1738
    println!(
1739
        "  {} {}: {}{tail}",
1740
        ui::dim(format!("{:<13}", "Bookmark")),
1741
        ui::bookmark(&repo.config.bookmark),
1742
        plural(heads.len(), "head")
1743
    );
fce44ae9 Andy McCall ↶ 1744
}
d02e545d Andy McCall ↶ 1745
b689779e Andy McCall ↶ 1746
pub fn gc(now: bool, expire_snapshots: bool) -> Result<i32> {
1747
    let mut repo = open()?;
1748
    if expire_snapshots {
1749
        repo.config.snapshot_retention_days = 0;
1750
    }
135212de Andy McCall ↶ 1751
    let sp = ui::spinner("Packing and pruning…");
9ae4d15a Andy McCall ↶ 1752
    let r = repo.gc_with(now, &mut |phase| sp.set(&format!("{phase}…")))?;
135212de Andy McCall ↶ 1753
    sp.clear();
ba160f60 Andy McCall ↶ 1754
    if r.packed.objects > 0 {
d02e545d Andy McCall ↶ 1755
        println!(
da652d63 Andy McCall ↶ 1756
            "Packed {} into one pack: {} bytes -> {} bytes ({:.0}%).",
1757
            plural(r.packed.objects, "object"),
ba160f60 Andy McCall ↶ 1758
            r.packed.raw_bytes,
1759
            r.packed.packed_bytes,
1760
            100.0 * r.packed.packed_bytes as f64 / r.packed.raw_bytes.max(1) as f64
d02e545d Andy McCall ↶ 1761
        );
1762
    }
75167672 Andy McCall 1763
    println!(
1764
        "{} reachable; pruned {} ({} bytes){}{}; {} present.",
ba160f60 Andy McCall ↶ 1765
        plural(r.reachable, "object"),
1766
        plural(r.pruned.objects, "unreachable object"),
1767
        r.pruned.bytes,
75167672 Andy McCall 1768
        if r.pruned.deferred > 0 {
1769
            format!(
1770
                "; {} deferred by the grace period (use --now)",
1771
                plural(r.pruned.deferred, "object")
1772
            )
1773
        } else {
1774
            String::new()
1775
        },
1776
        if r.merged_packs > 0 {
1777
            format!("; merged {} into one", plural(r.merged_packs, "pack"))
1778
        } else {
1779
            String::new()
1780
        },
ba160f60 Andy McCall ↶ 1781
        plural(repo.store().pack_count(), "pack")
1782
    );
767ac6cd Andy McCall 1783
    // What earlier repacks replaced stays an hour for readers still using
1784
    // it (oxygenhub#249); a collection after that deletes it.
1785
    let kept = if r.retired.kept > 0 {
1786
        format!(
1787
            "{} retired less than an hour ago kept for readers",
1788
            plural(r.retired.kept, "file")
1789
        )
1790
    } else {
1791
        String::new()
1792
    };
1793
    if r.retired.packs + r.retired.loose > 0 {
1794
        println!(
1795
            "Deleted {} and {} retired by earlier repacks{}.",
1796
            plural(r.retired.packs, "pack"),
1797
            plural(r.retired.loose, "loose object"),
1798
            if kept.is_empty() {
1799
                String::new()
1800
            } else {
1801
                format!("; {kept}")
1802
            }
1803
        );
1804
    } else if !kept.is_empty() {
1805
        println!("{kept}.");
1806
    }
d02e545d Andy McCall ↶ 1807
    Ok(0)
1808
}
614541c3 Andy McCall ↶ 1809
d8d93678 Andy McCall ↶ 1810
/// `oxy gc --remote NAME`: fold a store remote's bundles into one.
1811
pub fn gc_remote(name: &str, now: bool) -> Result<i32> {
1812
    let repo = open()?;
1813
    let sp = ui::spinner(&format!("Folding {name}'s bundles…"));
1814
    let r = repo.gc_remote(name, now)?;
1815
    sp.clear();
1816
    if r.merged > 0 {
1817
        println!(
1818
            "{} {} into one of {} bytes ({})",
1819
            ui::done("Folded"),
1820
            plural(r.merged, "bundle"),
1821
            r.bundle_bytes,
1822
            plural(r.changes, "change")
1823
        );
1824
    } else {
1825
        println!(
1826
            "{} {}",
1827
            ui::note(),
1828
            ui::dim("nothing to fold: one bundle holds everything")
1829
        );
1830
    }
1831
    println!(
1832
        "  {} {}",
1833
        ui::note(),
1834
        ui::dim(format!(
1835
            "removed {}{}",
1836
            plural(r.deleted, "superseded bundle"),
1837
            if r.deferred > 0 {
1838
                format!(
1839
                    "; {} kept for readers that may still fetch them (use --now)",
1840
                    plural(r.deferred, "bundle")
1841
                )
1842
            } else {
1843
                String::new()
1844
            }
1845
        ))
1846
    );
1847
    Ok(0)
1848
}
1849
3e5772cc Andy McCall ↶ 1850
/// `oxy doctor`: exit 1 when a check fails, 0 otherwise (warnings allowed).
1851
pub fn doctor(offline: bool, as_json: bool) -> Result<i32> {
1852
    use oxy_core::repo::Health;
1853
    let repo = open()?;
1854
    let sp = ui::spinner(if offline {
1855
        "Checking…"
1856
    } else {
1857
        "Checking, and contacting remotes…"
1858
    });
1859
    let r = repo.doctor(!offline);
1860
    sp.clear();
1861
    let code = if r.worst() == Health::Fail { 1 } else { 0 };
1862
    let word = |h: Health| match h {
1863
        Health::Ok => "ok",
1864
        Health::Warn => "warning",
1865
        Health::Fail => "fail",
1866
    };
1867
    if as_json {
1868
        let checks: Vec<String> = r
1869
            .checks
1870
            .iter()
1871
            .map(|c| {
1872
                json::obj([
1873
                    ("area", json::s(&c.area)),
1874
                    ("health", json::s(word(c.health))),
1875
                    ("message", json::s(&c.message)),
1876
                    ("hint", json::opt(c.hint.as_deref(), json::s)),
1877
                ])
1878
            })
1879
            .collect();
1880
        println!(
1881
            "{}",
1882
            json::obj([("ok", json::b(code == 0)), ("checks", json::arr(checks))])
1883
        );
1884
        return Ok(code);
1885
    }
1886
    let mut last = "";
1887
    for c in &r.checks {
1888
        if c.area != last {
1889
            println!("{}", ui::section(&c.area));
1890
            last = &c.area;
1891
        }
1892
        let mark = match c.health {
1893
            Health::Ok => ui::ok(ui::glyph("✔", "v")),
1894
            Health::Warn => ui::caution(),
1895
            Health::Fail => ui::err(ui::glyph("✖", "x")),
1896
        };
1897
        println!("  {mark} {}", c.message);
1898
        if let Some(h) = &c.hint {
1899
            println!("    {} {}", ui::dim("→"), ui::dim(h));
1900
        }
1901
    }
1902
    let fails = r.checks.iter().filter(|c| c.health == Health::Fail).count();
1903
    let warns = r.checks.iter().filter(|c| c.health == Health::Warn).count();
1904
    if fails + warns == 0 {
1905
        println!("{}", ui::done("Everything looks right."));
1906
    } else {
1907
        println!(
1908
            "{} {}, {}.",
1909
            if fails > 0 {
1910
                ui::err(ui::glyph("✖", "x"))
1911
            } else {
1912
                ui::caution()
1913
            },
1914
            plural(fails, "problem"),
1915
            plural(warns, "warning")
1916
        );
1917
    }
1918
    Ok(code)
1919
}
1920
387335ff Andy McCall ↶ 1921
/// `oxy fsck`: exit 0 when sound (warnings allowed), 1 on any error.
1922
pub fn fsck(as_json: bool) -> Result<i32> {
1923
    use oxy_core::repo::Severity;
1924
    let repo = open()?;
1925
    let sp = ui::spinner("Checking every object…");
1926
    let r = repo.fsck()?;
1927
    sp.clear();
1928
    let code = if r.errors() > 0 { 1 } else { 0 };
1929
    if as_json {
1930
        let problems: Vec<String> = r
1931
            .problems
1932
            .iter()
1933
            .map(|p| {
1934
                json::obj([
1935
                    (
1936
                        "severity",
1937
                        json::s(match p.severity {
1938
                            Severity::Error => "error",
1939
                            Severity::Warning => "warning",
1940
                        }),
1941
                    ),
1942
                    (
1943
                        "object",
1944
                        json::opt(p.object.map(|o| o.to_hex()).as_deref(), json::s),
1945
                    ),
1946
                    ("what", json::s(&p.what)),
1947
                ])
1948
            })
1949
            .collect();
1950
        println!(
1951
            "{}",
1952
            json::obj([
1953
                ("ok", json::b(code == 0)),
1954
                ("objects", json::n(r.objects)),
1955
                ("ops", json::n(r.ops)),
1956
                ("changes", json::n(r.changes)),
1957
                ("signed", json::n(r.signed)),
1958
                ("unsigned", json::n(r.unsigned)),
1959
                ("checkpoints", json::n(r.checkpoints)),
1960
                ("attestations", json::n(r.attestations)),
1961
                ("errors", json::n(r.errors())),
1962
                ("warnings", json::n(r.warnings())),
1963
                ("problems", json::arr(problems)),
1964
            ])
1965
        );
1966
        return Ok(code);
1967
    }
1968
    println!(
1969
        "Checked {}, {}, {} ({} signed or sealed, {} unsigned), {}, {}.",
1970
        plural(r.objects, "object"),
1971
        plural(r.ops, "operation"),
1972
        plural(r.changes, "change"),
1973
        r.signed,
1974
        r.unsigned,
1975
        plural(r.checkpoints, "checkpoint"),
1976
        plural(r.attestations, "attestation"),
1977
    );
1978
    for p in &r.problems {
1979
        let obj = p
1980
            .object
1981
            .map(|o| format!("{} ", ui::hash(format!("#{}", o.short(12)))))
1982
            .unwrap_or_default();
1983
        match p.severity {
1984
            Severity::Error => println!("{} {obj}{}", ui::err(ui::glyph("✖", "x")), p.what),
1985
            Severity::Warning => println!("{} {obj}{}", ui::caution(), p.what),
1986
        }
1987
    }
1988
    if r.problems.is_empty() {
1989
        println!("{}", ui::done("No problems found."));
1990
    } else if code == 0 {
1991
        println!(
1992
            "{}",
1993
            ui::done(format!("Sound, with {}.", plural(r.warnings(), "warning")))
1994
        );
1995
    } else {
1996
        println!(
1997
            "{} {} and {}. Fetch missing objects from a remote with `oxy pull`, or clone afresh and compare.",
1998
            ui::err(ui::glyph("✖", "x")),
1999
            plural(r.errors(), "error"),
2000
            plural(r.warnings(), "warning")
2001
        );
2002
    }
2003
    Ok(code)
2004
}
2005
79d03f82 Andy McCall ↶ 2006
/// The head the working copy sits on, for commands that default to it. With
2007
/// several heads beneath the working copy, the most recently made one is
2008
/// chosen and announced.
614541c3 Andy McCall ↶ 2009
fn current_head(repo: &Repo) -> Result<oxy_core::hash::ChangeHash> {
cc7a0056 Andy McCall ↶ 2010
    let snap = repo.snapshot_status("oxy")?;
614541c3 Andy McCall ↶ 2011
    let base = snap.base.maximal(&snap.wc.deps);
2012
    match base.as_slice() {
79d03f82 Andy McCall ↶ 2013
        [] => anyhow::bail!("the working copy sits on no change yet"),
614541c3 Andy McCall ↶ 2014
        [h] => Ok(*h),
79d03f82 Andy McCall ↶ 2015
        many => {
2016
            let latest = many
2017
                .iter()
2018
                .max_by_key(|h| snap.base.get(h).map(|c| (c.hlc, **h)))
2019
                .copied()
2020
                .unwrap();
2021
            let c = repo.store().get_change(&latest)?;
75167672 Andy McCall 2022
            println!(
2023
                "Defaulting to {} #{} (most recent of the {} heads under the working copy)",
2024
                short_id(&c),
2025
                latest.short(12),
2026
                many.len()
2027
            );
79d03f82 Andy McCall ↶ 2028
            Ok(latest)
2029
        }
614541c3 Andy McCall ↶ 2030
    }
2031
}
2032
6a81dbab Andy McCall ↶ 2033
/// `oxy describe` without -m: the working copy's name for a build.
2034
pub fn describe_name(as_json: bool) -> Result<i32> {
2035
    let repo = open()?;
2036
    let d = repo.describe_working_copy()?;
2037
    if as_json {
2038
        println!(
2039
            "{}",
2040
            json::obj([
2041
                ("describe", json::s(&d.line())),
2042
                ("tag", json::opt(d.tag.as_ref(), |t| json::s(t))),
2043
                ("distance", json::n(d.distance)),
2044
                ("bookmark", json::s(&d.bookmark)),
2045
                ("change", json::s(&d.change.to_hex())),
2046
                ("hash", json::s(&d.hash.to_hex())),
2047
                ("dirty", json::b(d.dirty)),
2048
            ])
2049
        );
2050
    } else {
2051
        println!("{}", d.line());
2052
    }
2053
    Ok(0)
2054
}
2055
614541c3 Andy McCall ↶ 2056
pub fn describe(message: &str, change: Option<&str>) -> Result<i32> {
2057
    let repo = open()?;
2058
    let h = match change {
2059
        Some(c) => repo.resolve_change(c)?,
2060
        None => current_head(&repo)?,
2061
    };
2062
    let new = repo.describe(&h, message)?;
631fc897 Andy McCall ↶ 2063
    let c = repo.store().get_change_meta(&new)?;
75167672 Andy McCall 2064
    println!(
2065
        "{} {} {}  {}",
2066
        ui::done("Described"),
631fc897 Andy McCall ↶ 2067
        ui::id(c.change_id.short(8)),
75167672 Andy McCall 2068
        ui::hash(format!("#{}", new.short(12))),
2069
        ui::subject(subject_of(message))
2070
    );
614541c3 Andy McCall ↶ 2071
    Ok(0)
2072
}
2073
69953b50 Andy McCall ↶ 2074
pub fn amend(message: Option<&str>, into: Option<&str>, paths: &[String]) -> Result<i32> {
614541c3 Andy McCall ↶ 2075
    let repo = open()?;
530a0fc6 Andy McCall ↶ 2076
    let before = repo.view()?;
23b217d7 Andy McCall ↶ 2077
    let new = match into {
2078
        Some(spec) => {
2079
            let target = repo.resolve_change(spec)?;
2080
            repo.amend_into(&target, message)?
2081
        }
69953b50 Andy McCall ↶ 2082
        None if !paths.is_empty() => repo.amend_paths(message, paths)?,
23b217d7 Andy McCall ↶ 2083
        None => repo.amend(message)?,
2084
    };
631fc897 Andy McCall ↶ 2085
    let c = repo.store().get_change_meta(&new)?;
75167672 Andy McCall 2086
    println!(
2087
        "{} {} {}  {}",
2088
        ui::done("Amended"),
631fc897 Andy McCall ↶ 2089
        ui::id(c.change_id.short(8)),
75167672 Andy McCall 2090
        ui::hash(format!("#{}", new.short(12))),
2091
        ui::subject(subject_of(&c.message))
2092
    );
530a0fc6 Andy McCall ↶ 2093
    // Rewriting what a remote already has means the next push replaces it
2094
    // there (oxygen#99): say so now, not when the push is refused.
2095
    let bookmark = repo.config.bookmark.clone();
2096
    let (had, has) = (
2097
        repo.heads_of(&before, &bookmark)?,
2098
        repo.heads_of(&repo.view()?, &bookmark)?,
2099
    );
2100
    for (remote, bookmarks) in &before.remotes {
2101
        let Some(st) = bookmarks.get(&bookmark) else {
2102
            continue;
2103
        };
2104
        let theirs = repo.store().get_state(st)?.heads;
2105
        if theirs.iter().any(|h| had.contains(h) && !has.contains(h)) {
2106
            println!(
2107
                "  {} {}",
2108
                ui::caution(),
2109
                ui::warn(format!(
2110
                    "{remote}/{bookmark} has the change as it was: the next push of {bookmark} needs --force to replace it"
2111
                ))
2112
            );
2113
        }
2114
    }
98a1620a Andy McCall ↶ 2115
    where_now(&repo);
614541c3 Andy McCall ↶ 2116
    Ok(0)
2117
}
2118
2c9ddac3 Andy McCall ↶ 2119
pub fn abandon(change: Option<&str>, revset: Option<&str>, yes: bool) -> Result<i32> {
614541c3 Andy McCall ↶ 2120
    let repo = open()?;
2c9ddac3 Andy McCall ↶ 2121
    let Some(r) = revset else {
2122
        let h = repo.resolve_change(change.unwrap_or_default())?;
2123
        abandon_one(&repo, h)?;
2124
        where_now(&repo);
2125
        return Ok(0);
2126
    };
2127
    // A revset: say what it selects, refuse an empty set, ask above a few.
2128
    let selected = repo.revset(r)?;
2129
    let wc = repo.working_copy().ok().map(|(h, _)| h);
2130
    let hashes: Vec<ChangeHash> = selected
2131
        .hashes
2132
        .iter()
2133
        .copied()
2134
        .filter(|h| Some(*h) != wc)
2135
        .collect();
2136
    if hashes.is_empty() {
2137
        anyhow::bail!("{r:?} selects no change to abandon");
2138
    }
2139
    println!(
2140
        "{} {} selects {}:",
2141
        ui::section("abandon"),
2142
        ui::id(r),
2143
        plural(hashes.len(), "change")
2144
    );
2145
    for h in &hashes {
2146
        if let Some(m) = selected.metas.get(h) {
2147
            println!(
2148
                "  {} {}  {}",
2149
                ui::id(m.change_id.short(8)),
2150
                ui::hash(format!("#{}", h.short(12))),
2151
                subject_of(&m.message)
2152
            );
2153
        }
2154
    }
2155
    const ASK_ABOVE: usize = 5;
2156
    if hashes.len() > ASK_ABOVE && !yes {
2157
        anyhow::bail!("that is more than {ASK_ABOVE} changes; pass --yes to abandon them all");
2158
    }
2159
    // Newest first: abandoning a change rewrites what depends on it, so a
2160
    // later member is found again by its ChangeId.
2161
    let mut done = 0;
2162
    for h in &hashes {
2163
        let Some(m) = selected.metas.get(h) else {
2164
            continue;
2165
        };
2166
        let now = match repo.resolve_change(&m.change_id.to_hex()) {
2167
            Ok(h) => h,
2168
            Err(_) => continue,
2169
        };
2170
        abandon_one(&repo, now)?;
2171
        done += 1;
2172
    }
2173
    println!("{} {}", ui::done("Abandoned"), plural(done, "change"));
2174
    where_now(&repo);
2175
    Ok(0)
2176
}
2177
2178
fn abandon_one(repo: &Repo, h: ChangeHash) -> Result<()> {
631fc897 Andy McCall ↶ 2179
    let c = repo.store().get_change_meta(&h)?;
6706bdcf Andy McCall ↶ 2180
    let notes = repo.abandon(&h)?;
75167672 Andy McCall 2181
    println!(
2182
        "{} {} {}  {}",
2183
        ui::done("Abandoned"),
631fc897 Andy McCall ↶ 2184
        ui::id(c.change_id.short(8)),
75167672 Andy McCall 2185
        ui::hash(format!("#{}", h.short(12))),
2186
        ui::dim("its content remains on disk as uncommitted work if the working copy sat on it")
2187
    );
6706bdcf Andy McCall ↶ 2188
    for n in notes {
94084087 Andy McCall ↶ 2189
        println!("  {} {n}", ui::note());
6706bdcf Andy McCall ↶ 2190
    }
2c9ddac3 Andy McCall ↶ 2191
    Ok(())
614541c3 Andy McCall ↶ 2192
}
252e8cb2 Andy McCall ↶ 2193
2194
pub fn checkpoint() -> Result<i32> {
2195
    let repo = open()?;
135212de Andy McCall ↶ 2196
    let sp = ui::spinner("Writing checkpoint…");
252e8cb2 Andy McCall ↶ 2197
    let (r, changes, bytes) = repo.checkpoint()?;
75167672 Andy McCall 2198
    sp.done(&format!(
2199
        "{} {} of {} ({} bytes) for {}",
2200
        ui::done("Checkpoint"),
2201
        ui::hash(format!("#{}", r.short(12))),
2202
        plural(changes, "change"),
2203
        bytes,
2204
        ui::id(&repo.config.bookmark)
2205
    ));
135212de Andy McCall ↶ 2206
    Ok(0)
2207
}
2208
2209
/// For shell prompts: silent (exit 1) outside a repository, otherwise one line.
694805cc Andy McCall ↶ 2210
pub fn prompt(format: &str, glyphs: bool, color: Option<&str>) -> Result<i32> {
75167672 Andy McCall 2211
    let Ok(repo) = Repo::open(&std::env::current_dir()?) else {
2212
        return Ok(1);
2213
    };
135212de Andy McCall ↶ 2214
    let info = repo.quick_status()?;
75167672 Andy McCall 2215
    let fmt = if glyphs && format == oxy_core::repo::DEFAULT_PROMPT_FORMAT {
2216
        oxy_core::repo::GLYPH_PROMPT_FORMAT
2217
    } else {
2218
        format
2219
    };
694805cc Andy McCall ↶ 2220
    let line = match color {
2221
        Some("bash") => info.format_colored(fmt, ("\\[", "\\]")),
2222
        Some("zsh") => info.format_colored(fmt, ("%{", "%}")),
2223
        Some(_) => info.format_colored(fmt, ("", "")),
2224
        None => info.format(fmt),
2225
    };
2226
    println!("{line}");
252e8cb2 Andy McCall ↶ 2227
    Ok(0)
2228
}
7421309f Andy McCall ↶ 2229
2230
pub fn sparse(paths: &[String]) -> Result<i32> {
2231
    let mut repo = open()?;
2232
    if paths.is_empty() {
2233
        if repo.config.sparse_paths.is_empty() {
2234
            println!("No sparse filter: the whole tree is materialized.");
2235
        } else {
2236
            println!("{}", ui::heading("Sparse filter"));
2237
            for p in &repo.config.sparse_paths {
2238
                println!("  {}", ui::path(p));
2239
            }
2240
        }
2241
        return Ok(0);
2242
    }
2243
    let snap = repo.snapshot("oxy sparse")?;
2244
    if !snap.wc.tree_ops.is_empty() || !snap.wc.edits.is_empty() {
75167672 Andy McCall 2245
        anyhow::bail!(
2246
            "the working copy has changes; commit them before changing the sparse filter"
2247
        );
7421309f Andy McCall ↶ 2248
    }
2249
    repo.config.set("filter.paths", &paths.join(","))?;
2250
    repo.save_config()?;
2251
    let heads = repo.bookmark_heads()?;
2252
    let sp = ui::spinner("Re-checking out with the new filter…");
2253
    let r = repo.checkout(&heads, true, "oxy sparse (checkout)")?;
75167672 Andy McCall 2254
    sp.done(&format!(
2255
        "{} {}: {} written, {} removed",
c07406bc Andy McCall ↶ 2256
        ui::done("Sparse filter set to"),
75167672 Andy McCall 2257
        ui::path(paths.join(",")),
2258
        plural(r.written, "file"),
2259
        r.removed
2260
    ));
7421309f Andy McCall ↶ 2261
    Ok(0)
2262
}
132b884b Andy McCall ↶ 2263
2264
pub fn tag_create(name: &str, changes: &[String], message: &str) -> Result<i32> {
2265
    let repo = open()?;
cc7a0056 Andy McCall ↶ 2266
    repo.snapshot_status("oxy tag")?;
132b884b Andy McCall ↶ 2267
    let heads = if changes.is_empty() {
2268
        repo.bookmark_heads()?
2269
    } else {
2270
        let mut heads = Vec::new();
2271
        for c in changes {
2272
            heads.extend(repo.resolve_heads(c)?);
2273
        }
2274
        heads
2275
    };
2276
    let t = repo.tag_create(name, &heads, message)?;
2277
    println!(
94084087 Andy McCall ↶ 2278
        "{} {} {} {} ({}){}",
2279
        ui::done("Tagged"),
2280
        ui::bookmark(&t.name),
2281
        ui::dim(ui::glyph("→", "->")),
132b884b Andy McCall ↶ 2282
        ui::hash(format!("#{}", t.state.short(12))),
2283
        plural(t.heads.len(), "head"),
75167672 Andy McCall 2284
        t.signer
2285
            .map(|s| format!(", signed by {s}"))
2286
            .unwrap_or_else(|| ui::dim(", unsigned"))
132b884b Andy McCall ↶ 2287
    );
2288
    Ok(0)
2289
}
2290
1196b29b Andy McCall ↶ 2291
pub fn tag_list(as_json: bool) -> Result<i32> {
132b884b Andy McCall ↶ 2292
    let repo = open()?;
2293
    let tags = repo.tags()?;
1196b29b Andy McCall ↶ 2294
    if as_json {
2295
        println!(
2296
            "{}",
2297
            json::arr(tags.iter().map(|t| {
2298
                json::obj([
2299
                    ("name", json::s(&t.name)),
2300
                    ("state", json::s(&t.state.to_hex())),
2301
                    ("heads", json::hashes(&t.heads)),
2302
                    ("message", json::s(&t.message)),
75167672 Andy McCall 2303
                    (
2304
                        "signer",
2305
                        json::opt(t.signer.as_ref(), |s| json::s(&s.to_string())),
2306
                    ),
1196b29b Andy McCall ↶ 2307
                ])
2308
            }))
2309
        );
2310
        return Ok(0);
2311
    }
132b884b Andy McCall ↶ 2312
    if tags.is_empty() {
2313
        println!("{}", ui::dim("No tags."));
2314
    }
2315
    for t in tags {
2316
        println!(
94084087 Andy McCall ↶ 2317
            "{} {} {}{}{}{}",
2318
            ui::ok(ui::glyph("⚑", "#")),
2319
            ui::bookmark(&t.name),
6a81dbab Andy McCall ↶ 2320
            // The heads, as `oxy log` shows changes, not the state hash.
2321
            ui::hash(
2322
                t.heads
2323
                    .iter()
2324
                    .map(|h| format!("#{}", h.short(12)))
2325
                    .collect::<Vec<_>>()
2326
                    .join(" ")
2327
            ),
94084087 Andy McCall ↶ 2328
            ui::sep() + &ui::dim(plural(t.heads.len(), "head")),
75167672 Andy McCall 2329
            if t.message.is_empty() {
2330
                String::new()
2331
            } else {
2332
                format!("  {}", ui::subject(subject_of(&t.message)))
2333
            },
2334
            t.signer
2335
                .map(|s| ui::sep() + &ui::ok(format!("{} {s}", ui::glyph("✔", "v"))))
2336
                .unwrap_or_default()
132b884b Andy McCall ↶ 2337
        );
2338
    }
2339
    Ok(0)
a46f3e4f Andy McCall ↶ 2340
}
2341
2342
/// Whether `name` is a bookmark, here or only on a remote: what checkout
2343
/// and workspace add take as "be on this bookmark" (oxygen#60).
2344
fn names_bookmark(repo: &oxy_core::repo::Repo, name: &str) -> Result<bool> {
2345
    Ok(repo.view()?.bookmarks.contains_key(name)
2346
        || repo
2347
            .remote_only_bookmarks()?
2348
            .iter()
2349
            .any(|b| b.name == name || format!("{}/{}", b.remote, b.name) == name))
132b884b Andy McCall ↶ 2350
}
4e88e774 Andy McCall ↶ 2351
2352
pub fn workspace_add(path: &std::path::Path, changes: &[String]) -> Result<i32> {
2353
    let repo = open()?;
a46f3e4f Andy McCall ↶ 2354
    // One bookmark named: the new workspace is on it (oxygen#60), not on
2355
    // this workspace's bookmark with that one's files.
2356
    if let [name] = changes {
2357
        if names_bookmark(&repo, name)? {
2358
            let (local, _) = repo.local_bookmark_for(name)?;
2359
            let heads = repo.resolve_heads(&local)?;
2360
            let sp = ui::spinner(&format!("Creating workspace at {}…", path.display()));
2361
            let ws = repo.workspace_add_on(path, Some(&heads), Some(&local))?;
2362
            sp.done(&format!(
2363
                "{} {} (workspace {}) on {}",
2364
                ui::done("Workspace created at"),
2365
                ui::path(ws.root().display()),
2366
                ui::id(ws.workspace().short(8)),
2367
                ui::bookmark(&local)
2368
            ));
2369
            return Ok(0);
2370
        }
2371
    }
4e88e774 Andy McCall ↶ 2372
    let heads = if changes.is_empty() {
2373
        None
2374
    } else {
2375
        let mut heads = Vec::new();
2376
        for c in changes {
2377
            heads.extend(repo.resolve_heads(c)?);
2378
        }
2379
        Some(heads)
2380
    };
2381
    let sp = ui::spinner(&format!("Creating workspace at {}…", path.display()));
2382
    let ws = repo.workspace_add(path, heads.as_deref())?;
75167672 Andy McCall 2383
    sp.done(&format!(
2384
        "{} {} (workspace {})",
2385
        ui::done("Workspace created at"),
2386
        ui::path(ws.root().display()),
2387
        ui::id(ws.workspace().short(8))
2388
    ));
4e88e774 Andy McCall ↶ 2389
    Ok(0)
2390
}
2391
1196b29b Andy McCall ↶ 2392
pub fn workspace_list(as_json: bool) -> Result<i32> {
4e88e774 Andy McCall ↶ 2393
    let repo = open()?;
1196b29b Andy McCall ↶ 2394
    if as_json {
2395
        println!(
2396
            "{}",
2397
            json::arr(repo.workspaces()?.iter().map(|w| {
2398
                json::obj([
2399
                    ("id", json::s(&w.id.to_hex())),
2400
                    ("path", json::s(&w.path.to_string_lossy())),
2401
                    ("current", json::b(w.current)),
75167672 Andy McCall 2402
                    (
2403
                        "working_copy",
2404
                        json::opt(w.working_copy, |h| json::s(&h.to_hex())),
2405
                    ),
1196b29b Andy McCall ↶ 2406
                ])
2407
            }))
2408
        );
2409
        return Ok(0);
2410
    }
4e88e774 Andy McCall ↶ 2411
    for w in repo.workspaces()? {
2412
        println!(
2413
            "{} {} {}{}",
75167672 Andy McCall 2414
            if w.current {
2415
                ui::ok(ui::glyph("●", "*"))
2416
            } else {
2417
                ui::dim(ui::glyph("○", "o"))
2418
            },
4e88e774 Andy McCall ↶ 2419
            ui::id(w.id.short(8)),
2420
            ui::path(w.path.display()),
75167672 Andy McCall 2421
            w.working_copy
2422
                .map(|h| ui::sep() + &ui::dim(format!("working copy #{}", h.short(12))))
2423
                .unwrap_or_else(|| ui::sep() + &ui::warn("not in view"))
4e88e774 Andy McCall ↶ 2424
        );
2425
    }
2426
    Ok(0)
2427
}
2428
2429
pub fn workspace_forget(spec: &str) -> Result<i32> {
2430
    let repo = open()?;
2431
    let all = repo.workspaces()?;
2432
    let target = std::fs::canonicalize(spec).ok();
75167672 Andy McCall 2433
    let matches: Vec<_> = all
2434
        .iter()
2435
        .filter(|w| w.id.starts_with_hex(spec) || target.as_ref().is_some_and(|t| *t == w.path))
2436
        .collect();
4e88e774 Andy McCall ↶ 2437
    let w = match matches.as_slice() {
2438
        [w] => *w,
2439
        [] => anyhow::bail!("no workspace matches {spec:?}"),
2440
        _ => anyhow::bail!("{spec:?} is ambiguous"),
2441
    };
2442
    repo.workspace_forget(&w.id)?;
75167672 Andy McCall 2443
    println!(
2444
        "{} workspace {} at {}; its files were left in place.",
2445
        ui::done("Forgot"),
2446
        ui::id(w.id.short(8)),
2447
        ui::path(w.path.display())
2448
    );
4e88e774 Andy McCall ↶ 2449
    Ok(0)
2450
}
cdb5beba Andy McCall ↶ 2451
2452
pub fn lock(path: &str) -> Result<i32> {
2453
    let repo = open()?;
2454
    repo.lock_path(path)?;
75167672 Andy McCall 2455
    println!(
2456
        "{} {} {}",
2457
        ui::done("Locked"),
2458
        ui::path(path),
2459
        ui::dim("(advisory, local)")
2460
    );
cdb5beba Andy McCall ↶ 2461
    Ok(0)
2462
}
2463
2464
pub fn unlock(path: &str) -> Result<i32> {
2465
    let repo = open()?;
2466
    repo.unlock_path(path)?;
94084087 Andy McCall ↶ 2467
    println!("{} {}", ui::done("Unlocked"), ui::path(path));
cdb5beba Andy McCall ↶ 2468
    Ok(0)
2469
}
23b217d7 Andy McCall ↶ 2470
2471
pub fn split(change: &str, paths: &[String]) -> Result<i32> {
2472
    let repo = open()?;
2473
    let h = repo.resolve_change(change)?;
2474
    let (h1, h2) = repo.split(&h, paths)?;
631fc897 Andy McCall ↶ 2475
    let c1 = repo.store().get_change_meta(&h1)?;
2476
    let c2 = repo.store().get_change_meta(&h2)?;
75167672 Andy McCall 2477
    println!(
2478
        "{} into {} {} and {} {}",
2479
        ui::done("Split"),
631fc897 Andy McCall ↶ 2480
        ui::id(c1.change_id.short(8)),
75167672 Andy McCall 2481
        ui::hash(format!("#{}", h1.short(12))),
631fc897 Andy McCall ↶ 2482
        ui::id(c2.change_id.short(8)),
75167672 Andy McCall 2483
        ui::hash(format!("#{}", h2.short(12)))
2484
    );
23b217d7 Andy McCall ↶ 2485
    Ok(0)
2486
}
7b9809e6 Andy McCall ↶ 2487
2488
fn heads_from(repo: &Repo, changes: &[String]) -> Result<Vec<oxy_core::hash::ChangeHash>> {
2489
    let mut heads = Vec::new();
2490
    for c in changes {
2491
        heads.extend(repo.resolve_heads(c)?);
2492
    }
2493
    Ok(heads)
2494
}
2495
4fe7d604 Andy McCall ↶ 2496
pub fn bookmark_create(name: &str, changes: &[String], upto: bool, only: bool) -> Result<i32> {
7b9809e6 Andy McCall ↶ 2497
    let repo = open()?;
cc7a0056 Andy McCall ↶ 2498
    repo.snapshot_status("oxy bookmark create")?;
75167672 Andy McCall 2499
    let heads = if changes.is_empty() {
2500
        None
2501
    } else {
4fe7d604 Andy McCall ↶ 2502
        let named = heads_from(&repo, changes)?;
2503
        let (_, wc) = repo.working_copy()?;
2504
        let gap = repo.state_gap(&wc.deps, &named)?;
2505
        if upto {
2506
            Some(gap.upto)
2507
        } else if only || gap.earlier.is_empty() {
2508
            Some(named)
2509
        } else {
2510
            anyhow::bail!(
2511
                "{name} at {} would lack {} of the current state recorded before it:\n{}\n\
2512
                 A state is its heads and what they depend on, not everything recorded before them.\n\
2513
                 Pass --upto for everything of the current state up to {0} (as a Git branch would hold),\n\
2514
                 or --only for just {0} and what it depends on.",
2515
                changes.join(" "),
2516
                plural(gap.earlier.len(), "change"),
2517
                listed_changes(&repo, &gap.earlier)
2518
            );
2519
        }
75167672 Andy McCall 2520
    };
7b9809e6 Andy McCall ↶ 2521
    let heads = repo.bookmark_create(name, heads.as_deref())?;
75167672 Andy McCall 2522
    println!(
2523
        "{} bookmark {} {} {}",
2524
        ui::done("Created"),
2525
        ui::bookmark(name),
2526
        ui::dim(ui::glyph("→", "->")),
2527
        ui::dim(plural(heads.len(), "head"))
2528
    );
2529
    println!(
2530
        "  {} {}",
2531
        ui::note(),
2532
        ui::dim(format!("switch to it with oxy bookmark switch {name}"))
2533
    );
7b9809e6 Andy McCall ↶ 2534
    Ok(0)
2535
}
2536
4fe7d604 Andy McCall ↶ 2537
/// Up to five changes, newest last, as `  #hash subject` lines.
2538
fn listed_changes(repo: &Repo, changes: &[oxy_core::hash::ChangeHash]) -> String {
2539
    let shown = changes.len().min(5);
2540
    let mut lines: Vec<String> = changes[changes.len() - shown..]
2541
        .iter()
2542
        .map(|h| {
2543
            let subject = repo
2544
                .store()
2545
                .get_change(h)
2546
                .map(|c| subject_of(&c.message).to_string())
2547
                .unwrap_or_default();
2548
            format!("  #{} {subject}", h.short(12))
2549
        })
2550
        .collect();
2551
    if changes.len() > shown {
2552
        lines.insert(0, format!("  … and {} earlier", changes.len() - shown));
2553
    }
2554
    lines.join("\n")
2555
}
2556
06d16dc2 Andy McCall ↶ 2557
pub fn bookmark_set(name: &str, changes: &[String], upto: bool, only: bool) -> Result<i32> {
7b9809e6 Andy McCall ↶ 2558
    let repo = open()?;
cc7a0056 Andy McCall ↶ 2559
    repo.snapshot_status("oxy bookmark set")?;
06d16dc2 Andy McCall ↶ 2560
    let named = heads_from(&repo, changes)?;
2561
    // What the bookmark holds now and the new heads would leave out.
2562
    let current = repo
2563
        .bookmarks()?
2564
        .into_iter()
2565
        .find(|b| b.name == name)
2566
        .map(|b| b.heads)
2567
        .ok_or_else(|| anyhow::anyhow!("no bookmark {name}"))?;
2568
    let gap = repo.state_gap(&current, &named)?;
2569
    let heads = if upto {
2570
        gap.upto
2571
    } else if only || gap.earlier.is_empty() {
2572
        named
2573
    } else {
2574
        anyhow::bail!(
2575
            "{name} at {} would lose {} it holds now, recorded before it:\n{}\n\
2576
             A state is its heads and what they depend on, not everything recorded before them.\n\
2577
             Pass --upto to keep everything {name} holds up to {0} (as a Git branch would),\n\
2578
             or --only for just {0} and what it depends on.",
2579
            changes.join(" "),
2580
            plural(gap.earlier.len(), "change"),
2581
            listed_changes(&repo, &gap.earlier)
2582
        );
2583
    };
7b9809e6 Andy McCall ↶ 2584
    let heads = repo.bookmark_set(name, &heads)?;
75167672 Andy McCall 2585
    println!(
2586
        "{} bookmark {} {} {}",
2587
        ui::done("Moved"),
2588
        ui::bookmark(name),
2589
        ui::dim(ui::glyph("→", "->")),
2590
        ui::dim(plural(heads.len(), "head"))
2591
    );
7b9809e6 Andy McCall ↶ 2592
    Ok(0)
2593
}
2594
2595
pub fn bookmark_delete(name: &str) -> Result<i32> {
2596
    let repo = open()?;
2597
    repo.bookmark_delete(name)?;
75167672 Andy McCall 2598
    println!(
2599
        "{} bookmark {} {}",
2600
        ui::done("Deleted"),
2601
        ui::bookmark(name),
2602
        ui::dim("(its changes stay until unreachable)")
2603
    );
7b9809e6 Andy McCall ↶ 2604
    Ok(0)
2605
}
2606
bf3a6781 Andy McCall ↶ 2607
/// Heads with their metadata, newest first.
2608
fn heads_newest_first(
2609
    repo: &Repo,
2610
    heads: &[oxy_core::hash::ChangeHash],
2611
) -> Vec<(oxy_core::hash::ChangeHash, oxy_core::state::ChangeMeta)> {
2612
    let mut out: Vec<_> = heads
2613
        .iter()
2614
        .filter_map(|h| repo.store().get_change_meta(h).ok().map(|m| (*h, m)))
2615
        .collect();
2616
    out.sort_by_key(|(_, m)| std::cmp::Reverse(m.hlc));
2617
    out
2618
}
2619
2620
/// One head as a line: id, hash, author, when, subject.
2621
fn print_head(
2622
    repo: &Repo,
2623
    indent: &str,
2624
    h: &oxy_core::hash::ChangeHash,
2625
    m: &oxy_core::state::ChangeMeta,
2626
) {
2627
    println!(
2628
        "{indent}{} {} {}{}{}{}{}",
2629
        ui::id(m.change_id.short(8)),
2630
        ui::dim(format!("#{}", &h.to_hex()[..8])),
2631
        author_name(repo, m),
2632
        ui::sep(),
2633
        ui::dim(ui::ago(m.hlc.wall_ms)),
2634
        ui::sep(),
2635
        ui::subject(subject_of(&m.message))
2636
    );
2637
}
2638
2639
pub fn bookmark_show(name: Option<&str>, as_json: bool) -> Result<i32> {
7b9809e6 Andy McCall ↶ 2640
    let repo = open()?;
bf3a6781 Andy McCall ↶ 2641
    let name = name.unwrap_or(&repo.config.bookmark).to_string();
2642
    let view = repo.view()?;
2643
    let st = match view.bookmarks.get(&name) {
2644
        Some(st) => *st,
2645
        None => match oxy_core::repo::remote_bookmark(&view, &name) {
2646
            Some((_, _, st)) => st,
2647
            None => anyhow::bail!("no bookmark {name}: oxy bookmark list --all lists them"),
2648
        },
2649
    };
2650
    let heads = repo.store().get_state(&st)?.heads;
2651
    let shown = heads_newest_first(&repo, &heads);
2652
    if as_json {
2653
        println!(
2654
            "{}",
2655
            json::obj([
2656
                ("name", json::s(&name)),
2657
                ("current", json::b(name == repo.config.bookmark)),
2658
                (
2659
                    "heads",
2660
                    json::arr(shown.iter().map(|(h, m)| {
2661
                        json::obj([
2662
                            ("hash", json::s(&h.to_hex())),
2663
                            ("id", json::s(&m.change_id.to_string())),
2664
                            ("author", json::s(&author_name(&repo, m))),
2665
                            ("time_ms", json::n(m.hlc.wall_ms)),
2666
                            ("subject", json::s(subject_of(&m.message))),
2667
                        ])
2668
                    }))
2669
                ),
2670
            ])
2671
        );
2672
        return Ok(0);
2673
    }
2674
    println!(
2675
        "{} {}{}{}",
2676
        ui::ok(ui::glyph("●", "*")),
2677
        ui::bookmark(&name),
2678
        ui::sep(),
2679
        ui::dim(plural(heads.len(), "head"))
2680
    );
2681
    for (h, m) in &shown {
2682
        print_head(&repo, "  ", h, m);
2683
    }
2684
    if heads.len() > 1 {
2685
        println!(
2686
            "  {}",
2687
            ui::dim("None depends on another: the bookmark is all of them and their dependencies.")
2688
        );
2689
    }
2690
    Ok(0)
2691
}
2692
2693
pub fn bookmark_list(all: bool, verbose: bool, as_json: bool) -> Result<i32> {
2694
    let repo = open()?;
14edbaaf Andy McCall ↶ 2695
    let remote_only = repo.remote_only_bookmarks()?;
c6aa5bd9 Andy McCall ↶ 2696
    if as_json {
14edbaaf Andy McCall ↶ 2697
        let mut items: Vec<String> = repo
2698
            .bookmarks()?
2699
            .into_iter()
2700
            .map(|b| {
2701
                json::obj([
2702
                    ("name", json::s(&b.name)),
2703
                    ("heads", json::hashes(&b.heads)),
2704
                    ("current", json::b(b.current)),
2705
                    ("ahead", json::opt(b.ahead, json::n)),
2706
                    ("behind", json::opt(b.behind, json::n)),
42ccf504 Andy McCall ↶ 2707
                    ("remote", json::opt(b.remote.as_deref(), json::s)),
14edbaaf Andy McCall ↶ 2708
                ])
2709
            })
2710
            .collect();
2711
        if all {
2712
            items.extend(remote_only.iter().map(|b| {
2713
                json::obj([
2714
                    ("name", json::s(&format!("{}/{}", b.remote, b.name))),
2715
                    ("heads", json::hashes(&b.heads)),
2716
                    ("current", json::b(false)),
2717
                    ("ahead", "null".into()),
2718
                    ("behind", "null".into()),
2719
                    ("remote", json::s(&b.remote)),
2720
                ])
2721
            }));
2722
        }
c6aa5bd9 Andy McCall ↶ 2723
        println!("{}", json::arr(items));
2724
        return Ok(0);
2725
    }
7b9809e6 Andy McCall ↶ 2726
    for b in repo.bookmarks()? {
75167672 Andy McCall 2727
        let mark = if b.current {
2728
            ui::ok(ui::glyph("●", "*"))
2729
        } else {
2730
            ui::dim(ui::glyph("○", "o"))
2731
        };
7b9809e6 Andy McCall ↶ 2732
        let mut tail = String::new();
2733
        if let Some(n) = b.ahead.filter(|n| *n > 0) {
2734
            tail += &ui::ok(format!(" {}{n}", ui::glyph("↑", "+")));
2735
        }
2736
        if let Some(n) = b.behind.filter(|n| *n > 0) {
2737
            tail += &ui::warn(format!(" {}{n}", ui::glyph("↓", "-")));
2738
        }
42ccf504 Andy McCall ↶ 2739
        // Which remote, when there is a choice (oxygen#83).
2740
        if !tail.is_empty() && repo.config.remotes.len() > 1 {
2741
            if let Some(r) = &b.remote {
2742
                tail += &ui::dim(format!(" {r}"));
2743
            }
2744
        }
bf3a6781 Andy McCall ↶ 2745
        let shown = heads_newest_first(&repo, &b.heads);
2746
        // The newest head's subject, so a busy bookmark shows its latest work.
2747
        let head = match shown.first() {
2748
            Some(_) if verbose => String::new(),
2749
            Some((_, m)) if b.heads.len() > 1 => format!(
2750
                "{} {}",
2751
                ui::dim("newest"),
2752
                ui::subject(subject_of(&m.message))
2753
            ),
2754
            Some((_, m)) => ui::subject(subject_of(&m.message)),
2755
            None => ui::dim("(empty)"),
2756
        };
2757
        let head = if head.is_empty() {
2758
            head
2759
        } else {
2760
            format!("  {head}")
2761
        };
75167672 Andy McCall 2762
        println!(
bf3a6781 Andy McCall ↶ 2763
            "{mark} {}{}{}{tail}{head}",
75167672 Andy McCall 2764
            ui::bookmark(&b.name),
14edbaaf Andy McCall ↶ 2765
            ui::sep(),
2766
            ui::dim(plural(b.heads.len(), "head"))
2767
        );
bf3a6781 Andy McCall ↶ 2768
        if verbose {
2769
            for (h, m) in &shown {
2770
                print_head(&repo, "    ", h, m);
2771
            }
2772
        }
14edbaaf Andy McCall ↶ 2773
    }
2774
    if remote_only.is_empty() {
2775
        return Ok(0);
2776
    }
2777
    if !all {
2778
        println!(
2779
            "{}",
2780
            ui::dim(format!(
2781
                "  and {} only on a remote: oxy bookmark list --all",
2782
                plural(remote_only.len(), "bookmark")
2783
            ))
2784
        );
2785
        return Ok(0);
2786
    }
2787
    for b in &remote_only {
2788
        println!(
2789
            "{} {}{}{}",
2790
            ui::dim(ui::glyph("◌", ".")),
2791
            ui::dim(format!("{}/{}", b.remote, b.name)),
75167672 Andy McCall 2792
            ui::sep(),
2793
            ui::dim(plural(b.heads.len(), "head"))
2794
        );
7b9809e6 Andy McCall ↶ 2795
    }
14edbaaf Andy McCall ↶ 2796
    println!(
2797
        "  {} {}",
2798
        ui::note(),
2799
        ui::dim("oxy bookmark switch NAME makes a local bookmark from a remote's")
2800
    );
7b9809e6 Andy McCall ↶ 2801
    Ok(0)
2802
}
2803
61e3bce7 Andy McCall ↶ 2804
pub fn bookmark_switch(name: &str, discard: bool, carry: bool) -> Result<i32> {
7b9809e6 Andy McCall ↶ 2805
    let mut repo = open()?;
4fe7d604 Andy McCall ↶ 2806
    let from = repo.config.bookmark.clone();
14edbaaf Andy McCall ↶ 2807
    let (local, created_from) = repo.local_bookmark_for(name)?;
2808
    if let Some(remote) = created_from {
2809
        println!(
2810
            "{} {} from {}",
2811
            ui::done("Created"),
2812
            ui::bookmark(&local),
2813
            ui::dim(format!("{remote}/{local}"))
2814
        );
2815
    }
2816
    let name = local.as_str();
72484195 Andy McCall ↶ 2817
    // Only behind its remote: take the remote's state, so the switch does
2818
    // not check out a stale one (oxygen#74).
2819
    if let Some(n) = repo.catch_up_bookmark(name)? {
2820
        println!(
2821
            "{} {} with {}: {} it lacked",
2822
            ui::done("Caught up"),
2823
            ui::bookmark(name),
2824
            ui::dim(format!("{}/{name}", repo.default_remote())),
2825
            plural(n, "change")
2826
        );
2827
    }
7b9809e6 Andy McCall ↶ 2828
    let sp = ui::spinner(&format!("Switching to {name}…"));
61e3bce7 Andy McCall ↶ 2829
    let (r, carried) = if carry {
2830
        repo.bookmark_switch_carry(name)?
2831
    } else {
2832
        (repo.bookmark_switch(name, discard)?, vec![])
2833
    };
75167672 Andy McCall 2834
    sp.done(&format!(
61e3bce7 Andy McCall ↶ 2835
        "{} to {}: {} written, {} removed{}",
75167672 Andy McCall 2836
        ui::done("Switched"),
2837
        ui::bookmark(name),
2838
        plural(r.written, "file"),
61e3bce7 Andy McCall ↶ 2839
        r.removed,
2840
        if carried.is_empty() {
2841
            String::new()
2842
        } else {
2843
            format!("; carried {}", plural(carried.len(), "changed path"))
2844
        }
75167672 Andy McCall 2845
    ));
9b723f69 Andy McCall ↶ 2846
    print_removed(&r);
7b9809e6 Andy McCall ↶ 2847
    if r.conflicted {
75167672 Andy McCall 2848
        println!(
2849
            "  {} {}",
2850
            ui::caution(),
2851
            ui::warn("this state has conflicts; see oxy status")
2852
        );
4fe7d604 Andy McCall ↶ 2853
    }
2854
    // Say when the new state leaves out work the old one had: a state is
2855
    // its heads and what they depend on, which can be far less than a Git
2856
    // user expects.
d3cd7ef9 Andy McCall ↶ 2857
    if r.lacks > 0 {
2858
        println!(
2859
            "  {} {}",
2860
            ui::note(),
2861
            ui::dim(format!(
2862
                "{name} lacks {} that {from} has",
2863
                plural(r.lacks, "change")
2864
            ))
2865
        );
7b9809e6 Andy McCall ↶ 2866
    }
14edbaaf Andy McCall ↶ 2867
    super::sync::subrepos_behind(&repo);
98a1620a Andy McCall ↶ 2868
    where_now(&repo);
7b9809e6 Andy McCall ↶ 2869
    Ok(0)
2870
}
2871
2872
pub fn merge(bookmark: &str) -> Result<i32> {
2873
    let repo = open()?;
2874
    let sp = ui::spinner(&format!("Merging {bookmark}…"));
2875
    let r = repo.merge(bookmark)?;
2876
    if !r.changed {
75167672 Andy McCall 2877
        sp.done(&format!(
2878
            "{} {}",
2879
            ui::ok(ui::glyph("✔", "*")),
2880
            ui::dim(format!(
2881
                "{} already contains {bookmark}",
2882
                repo.config.bookmark
2883
            ))
2884
        ));
7b9809e6 Andy McCall ↶ 2885
        return Ok(0);
2886
    }
2887
    sp.done(&format!(
2888
        "{} {} into {}: now {}{}",
2889
        ui::done("Merged"),
2890
        ui::bookmark(bookmark),
2891
        ui::bookmark(&repo.config.bookmark),
2892
        plural(r.heads.len(), "head"),
75167672 Andy McCall 2893
        ui::dim(if r.checked_out {
2894
            "; working copy updated"
2895
        } else {
2896
            "; run oxy checkout to update the working copy"
2897
        })
7b9809e6 Andy McCall ↶ 2898
    ));
2899
    let snap = repo.snapshot("oxy merge")?;
2900
    if snap.base_snapshot.is_conflicted() {
75167672 Andy McCall 2901
        println!(
2902
            "  {} {}",
2903
            ui::caution(),
2904
            ui::warn("the result has conflicts; see oxy status")
2905
        );
7b9809e6 Andy McCall ↶ 2906
    }
98a1620a Andy McCall ↶ 2907
    where_now(&repo);
7b9809e6 Andy McCall ↶ 2908
    Ok(0)
2909
}
2910
728da0f9 Andy McCall ↶ 2911
pub fn squash(change: &str, into: &str) -> Result<i32> {
2912
    let repo = open()?;
2913
    let c = repo.resolve_change(change)?;
2914
    let d = repo.resolve_change(into)?;
2915
    let sp = ui::spinner("Squashing…");
2916
    let h = repo.squash(&c, &d)?;
631fc897 Andy McCall ↶ 2917
    let new = repo.store().get_change_meta(&h)?;
75167672 Andy McCall 2918
    sp.done(&format!(
2919
        "{} into {} {}  {}",
2920
        ui::done("Squashed"),
631fc897 Andy McCall ↶ 2921
        ui::id(new.change_id.short(8)),
75167672 Andy McCall 2922
        ui::hash(format!("#{}", h.short(12))),
2923
        ui::subject(subject_of(&new.message))
2924
    ));
728da0f9 Andy McCall ↶ 2925
    Ok(0)
2926
}
2927
e266b7d4 Andy McCall ↶ 2928
pub fn rebase(change: &str, onto: &[String], into_working_copy: bool) -> Result<i32> {
728da0f9 Andy McCall ↶ 2929
    let repo = open()?;
2930
    let c = repo.resolve_change(change)?;
2931
    let mut heads = Vec::new();
2932
    for o in onto {
2933
        heads.extend(repo.resolve_heads(o)?);
2934
    }
2935
    let sp = ui::spinner("Rebasing…");
e266b7d4 Andy McCall ↶ 2936
    let h = match repo.rebase(&c, &heads) {
2937
        Err(oxy_core::repo::RepoError::Conflicted(_)) if into_working_copy => {
2938
            let r = repo.rebase_into_working_copy(&c, &heads)?;
2939
            sp.done(&format!(
2940
                "{} {} into {}: now {}",
2941
                ui::done("Merged"),
2942
                onto.join(" "),
2943
                ui::bookmark(&repo.config.bookmark),
2944
                plural(r.heads.len(), "head"),
2945
            ));
2946
            if r.checked_out {
2947
                println!(
2948
                    "  {} {}",
2949
                    ui::caution(),
2950
                    ui::warn("the conflict is in the working copy: fix the marked files (see oxy status), then oxy commit")
2951
                );
2952
            } else {
2953
                println!(
2954
                    "  {} {}",
2955
                    ui::caution(),
2956
                    ui::warn("the working copy has changes of its own: commit or discard them, then oxy checkout to take the conflict in")
2957
                );
2958
            }
98a1620a Andy McCall ↶ 2959
            where_now(&repo);
e266b7d4 Andy McCall ↶ 2960
            return Ok(0);
2961
        }
2962
        other => other?,
2963
    };
631fc897 Andy McCall ↶ 2964
    let new = repo.store().get_change_meta(&h)?;
75167672 Andy McCall 2965
    sp.done(&format!(
2966
        "{} {} {}  {}",
2967
        ui::done("Rebased"),
631fc897 Andy McCall ↶ 2968
        ui::id(new.change_id.short(8)),
75167672 Andy McCall 2969
        ui::hash(format!("#{}", h.short(12))),
2970
        ui::dim(format!(
2971
            "now depends on {}",
2972
            plural(new.deps.len(), "change")
2973
        ))
2974
    ));
98a1620a Andy McCall ↶ 2975
    where_now(&repo);
728da0f9 Andy McCall ↶ 2976
    Ok(0)
79b5a90f Andy McCall ↶ 2977
}
2978
2979
#[cfg(test)]
2980
mod tests {
2981
    use super::*;
2982
2983
    #[test]
2984
    fn id_prefixes_agree_with_the_whole_list() {
2985
        // Ids that share long prefixes, and some that differ only in a
2986
        // half byte, from a fixed sequence.
2987
        let mut x: u64 = 0x9e37_79b9_7f4a_7c15;
2988
        let mut next = || {
2989
            x ^= x << 13;
2990
            x ^= x >> 7;
2991
            x ^= x << 17;
2992
            x
2993
        };
2994
        let mut ids: Vec<[u8; 16]> = Vec::new();
2995
        for _ in 0..2000 {
2996
            let mut id = [0u8; 16];
2997
            id[..8].copy_from_slice(&next().to_le_bytes());
2998
            id[8..].copy_from_slice(&next().to_le_bytes());
2999
            let share = (next() % 4) as usize;
3000
            if let Some(prev) = ids.last() {
3001
                id[..share].copy_from_slice(&prev[..share]);
3002
                if next() % 2 == 0 {
3003
                    id[share] = (prev[share] & 0xf0) | (id[share] & 0x0f);
3004
                }
3005
            }
3006
            ids.push(id);
3007
        }
3008
        let hex = |id: &[u8; 16]| oxy_core::ids::ChangeId(*id).to_hex();
3009
        let whole = unique_prefixes(ids.iter().map(hex));
3010
        let some = id_prefixes(ids.clone(), ids.iter().copied());
3011
        for id in &ids {
3012
            assert_eq!(some[id], whole[&hex(id)], "{}", hex(id));
3013
        }
3014
    }
728da0f9 Andy McCall ↶ 3015
}