spark-sql-language-server 0.0.1-beta.5 → 0.2.0-beta.0

package/out-tsc/visitors/completion.visitor.js

@@ -0,0 +1,379 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CompletionVisitor = void 0;
+const sparksql_relation_visitor_1 = require("./sparksql-relation.visitor");
+const cursor_1 = require("../cursor");
+const SparkSQLParser_1 = require("../lib/SparkSQLParser");
+const utils_1 = require("../utils");
+const getRelationName = (relationPrimaryString) => {
+    const splits = relationPrimaryString.split(sparksql_relation_visitor_1.RELATION_SEPARATOR);
+    return splits[splits.length - 1];
+};
+function availableColumns(relation) {
+    const columns = [];
+    if (relation instanceof sparksql_relation_visitor_1.QueryRelation) {
+        relation.relations.forEach((rel, name) => {
+            const relationName = name !== rel.id ? name : undefined;
+            rel.columns.forEach(col => {
+                if (col.label !== sparksql_relation_visitor_1.STAR_LABEL && !col.label.includes(cursor_1.defaultCursorPlaceholder)) {
+                    columns.push({
+                        relation: relationName !== undefined ? { category: 'table', label: relationName } : undefined,
+                        name: col.label,
+                        desc: col.data !== undefined ? col.data : undefined
+                    });
+                }
+            });
+        });
+    }
+    else {
+        if (relation.sinkTableRelation !== undefined) {
+            relation.sinkTableRelation.columns.forEach(col => {
+                if (col.label !== sparksql_relation_visitor_1.STAR_LABEL && !col.label.includes(cursor_1.defaultCursorPlaceholder)) {
+                    columns.push({
+                        relation: { category: 'table', label: relation.sinkTableRelation.tablePrimary.relationName },
+                        name: col.label,
+                        desc: col.data !== undefined ? col.data : undefined
+                    });
+                }
+            });
+        }
+    }
+    return columns;
+}
+function getTableMetadataFnFactory(getColumns) {
+    return (relationPrimary) => {
+        if (relationPrimary.type !== 'table') {
+            return undefined;
+        }
+        const columns = getColumns({
+            table: relationPrimary.relationName,
+            catalog: relationPrimary.catalogName,
+            database: relationPrimary.databaseName
+        });
+        return {
+            columns: columns.map(c => {
+                return {
+                    id: `${(0, sparksql_relation_visitor_1.relationPrimaryToString)(relationPrimary)}_${c.label}`,
+                    label: c.label,
+                    data: c
+                };
+            })
+        };
+    };
+}
+const SQL_SNIPPETS = [
+    {
+        label: 'CREATE TABLE ?',
+        filterText: 'CREATE',
+        template: `CREATE TABLE \${1:table_name}
+(
+    \${2:column_name} \${3:column_type}
+)
+USING \${4:parquet}
+;`
+    },
+    {
+        label: 'CREATE VIEW ?',
+        filterText: 'CREATE',
+        template: `CREATE VIEW \${1:view_name}
+AS SELECT \${2:columns} FROM \${3:table_name}
+;`
+    },
+    {
+        label: 'INSERT INTO ?',
+        filterText: 'INSERT',
+        template: `INSERT INTO \${1:table_name}
+SELECT \${2:columns} FROM \${3:source_table}
+;`
+    },
+    {
+        label: 'SELECT ? FROM ?',
+        filterText: 'SELECT',
+        template: `SELECT \${1:columns} FROM \${2:table_name}`
+    }
+];
+/**
+ * An Antlr4ts visitor which builds beyond the query contextual analysis. It also
+ * captures contexts of any TEMPORARY VIEW/TABLE DDL statements if exists.
+ */
+class CompletionVisitor extends sparksql_relation_visitor_1.SparkSQLRelationVisitor {
+    constructor(cursor, getColumns) {
+        super(getColumns && getTableMetadataFnFactory(getColumns));
+        this.cursor = cursor;
+        this.getColumns = getColumns;
+        this.hasCompletions = false;
+        this.completions = { type: 'other', snippets: [] };
+    }
+    defaultResult() {
+        return;
+    }
+    aggregateResult() {
+        return;
+    }
+    onColumnReference() {
+        return;
+    }
+    updateQueryCompletionItems(relation) {
+        // Is this the innermost query with the cursor inside
+        if (!this.hasCompletions && this.caretScope !== undefined) {
+            this.hasCompletions = true;
+            switch (this.caretScope.type) {
+                case 'select-column':
+                case 'spec-column': {
+                    const columns = availableColumns(relation);
+                    // unique relations to avoid duplicated tables
+                    const relationMap = new Map();
+                    columns.forEach(c => {
+                        if (c.relation) {
+                            const key = `${c.relation.category}:${c.relation.label}`;
+                            if (!relationMap.has(key)) {
+                                relationMap.set(key, c.relation);
+                            }
+                        }
+                    });
+                    const uniqueRelations = [...relationMap.values()];
+                    this.completions = {
+                        type: 'column',
+                        columns,
+                        relations: uniqueRelations,
+                        snippets: this.completions.snippets
+                    };
+                    break;
+                }
+                case 'scoped-column': {
+                    let newCompletions = [];
+                    const relationName = this.caretScope.relation;
+                    const relationFound = this.relationDDLColumns.get(relationName);
+                    if (relationFound) {
+                        newCompletions = relationFound.map(c => ({
+                            relation: { category: 'table', label: getRelationName(relationName) },
+                            name: c.label
+                        }));
+                    }
+                    else {
+                        const cteFound = relation.findCTE(getRelationName(relationName));
+                        if (cteFound) {
+                            newCompletions = cteFound.columns.map(c => ({
+                                relation: { category: 'CTE', label: getRelationName(relationName) },
+                                name: c.label
+                            }));
+                        }
+                        else {
+                            newCompletions =
+                                relation
+                                    .findLocalRelation(relationName)
+                                    ?.columns.filter(c => c.label !== sparksql_relation_visitor_1.STAR_LABEL && !c.label.includes(cursor_1.defaultCursorPlaceholder))
+                                    .map(c => ({
+                                    relation: { category: 'table', label: getRelationName(relationName) },
+                                    name: c.label
+                                })) ?? [];
+                        }
+                    }
+                    this.completions = {
+                        type: 'column',
+                        relations: [], // no need to suggest relation if already scoped
+                        columns: newCompletions,
+                        snippets: this.completions.snippets
+                    };
+                    break;
+                }
+                case 'relation': {
+                    const ctes = relation.getCTENames().map(label => ({
+                        label,
+                        category: 'CTE'
+                    }));
+                    const tableDDLNames = [...this.tableDDLs.keys()].map(label => ({
+                        label: getRelationName(label),
+                        category: 'table'
+                    }));
+                    const viewDDLNames = [...this.viewDDLs.keys()].map(label => ({
+                        label: getRelationName(label),
+                        category: 'view'
+                    }));
+                    // 对 ctes / tableDDLNames / viewDDLNames 统一按 category+label 去重
+                    const relationMap = new Map();
+                    [...ctes, ...tableDDLNames, ...viewDDLNames].forEach(r => {
+                        const key = `${r.category}:${r.label}`;
+                        if (!relationMap.has(key)) {
+                            relationMap.set(key, r);
+                        }
+                    });
+                    const uniqueRelations = [...relationMap.values()];
+                    this.completions = {
+                        type: 'relation',
+                        relations: uniqueRelations,
+                        snippets: this.completions.snippets,
+                        incompleteReference: this.caretScope.prefix.length > 0 ? this.caretScope.prefix : undefined
+                    };
+                    break;
+                }
+                default:
+                    break;
+            }
+        }
+    }
+    handleRelationContext(ctx) {
+        let multipartRelationName;
+        if (ctx instanceof SparkSQLParser_1.IdentifierReferenceContext) {
+            const multipartId = ctx.multipartIdentifier();
+            if (!multipartId)
+                return;
+            multipartRelationName = multipartId.errorCapturingIdentifier().map(t => (0, utils_1.sanitizeText)(t.text));
+        }
+        else {
+            multipartRelationName = ctx.errorCapturingIdentifier().map(t => (0, utils_1.sanitizeText)(t.text));
+        }
+        const lastPart = multipartRelationName[multipartRelationName.length - 1];
+        const category = 'table';
+        this.relationInStatementContext = this.relationPrimaryFromMultipart(multipartRelationName, category);
+        if (!this.hasCompletions && this.cursor.isIn(lastPart)) {
+            this.hasCompletions = true;
+            this.caretScope = { type: 'relation', prefix: multipartRelationName.slice(0, -1) };
+            this.completions = {
+                type: 'relation',
+                relations: [...this.tableDDLs.keys()].map(label => ({
+                    label: getRelationName(label),
+                    category
+                })),
+                snippets: this.completions.snippets,
+                incompleteReference: this.caretScope.prefix.length > 0 ? this.caretScope.prefix : undefined
+            };
+        }
+    }
+    handleScopedColumn() {
+        if (!this.hasCompletions && this.relationInStatementContext) {
+            this.hasCompletions = true;
+            this.caretScope = {
+                type: 'scoped-column',
+                relation: (0, sparksql_relation_visitor_1.relationPrimaryToString)(this.relationInStatementContext)
+            };
+            let newCompletions = [];
+            const relationName = this.relationInStatementContext.relationName;
+            const relationDDL = this.relationDDLColumns.get(relationName);
+            if (relationDDL) {
+                newCompletions = relationDDL.map(c => ({
+                    relation: { category: 'table', label: relationName },
+                    name: c.label
+                }));
+            }
+            else if (this.getColumns) {
+                newCompletions = this.getColumns({
+                    catalog: this.relationInStatementContext.catalogName,
+                    database: this.relationInStatementContext.databaseName,
+                    table: this.relationInStatementContext.relationName
+                }).map(c => ({
+                    relation: { category: 'table', label: relationName },
+                    name: c.label
+                }));
+            }
+            this.completions = {
+                type: 'column',
+                relations: [], // no need to suggest relation if already scoped
+                columns: newCompletions,
+                snippets: this.completions.snippets
+            };
+        }
+    }
+    handleDatabaseContext(ctx) {
+        const multipartDatabaseName = ctx.errorCapturingIdentifier().map(t => (0, utils_1.sanitizeText)(t.text));
+        const lastPart = multipartDatabaseName[multipartDatabaseName.length - 1];
+        if (this.cursor.isIn(lastPart)) {
+            this.hasCompletions = true;
+            const incompleteReference = multipartDatabaseName.length > 1 ? multipartDatabaseName[0] : undefined;
+            this.completions = { type: 'database', incompleteReference, snippets: [] };
+        }
+    }
+    getSuggestions() {
+        return this.completions;
+    }
+    onRelation(relation) {
+        if (relation instanceof sparksql_relation_visitor_1.TableRelation || relation instanceof sparksql_relation_visitor_1.InsertRelation || relation instanceof sparksql_relation_visitor_1.ViewRelation) {
+            return;
+        }
+        this.updateQueryCompletionItems(relation);
+    }
+    visitErrorNode(node) {
+        super.visitErrorNode(node);
+        if (this.cursor.isIn(node.text)) {
+            if (node.parent instanceof SparkSQLParser_1.StatementContext || node.parent instanceof SparkSQLParser_1.ProgramContext) {
+                this.completions.snippets.push(...SQL_SNIPPETS.filter(s => s.filterText.toUpperCase().indexOf(this.cursor.revert(node.text).toUpperCase()) === 0) // must match snippet's filterText from the beginning
+                );
+                this.caretScope = { type: 'other' };
+            }
+        }
+    }
+    visitStatement(ctx) {
+        this.relationInStatementContext = undefined;
+        super.visitStatement(ctx);
+    }
+    visitUse(ctx) {
+        super.visitUse(ctx);
+        const identifierRef = ctx.identifierReference();
+        if (identifierRef) {
+            const multipartId = identifierRef.multipartIdentifier();
+            if (multipartId) {
+                this.handleDatabaseContext(multipartId);
+            }
+        }
+    }
+    visitCreateTable(ctx) {
+        const tableHeader = ctx.createTableHeader();
+        if (tableHeader) {
+            const identifierRef = tableHeader.identifierReference();
+            if (identifierRef) {
+                this.handleRelationContext(identifierRef);
+            }
+        }
+        super.visitCreateTable(ctx);
+    }
+    visitCreateView(ctx) {
+        const identifierRef = ctx.identifierReference();
+        if (identifierRef) {
+            this.handleRelationContext(identifierRef);
+        }
+        super.visitCreateView(ctx);
+    }
+    visitQuerySpecification(ctx) {
+        super.visitQuerySpecification(ctx);
+        if (this.currentQueryRelation !== undefined) {
+            this.updateQueryCompletionItems(this.currentQueryRelation);
+        }
+    }
+    visitRelation(ctx) {
+        const relationPrimary = ctx.relationPrimary();
+        if (relationPrimary instanceof SparkSQLParser_1.TableNameContext) {
+            const identifierRef = relationPrimary.identifierReference();
+            const multipartId = identifierRef.multipartIdentifier();
+            if (multipartId) {
+                const multipartTableName = multipartId
+                    .errorCapturingIdentifier()
+                    .map((t) => (0, utils_1.sanitizeText)(t.text));
+                const lastPart = multipartTableName[multipartTableName.length - 1];
+                if (this.cursor.isIn(lastPart)) {
+                    this.caretScope = { type: 'relation', prefix: multipartTableName.slice(0, -1) };
+                }
+            }
+        }
+        super.visitRelation(ctx);
+    }
+    visitColumnReference(ctx) {
+        super.visitColumnReference(ctx);
+        const identifier = ctx.identifier();
+        if (this.currentQueryRelation !== undefined && identifier && this.cursor.isIn(identifier.text)) {
+            if (this.currentQueryRelation.currentClause === 'select') {
+                this.caretScope = { type: 'select-column' };
+            }
+            else {
+                this.caretScope = { type: 'spec-column' };
+            }
+        }
+    }
+    visitSelectClause(ctx) {
+        super.visitSelectClause(ctx);
+    }
+    visitFromClause(ctx) {
+        super.visitFromClause(ctx);
+    }
+}
+exports.CompletionVisitor = CompletionVisitor;
+//# sourceMappingURL=completion.visitor.js.map

