Fix publish error with colored output and crate is already published

Author: syl20bnr

crates/tracel-xtask/src/utils/cargo.rs

@@ -49,12 +49,16 @@ pub fn is_cargo_crate_installed(crate_name: &str) -> bool {
     output_str.lines().any(|line| line.contains(crate_name))
 }
 
-pub fn parse_cargo_search_output(output: &str) -> Option<(&str, &str)> {
+pub fn parse_cargo_search_output(output: &str) -> Option<(String, String)> {
+    // First strip ANSI color codes
+    let ansi_re = Regex::new(r"\x1b\[[0-9;]*m").expect("should compile regex for ANSI codes");
+    let cleaned_output = ansi_re.replace_all(output, "");
+    // Then retrieve the crate name and version in the output
     let re = Regex::new(r#"(?P<name>[a-zA-Z0-9_-]+)\s*=\s*"(?P<version>\d+\.\d+\.\d+)""#)
         .expect("should compile regex");
-    if let Some(captures) = re.captures(output) {
+    if let Some(captures) = re.captures(&cleaned_output) {
         if let (Some(name), Some(version)) = (captures.name("name"), captures.name("version")) {
-            return Some((name.as_str(), version.as_str()));
+            return Some((name.as_str().to_owned(), version.as_str().to_owned()));
         }
     }
     None
@@ -67,12 +71,15 @@ mod tests {
 
     #[rstest]
     #[case::valid_input("tracel-xtask-macros = \"1.0.1\"", Some(("tracel-xtask-macros", "1.0.1")))]
+    #[case::valid_input_with_comments("heat-sdk-cli-macros = \"0.1.0\"    # Macros for Tracel Heat SDK CLI.\n", Some(("heat-sdk-cli-macros", "0.1.0")))]
+    #[case::valid_input_with_comments_and_color_codes("\u{1b}[1m\u{1b}[32mheat-sdk-cli-macros\u{1b}[0m = \"0.1.0\"    # Macros for Tracel Heat SDK CLI.\n", Some(("heat-sdk-cli-macros", "0.1.0")))]
     #[case::missing_version("tracel-xtask-macros =", None)]
     #[case::invalid_format("tracel-xtask-macros: \"1.0.1\"", None)]
     #[case::extra_whitespace("   tracel-xtask-macros    =    \"1.0.1\"  ", Some(("tracel-xtask-macros", "1.0.1")))]
     #[case::no_quotes("tracel-xtask-macros = 1.0.1", None)]
     #[case::wrong_version_format("tracel-xtask-macros = \"1.0\"", None)]
     fn test_parse_cargo_search_output(#[case] input: &str, #[case] expected: Option<(&str, &str)>) {
+        let expected = expected.map(|(name, version)| (name.to_string(), version.to_string()));
         let result = parse_cargo_search_output(input);
         assert_eq!(result, expected);
     }