wip

Author: functora

rust/rustell/src/decode.rs

@@ -9,8 +9,7 @@ pub fn expr<'src>() -> impl Parser<
     Vec<Expr<'src>>,
     extra::Err<Rich<'src, char>>,
 > {
-    expr_use()
-        .or(expr_other())
+    choice((expr_use(), expr_other()))
         .repeated()
         .collect::<Vec<_>>()
 }

rust/rustell/src/encode.rs

@@ -10,6 +10,7 @@ pub fn expr<'a>(
 
 fn expr_one<'a>(ast: &'a Expr<'a>) -> IntoIter<&'a str> {
     match ast {
+        Expr::Mod(x) => vec!["mod ", x, ";"].into_iter(),
         Expr::Use(x) => expr_use(true, x)
             .chain(once(";"))
             .collect::<Vec<_>>()

rust/rustell/src/lib.rs

@@ -4,6 +4,7 @@ pub use chumsky::prelude::Parser;
 
 #[derive(Eq, PartialEq, Debug, Clone)]
 pub enum Expr<'src> {
+    Mod(&'src str),
     Use(ExprUse<'src>),
     Other(&'src str),
 }

rust/rustell/tests/integration.rs

@@ -325,6 +325,14 @@ fn test_parser_mixed_all_cases() {
     assert_eq!(decode(&encode(&rhs)), rhs)
 }
 
+#[test]
+fn test_rountrip_lib() {
+    let src =
+        std::fs::read_to_string("./src/lib.rs").unwrap();
+    let ast = decode(&src);
+    assert_eq!(decode(&encode(&ast)), ast);
+}
+
 fn sloppy(src: &str) -> String {
     src.replace(";", "")
 }