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
use std::ffi::OsStr;
use std::io::BufRead as _;
use std::io::BufReader;
use std::io::Error;
use std::io::ErrorKind;
use std::io::Result;
use std::process::Child;
use std::process::ChildStdout;
use std::process::Command;
use std::process::Stdio;
use diff_parse::File;
pub const CLANG_FORMAT: &str = "clang-format";
pub fn await_child<S>(program: S, child: Child) -> Result<Option<ChildStdout>>
where
S: AsRef<OsStr>,
{
let mut child = child;
let status = child.wait()?;
if !status.success() {
let error = format!("process `{}` failed", program.as_ref().to_string_lossy());
if let Some(stderr) = child.stderr {
let mut stderr = BufReader::new(stderr);
let mut line = String::new();
if stderr.read_line(&mut line).is_ok() {
let line = line.trim();
return Err(Error::new(ErrorKind::Other, format!("{error}: {line}")))
}
}
return Err(Error::new(ErrorKind::Other, error))
}
Ok(child.stdout)
}
pub fn format(diffs: &[(File, File)]) -> Result<()> {
fn format_now(file: &str, lines_args: &[String]) -> Result<()> {
let child = Command::new(CLANG_FORMAT)
.arg("-i")
.arg(file)
.args(lines_args)
.stdin(Stdio::null())
.stdout(Stdio::null())
.stderr(Stdio::piped())
.spawn()?;
let _ = await_child(CLANG_FORMAT, child)?;
Ok(())
}
let mut last_dst = Option::<String>::None;
let mut lines = Vec::new();
for (_, dst) in diffs {
match last_dst {
Some(prev_dst) if prev_dst != *dst.file => {
let () = format_now(&prev_dst, &lines)?;
last_dst = Some(dst.file.to_string());
lines.clear();
},
_ => (),
}
let start_line = dst.line;
let end_line = dst.line + dst.count;
lines.push(format!("--lines={start_line}:{end_line}"));
if last_dst.is_none() {
last_dst = Some(dst.file.to_string());
}
}
if let Some(prev_dst) = last_dst {
let () = format_now(&prev_dst, &lines)?;
}
Ok(())
}