Test and fix lexing of redirects

ProjectLighthouseCAU · Mar 21, 2024 · 5cff29e · 5cff29e
1 parent 6ada4f8
commit 5cff29e
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/src/line/lex.rs b/src/line/lex.rs
@@ -6,7 +6,7 @@ const REGEX: Lazy<Regex> = Lazy::new(|| Regex::new(&[
     r"'(?<singlequoted>[^']*)'",
     r#""(?<doublequoted>[^"]*)""#,
     r#"(?<strayquotes>['"]+)"#,
-    r#"(?<unquoted>[^'"\s]+)"#,
+    r#"(?<unquoted>[^'">\s]+)"#,
 ].join("|")).unwrap());
 
 #[derive(Debug, Clone, PartialEq, Eq)]
@@ -73,4 +73,15 @@ mod tests {
         // TODO: Should we parse adjacent stuff as one arg?
         assert_eq!(lex(r#"echo '"Hello 'world   1234"#), vec![arg("echo"), arg("\"Hello "), arg("world"), arg("1234")]);
     }
+
+    #[test]
+    fn redirects() {
+        assert_eq!(lex(">"), vec![redirect()]);
+        assert_eq!(lex(">>"), vec![redirect(), redirect()]);
+        assert_eq!(lex(">a"), vec![redirect(), arg("a")]);
+        assert_eq!(lex(">1"), vec![redirect(), arg("1")]);
+        assert_eq!(lex("  >0>  1"), vec![redirect(), arg("0"), redirect(), arg("1")]);
+        assert_eq!(lex("echo Test > a"), vec![arg("echo"), arg("Test"), redirect(), arg("a")]);
+        assert_eq!(lex(r#"echo '{"x": 23,"y":3}' > /dev/null"#), vec![arg("echo"), arg(r#"{"x": 23,"y":3}"#), redirect(), arg("/dev/null")])
+    }
 }