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
|
use std::{path::PathBuf, fs, sync::{Arc, Mutex}};
use rowan::ast::AstNode;
use threadpool::ThreadPool;
use crate::{status_reporter::*, queries::Query, parse_nixfile, changes::apply_changes};
#[allow(unreachable_code, unused)]
pub fn batchmode(tasks: Vec<PathBuf>, query: Query, debug: bool) {
fn do_task(path: PathBuf, query: Query, debug: bool) -> anyhow::Result<(PathBuf, Option<String>)> {
let (content, nexp) = match parse_nixfile(&path) {
Err(e) => {
anyhow::bail!("could not parse file {path:?}")
},
Ok(exp) => exp
};
let (changes, _) = query.apply(&content, nexp.syntax().clone())?;
let changed = apply_changes(&content, changes, debug);
Ok((path, if changed != content { Some(changed) } else { None }))
}
let pool = ThreadPool::new(16);
let results = Arc::new(Mutex::new(vec![]));
let printer = Arc::new(StatusReport::new(tasks.len(), tasks.len()));
for path in tasks {
pool.execute({
let results = Arc::clone(&results);
let printer = Arc::clone(&printer);
let query = query.clone();
move || {
printer.enter_file(&format!("{path:?}"));
let result = do_task(path, query, debug);
results.lock().unwrap().push(result);
}
});
}
pool.join();
println!("\n\nSummary:");
let mut c_errors = 0;
let mut c_total = 0;
let mut c_changes = 0;
for r in results.lock().unwrap().iter() {
match r {
Err(e) => {
println!(" {}", e);
c_errors += 1;
},
Ok((_, Some(_))) => c_changes += 1,
_ => ()
}
c_total += 1;
}
println!("\n ({c_total} sites total, {c_errors} errors, generated {} edits)", c_changes);
let edits: Vec<_> = Arc::into_inner(results).unwrap().into_inner().unwrap()
.into_iter()
.filter_map(|r| r.ok())
.collect();
println!("applying changes ...");
for (filename, content) in edits {
if let Some(content) = content {
fs::write(&filename, content.as_bytes()).unwrap();
}
}
}
|
