Add source mappings for all nodes on parse()

Author: matthew-canestraro

src/expression/node/AccessorNode.js

@@ -136,7 +136,9 @@ export const createAccessorNode = /* #__PURE__ */ factory(name, dependencies, ({
      * @return {AccessorNode}
      */
     clone () {
-      return new AccessorNode(this.object, this.index)
+      const cloned = new AccessorNode(this.object, this.index)
+      cloned.sources = this.sources
+      return cloned
     }
 
     /**

src/expression/node/ArrayNode.js

@@ -94,7 +94,9 @@ export const createArrayNode = /* #__PURE__ */ factory(name, dependencies, ({ No
      * @return {ArrayNode}
      */
     clone () {
-      return new ArrayNode(this.items.slice(0))
+      const cloned = new ArrayNode(this.items.slice(0))
+      cloned.sources = this.sources
+      return cloned
     }
 
     /**

src/expression/node/AssignmentNode.js

@@ -231,7 +231,9 @@ export const createAssignmentNode = /* #__PURE__ */ factory(name, dependencies,
      * @return {AssignmentNode}
      */
     clone () {
-      return new AssignmentNode(this.object, this.index, this.value)
+      const cloned = new AssignmentNode(this.object, this.index, this.value)
+      cloned.sources = this.sources
+      return cloned
     }
 
     /**

src/expression/node/BlockNode.js

@@ -119,7 +119,9 @@ export const createBlockNode = /* #__PURE__ */ factory(name, dependencies, ({ Re
         }
       })
 
-      return new BlockNode(blocks)
+      const cloned = new BlockNode(blocks)
+      cloned.sources = this.sources
+      return cloned
     }
 
     /**

src/expression/node/ConditionalNode.js

@@ -121,7 +121,9 @@ export const createConditionalNode = /* #__PURE__ */ factory(name, dependencies,
      * @return {ConditionalNode}
      */
     clone () {
-      return new ConditionalNode(this.condition, this.trueExpr, this.falseExpr)
+      const cloned = new ConditionalNode(this.condition, this.trueExpr, this.falseExpr)
+      cloned.sources = this.sources
+      return cloned
     }
 
     /**

src/expression/node/ConstantNode.js

@@ -75,7 +75,9 @@ export const createConstantNode = /* #__PURE__ */ factory(name, dependencies, ({
      * @return {ConstantNode}
      */
     clone () {
-      return new ConstantNode(this.value)
+      const cloned = new ConstantNode(this.value)
+      cloned.sources = this.sources
+      return cloned
     }
 
     /**

src/expression/node/FunctionAssignmentNode.js

@@ -149,8 +149,10 @@ export const createFunctionAssignmentNode = /* #__PURE__ */ factory(name, depend
      * @return {FunctionAssignmentNode}
      */
     clone () {
-      return new FunctionAssignmentNode(
+      const cloned = new FunctionAssignmentNode(
         this.name, this.params.slice(0), this.expr)
+      cloned.sources = this.sources
+      return cloned
     }
 
     /**

src/expression/node/FunctionNode.js

@@ -310,7 +310,9 @@ export const createFunctionNode = /* #__PURE__ */ factory(name, dependencies, ({
      * @return {FunctionNode}
      */
     clone () {
-      return new FunctionNode(this.fn, this.args.slice(0))
+      const cloned = new FunctionNode(this.fn, this.args.slice(0))
+      cloned.sources = this.sources
+      return cloned
     }
 
     /**

src/expression/node/IndexNode.js

@@ -141,7 +141,9 @@ export const createIndexNode = /* #__PURE__ */ factory(name, dependencies, ({ No
      * @return {IndexNode}
      */
     clone () {
-      return new IndexNode(this.dimensions.slice(0), this.dotNotation)
+      const cloned = new IndexNode(this.dimensions.slice(0), this.dotNotation)
+      cloned.sources = this.sources
+      return cloned
     }
 
     /**

src/expression/node/Node.js

@@ -23,6 +23,8 @@ export const createNode = /* #__PURE__ */ factory(name, dependencies, ({ mathWit
   }
 
   class Node {
+    sources = []
+
     get type () { return 'Node' }
     get isNode () { return true }
 
@@ -204,6 +206,15 @@ export const createNode = /* #__PURE__ */ factory(name, dependencies, ({ mathWit
       throw new Error('Cannot clone a Node interface')
     }
 
+    /**
+     * Set the source indices mapping this node back to its
+     * location in the original source string
+     * @param {SourceMapping[]} sources - the data mapping this node back to its source string
+     */
+    setSources (sources) {
+      this.sources = sources
+    }
+
     /**
      * Create a deep clone of this node
      * @return {Node}