package/out-tsc/visitors/completion.visitor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"completion.visitor.js","sourceRoot":"","sources":["../../src/visitors/completion.visitor.ts"],"names":[],"mappings":";;;AAEA,2EASqC;AACrC,sCAAkE;AAClE,0DAc+B;AAG/B,oCAAwC;AAsDxC,MAAM,eAAe,GAAG,CAAC,qBAA6B,EAAU,EAAE;IAChE,MAAM,MAAM,GAAG,qBAAqB,CAAC,KAAK,CAAC,8CAAkB,CAAC,CAAC;IAC/D,OAAO,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;AACnC,CAAC,CAAC;AAEF,SAAS,gBAAgB,CACvB,QAAwC;IAExC,MAAM,OAAO,GAAqE,EAAE,CAAC;IAErF,IAAI,QAAQ,YAAY,yCAAa,EAAE,CAAC;QACtC,QAAQ,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE;YACvC,MAAM,YAAY,GAAG,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;YAExD,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE;gBACxB,IAAI,GAAG,CAAC,KAAK,KAAK,sCAAU,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,iCAAwB,CAAC,EAAE,CAAC;oBAC9E,OAAO,CAAC,IAAI,CAAC;wBACX,QAAQ,EAAE,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,SAAS;wBAC7F,IAAI,EAAE,GAAG,CAAC,KAAK;wBACf,IAAI,EAAE,GAAG,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAE,GAAG,CAAC,IAAU,CAAC,CAAC,CAAC,SAAS;qBAC3D,CAAC,CAAC;gBACL,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;SAAM,CAAC;QACN,IAAI,QAAQ,CAAC,iBAAiB,KAAK,SAAS,EAAE,CAAC;YAC7C,QAAQ,CAAC,iBAAiB,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE;gBAC/C,IAAI,GAAG,CAAC,KAAK,KAAK,sCAAU,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,iCAAwB,CAAC,EAAE,CAAC;oBAC9E,OAAO,CAAC,IAAI,CAAC;wBACX,QAAQ,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,CAAC,iBAAkB,CAAC,YAAY,CAAC,YAAY,EAAE;wBAC7F,IAAI,EAAE,GAAG,CAAC,KAAK;wBACf,IAAI,EAAE,GAAG,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAE,GAAG,CAAC,IAAU,CAAC,CAAC,CAAC,SAAS;qBAC3D,CAAC,CAAC;gBACL,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAS,yBAAyB,CAChC,UAAmD;IAEnD,OAAO,CAAC,eAAgC,EAAE,EAAE;QAC1C,IAAI,eAAe,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;YACrC,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,MAAM,OAAO,GAAG,UAAU,CAAC;YACzB,KAAK,EAAE,eAAe,CAAC,YAAY;YACnC,OAAO,EAAE,eAAe,CAAC,WAAW;YACpC,QAAQ,EAAE,eAAe,CAAC,YAAY;SACvC,CAAC,CAAC;QACH,OAAO;YACL,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE;gBACvB,OAAO;oBACL,EAAE,EAAE,GAAG,IAAA,mDAAuB,EAAC,eAAe,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE;oBAC5D,KAAK,EAAE,CAAC,CAAC,KAAK;oBACd,IAAI,EAAE,CAAC;iBACR,CAAC;YACJ,CAAC,CAAC;SACH,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC;AAED,MAAM,YAAY,GAAc;IAC9B;QACE,KAAK,EAAE,gBAAgB;QACvB,UAAU,EAAE,QAAQ;QACpB,QAAQ,EAAE;;;;;EAKZ;KACC;IACD;QACE,KAAK,EAAE,eAAe;QACtB,UAAU,EAAE,QAAQ;QACpB,QAAQ,EAAE;;EAEZ;KACC;IACD;QACE,KAAK,EAAE,eAAe;QACtB,UAAU,EAAE,QAAQ;QACpB,QAAQ,EAAE;;EAEZ;KACC;IACD;QACE,KAAK,EAAE,iBAAiB;QACxB,UAAU,EAAE,QAAQ;QACpB,QAAQ,EAAE,4CAA4C;KACvD;CACF,CAAC;AAEF;;;GAGG;AACH,MAAa,iBAAmC,SAAQ,mDAA6B;IAMnF,YAAoB,MAAmB,EAAU,UAAoD;QACnG,KAAK,CAAC,UAAU,IAAI,yBAAyB,CAAC,UAAU,CAAC,CAAC,CAAC;QADzC,WAAM,GAAN,MAAM,CAAa;QAAU,eAAU,GAAV,UAAU,CAA0C;QAJ7F,mBAAc,GAAG,KAAK,CAAC;QACvB,gBAAW,GAAmB,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;IAKtE,CAAC;IAED,aAAa;QACX,OAAO;IACT,CAAC;IAED,eAAe;QACb,OAAO;IACT,CAAC;IAED,iBAAiB;QACf,OAAO;IACT,CAAC;IAEO,0BAA0B,CAAC,QAAuB;QACxD,qDAAqD;QACrD,IAAI,CAAC,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;YAC1D,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;YAC3B,QAAQ,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;gBAC7B,KAAK,eAAe,CAAC;gBACrB,KAAK,aAAa,CAAC,CAAC,CAAC;oBACnB,MAAM,OAAO,GAAG,gBAAgB,CAAI,QAAQ,CAAC,CAAC;oBAC9C,8CAA8C;oBAC9C,MAAM,WAAW,GAAG,IAAI,GAAG,EAA8B,CAAC;oBAC1D,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;wBAClB,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC;4BACf,MAAM,GAAG,GAAG,GAAG,CAAC,CAAC,QAAQ,CAAC,QAAQ,IAAI,CAAC,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;4BACzD,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;gCAC1B,WAAW,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC;4BACnC,CAAC;wBACH,CAAC;oBACH,CAAC,CAAC,CAAC;oBACH,MAAM,eAAe,GAAG,CAAC,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC;oBAElD,IAAI,CAAC,WAAW,GAAG;wBACjB,IAAI,EAAE,QAAQ;wBACd,OAAO;wBACP,SAAS,EAAE,eAAe;wBAC1B,QAAQ,EAAE,IAAI,CAAC,WAAW,CAAC,QAAQ;qBACpC,CAAC;oBACF,MAAM;gBACR,CAAC;gBACD,KAAK,eAAe,CAAC,CAAC,CAAC;oBACrB,IAAI,cAAc,GAA2D,EAAE,CAAC;oBAChF,MAAM,YAAY,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC;oBAC9C,MAAM,aAAa,GAAG,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;oBAChE,IAAI,aAAa,EAAE,CAAC;wBAClB,cAAc,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;4BACvC,QAAQ,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,eAAe,CAAC,YAAY,CAAC,EAAE;4BACrE,IAAI,EAAE,CAAC,CAAC,KAAK;yBACd,CAAC,CAAC,CAAC;oBACN,CAAC;yBAAM,CAAC;wBACN,MAAM,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,eAAe,CAAC,YAAY,CAAC,CAAC,CAAC;wBACjE,IAAI,QAAQ,EAAE,CAAC;4BACb,cAAc,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gCAC1C,QAAQ,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,eAAe,CAAC,YAAY,CAAC,EAAE;gCACnE,IAAI,EAAE,CAAC,CAAC,KAAK;6BACd,CAAC,CAAC,CAAC;wBACN,CAAC;6BAAM,CAAC;4BACN,cAAc;gCACZ,QAAQ;qCACL,iBAAiB,CAAC,YAAY,CAAC;oCAChC,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,sCAAU,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,iCAAwB,CAAC,CAAC;qCAC3F,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;oCACT,QAAQ,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,eAAe,CAAC,YAAY,CAAC,EAAE;oCACrE,IAAI,EAAE,CAAC,CAAC,KAAK;iCACd,CAAC,CAAC,IAAI,EAAE,CAAC;wBAChB,CAAC;oBACH,CAAC;oBACD,IAAI,CAAC,WAAW,GAAG;wBACjB,IAAI,EAAE,QAAQ;wBACd,SAAS,EAAE,EAAE,EAAE,gDAAgD;wBAC/D,OAAO,EAAE,cAAc;wBACvB,QAAQ,EAAE,IAAI,CAAC,WAAW,CAAC,QAAQ;qBACpC,CAAC;oBACF,MAAM;gBACR,CAAC;gBACD,KAAK,UAAU,CAAC,CAAC,CAAC;oBAChB,MAAM,IAAI,GAAyB,QAAQ,CAAC,WAAW,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;wBACtE,KAAK;wBACL,QAAQ,EAAE,KAAK;qBAChB,CAAC,CAAC,CAAC;oBACJ,MAAM,aAAa,GAAyB,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;wBACnF,KAAK,EAAE,eAAe,CAAC,KAAK,CAAC;wBAC7B,QAAQ,EAAE,OAAO;qBAClB,CAAC,CAAC,CAAC;oBACJ,MAAM,YAAY,GAAyB,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;wBACjF,KAAK,EAAE,eAAe,CAAC,KAAK,CAAC;wBAC7B,QAAQ,EAAE,MAAM;qBACjB,CAAC,CAAC,CAAC;oBAEJ,8DAA8D;oBAC9D,MAAM,WAAW,GAAG,IAAI,GAAG,EAA8B,CAAC;oBAC1D,CAAC,GAAG,IAAI,EAAE,GAAG,aAAa,EAAE,GAAG,YAAY,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;wBACvD,MAAM,GAAG,GAAG,GAAG,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,KAAK,EAAE,CAAC;wBACvC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;4BAC1B,WAAW,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;wBAC1B,CAAC;oBACH,CAAC,CAAC,CAAC;oBACH,MAAM,eAAe,GAAG,CAAC,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC;oBAClD,IAAI,CAAC,WAAW,GAAG;wBACjB,IAAI,EAAE,UAAU;wBAChB,SAAS,EAAE,eAAe;wBAC1B,QAAQ,EAAE,IAAI,CAAC,WAAW,CAAC,QAAQ;wBACnC,mBAAmB,EAAE,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS;qBAC5F,CAAC;oBACF,MAAM;gBACR,CAAC;gBACD;oBACE,MAAM;YACV,CAAC;QACH,CAAC;IACH,CAAC;IAEO,qBAAqB,CAAC,GAA4D;QACxF,IAAI,qBAA+B,CAAC;QAEpC,IAAI,GAAG,YAAY,2CAA0B,EAAE,CAAC;YAC9C,MAAM,WAAW,GAAG,GAAG,CAAC,mBAAmB,EAAE,CAAC;YAC9C,IAAI,CAAC,WAAW;gBAAE,OAAO;YACzB,qBAAqB,GAAG,WAAW,CAAC,wBAAwB,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAA,oBAAY,EAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QAChG,CAAC;aAAM,CAAC;YACN,qBAAqB,GAAG,GAAG,CAAC,wBAAwB,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAA,oBAAY,EAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QACxF,CAAC;QAED,MAAM,QAAQ,GAAG,qBAAqB,CAAC,qBAAqB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACzE,MAAM,QAAQ,GAAG,OAAO,CAAC;QACzB,IAAI,CAAC,0BAA0B,GAAG,IAAI,CAAC,4BAA4B,CAAC,qBAAqB,EAAE,QAAQ,CAAC,CAAC;QAErG,IAAI,CAAC,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;YACvD,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;YAC3B,IAAI,CAAC,UAAU,GAAG,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,qBAAqB,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACnF,IAAI,CAAC,WAAW,GAAG;gBACjB,IAAI,EAAE,UAAU;gBAChB,SAAS,EAAE,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;oBAClD,KAAK,EAAE,eAAe,CAAC,KAAK,CAAC;oBAC7B,QAAQ;iBACT,CAAC,CAAC;gBACH,QAAQ,EAAE,IAAI,CAAC,WAAW,CAAC,QAAQ;gBACnC,mBAAmB,EAAE,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS;aAC5F,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,kBAAkB;QACxB,IAAI,CAAC,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,0BAA0B,EAAE,CAAC;YAC5D,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;YAC3B,IAAI,CAAC,UAAU,GAAG;gBAChB,IAAI,EAAE,eAAe;gBACrB,QAAQ,EAAE,IAAA,mDAAuB,EAAC,IAAI,CAAC,0BAA0B,CAAC;aACnE,CAAC;YAEF,IAAI,cAAc,GAA2D,EAAE,CAAC;YAChF,MAAM,YAAY,GAAG,IAAI,CAAC,0BAA0B,CAAC,YAAY,CAAC;YAClE,MAAM,WAAW,GAAG,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;YAC9D,IAAI,WAAW,EAAE,CAAC;gBAChB,cAAc,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;oBACrC,QAAQ,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE;oBACpD,IAAI,EAAE,CAAC,CAAC,KAAK;iBACd,CAAC,CAAC,CAAC;YACN,CAAC;iBAAM,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;gBAC3B,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC;oBAC/B,OAAO,EAAE,IAAI,CAAC,0BAA0B,CAAC,WAAW;oBACpD,QAAQ,EAAE,IAAI,CAAC,0BAA0B,CAAC,YAAY;oBACtD,KAAK,EAAE,IAAI,CAAC,0BAA0B,CAAC,YAAY;iBACpD,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;oBACX,QAAQ,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE;oBACpD,IAAI,EAAE,CAAC,CAAC,KAAK;iBACd,CAAC,CAAC,CAAC;YACN,CAAC;YACD,IAAI,CAAC,WAAW,GAAG;gBACjB,IAAI,EAAE,QAAQ;gBACd,SAAS,EAAE,EAAE,EAAE,gDAAgD;gBAC/D,OAAO,EAAE,cAAc;gBACvB,QAAQ,EAAE,IAAI,CAAC,WAAW,CAAC,QAAQ;aACpC,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,qBAAqB,CAAC,GAA+B;QAC3D,MAAM,qBAAqB,GAAG,GAAG,CAAC,wBAAwB,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAA,oBAAY,EAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QAC5F,MAAM,QAAQ,GAAG,qBAAqB,CAAC,qBAAqB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACzE,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC/B,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;YAC3B,MAAM,mBAAmB,GAAG,qBAAqB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YACpG,IAAI,CAAC,WAAW,GAAG,EAAE,IAAI,EAAE,UAAU,EAAE,mBAAmB,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;QAC7E,CAAC;IACH,CAAC;IAED,cAAc;QACZ,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAED,UAAU,CAAC,QAAuE;QAChF,IAAI,QAAQ,YAAY,yCAAa,IAAI,QAAQ,YAAY,0CAAc,IAAI,QAAQ,YAAY,wCAAY,EAAE,CAAC;YAChH,OAAO;QACT,CAAC;QACD,IAAI,CAAC,0BAA0B,CAAC,QAAQ,CAAC,CAAC;IAC5C,CAAC;IAED,cAAc,CAAC,IAAe;QAC5B,KAAK,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;QAE3B,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YAChC,IAAI,IAAI,CAAC,MAAM,YAAY,iCAAgB,IAAI,IAAI,CAAC,MAAM,YAAY,+BAAc,EAAE,CAAC;gBACrF,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,IAAI,CAC5B,GAAG,YAAY,CAAC,MAAM,CACpB,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,CAC3F,CAAC,qDAAqD;iBACxD,CAAC;gBACF,IAAI,CAAC,UAAU,GAAG,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;YACtC,CAAC;QACH,CAAC;IACH,CAAC;IAED,cAAc,CAAC,GAAqB;QAClC,IAAI,CAAC,0BAA0B,GAAG,SAAS,CAAC;QAC5C,KAAK,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC;IAC5B,CAAC;IAED,QAAQ,CAAC,GAAe;QACtB,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;QACpB,MAAM,aAAa,GAAG,GAAG,CAAC,mBAAmB,EAAE,CAAC;QAChD,IAAI,aAAa,EAAE,CAAC;YAClB,MAAM,WAAW,GAAG,aAAa,CAAC,mBAAmB,EAAE,CAAC;YACxD,IAAI,WAAW,EAAE,CAAC;gBAChB,IAAI,CAAC,qBAAqB,CAAC,WAAW,CAAC,CAAC;YAC1C,CAAC;QACH,CAAC;IACH,CAAC;IAED,gBAAgB,CAAC,GAAuB;QACtC,MAAM,WAAW,GAAG,GAAG,CAAC,iBAAiB,EAAE,CAAC;QAC5C,IAAI,WAAW,EAAE,CAAC;YAChB,MAAM,aAAa,GAAG,WAAW,CAAC,mBAAmB,EAAE,CAAC;YACxD,IAAI,aAAa,EAAE,CAAC;gBAClB,IAAI,CAAC,qBAAqB,CAAC,aAAa,CAAC,CAAC;YAC5C,CAAC;QACH,CAAC;QACD,KAAK,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC;IAC9B,CAAC;IAED,eAAe,CAAC,GAAsB;QACpC,MAAM,aAAa,GAAG,GAAG,CAAC,mBAAmB,EAAE,CAAC;QAChD,IAAI,aAAa,EAAE,CAAC;YAClB,IAAI,CAAC,qBAAqB,CAAC,aAAa,CAAC,CAAC;QAC5C,CAAC;QACD,KAAK,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;IAC7B,CAAC;IAED,uBAAuB,CAAC,GAA8B;QACpD,KAAK,CAAC,uBAAuB,CAAC,GAAG,CAAC,CAAC;QAEnC,IAAI,IAAI,CAAC,oBAAoB,KAAK,SAAS,EAAE,CAAC;YAC5C,IAAI,CAAC,0BAA0B,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QAC7D,CAAC;IACH,CAAC;IAED,aAAa,CAAC,GAAoB;QAChC,MAAM,eAAe,GAAG,GAAG,CAAC,eAAe,EAAE,CAAC;QAC9C,IAAI,eAAe,YAAY,iCAAgB,EAAE,CAAC;YAChD,MAAM,aAAa,GAAG,eAAe,CAAC,mBAAmB,EAAE,CAAC;YAC5D,MAAM,WAAW,GAAG,aAAa,CAAC,mBAAmB,EAAE,CAAC;YACxD,IAAI,WAAW,EAAE,CAAC;gBAChB,MAAM,kBAAkB,GAAG,WAAW;qBACnC,wBAAwB,EAAE;qBAC1B,GAAG,CAAC,CAAC,CAAmB,EAAE,EAAE,CAAC,IAAA,oBAAY,EAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;gBACtD,MAAM,QAAQ,GAAG,kBAAkB,CAAC,kBAAkB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;gBAEnE,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;oBAC/B,IAAI,CAAC,UAAU,GAAG,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,kBAAkB,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAClF,CAAC;YACH,CAAC;QACH,CAAC;QAED,KAAK,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;IAC3B,CAAC;IAED,oBAAoB,CAAC,GAA2B;QAC9C,KAAK,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC;QAEhC,MAAM,UAAU,GAAG,GAAG,CAAC,UAAU,EAAE,CAAC;QACpC,IAAI,IAAI,CAAC,oBAAoB,KAAK,SAAS,IAAI,UAAU,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;YAC/F,IAAI,IAAI,CAAC,oBAAoB,CAAC,aAAa,KAAK,QAAQ,EAAE,CAAC;gBACzD,IAAI,CAAC,UAAU,GAAG,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC;YAC9C,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,UAAU,GAAG,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;YAC5C,CAAC;QACH,CAAC;IACH,CAAC;IAED,iBAAiB,CAAC,GAAwB;QACxC,KAAK,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC;IAC/B,CAAC;IAED,eAAe,CAAC,GAAsB;QACpC,KAAK,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;IAC7B,CAAC;CACF;AAlTD,8CAkTC"}

