feat: control filenames in DiffOptions

src/diff/mod.rs

@@ -3,7 +3,7 @@ use crate::{
     range::{DiffRange, SliceLike},
     utils::Classifier,
 };
-use std::{cmp, ops};
+use std::{borrow::Cow, cmp, ops};
 
 mod cleanup;
 mod myers;
@@ -46,6 +46,8 @@ where
 pub struct DiffOptions {
     compact: bool,
     context_len: usize,
+    original_filename: Option<String>,
+    modified_filename: Option<String>,
 }
 
 impl DiffOptions {
@@ -57,6 +59,8 @@ impl DiffOptions {
         Self {
             compact: true,
             context_len: 3,
+            original_filename: None,
+            modified_filename: None,
         }
     }
 
@@ -76,6 +80,22 @@ impl DiffOptions {
         self
     }
 
+    /// Set the filename to be used in the patch for the original text
+    ///
+    /// If not set, the default value is "original".
+    pub fn set_original_filename(&mut self, filename: impl ToString) -> &mut Self {
+        self.original_filename = Some(filename.to_string());
+        self
+    }
+
+    /// Set the filename to be used in the patch for the modified text
+    ///
+    /// If not set, the default value is "modified".
+    pub fn set_modified_filename(&mut self, filename: impl ToString) -> &mut Self {
+        self.modified_filename = Some(filename.to_string());
+        self
+    }
+
     // TODO determine if this should be exposed in the public API
     #[allow(dead_code)]
     fn diff<'a>(&self, original: &'a str, modified: &'a str) -> Vec<Diff<'a, str>> {
@@ -102,7 +122,19 @@ impl DiffOptions {
         let solution = self.diff_slice(&old_ids, &new_ids);
 
         let hunks = to_hunks(&old_lines, &new_lines, &solution, self.context_len);
-        Patch::new(Some("original"), Some("modified"), hunks)
+        Patch::new(
+            self.original_filename
+                .as_ref()
+                .map_or(Some(Cow::Borrowed("original")), |s| {
+                    Some(Cow::Owned(s.clone()))
+                }),
+            self.modified_filename
+                .as_ref()
+                .map_or(Some(Cow::Borrowed("modified")), |s| {
+                    Some(Cow::Owned(s.clone()))
+                }),
+            hunks,
+        )
     }
 
     /// Create a patch between two potentially non-utf8 texts
@@ -358,3 +390,42 @@ fn build_edit_script<T>(solution: &[DiffRange<[T]>]) -> Vec<EditRange> {
 
     edit_script
 }
+
+#[cfg(test)]
+mod test {
+    use super::DiffOptions;
+
+    #[test]
+    fn set_original_and_modified_filenames() {
+        let original = "\
+I am afraid, however, that all I have known - that my story - will be forgotten.
+I am afraid for the world that is to come.
+Afraid that my plans will fail.
+Afraid of a doom worse than the Deepness.
+";
+        let modified = "\
+I am afraid, however, that all I have known - that my story - will be forgotten.
+I am afraid for the world that is to come.
+Afraid that Alendi will fail.
+Afraid of a doom brought by the Deepness.
+";
+        let expected = "\
+--- the old version
++++ the better version
+@@ -1,4 +1,4 @@
+ I am afraid, however, that all I have known - that my story - will be forgotten.
+ I am afraid for the world that is to come.
+-Afraid that my plans will fail.
+-Afraid of a doom worse than the Deepness.
++Afraid that Alendi will fail.
++Afraid of a doom brought by the Deepness.
+";
+
+        let patch = DiffOptions::new()
+            .set_original_filename("the old version")
+            .set_modified_filename("the better version")
+            .create_patch(original, modified);
+
+        assert_eq!(patch.to_string(), expected);
+    }
+}