uucore: don't show error for ambiguous value

that's a direct match in ShortcutValueParser
uutils · Aug 1, 2023 · 97786a5 · 97786a5
1 parent f37318d
commit 97786a5
Showing 1 changed file with 45 additions and 22 deletions.
diff --git a/src/uucore/src/lib/parser/shortcut_value_parser.rs b/src/uucore/src/lib/parser/shortcut_value_parser.rs
@@ -15,6 +15,34 @@ impl ShortcutValueParser {
     pub fn new(values: impl Into<Self>) -> Self {
         values.into()
     }
+
+    fn generate_clap_error(
+        &self,
+        cmd: &clap::Command,
+        arg: Option<&clap::Arg>,
+        value: &str,
+    ) -> clap::Error {
+        let mut err = clap::Error::new(ErrorKind::InvalidValue).with_cmd(cmd);
+
+        if let Some(arg) = arg {
+            err.insert(
+                ContextKind::InvalidArg,
+                ContextValue::String(arg.to_string()),
+            );
+        }
+
+        err.insert(
+            ContextKind::InvalidValue,
+            ContextValue::String(value.to_string()),
+        );
+
+        err.insert(
+            ContextKind::ValidValue,
+            ContextValue::Strings(self.0.iter().map(|x| x.get_name().to_string()).collect()),
+        );
+
+        err
+    }
 }
 
 impl TypedValueParser for ShortcutValueParser {
@@ -36,29 +64,16 @@ impl TypedValueParser for ShortcutValueParser {
             .filter(|x| x.get_name().starts_with(value))
             .collect();
 
-        if matched_values.len() == 1 {
-            Ok(matched_values[0].get_name().to_string())
-        } else {
-            let mut err = clap::Error::new(ErrorKind::InvalidValue).with_cmd(cmd);
-
-            if let Some(arg) = arg {
-                err.insert(
-                    ContextKind::InvalidArg,
-                    ContextValue::String(arg.to_string()),
-                );
+        match matched_values.len() {
+            0 => Err(self.generate_clap_error(cmd, arg, value)),
+            1 => Ok(matched_values[0].get_name().to_string()),
+            _ => {
+                if let Some(direct_match) = matched_values.iter().find(|x| x.get_name() == value) {
+                    Ok(direct_match.get_name().to_string())
+                } else {
+                    Err(self.generate_clap_error(cmd, arg, value))
+                }
             }
-
-            err.insert(
-                ContextKind::InvalidValue,
-                ContextValue::String(value.to_string()),
-            );
-
-            err.insert(
-                ContextKind::ValidValue,
-                ContextValue::Strings(self.0.iter().map(|x| x.get_name().to_string()).collect()),
-            );
-
-            Err(err)
         }
     }
 
@@ -127,6 +142,14 @@ mod tests {
         assert_eq!("abef", result.unwrap());
     }
 
+    #[test]
+    fn test_parse_ref_with_ambiguous_value_that_is_a_possible_value() {
+        let cmd = Command::new("cmd");
+        let parser = ShortcutValueParser::new(["abcd", "abcdefgh"]);
+        let result = parser.parse_ref(&cmd, None, OsStr::new("abcd"));
+        assert_eq!("abcd", result.unwrap());
+    }
+
     #[test]
     #[cfg(unix)]
     fn test_parse_ref_with_invalid_utf8() {