package/out-tsc/visitors/lineage.visitor.d.ts

@@ -0,0 +1,34 @@
+import { AbstractParseTreeVisitor } from 'antlr4ts/tree';
+import { CreateTableContext, CreateViewContext, FromClauseContext, InsertIntoTableContext, QueryContext, RelationContext, SelectClauseContext } from '../lib/SparkSQLParser';
+import { SparkSQLParserVisitor } from '../lib/SparkSQLParserVisitor';
+import { Lineage, LineageEdge, LineageOptions, Relation, RelationPrimary } from '../lineage.typing';
+export declare const STAR_LABEL = "*";
+export declare const COUNT_STAR_LABEL = "count(*)";
+export declare class LineageVisitor extends AbstractParseTreeVisitor<void> implements SparkSQLParserVisitor<void> {
+    private options;
+    relations: Array<{
+        relationPrimary?: RelationPrimary;
+        relation: Relation;
+    }>;
+    edges: LineageEdge[];
+    private edgeIdSeq;
+    private queryIdSeq;
+    private tableIdSeq;
+    private viewIdSeq;
+    private insertIdSeq;
+    private columnIdSeq;
+    constructor(options: LineageOptions);
+    protected defaultResult(): void;
+    getLineage(): Lineage;
+    private generateId;
+    private extractTableName;
+    private extractNameFromIdentifierReference;
+    visitQuery(ctx: QueryContext): void;
+    visitRelation(ctx: RelationContext): void;
+    visitInsertIntoTable(ctx: InsertIntoTableContext): void;
+    visitCreateTable(ctx: CreateTableContext): void;
+    visitCreateView(ctx: CreateViewContext): void;
+    visitSelectClause(ctx: SelectClauseContext): void;
+    visitFromClause(ctx: FromClauseContext): void;
+}
+//# sourceMappingURL=lineage.visitor.d.ts.map

