feat: add location info to root node

Author: gengjiawen

parser/ast.rs

@@ -27,6 +27,7 @@ impl fmt::Display for Node {
 #[derive(Clone, Debug, Eq, Deserialize, Hash, PartialEq)]
 pub struct Program {
     pub body: Vec<Statement>,
+    pub span: Span,
 }
 
 // remove this if serde resolve resolve https://github.com/serde-rs/serde/issues/760
@@ -37,14 +38,18 @@ impl Serialize for Program {
     {
         let mut state = serializer.serialize_struct("Program", 2)?;
         state.serialize_field("body", &self.body)?;
+        state.serialize_field("span", &self.span)?;
         state.serialize_field("type", "Program")?;
         state.end()
     }
 }
 
 impl Program {
     pub fn new() -> Self {
-        Program { body: vec![] }
+        Program { body: vec![], span: Span {
+            start: 0,
+            end: 0,
+        } }
     }
 }
 
@@ -103,7 +108,7 @@ impl fmt::Display for BlockStatement {
 #[serde(tag = "expr_type")]
 pub enum Expression {
     IDENTIFIER(IDENTIFIER),
-    LITERAL(Literal),
+    LITERAL(Literal), // need to flatten
     PREFIX(UnaryExpression),
     INFIX(BinaryExpression),
     IF(IF),

parser/lib.rs

@@ -75,6 +75,7 @@ impl<'a> Parser<'a> {
             }
             self.next_token();
         }
+        program.span.end = self.current_token.span.end;
 
         if self.errors.is_empty() {
             return Ok(program);

parser/snapshots/parser__ast_tree_test__tests__test_array.snap

@@ -35,6 +35,10 @@ expression: "[1, true]"
         }
       }
     ],
+    "span": {
+      "start": 0,
+      "end": 10
+    },
     "type": "Program"
   }
 }

parser/snapshots/parser__ast_tree_test__tests__test_binary.snap

@@ -66,6 +66,10 @@ expression: 1 + 2 * 3
         }
       }
     ],
+    "span": {
+      "start": 0,
+      "end": 10
+    },
     "type": "Program"
   }
 }

parser/snapshots/parser__ast_tree_test__tests__test_func_call.snap

@@ -42,6 +42,10 @@ expression: "add(1, 2)"
         }
       }
     ],
+    "span": {
+      "start": 0,
+      "end": 10
+    },
     "type": "Program"
   }
 }

parser/snapshots/parser__ast_tree_test__tests__test_func_declaration.snap

@@ -28,6 +28,10 @@ expression: "fn(x) { x };"
         }
       }
     ],
+    "span": {
+      "start": 0,
+      "end": 13
+    },
     "type": "Program"
   }
 }

parser/snapshots/parser__ast_tree_test__tests__test_hash.snap

@@ -37,6 +37,10 @@ expression: "{\"a\": 1}"
         }
       }
     ],
+    "span": {
+      "start": 0,
+      "end": 9
+    },
     "type": "Program"
   }
 }

parser/snapshots/parser__ast_tree_test__tests__test_if.snap

@@ -68,6 +68,10 @@ expression: "if (x < y) { x } else { y }"
         }
       }
     ],
+    "span": {
+      "start": 0,
+      "end": 28
+    },
     "type": "Program"
   }
 }

parser/snapshots/parser__ast_tree_test__tests__test_index.snap

@@ -31,6 +31,10 @@ expression: "a[1]"
         }
       }
     ],
+    "span": {
+      "start": 0,
+      "end": 5
+    },
     "type": "Program"
   }
 }

parser/snapshots/parser__ast_tree_test__tests__test_let.snap

@@ -32,6 +32,10 @@ expression: let a = 3
         }
       }
     ],
+    "span": {
+      "start": 0,
+      "end": 10
+    },
     "type": "Program"
   }
 }