@narrative.io/data-collaboration-sdk-ts 0.0.150 → 0.0.151

package/build/nql/AstParser.d.ts

@@ -194,6 +194,7 @@ export interface Concat {
     as?: string;
     operands: Expression[];
 }
+export type Node = Select | Expression | Deduplication | Table | BooleanExpression | ArithmeticExpression | StringExpression | Join | AttributeRef | DatasetColumnRef | RawRef | Lit | FunctionCall | Add | Divide | Multiply | Subtract | UnaryMinus | And | Or | Not | Between | Equals | NotEquals | Lt | Lte | Gt | Gte | In | IsNull | Like | Concat;
 export declare function parseStatement(n: ast.NqlAst): Statement | Raw;
 export declare function parseCreateMaterializedView(n: ast.NqlCreateMaterializedView): CreateMaterializedView;
 export declare function parseExplain(n: ast.NqlExplain): Explain | Raw;

package/build/nql/AstTraverse.d.ts

@@ -0,0 +1,2 @@
+import { type Node } from "./AstParser";
+export declare function traverseParsedAst(node: Node, callback: (node: Node) => void): void;

package/build/nql/AstTraverse.js

@@ -0,0 +1,80 @@
+function findTraversalTargets(node) {
+    switch (node.type) {
+        case "function":
+            return node.args;
+        case "+":
+        case "/":
+        case "*":
+        case "-":
+        case "and":
+        case "or":
+        case "||":
+            return node.operands;
+        case "unary_-":
+            return [node.operand];
+        case "is_null":
+            return [node.operand];
+        case "like":
+            return [node.value];
+        case "not":
+            return node.operand.type !== "nql" ? [node.operand] : [];
+        case "between":
+            return [node.operand, node.lower, node.upper];
+        case "=":
+        case "<>":
+        case "<":
+        case "<=":
+        case ">":
+        case ">=":
+            return [node.left, node.right];
+        case "in": {
+            const inTargets = [node.expression];
+            if (Array.isArray(node.values)) {
+                inTargets.push(...node.values);
+            }
+            return inTargets;
+        }
+        case "deduplication":
+            return node.expressions;
+        case "select": {
+            const selectTargets = [...node.columns];
+            if (node.where !== null &&
+                node.where !== undefined &&
+                node.where.type !== "nql") {
+                selectTargets.push(node.where);
+            }
+            if (node.deduplication !== null &&
+                node.deduplication !== undefined &&
+                node.deduplication.type !== "nql") {
+                selectTargets.push(node.deduplication);
+            }
+            if (Array.isArray(node.from)) {
+                node.from.forEach((table) => {
+                    selectTargets.push(table);
+                });
+            }
+            return selectTargets;
+        }
+        case "rosetta_stone":
+        case "dataset":
+        case "raw_table": {
+            if (node.join?.condition != null) {
+                return [node.join];
+            }
+            return [];
+        }
+        case "join":
+            if (node.condition !== null && node.condition !== undefined) {
+                return [node.condition];
+            }
+            return [];
+        default:
+            return [];
+    }
+}
+export function traverseParsedAst(node, callback) {
+    callback(node);
+    findTraversalTargets(node).forEach((childNode) => {
+        traverseParsedAst(childNode, callback);
+    });
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@narrative.io/data-collaboration-sdk-ts",
-  "version": "0.0.150",
+  "version": "0.0.151",
   "main": "build/index.js",
   "repository": "github:narrative-io/data-collaboration-sdk-ts",
   "source": "src/index.ts",