package/out-tsc/visitors/lineage.visitor.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"lineage.visitor.d.ts","sourceRoot":"","sources":["../../src/visitors/lineage.visitor.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,wBAAwB,EAAE,MAAM,eAAe,CAAC;AAEzD,OAAO,EACL,kBAAkB,EAClB,iBAAiB,EACjB,iBAAiB,EAEjB,sBAAsB,EAEtB,YAAY,EACZ,eAAe,EACf,mBAAmB,EACpB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAE,qBAAqB,EAAE,MAAM,8BAA8B,CAAC;AACrE,OAAO,EAAE,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,QAAQ,EAAkB,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAEpH,eAAO,MAAM,UAAU,MAAM,CAAC;AAC9B,eAAO,MAAM,gBAAgB,aAAa,CAAC;AAE3C,qBAAa,cAAe,SAAQ,wBAAwB,CAAC,IAAI,CAAE,YAAW,qBAAqB,CAAC,IAAI,CAAC;IAW3F,OAAO,CAAC,OAAO;IAV3B,SAAS,EAAE,KAAK,CAAC;QAAE,eAAe,CAAC,EAAE,eAAe,CAAC;QAAC,QAAQ,EAAE,QAAQ,CAAA;KAAE,CAAC,CAAM;IACjF,KAAK,EAAE,WAAW,EAAE,CAAM;IAE1B,OAAO,CAAC,SAAS,CAAK;IACtB,OAAO,CAAC,UAAU,CAAK;IACvB,OAAO,CAAC,UAAU,CAAK;IACvB,OAAO,CAAC,SAAS,CAAK;IACtB,OAAO,CAAC,WAAW,CAAK;IACxB,OAAO,CAAC,WAAW,CAAK;gBAEJ,OAAO,EAAE,cAAc;IAI3C,SAAS,CAAC,aAAa,IAAI,IAAI;IAI/B,UAAU,IAAI,OAAO;IAOrB,OAAO,CAAC,UAAU;IAIlB,OAAO,CAAC,gBAAgB;IAaxB,OAAO,CAAC,kCAAkC;IAQ1C,UAAU,CAAC,GAAG,EAAE,YAAY,GAAG,IAAI;IAkBnC,aAAa,CAAC,GAAG,EAAE,eAAe,GAAG,IAAI;IA4BzC,oBAAoB,CAAC,GAAG,EAAE,sBAAsB,GAAG,IAAI;IA4BvD,gBAAgB,CAAC,GAAG,EAAE,kBAAkB,GAAG,IAAI;IAgC/C,eAAe,CAAC,GAAG,EAAE,iBAAiB,GAAG,IAAI;IA4B7C,iBAAiB,CAAC,GAAG,EAAE,mBAAmB,GAAG,IAAI;IAYjD,eAAe,CAAC,GAAG,EAAE,iBAAiB,GAAG,IAAI;CAG9C"}

