// Copyright 2022 The Jujutsu Authors//// Licensed under the Apache License, Version 2.0 (the "License");// you may not use this file except in compliance with the License.// You may obtain a copy of the License at//// https://www.apache.org/licenses/LICENSE-2.0//// Unless required by applicable law or agreed to in writing, software// distributed under the License is distributed on an "AS IS" BASIS,// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.// See the License for the specific language governing permissions and// limitations under the License.use std::collections::HashSet;use std::env;use std::path::Path;use std::path::PathBuf;use std::process::exit;use clap::Parser;use itertools::Itertools as _;/// A fake diff-editor, useful for testing#[derive(Parser, Debug)]#[clap()]struct Args { /// Path to the "before" directory before: PathBuf, /// Path to the "after" directory after: PathBuf, /// Ignored argument #[arg(long)] _ignore: Vec<String>,}fn files_recursively(p: &Path) -> HashSet<String> { let mut files = HashSet::new(); if !p.is_dir() { files.insert(p.file_name().unwrap().to_str().unwrap().to_string()); } else { for dir_entry in std::fs::read_dir(p).unwrap() { let dir_entry = dir_entry.unwrap(); let base_name = dir_entry.file_name().to_str().unwrap().to_string(); if !dir_entry.path().is_dir() { files.insert(base_name); } else { for sub_path in files_recursively(&dir_entry.path()) { files.insert(format!("{base_name}/{sub_path}")); } } } } files}fn main() { let args: Args = Args::parse(); let edit_script_path = PathBuf::from(std::env::var_os("DIFF_EDIT_SCRIPT").unwrap()); let edit_script = String::from_utf8(std::fs::read(&edit_script_path).unwrap()).unwrap(); for instruction in edit_script.split('\0') { let (command, payload) = instruction.split_once('\n').unwrap_or((instruction, "")); let parts = command.split(' ').collect_vec(); match parts.as_slice() { [""] => {} ["fail"] => exit(1), ["files-before", ..] => { let expected = parts[1..].iter().copied().map(str::to_string).collect(); let actual = files_recursively(&args.before); if actual != expected { eprintln!( "fake-diff-editor: unexpected files before. EXPECTED: {:?} ACTUAL: {:?}", expected.iter().sorted().collect_vec(), actual.iter().sorted().collect_vec(), ); exit(1) } } ["files-after", ..] => { let expected = parts[1..].iter().copied().map(str::to_string).collect(); let actual = files_recursively(&args.after); if actual != expected { eprintln!( "fake-diff-editor: unexpected files after. EXPECTED: {:?} ACTUAL: {:?}", expected.iter().sorted().collect_vec(), actual.iter().sorted().collect_vec(), ); exit(1) } } ["print", message] => { println!("{message}"); } ["print-current-dir"] => { println!("{}", env::current_dir().unwrap().display()); } ["print-files-before"] => { for base_name in files_recursively(&args.before).iter().sorted() { println!("{base_name}"); } } ["print-files-after"] => { for base_name in files_recursively(&args.after).iter().sorted() { println!("{base_name}"); } } ["rm", file] => { std::fs::remove_file(args.after.join(file)).unwrap(); } ["reset", file] => { if args.before.join(file).exists() { std::fs::copy(args.before.join(file), args.after.join(file)).unwrap(); } else { std::fs::remove_file(args.after.join(file)).unwrap(); } } ["dump", file, dest] => { let dest_path = edit_script_path.parent().unwrap().join(dest); std::fs::copy(args.after.join(file), dest_path).unwrap(); } ["write", file] => { std::fs::write(args.after.join(file), payload).unwrap(); } _ => { eprintln!("fake-diff-editor: unexpected command: {command}"); exit(1) } } }}
// Copyright 2022 The Jujutsu Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
use std::collections::HashSet;
use std::env;
use std::path::Path;
use std::path::PathBuf;
use std::process::exit;
use clap::Parser;
use itertools::Itertools as _;
/// A fake diff-editor, useful for testing
#[derive(Parser, Debug)]
#[clap()]
struct Args {
/// Path to the "before" directory
before: PathBuf,
/// Path to the "after" directory
after: PathBuf,
/// Ignored argument
#[arg(long)]
_ignore: Vec<String>,
}
fn files_recursively(p: &Path) -> HashSet<String> {
let mut files = HashSet::new();
if !p.is_dir() {
files.insert(p.file_name().unwrap().to_str().unwrap().to_string());
} else {
for dir_entry in std::fs::read_dir(p).unwrap() {
let dir_entry = dir_entry.unwrap();
let base_name = dir_entry.file_name().to_str().unwrap().to_string();
if !dir_entry.path().is_dir() {
files.insert(base_name);
} else {
for sub_path in files_recursively(&dir_entry.path()) {
files.insert(format!("{base_name}/{sub_path}"));
}
}
}
}
files
}
fn main() {
let args: Args = Args::parse();
let edit_script_path = PathBuf::from(std::env::var_os("DIFF_EDIT_SCRIPT").unwrap());
let edit_script = String::from_utf8(std::fs::read(&edit_script_path).unwrap()).unwrap();
for instruction in edit_script.split('\0') {
let (command, payload) = instruction.split_once('\n').unwrap_or((instruction, ""));
let parts = command.split(' ').collect_vec();
match parts.as_slice() {
[""] => {}
["fail"] => exit(1),
["files-before", ..] => {
let expected = parts[1..].iter().copied().map(str::to_string).collect();
let actual = files_recursively(&args.before);
if actual != expected {
eprintln!(
"fake-diff-editor: unexpected files before. EXPECTED: {:?} ACTUAL: {:?}",
expected.iter().sorted().collect_vec(),
actual.iter().sorted().collect_vec(),
);
exit(1)
}
}
["files-after", ..] => {
let expected = parts[1..].iter().copied().map(str::to_string).collect();
let actual = files_recursively(&args.after);
if actual != expected {
eprintln!(
"fake-diff-editor: unexpected files after. EXPECTED: {:?} ACTUAL: {:?}",
expected.iter().sorted().collect_vec(),
actual.iter().sorted().collect_vec(),
);
exit(1)
}
}
["print", message] => {
println!("{message}");
}
["print-current-dir"] => {
println!("{}", env::current_dir().unwrap().display());
}
["print-files-before"] => {
for base_name in files_recursively(&args.before).iter().sorted() {
println!("{base_name}");
}
}
["print-files-after"] => {
for base_name in files_recursively(&args.after).iter().sorted() {
println!("{base_name}");
}
}
["rm", file] => {
std::fs::remove_file(args.after.join(file)).unwrap();
}
["reset", file] => {
if args.before.join(file).exists() {
std::fs::copy(args.before.join(file), args.after.join(file)).unwrap();
} else {
std::fs::remove_file(args.after.join(file)).unwrap();
}
}
["dump", file, dest] => {
let dest_path = edit_script_path.parent().unwrap().join(dest);
std::fs::copy(args.after.join(file), dest_path).unwrap();
}
["write", file] => {
std::fs::write(args.after.join(file), payload).unwrap();
}
_ => {
eprintln!("fake-diff-editor: unexpected command: {command}");
exit(1)
}
}
}
}