Add --asset_dir to blame-copy-royal

Author: cruessler

tests/it/src/args.rs

@@ -4,6 +4,7 @@ use clap::{
     builder::{OsStringValueParser, TypedValueParser},
     Arg, Command, Error,
 };
+use gix::bstr::BString;
 
 #[derive(Debug, clap::Parser)]
 #[clap(name = "it", about = "internal tools to help create test cases")]
@@ -51,6 +52,9 @@ pub enum Subcommands {
         worktree_dir: PathBuf,
         /// The directory into which to copy the files.
         destination_dir: PathBuf,
+        /// The directory to place assets in.
+        #[clap(long)]
+        asset_dir: Option<BString>,
         /// The file to extract the history for.
         file: std::ffi::OsString,
         /// Do not use `copy-royal` to obfuscate the content of blobs, but copy it verbatim.

tests/it/src/commands/blame_copy_royal.rs

@@ -23,6 +23,7 @@ pub(super) mod function {
         dry_run: bool,
         worktree_dir: &Path,
         destination_dir: PathBuf,
+        asset_dir: Option<BString>,
         file: &OsStr,
         Options { verbatim }: Options,
     ) -> anyhow::Result<()> {
@@ -76,7 +77,8 @@ pub(super) mod function {
             .blame_path
             .expect("blame path to be present as `debug_track_path == true`");
 
-        let assets = destination_dir.join("assets");
+        let asset_dir = asset_dir.unwrap_or("assets".into());
+        let assets = destination_dir.join(asset_dir.to_os_str()?);
         eprintln!("{prefix} create directory '{assets}'", assets = assets.display());
 
         if !dry_run {
@@ -107,7 +109,7 @@ pub(super) mod function {
             }
         }
 
-        let mut blame_script = BlameScript::new(blame_infos, Options { verbatim });
+        let mut blame_script = BlameScript::new(blame_infos, asset_dir, Options { verbatim });
         blame_script.generate()?;
 
         let script_file = destination_dir.join("create-history.sh");
@@ -130,9 +132,9 @@ pub(super) mod function {
     }
 
     enum BlameScriptOperation {
-        InitRepository,
+        InitRepository(BString),
         RemoveFile(String),
-        CommitFile(BString, ObjectId),
+        CommitFile(BString, BString, ObjectId),
         CheckoutTag(ObjectId),
         PrepareMerge(Vec<ObjectId>),
         CreateTag(ObjectId),
@@ -141,15 +143,15 @@ pub(super) mod function {
     impl Display for BlameScriptOperation {
         fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
             match self {
-                BlameScriptOperation::InitRepository => write!(
+                BlameScriptOperation::InitRepository(asset_dir) => write!(
                     f,
                     r"#!/bin/sh
 
 set -e
 
 git init
 echo .gitignore >> .gitignore
-echo assets/ >> .gitignore
+echo {asset_dir}/ >> .gitignore
 echo create-history.sh >> .gitignore
 
 "
@@ -160,16 +162,15 @@ echo create-history.sh >> .gitignore
 git rm {src}
 "
                 ),
-                BlameScriptOperation::CommitFile(src, commit_id) => {
+                BlameScriptOperation::CommitFile(asset_dir, src, commit_id) => {
                     writeln!(f, r"# make file {src} contain content at commit {commit_id}")?;
                     if let Some(pos) = src.rfind_byte(b'/') {
                         let dirname = src[..pos].as_bstr();
                         writeln!(f, "mkdir -p \"{dirname}\"")?;
                     }
                     write!(
                         f,
-                        r"#
-cp ./assets/{commit_id}.commit ./{src}
+                        r"cp ./{asset_dir}/{commit_id}.commit ./{src}
 git add {src}
 git commit -m {commit_id}
 "
@@ -194,17 +195,19 @@ git commit -m {commit_id}
         blame_infos: Vec<BlamePathEntry>,
         seen: BTreeSet<ObjectId>,
         script: Vec<BlameScriptOperation>,
+        asset_dir: BString,
         options: Options,
     }
 
     impl BlameScript {
-        fn new(blame_infos: Vec<BlamePathEntry>, options: Options) -> Self {
-            let script = vec![BlameScriptOperation::InitRepository];
+        fn new(blame_infos: Vec<BlamePathEntry>, asset_dir: BString, options: Options) -> Self {
+            let script = vec![BlameScriptOperation::InitRepository(asset_dir.clone())];
 
             Self {
                 blame_infos,
                 seen: BTreeSet::default(),
                 script,
+                asset_dir,
                 options,
             }
         }
@@ -265,7 +268,8 @@ git commit -m {commit_id}
                 if let Some(delete_previous_file_operation) = delete_previous_file_operation {
                     self.script.push(delete_previous_file_operation);
                 }
-                self.script.push(BlameScriptOperation::CommitFile(src, commit_id));
+                self.script
+                    .push(BlameScriptOperation::CommitFile(self.asset_dir.clone(), src, commit_id));
             } else {
                 let ([first], rest) = parents.split_at(1) else {
                     unreachable!();
@@ -277,7 +281,8 @@ git commit -m {commit_id}
                     if let Some(delete_previous_file_operation) = delete_previous_file_operation {
                         self.script.push(delete_previous_file_operation);
                     }
-                    self.script.push(BlameScriptOperation::CommitFile(src, commit_id));
+                    self.script
+                        .push(BlameScriptOperation::CommitFile(self.asset_dir.clone(), src, commit_id));
                 } else {
                     self.script.push(BlameScriptOperation::PrepareMerge(
                         rest.iter().map(|blame_path_entry| blame_path_entry.commit_id).collect(),
@@ -286,7 +291,8 @@ git commit -m {commit_id}
                     if let Some(delete_previous_file_operation) = delete_previous_file_operation {
                         self.script.push(delete_previous_file_operation);
                     }
-                    self.script.push(BlameScriptOperation::CommitFile(src, commit_id));
+                    self.script
+                        .push(BlameScriptOperation::CommitFile(self.asset_dir.clone(), src, commit_id));
                 }
             }
 

tests/it/src/main.rs

@@ -29,12 +29,14 @@ fn main() -> anyhow::Result<()> {
             dry_run,
             worktree_dir: worktree_root,
             destination_dir,
+            asset_dir,
             file,
             verbatim,
         } => commands::blame_copy_royal(
             dry_run,
             &worktree_root,
             destination_dir,
+            asset_dir,
             &file,
             commands::blame_copy_royal::Options { verbatim },
         ),