package/out-tsc/visitors/lineage.visitor.js

@@ -0,0 +1,181 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LineageVisitor = exports.COUNT_STAR_LABEL = exports.STAR_LABEL = void 0;
+const tree_1 = require("antlr4ts/tree");
+exports.STAR_LABEL = '*';
+exports.COUNT_STAR_LABEL = 'count(*)';
+class LineageVisitor extends tree_1.AbstractParseTreeVisitor {
+    constructor(options) {
+        super();
+        this.options = options;
+        this.relations = [];
+        this.edges = [];
+        this.edgeIdSeq = 1;
+        this.queryIdSeq = 1;
+        this.tableIdSeq = 1;
+        this.viewIdSeq = 1;
+        this.insertIdSeq = 1;
+        this.columnIdSeq = 1;
+    }
+    defaultResult() {
+        return;
+    }
+    getLineage() {
+        return {
+            nodes: this.relations.map(r => r.relation),
+            edges: this.edges
+        };
+    }
+    generateId(prefix, seq) {
+        return `${prefix}_${seq}`;
+    }
+    extractTableName(ctx) {
+        const identifiers = ctx.errorCapturingIdentifier();
+        const relationName = identifiers[identifiers.length - 1].text;
+        const databaseName = identifiers.length > 1 ? identifiers[identifiers.length - 2].text : undefined;
+        const catalogName = identifiers.length > 2 ? identifiers[identifiers.length - 3].text : undefined;
+        return {
+            type: 'table',
+            relationName,
+            databaseName,
+            catalogName
+        };
+    }
+    extractNameFromIdentifierReference(ctx) {
+        const multipartId = ctx.multipartIdentifier();
+        if (multipartId) {
+            return multipartId.text;
+        }
+        return ctx.text;
+    }
+    visitQuery(ctx) {
+        const queryId = this.generateId('query', this.queryIdSeq++);
+        const columns = [];
+        this.visitChildren(ctx);
+        this.relations.push({
+            relation: {
+                type: 'query',
+                id: queryId,
+                label: `query_${this.queryIdSeq - 1}`,
+                columns,
+                isSourceOnly: false,
+                isTargetOnly: false
+            }
+        });
+    }
+    visitRelation(ctx) {
+        const tableId = this.generateId('table', this.tableIdSeq++);
+        const relationPrimary = ctx.relationPrimary();
+        if (relationPrimary) {
+            const text = relationPrimary.text;
+            const tablePrimary = {
+                type: 'table',
+                relationName: text
+            };
+            this.relations.push({
+                relationPrimary: tablePrimary,
+                relation: {
+                    type: 'table',
+                    id: tableId,
+                    label: text,
+                    columns: [],
+                    data: tablePrimary,
+                    isSourceOnly: true,
+                    isTargetOnly: false
+                }
+            });
+        }
+        this.visitChildren(ctx);
+    }
+    visitInsertIntoTable(ctx) {
+        const insertId = this.generateId('insert', this.insertIdSeq++);
+        const identifierRef = ctx.identifierReference();
+        if (identifierRef) {
+            const tableName = this.extractNameFromIdentifierReference(identifierRef);
+            const tablePrimary = {
+                type: 'insert',
+                relationName: tableName
+            };
+            this.relations.push({
+                relationPrimary: tablePrimary,
+                relation: {
+                    type: 'insert',
+                    id: insertId,
+                    label: tableName,
+                    columns: [],
+                    data: tablePrimary,
+                    isSourceOnly: false,
+                    isTargetOnly: true
+                }
+            });
+        }
+        this.visitChildren(ctx);
+    }
+    visitCreateTable(ctx) {
+        const tableId = this.generateId('table', this.tableIdSeq++);
+        // Get table name from createTableHeader
+        const tableHeader = ctx.createTableHeader();
+        if (tableHeader) {
+            const identifierRef = tableHeader.identifierReference();
+            if (identifierRef) {
+                const tableName = this.extractNameFromIdentifierReference(identifierRef);
+                const tablePrimary = {
+                    type: 'table',
+                    relationName: tableName
+                };
+                this.relations.push({
+                    relationPrimary: tablePrimary,
+                    relation: {
+                        type: 'table',
+                        id: tableId,
+                        label: tableName,
+                        columns: [],
+                        data: tablePrimary,
+                        isSourceOnly: false,
+                        isTargetOnly: true
+                    }
+                });
+            }
+        }
+        this.visitChildren(ctx);
+    }
+    visitCreateView(ctx) {
+        const viewId = this.generateId('view', this.viewIdSeq++);
+        const identifierRef = ctx.identifierReference();
+        if (identifierRef) {
+            const viewName = this.extractNameFromIdentifierReference(identifierRef);
+            const tablePrimary = {
+                type: 'view',
+                relationName: viewName
+            };
+            this.relations.push({
+                relationPrimary: tablePrimary,
+                relation: {
+                    type: 'view',
+                    id: viewId,
+                    label: viewName,
+                    columns: [],
+                    data: tablePrimary,
+                    isSourceOnly: false,
+                    isTargetOnly: true
+                }
+            });
+        }
+        this.visitChildren(ctx);
+    }
+    visitSelectClause(ctx) {
+        const namedExprSeq = ctx.namedExpressionSeq();
+        if (namedExprSeq) {
+            const namedExprs = namedExprSeq.namedExpression();
+            namedExprs.forEach((_expr, _index) => {
+                // TODO: Implement column-level lineage tracking
+            });
+        }
+        this.visitChildren(ctx);
+    }
+    visitFromClause(ctx) {
+        this.visitChildren(ctx);
+    }
+}
+exports.LineageVisitor = LineageVisitor;
+//# sourceMappingURL=lineage.visitor.js.map

