merge_tools: function to extract all variables from tool arguments

To be used in the next commit

cli/src/merge_tools/external.rs

@@ -344,27 +344,39 @@ pub fn run_mergetool_external(
     Ok(tree_builder.write_tree())
 }
 
+// Not interested in $UPPER_CASE_VARIABLES
+static VARIABLE_REGEX: once_cell::sync::Lazy<Regex> =
+    once_cell::sync::Lazy::new(|| Regex::new(r"\$([a-z0-9_]+)\b").unwrap());
+
 fn interpolate_variables<V: AsRef<str>>(
     args: &[String],
     variables: &HashMap<&str, V>,
 ) -> Vec<String> {
-    // Not interested in $UPPER_CASE_VARIABLES
-    let re = Regex::new(r"\$([a-z0-9_]+)\b").unwrap();
     args.iter()
         .map(|arg| {
-            re.replace_all(arg, |caps: &Captures| {
+            VARIABLE_REGEX
-                let name = &caps[1];
+                .replace_all(arg, |caps: &Captures| {
-                if let Some(subst) = variables.get(name) {
+                    let name = &caps[1];
-                    subst.as_ref().to_owned()
+                    if let Some(subst) = variables.get(name) {
-                } else {
+                        subst.as_ref().to_owned()
-                    caps[0].to_owned()
+                    } else {
-                }
+                        caps[0].to_owned()
-            })
+                    }
-            .into_owned()
+                })
+                .into_owned()
         })
         .collect()
 }
 
+/// Return all variable names found in the args, without the dollar sign
+#[allow(unused)]
+fn find_all_variables(args: &[String]) -> Vec<String> {
+    args.iter()
+        .flat_map(|arg| VARIABLE_REGEX.find_iter(arg))
+        .map(|single_match| single_match.as_str()[1..].to_owned())
+        .collect()
+}
+
 pub fn edit_diff_external(
     editor: ExternalMergeTool,
     left_tree: &Tree,
@@ -515,4 +527,22 @@ mod tests {
             ["$left $right"],
         );
     }
+
+    #[test]
+    fn test_find_all_variables() {
+        assert_eq!(
+            find_all_variables(
+                &[
+                    "$left",
+                    "$right",
+                    "--two=$1 and $2",
+                    "--can-be-part-of-string=$output",
+                    "$NOT_CAPITALS",
+                    "--can-repeat=$right"
+                ]
+                .map(ToOwned::to_owned),
+            ),
+            ["left", "right", "1", "2", "output", "right"],
+        );
+    }
 }