@devrev/meerkat-core 0.0.84 → 0.0.86

package/src/utils/get-column-names-from-ast.js

@@ -0,0 +1,45 @@
+"use strict";
+Object.defineProperty(exports, "getColumnNamesFromAst", {
+    enumerable: true,
+    get: function() {
+        return getColumnNamesFromAst;
+    }
+});
+const _constant = require("../cube-filter-transformer/constant");
+const _utils = require("../types/utils");
+const getColumnNamesFromAst = (node, columnSet = new Set())=>{
+    if (!node) return Array.from(columnSet);
+    if ((0, _utils.isBetweenExpression)(node)) {
+        getColumnNamesFromAst(node.input, columnSet);
+        getColumnNamesFromAst(node.lower, columnSet);
+        getColumnNamesFromAst(node.upper, columnSet);
+    }
+    if ((0, _utils.isColumnRefExpression)(node)) {
+        columnSet.add(node.column_names.join(_constant.COLUMN_NAME_DELIMITER));
+    }
+    if ((0, _utils.isCastExpression)(node) || (0, _utils.isCollateExpression)(node)) {
+        getColumnNamesFromAst(node.child, columnSet);
+    }
+    if ((0, _utils.isComparisonExpression)(node)) {
+        getColumnNamesFromAst(node.left, columnSet);
+        getColumnNamesFromAst(node.right, columnSet);
+    }
+    if ((0, _utils.isFunctionExpression)(node) || (0, _utils.isWindowExpression)(node) || (0, _utils.isOperatorExpression)(node) || (0, _utils.isConjunctionExpression)(node)) {
+        node.children.forEach((child)=>getColumnNamesFromAst(child, columnSet));
+    }
+    if ((0, _utils.isCaseExpression)(node)) {
+        node.case_checks.forEach((check)=>{
+            getColumnNamesFromAst(check.then_expr, columnSet);
+            getColumnNamesFromAst(check.when_expr, columnSet);
+        });
+        getColumnNamesFromAst(node.else_expr, columnSet);
+    }
+    if ((0, _utils.isSubqueryExpression)(node) && (0, _utils.isSelectNode)(node.subquery.node)) {
+        node.subquery.node.select_list.forEach((node)=>{
+            getColumnNamesFromAst(node, columnSet);
+        });
+    }
+    return Array.from(columnSet);
+};
+
+//# sourceMappingURL=get-column-names-from-ast.js.map