package/out-tsc/visitors/lineage.visitor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"lineage.visitor.js","sourceRoot":"","sources":["../../src/visitors/lineage.visitor.ts"],"names":[],"mappings":";;;AAAA,wCAAyD;AAgB5C,QAAA,UAAU,GAAG,GAAG,CAAC;AACjB,QAAA,gBAAgB,GAAG,UAAU,CAAC;AAE3C,MAAa,cAAe,SAAQ,+BAA8B;IAWhE,YAAoB,OAAuB;QACzC,KAAK,EAAE,CAAC;QADU,YAAO,GAAP,OAAO,CAAgB;QAV3C,cAAS,GAAqE,EAAE,CAAC;QACjF,UAAK,GAAkB,EAAE,CAAC;QAElB,cAAS,GAAG,CAAC,CAAC;QACd,eAAU,GAAG,CAAC,CAAC;QACf,eAAU,GAAG,CAAC,CAAC;QACf,cAAS,GAAG,CAAC,CAAC;QACd,gBAAW,GAAG,CAAC,CAAC;QAChB,gBAAW,GAAG,CAAC,CAAC;IAIxB,CAAC;IAES,aAAa;QACrB,OAAO;IACT,CAAC;IAED,UAAU;QACR,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC;YAC1C,KAAK,EAAE,IAAI,CAAC,KAAK;SAClB,CAAC;IACJ,CAAC;IAEO,UAAU,CAAC,MAAc,EAAE,GAAW;QAC5C,OAAO,GAAG,MAAM,IAAI,GAAG,EAAE,CAAC;IAC5B,CAAC;IAEO,gBAAgB,CAAC,GAA+B;QACtD,MAAM,WAAW,GAAG,GAAG,CAAC,wBAAwB,EAAE,CAAC;QACnD,MAAM,YAAY,GAAG,WAAW,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC;QAC9D,MAAM,YAAY,GAAG,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;QACnG,MAAM,WAAW,GAAG,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;QAClG,OAAO;YACL,IAAI,EAAE,OAAO;YACb,YAAY;YACZ,YAAY;YACZ,WAAW;SACZ,CAAC;IACJ,CAAC;IAEO,kCAAkC,CAAC,GAA+B;QACxE,MAAM,WAAW,GAAG,GAAG,CAAC,mBAAmB,EAAE,CAAC;QAC9C,IAAI,WAAW,EAAE,CAAC;YAChB,OAAO,WAAW,CAAC,IAAI,CAAC;QAC1B,CAAC;QACD,OAAO,GAAG,CAAC,IAAI,CAAC;IAClB,CAAC;IAED,UAAU,CAAC,GAAiB;QAC1B,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC;QAC5D,MAAM,OAAO,GAAqB,EAAE,CAAC;QAErC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;QAExB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;YAClB,QAAQ,EAAE;gBACR,IAAI,EAAE,OAAO;gBACb,EAAE,EAAE,OAAO;gBACX,KAAK,EAAE,SAAS,IAAI,CAAC,UAAU,GAAG,CAAC,EAAE;gBACrC,OAAO;gBACP,YAAY,EAAE,KAAK;gBACnB,YAAY,EAAE,KAAK;aACpB;SACF,CAAC,CAAC;IACL,CAAC;IAED,aAAa,CAAC,GAAoB;QAChC,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC;QAE5D,MAAM,eAAe,GAAG,GAAG,CAAC,eAAe,EAAE,CAAC;QAC9C,IAAI,eAAe,EAAE,CAAC;YACpB,MAAM,IAAI,GAAG,eAAe,CAAC,IAAI,CAAC;YAClC,MAAM,YAAY,GAAoB;gBACpC,IAAI,EAAE,OAAO;gBACb,YAAY,EAAE,IAAI;aACnB,CAAC;YAEF,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;gBAClB,eAAe,EAAE,YAAY;gBAC7B,QAAQ,EAAE;oBACR,IAAI,EAAE,OAAO;oBACb,EAAE,EAAE,OAAO;oBACX,KAAK,EAAE,IAAI;oBACX,OAAO,EAAE,EAAE;oBACX,IAAI,EAAE,YAAY;oBAClB,YAAY,EAAE,IAAI;oBAClB,YAAY,EAAE,KAAK;iBACpB;aACF,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;IAC1B,CAAC;IAED,oBAAoB,CAAC,GAA2B;QAC9C,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;QAE/D,MAAM,aAAa,GAAG,GAAG,CAAC,mBAAmB,EAAE,CAAC;QAChD,IAAI,aAAa,EAAE,CAAC;YAClB,MAAM,SAAS,GAAG,IAAI,CAAC,kCAAkC,CAAC,aAAa,CAAC,CAAC;YACzE,MAAM,YAAY,GAAoB;gBACpC,IAAI,EAAE,QAAQ;gBACd,YAAY,EAAE,SAAS;aACxB,CAAC;YAEF,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;gBAClB,eAAe,EAAE,YAAY;gBAC7B,QAAQ,EAAE;oBACR,IAAI,EAAE,QAAQ;oBACd,EAAE,EAAE,QAAQ;oBACZ,KAAK,EAAE,SAAS;oBAChB,OAAO,EAAE,EAAE;oBACX,IAAI,EAAE,YAAY;oBAClB,YAAY,EAAE,KAAK;oBACnB,YAAY,EAAE,IAAI;iBACnB;aACF,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;IAC1B,CAAC;IAED,gBAAgB,CAAC,GAAuB;QACtC,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC;QAE5D,wCAAwC;QACxC,MAAM,WAAW,GAAG,GAAG,CAAC,iBAAiB,EAAE,CAAC;QAC5C,IAAI,WAAW,EAAE,CAAC;YAChB,MAAM,aAAa,GAAG,WAAW,CAAC,mBAAmB,EAAE,CAAC;YACxD,IAAI,aAAa,EAAE,CAAC;gBAClB,MAAM,SAAS,GAAG,IAAI,CAAC,kCAAkC,CAAC,aAAa,CAAC,CAAC;gBACzE,MAAM,YAAY,GAAoB;oBACpC,IAAI,EAAE,OAAO;oBACb,YAAY,EAAE,SAAS;iBACxB,CAAC;gBAEF,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;oBAClB,eAAe,EAAE,YAAY;oBAC7B,QAAQ,EAAE;wBACR,IAAI,EAAE,OAAO;wBACb,EAAE,EAAE,OAAO;wBACX,KAAK,EAAE,SAAS;wBAChB,OAAO,EAAE,EAAE;wBACX,IAAI,EAAE,YAAY;wBAClB,YAAY,EAAE,KAAK;wBACnB,YAAY,EAAE,IAAI;qBACnB;iBACF,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;IAC1B,CAAC;IAED,eAAe,CAAC,GAAsB;QACpC,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;QAEzD,MAAM,aAAa,GAAG,GAAG,CAAC,mBAAmB,EAAE,CAAC;QAChD,IAAI,aAAa,EAAE,CAAC;YAClB,MAAM,QAAQ,GAAG,IAAI,CAAC,kCAAkC,CAAC,aAAa,CAAC,CAAC;YACxE,MAAM,YAAY,GAAoB;gBACpC,IAAI,EAAE,MAAM;gBACZ,YAAY,EAAE,QAAQ;aACvB,CAAC;YAEF,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;gBAClB,eAAe,EAAE,YAAY;gBAC7B,QAAQ,EAAE;oBACR,IAAI,EAAE,MAAM;oBACZ,EAAE,EAAE,MAAM;oBACV,KAAK,EAAE,QAAQ;oBACf,OAAO,EAAE,EAAE;oBACX,IAAI,EAAE,YAAY;oBAClB,YAAY,EAAE,KAAK;oBACnB,YAAY,EAAE,IAAI;iBACnB;aACF,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;IAC1B,CAAC;IAED,iBAAiB,CAAC,GAAwB;QACxC,MAAM,YAAY,GAAG,GAAG,CAAC,kBAAkB,EAAE,CAAC;QAC9C,IAAI,YAAY,EAAE,CAAC;YACjB,MAAM,UAAU,GAAG,YAAY,CAAC,eAAe,EAAE,CAAC;YAClD,UAAU,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,MAAM,EAAE,EAAE;gBACnC,gDAAgD;YAClD,CAAC,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;IAC1B,CAAC;IAED,eAAe,CAAC,GAAsB;QACpC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;IAC1B,CAAC;CACF;AAxMD,wCAwMC"}