punktf 3.1.2

A cross-platform multi-target dotfiles manager
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121

//! Functions and utilities for the [`Diff`](`crate::opt::Diff`) command
//! and [`Diff`](`punktf_lib::visit::diff::Diff`) visitor.

use crate::opt::DiffFormat;
use console::{style, Style};
use punktf_lib::visit::diff::Event;
use similar::{ChangeTag, TextDiff};
use std::{fmt, path::Path};

/// Processes diff [`Event`s](`punktf_lib::visit::diff::Event`) from the visitor.
pub fn diff(format: DiffFormat, event: Event<'_>) {
	match event {
		Event::NewFile {
			relative_source_path,
			target_path,
		} => println!(
			"[{} => {}] New file",
			style(relative_source_path.display())
				.bold()
				.black()
				.bright(),
			style(target_path.display()).bold().bright()
		),
		Event::NewDirectory {
			relative_source_path,
			target_path,
		} => println!(
			"[{} => {}] New directory",
			style(relative_source_path.display())
				.bold()
				.black()
				.bright(),
			style(target_path.display()).bold().bright()
		),
		Event::Diff {
			relative_source_path,
			target_path,
			old_content,
			new_content,
		} => {
			if format == DiffFormat::Unified {
				print_udiff(target_path, &old_content, &new_content);
			} else {
				print_pretty(
					relative_source_path,
					target_path,
					&old_content,
					&new_content,
				);
			}
		}
	}
}

/// Prints a file diff with the gnu unified format.
fn print_udiff(target: &Path, old: &str, new: &str) {
	let diff = TextDiff::from_lines(old, new);

	println!("--- {path}\r\n+++ {path}", path = target.display());

	diff.unified_diff()
		.to_writer(std::io::stdout())
		.expect("Writing to stdout to never fail");
}

/// Used to pretty print diff line numbers.
struct Line(Option<usize>);

impl fmt::Display for Line {
	fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
		match self.0 {
			None => write!(f, "    "),
			Some(idx) => write!(f, "{:<4}", idx + 1),
		}
	}
}

/// Prints a file diff with ansii escape codes.
fn print_pretty(source: &Path, target: &Path, old: &str, new: &str) {
	let diff = TextDiff::from_lines(old, new);

	for (idx, group) in diff.grouped_ops(3).iter().enumerate() {
		if idx == 0 {
			println!(
				">> {} => {}",
				style(source.display()).bold().black().bright(),
				style(target.display()).bold().bright()
			);
		}

		if idx > 0 {
			println!("{:-^1$}", "-", 80);
		}

		for op in group {
			for change in diff.iter_inline_changes(op) {
				let (sign, s) = match change.tag() {
					ChangeTag::Delete => ("-", Style::new().red()),
					ChangeTag::Insert => ("+", Style::new().green()),
					ChangeTag::Equal => (" ", Style::new().dim()),
				};
				print!(
					"{}{} |{}",
					style(Line(change.old_index())).dim(),
					style(Line(change.new_index())).dim(),
					s.apply_to(sign).bold(),
				);
				for (emphasized, value) in change.iter_strings_lossy() {
					if emphasized {
						print!("{}", s.apply_to(value).underlined().on_black());
					} else {
						print!("{}", s.apply_to(value));
					}
				}
				if change.missing_newline() {
					println!();
				}
			}
		}
	}
}