expose declaration nodes

Author: mbostock

src/parse.js

@@ -23,6 +23,7 @@ export function parseCell(input, {tag, raw, globals, ...options} = {}) {
   }
   parseReferences(cell, input, globals);
   parseFeatures(cell, input);
+  parseDeclarations(cell);
   return cell;
 }
 
@@ -423,3 +424,40 @@ function parseFeatures(cell, input) {
   }
   return cell;
 }
+
+// Find declarations: things that this cell defines.
+function parseDeclarations(cell) {
+  if (!cell.body) {
+    cell.declarations = [];
+  } else if (cell.body.type === "ImportDeclaration") {
+    cell.declarations = cell.body.specifiers.map(s => s.local);
+  } else if (!cell.id) {
+    cell.declarations = [];
+  } else {
+    switch (cell.id.type) {
+      case "Identifier":
+      case "ViewExpression":
+      case "MutableExpression":
+        cell.declarations = [cell.id];
+        break;
+      case "ArrayPattern":
+      case "ObjectPattern":
+        cell.declarations = [];
+        declarePattern(cell.id, node => cell.declarations.push(node));
+        break;
+      default: throw new Error(`unexpected identifier: ${cell.id.type}`);
+    }
+  }
+}
+
+function declarePattern(node, callback) {
+  switch (node.type) {
+    case "Identifier": callback(node); break;
+    case "ObjectPattern": node.properties.forEach(node => declarePattern(node, callback)); break;
+    case "ArrayPattern": node.elements.forEach(node => node && declarePattern(node, callback)); break;
+    case "Property": declarePattern(node.value, callback); break;
+    case "RestElement": declarePattern(node.argument, callback); break;
+    case "AssignmentPattern": declarePattern(node.left, callback); break;
+    default: throw new Error(`illegal declaration: ${node.type}`);
+  }
+}

test/output/anonymous-block-cell.js.json

@@ -27,5 +27,6 @@
   "references": [],
   "fileAttachments": [],
   "databaseClients": [],
-  "secrets": []
+  "secrets": [],
+  "declarations": []
 }

test/output/anonymous-expression-cell.js.json

@@ -15,5 +15,6 @@
   "references": [],
   "fileAttachments": [],
   "databaseClients": [],
-  "secrets": []
+  "secrets": [],
+  "declarations": []
 }

test/output/anonymous-function.js.json

@@ -37,5 +37,6 @@
   "references": [],
   "fileAttachments": [],
   "databaseClients": [],
-  "secrets": []
+  "secrets": [],
+  "declarations": []
 }

test/output/await-block-cell.js.json

@@ -38,5 +38,6 @@
   ],
   "fileAttachments": [],
   "databaseClients": [],
-  "secrets": []
+  "secrets": [],
+  "declarations": []
 }

test/output/await-in-arrow-function-expression.js.json

@@ -48,5 +48,6 @@
   ],
   "fileAttachments": [],
   "databaseClients": [],
-  "secrets": []
+  "secrets": [],
+  "declarations": []
 }

test/output/await-in-arrow-function.js.json

@@ -74,5 +74,6 @@
   ],
   "fileAttachments": [],
   "databaseClients": [],
-  "secrets": []
+  "secrets": [],
+  "declarations": []
 }

test/output/await-in-class.js.json

@@ -88,5 +88,6 @@
   ],
   "fileAttachments": [],
   "databaseClients": [],
-  "secrets": []
+  "secrets": [],
+  "declarations": []
 }

test/output/await-in-function.js.json

@@ -60,5 +60,6 @@
   ],
   "fileAttachments": [],
   "databaseClients": [],
-  "secrets": []
+  "secrets": [],
+  "declarations": []
 }

test/output/await-in-markdown.md.json

@@ -88,5 +88,6 @@
   ],
   "fileAttachments": [],
   "databaseClients": [],
-  "secrets": []
+  "secrets": [],
+  "declarations": []
 }