package/src/utils/get-column-names-from-ast.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../meerkat-core/src/utils/get-column-names-from-ast.ts"],"sourcesContent":["import { COLUMN_NAME_DELIMITER } from '../cube-filter-transformer/constant';\nimport { ParsedExpression } from '../types/duckdb-serialization-types';\nimport {\n  isBetweenExpression,\n  isCaseExpression,\n  isCastExpression,\n  isCollateExpression,\n  isColumnRefExpression,\n  isComparisonExpression,\n  isConjunctionExpression,\n  isFunctionExpression,\n  isOperatorExpression,\n  isSelectNode,\n  isSubqueryExpression,\n  isWindowExpression,\n} from '../types/utils';\n\nexport const getColumnNamesFromAst = (\n  node: ParsedExpression,\n  columnSet: Set<string> = new Set()\n): string[] => {\n  if (!node) return Array.from(columnSet);\n\n  if (isBetweenExpression(node)) {\n    getColumnNamesFromAst(node.input, columnSet);\n    getColumnNamesFromAst(node.lower, columnSet);\n    getColumnNamesFromAst(node.upper, columnSet);\n  }\n\n  if (isColumnRefExpression(node)) {\n    columnSet.add(node.column_names.join(COLUMN_NAME_DELIMITER));\n  }\n\n  if (isCastExpression(node) || isCollateExpression(node)) {\n    getColumnNamesFromAst(node.child, columnSet);\n  }\n\n  if (isComparisonExpression(node)) {\n    getColumnNamesFromAst(node.left, columnSet);\n    getColumnNamesFromAst(node.right, columnSet);\n  }\n\n  if (\n    isFunctionExpression(node) ||\n    isWindowExpression(node) ||\n    isOperatorExpression(node) ||\n    isConjunctionExpression(node)\n  ) {\n    node.children.forEach((child) => getColumnNamesFromAst(child, columnSet));\n  }\n\n  if (isCaseExpression(node)) {\n    node.case_checks.forEach((check) => {\n      getColumnNamesFromAst(check.then_expr, columnSet);\n      getColumnNamesFromAst(check.when_expr, columnSet);\n    });\n    getColumnNamesFromAst(node.else_expr, columnSet);\n  }\n\n  if (isSubqueryExpression(node) && isSelectNode(node.subquery.node)) {\n    node.subquery.node.select_list.forEach((node) => {\n      getColumnNamesFromAst(node, columnSet);\n    });\n  }\n\n  return Array.from(columnSet);\n};\n"],"names":["getColumnNamesFromAst","node","columnSet","Set","Array","from","isBetweenExpression","input","lower","upper","isColumnRefExpression","add","column_names","join","COLUMN_NAME_DELIMITER","isCastExpression","isCollateExpression","child","isComparisonExpression","left","right","isFunctionExpression","isWindowExpression","isOperatorExpression","isConjunctionExpression","children","forEach","isCaseExpression","case_checks","check","then_expr","when_expr","else_expr","isSubqueryExpression","isSelectNode","subquery","select_list"],"mappings":";+BAiBaA;;;eAAAA;;;0BAjByB;uBAe/B;AAEA,MAAMA,wBAAwB,CACnCC,MACAC,YAAyB,IAAIC,KAAK;IAElC,IAAI,CAACF,MAAM,OAAOG,MAAMC,IAAI,CAACH;IAE7B,IAAII,IAAAA,0BAAmB,EAACL,OAAO;QAC7BD,sBAAsBC,KAAKM,KAAK,EAAEL;QAClCF,sBAAsBC,KAAKO,KAAK,EAAEN;QAClCF,sBAAsBC,KAAKQ,KAAK,EAAEP;IACpC;IAEA,IAAIQ,IAAAA,4BAAqB,EAACT,OAAO;QAC/BC,UAAUS,GAAG,CAACV,KAAKW,YAAY,CAACC,IAAI,CAACC,+BAAqB;IAC5D;IAEA,IAAIC,IAAAA,uBAAgB,EAACd,SAASe,IAAAA,0BAAmB,EAACf,OAAO;QACvDD,sBAAsBC,KAAKgB,KAAK,EAAEf;IACpC;IAEA,IAAIgB,IAAAA,6BAAsB,EAACjB,OAAO;QAChCD,sBAAsBC,KAAKkB,IAAI,EAAEjB;QACjCF,sBAAsBC,KAAKmB,KAAK,EAAElB;IACpC;IAEA,IACEmB,IAAAA,2BAAoB,EAACpB,SACrBqB,IAAAA,yBAAkB,EAACrB,SACnBsB,IAAAA,2BAAoB,EAACtB,SACrBuB,IAAAA,8BAAuB,EAACvB,OACxB;QACAA,KAAKwB,QAAQ,CAACC,OAAO,CAAC,CAACT,QAAUjB,sBAAsBiB,OAAOf;IAChE;IAEA,IAAIyB,IAAAA,uBAAgB,EAAC1B,OAAO;QAC1BA,KAAK2B,WAAW,CAACF,OAAO,CAAC,CAACG;YACxB7B,sBAAsB6B,MAAMC,SAAS,EAAE5B;YACvCF,sBAAsB6B,MAAME,SAAS,EAAE7B;QACzC;QACAF,sBAAsBC,KAAK+B,SAAS,EAAE9B;IACxC;IAEA,IAAI+B,IAAAA,2BAAoB,EAAChC,SAASiC,IAAAA,mBAAY,EAACjC,KAAKkC,QAAQ,CAAClC,IAAI,GAAG;QAClEA,KAAKkC,QAAQ,CAAClC,IAAI,CAACmC,WAAW,CAACV,OAAO,CAAC,CAACzB;YACtCD,sBAAsBC,MAAMC;QAC9B;IACF;IAEA,OAAOE,MAAMC,IAAI,CAACH;AACpB"}