@dbml/core 10.3.0-alpha.0 → 10.3.0-alpha.2

package/lib/parse/ANTLR/ASTGeneration/postgres/PostgresASTGen.mjs

@@ -1,17 +1,21 @@
+import e from "../../../../node_modules/lodash-es/flatten.mjs";
+import t from "../../../../node_modules/lodash-es/last.mjs";
+import n from "../../../../node_modules/lodash-es/flattenDepth.mjs";
+import { init_lodash as r } from "../../../../node_modules/lodash-es/lodash.mjs";
 import "../../../../model_structure/config.mjs";
-import e from "../../parsers/postgresql/PostgreSQLParserVisitor.mjs";
-import t from "../../parsers/postgresql/PostgreSQLParser.mjs";
-import { Enum as n, Field as r, Index as i, Table as a, TableRecord as o } from "../AST.mjs";
-import { COLUMN_CONSTRAINT_KIND as s, CONSTRAINT_TYPE as c, DATA_TYPE as l, TABLE_CONSTRAINT_KIND as u } from "../constants.mjs";
-import { getOriginalText as d } from "../helpers.mjs";
-import { flatten as f, flattenDepth as p, last as m } from "lodash-es";
-import { CARDINALITY_MANY as h, CARDINALITY_MAYBE as g, normalizeQualifiedName as _ } from "@dbml/parse";
-import { ParseTreeListener as v, ParseTreeWalker as y } from "antlr4";
+import i from "../../parsers/postgresql/PostgreSQLParserVisitor.mjs";
+import a from "../../parsers/postgresql/PostgreSQLParser.mjs";
+import { Enum as o, Field as s, Index as c, Table as l, TableRecord as u } from "../AST.mjs";
+import { COLUMN_CONSTRAINT_KIND as d, CONSTRAINT_TYPE as f, DATA_TYPE as p, TABLE_CONSTRAINT_KIND as m } from "../constants.mjs";
+import { getOriginalText as h } from "../helpers.mjs";
+import { CARDINALITY_MANY as g, CARDINALITY_MAYBE as _, normalizeQualifiedName as v } from "@dbml/parse";
+import { ParseTreeListener as y, ParseTreeWalker as b } from "antlr4";
 //#region src/parse/ANTLR/ASTGeneration/postgres/PostgresASTGen.js
-var b = { REF: "ref" }, x = { TABLE: "table" }, S = (e, t, n) => {
+r();
+var x = { REF: "ref" }, S = { TABLE: "table" }, C = (e, t, n) => {
 	let r = t ?? "public";
 	return e.find((e) => (e.schemaName ?? "public") === r && e.name === n);
-}, C = (e) => e && e.replaceAll("''", "'"), w = class extends e {
+}, w = (e) => e && e.replaceAll("''", "'"), T = class extends i {
 	constructor() {
 		super(), this.data = {
 			schemas: [],
@@ -26,39 +30,39 @@ var b = { REF: "ref" }, x = { TABLE: "table" }, S = (e, t, n) => {
 		};
 	}
 	visitViewstmt(e) {
-		let t = e.qualified_name().accept(this), n = m(t), r = t.length > 1 ? t[t.length - 2] : void 0, i = d(e);
+		let n = e.qualified_name().accept(this), r = t(n), i = n.length > 1 ? n[n.length - 2] : void 0, a = h(e);
 		this.data.tables.push({
-			name: n,
-			schemaName: r,
+			name: r,
+			schemaName: i,
 			fields: [],
 			indexes: [],
 			checks: [],
 			note: null,
 			headerColor: null,
 			custom: {
-				query: i,
+				query: a,
 				kind: "view"
 			}
-		}), T(this, e.selectstmt(), n, r);
+		}), E(this, e.selectstmt(), r, i);
 	}
 	visitCreatematviewstmt(e) {
-		let t = e.create_mv_target();
-		if (!t) return;
-		let n = t.qualified_name().accept(this), r = m(n), i = n.length > 1 ? n[n.length - 2] : void 0, a = d(e);
+		let n = e.create_mv_target();
+		if (!n) return;
+		let r = n.qualified_name().accept(this), i = t(r), a = r.length > 1 ? r[r.length - 2] : void 0, o = h(e);
 		this.data.tables.push({
-			name: r,
-			schemaName: i,
+			name: i,
+			schemaName: a,
 			fields: [],
 			indexes: [],
 			checks: [],
 			note: null,
 			headerColor: null,
 			custom: {
-				query: a,
+				query: o,
 				kind: "materialized_view",
 				materialized: !0
 			}
-		}), T(this, e.selectstmt(), r, i);
+		}), E(this, e.selectstmt(), i, a);
 	}
 	visitRoot(e) {
 		return e.stmtblock().accept(this), this.data;
@@ -76,7 +80,7 @@ var b = { REF: "ref" }, x = { TABLE: "table" }, S = (e, t, n) => {
 			return;
 		}
 		if (e.indexstmt()) {
-			let t = e.indexstmt().accept(this), { tableName: n, schemaName: r } = t.pathName, i = S(this.data.tables, r, n);
+			let t = e.indexstmt().accept(this), { tableName: n, schemaName: r } = t.pathName, i = C(this.data.tables, r, n);
 			i && i.indexes.push(t.index);
 			return;
 		}
@@ -94,21 +98,21 @@ var b = { REF: "ref" }, x = { TABLE: "table" }, S = (e, t, n) => {
 		}
 		e.insertstmt() && e.insertstmt().accept(this);
 	}
-	visitCreatestmt(e) {
-		let t = e.qualified_name(0).accept(this), n = m(t), r = t.length > 1 ? t[t.length - 2] : void 0;
-		if (!e.opttableelementlist()) return;
-		let [i, o, s, c] = e.opttableelementlist().accept(this).filter((e) => e).reduce((e, t) => (t.kind === u.FIELD ? e[0].push(t.value) : t.kind === u.INDEX ? e[1].push(t.value) : t.kind === u.FK ? e[2].push(t.value) : t.kind === u.UNIQUE ? e[1].push(t.value) : t.kind === u.CHECK && e[3].push(t.value), e), [
+	visitCreatestmt(n) {
+		let r = n.qualified_name(0).accept(this), i = t(r), a = r.length > 1 ? r[r.length - 2] : void 0;
+		if (!n.opttableelementlist()) return;
+		let [o, s, c, u] = n.opttableelementlist().accept(this).filter((e) => e).reduce((e, t) => (t.kind === m.FIELD ? e[0].push(t.value) : t.kind === m.INDEX ? e[1].push(t.value) : t.kind === m.FK ? e[2].push(t.value) : t.kind === m.UNIQUE ? e[1].push(t.value) : t.kind === m.CHECK && e[3].push(t.value), e), [
 			[],
 			[],
 			[],
 			[]
 		]);
-		return this.data.refs.push(...f(i.map((e) => e.inline_refs.map((t) => (t.endpoints[0].tableName = n, t.endpoints[0].schemaName = r, t.endpoints[0].fieldNames = [e.field.name], t))))), this.data.refs.push(...s.map((e) => (e.endpoints[0].tableName = n, e.endpoints[0].schemaName = r, e))), new a({
-			name: n,
-			schemaName: r,
-			fields: i.map((e) => e.field),
-			indexes: o,
-			checks: c
+		return this.data.refs.push(...e(o.map((e) => e.inline_refs.map((t) => (t.endpoints[0].tableName = i, t.endpoints[0].schemaName = a, t.endpoints[0].fieldNames = [e.field.name], t))))), this.data.refs.push(...c.map((e) => (e.endpoints[0].tableName = i, e.endpoints[0].schemaName = a, e))), new l({
+			name: i,
+			schemaName: a,
+			fields: o.map((e) => e.field),
+			indexes: s,
+			checks: u
 		});
 	}
 	visitOpttableelementlist(e) {
@@ -127,47 +131,47 @@ var b = { REF: "ref" }, x = { TABLE: "table" }, S = (e, t, n) => {
 	}
 	visitConstraintelem(e) {
 		if (e.PRIMARY()) return {
-			kind: u.INDEX,
-			value: new i({
+			kind: m.INDEX,
+			value: new c({
 				pk: !0,
 				columns: e.columnlist().accept(this)
 			})
 		};
 		if (e.FOREIGN()) {
-			let t = e.qualified_name().accept(this), n = m(t), r = t.length > 1 ? t[t.length - 2] : void 0, i = e.columnlist().accept(this).map((e) => e.value), a = e.column_list_()?.accept(this)?.map((e) => e.value), o = e.key_actions(), s = o ? o.accept(this) : {
+			let n = e.qualified_name().accept(this), r = t(n), i = n.length > 1 ? n[n.length - 2] : void 0, a = e.columnlist().accept(this).map((e) => e.value), o = e.column_list_()?.accept(this)?.map((e) => e.value), s = e.key_actions(), c = s ? s.accept(this) : {
 				onDelete: null,
 				onUpdate: null
 			};
 			return {
-				kind: u.FK,
+				kind: m.FK,
 				value: {
 					endpoints: [{
 						tableName: null,
 						schemaName: null,
-						fieldNames: i,
-						relation: h
-					}, {
-						tableName: n,
-						schemaName: r,
 						fieldNames: a,
 						relation: g
+					}, {
+						tableName: r,
+						schemaName: i,
+						fieldNames: o,
+						relation: _
 					}],
-					onDelete: s.onDelete,
-					onUpdate: s.onUpdate
+					onDelete: c.onDelete,
+					onUpdate: c.onUpdate
 				}
 			};
 		}
 		if (e.UNIQUE()) return {
-			kind: u.UNIQUE,
-			value: new i({
+			kind: m.UNIQUE,
+			value: new c({
 				unique: !0,
 				columns: e.columnlist().accept(this)
 			})
 		};
 		if (e.CHECK()) {
-			let t = d(e.a_expr());
+			let t = h(e.a_expr());
 			return {
-				kind: u.CHECK,
+				kind: m.CHECK,
 				value: { expression: t }
 			};
 		}
@@ -181,24 +185,24 @@ var b = { REF: "ref" }, x = { TABLE: "table" }, S = (e, t, n) => {
 	visitColumnElem(e) {
 		return {
 			value: e.colid().accept(this),
-			type: c.COLUMN
+			type: f.COLUMN
 		};
 	}
 	visitColumnDef(e) {
-		let t = e.colid().accept(this), n = e.typename().accept(this), i = e.colquallist().accept(this), a = new Set([
+		let t = e.colid().accept(this), n = e.typename().accept(this), r = e.colquallist().accept(this), i = new Set([
 			"serial",
 			"smallserial",
 			"bigserial"
-		]), o = n.type_name.toLowerCase();
-		return a.has(o) && (i.increment = !0), {
-			kind: u.FIELD,
+		]), a = n.type_name.toLowerCase();
+		return i.has(a) && (r.increment = !0), {
+			kind: m.FIELD,
 			value: {
-				field: new r({
+				field: new s({
 					name: t,
 					type: n,
-					...i
+					...r
 				}),
-				inline_refs: i.inline_refs
+				inline_refs: r.inline_refs
 			}
 		};
 	}
@@ -210,11 +214,11 @@ var b = { REF: "ref" }, x = { TABLE: "table" }, S = (e, t, n) => {
 		return e.colconstraint().forEach((e) => {
 			let n = e.accept(this);
 			if (n) {
-				if (n.kind === s.INLINE_REF) {
+				if (n.kind === d.INLINE_REF) {
 					t.inline_refs.push(n.value);
 					return;
 				}
-				if (n.kind === s.CHECK) {
+				if (n.kind === d.CHECK) {
 					t.checks.push(n.value);
 					return;
 				}
@@ -229,54 +233,54 @@ var b = { REF: "ref" }, x = { TABLE: "table" }, S = (e, t, n) => {
 		if (e.NULL_P()) {
 			let t = !1;
 			return e.NOT() && (t = !0), {
-				kind: s.NOT_NULL,
+				kind: d.NOT_NULL,
 				value: t
 			};
 		}
 		if (e.UNIQUE()) return {
-			kind: s.UNIQUE,
+			kind: d.UNIQUE,
 			value: !0
 		};
 		if (e.PRIMARY()) return {
-			kind: s.PK,
+			kind: d.PK,
 			value: !0
 		};
 		if (e.DEFAULT()) return {
-			kind: s.DEFAULT,
+			kind: d.DEFAULT,
 			value: e.b_expr().accept(this)
 		};
 		if (e.IDENTITY_P()) return {
-			kind: s.INCREMENT,
+			kind: d.INCREMENT,
 			value: !0
 		};
 		if (e.CHECK()) {
-			let t = d(e.a_expr());
+			let t = h(e.a_expr());
 			return {
-				kind: s.CHECK,
+				kind: d.CHECK,
 				value: { expression: t }
 			};
 		}
 		if (e.REFERENCES()) {
-			let t = e.qualified_name().accept(this), n = m(t), r = t.length > 1 ? t[t.length - 2] : void 0, i = e.column_list_()?.accept(this)?.map((e) => e.value), a = e.key_actions(), o = a ? a.accept(this) : {
+			let n = e.qualified_name().accept(this), r = t(n), i = n.length > 1 ? n[n.length - 2] : void 0, a = e.column_list_()?.accept(this)?.map((e) => e.value), o = e.key_actions(), s = o ? o.accept(this) : {
 				onDelete: null,
 				onUpdate: null
 			};
 			return {
-				kind: s.INLINE_REF,
+				kind: d.INLINE_REF,
 				value: {
 					endpoints: [{
 						tableName: null,
 						schemaName: null,
 						fieldNames: null,
-						relation: h
-					}, {
-						tableName: n,
-						schemaName: r,
-						fieldNames: i,
 						relation: g
+					}, {
+						tableName: r,
+						schemaName: i,
+						fieldNames: a,
+						relation: _
 					}],
-					onDelete: o.onDelete,
-					onUpdate: o.onUpdate
+					onDelete: s.onDelete,
+					onUpdate: s.onUpdate
 				}
 			};
 		}
@@ -284,46 +288,46 @@ var b = { REF: "ref" }, x = { TABLE: "table" }, S = (e, t, n) => {
 	visitB_expr(e) {
 		return e.c_expr() ? e.c_expr().accept(this) : {
 			value: e.getText(),
-			type: l.EXPRESSION
+			type: p.EXPRESSION
 		};
 	}
 	visitC_expr_expr(e) {
 		return e.aexprconst() ? e.aexprconst().accept(this) : e.a_expr() ? {
 			value: e.a_expr().getText(),
-			type: l.EXPRESSION
+			type: p.EXPRESSION
 		} : {
 			value: e.getText(),
-			type: l.EXPRESSION
+			type: p.EXPRESSION
 		};
 	}
 	visitC_expr_exists(e) {
 		return {
 			value: e.getText(),
-			type: l.EXPRESSION
+			type: p.EXPRESSION
 		};
 	}
 	visitC_expr_case(e) {
 		return {
 			value: e.getText(),
-			type: l.EXPRESSION
+			type: p.EXPRESSION
 		};
 	}
 	visitAexprconst(e) {
 		return e.sconst() && e.getChildCount() === 1 ? {
 			value: e.sconst().accept(this),
-			type: l.STRING
+			type: p.STRING
 		} : e.TRUE_P() || e.FALSE_P() || e.NULL_P() ? {
 			value: e.getText().toLowerCase(),
-			type: l.BOOLEAN
+			type: p.BOOLEAN
 		} : e.iconst() ? {
 			value: e.iconst().accept(this),
-			type: l.NUMBER
+			type: p.NUMBER
 		} : e.fconst() ? {
 			value: e.getText(),
-			type: l.NUMBER
+			type: p.NUMBER
 		} : {
 			value: e.getText(),
-			type: l.EXPRESSION
+			type: p.EXPRESSION
 		};
 	}
 	visitKey_actions(e) {
@@ -415,17 +419,17 @@ var b = { REF: "ref" }, x = { TABLE: "table" }, S = (e, t, n) => {
 	}
 	visitGenerictype(e) {
 		if (e.attrs()) {
-			let t = [e.getChild(0).getText(), ...e.attrs().accept(this)];
+			let n = [e.getChild(0).getText(), ...e.attrs().accept(this)];
 			return {
-				type: m(t),
-				schemaName: t.length > 1 ? t[t.length - 2] : void 0
+				type: t(n),
+				schemaName: n.length > 1 ? n[n.length - 2] : void 0
 			};
 		}
-		let t = "";
-		t = e.type_function_name() ? e.type_function_name().accept(this) : e.getText();
-		let n = e.type_modifiers_();
+		let n = "";
+		n = e.type_function_name() ? e.type_function_name().accept(this) : e.getText();
+		let r = e.type_modifiers_();
 		return {
-			type: t + (n ? n.getText() : ""),
+			type: n + (r ? r.getText() : ""),
 			schemaName: null
 		};
 	}
@@ -450,15 +454,15 @@ var b = { REF: "ref" }, x = { TABLE: "table" }, S = (e, t, n) => {
 		return Number(e.getText()).toString();
 	}
 	visitIndexstmt(e) {
-		let t = e.relation_expr().accept(this), n = m(t), r = t.length > 1 ? t[t.length - 2] : void 0, a = null;
+		let n = e.relation_expr().accept(this), r = t(n), i = n.length > 1 ? n[n.length - 2] : void 0, a = null;
 		e.index_name_() ? a = e.index_name_().accept(this) : e.name() && (a = e.name().accept(this));
 		let o = e.access_method_clause();
 		return {
 			pathName: {
-				tableName: n,
-				schemaName: r
+				tableName: r,
+				schemaName: i
 			},
-			index: new i({
+			index: new c({
 				name: a,
 				unique: e.unique_()?.accept(this) ?? !1,
 				type: o ? o.accept(this) : void 0,
@@ -484,10 +488,10 @@ var b = { REF: "ref" }, x = { TABLE: "table" }, S = (e, t, n) => {
 	visitIndex_elem(e) {
 		return e.colid() ? {
 			value: e.colid().accept(this),
-			type: c.STRING
+			type: f.STRING
 		} : {
 			value: e.getText(),
-			type: c.EXPRESSION
+			type: f.EXPRESSION
 		};
 	}
 	visitUnique_(e) {
@@ -495,35 +499,35 @@ var b = { REF: "ref" }, x = { TABLE: "table" }, S = (e, t, n) => {
 	}
 	visitAltertablestmt(e) {
 		if (e.TABLE() && e.relation_expr() && e.alter_table_cmds()) {
-			let t = e.relation_expr().accept(this), n = m(t), r = t.length > 1 ? t[t.length - 2] : void 0;
+			let n = e.relation_expr().accept(this), r = t(n), i = n.length > 1 ? n[n.length - 2] : void 0;
 			return e.alter_table_cmds().accept(this).map((e) => {
 				if (!e) return null;
 				let t = null;
 				switch (e.kind) {
-					case u.FK:
-						t = b.REF, e.value.endpoints[0].tableName = n, e.value.endpoints[0].schemaName = r, this.data.refs.push(e.value);
+					case m.FK:
+						t = x.REF, e.value.endpoints[0].tableName = r, e.value.endpoints[0].schemaName = i, this.data.refs.push(e.value);
 						break;
-					case u.CHECK: {
-						let t = S(this.data.tables, r, n);
+					case m.CHECK: {
+						let t = C(this.data.tables, i, r);
 						if (!t) break;
 						t.checks ||= [], t.checks.push(e.value);
 						break;
 					}
-					case u.UNIQUE:
-					case u.PK:
-					case u.INDEX: {
+					case m.UNIQUE:
+					case m.PK:
+					case m.INDEX: {
 						if (e.value.columns.length === 0 || !(e.value.pk || e.value.unique)) break;
 						t = e.kind;
-						let a = S(this.data.tables, r, n);
-						if (!a) break;
+						let n = C(this.data.tables, i, r);
+						if (!n) break;
 						if (e.value.columns.length === 1 && (e.value.unique || e.value.pk)) {
-							let t = e.value.columns[0].value, n = a.fields.find((e) => e.name === t);
-							if (n) {
-								n[e.value.unique ? "unique" : "pk"] = !0;
+							let t = e.value.columns[0].value, r = n.fields.find((e) => e.name === t);
+							if (r) {
+								r[e.value.unique ? "unique" : "pk"] = !0;
 								break;
 							}
 						}
-						let o = new i({
+						let a = new c({
 							name: e.value.name,
 							columns: e.value.columns,
 							note: e.value.note,
@@ -531,7 +535,7 @@ var b = { REF: "ref" }, x = { TABLE: "table" }, S = (e, t, n) => {
 							type: e.value.type,
 							unique: e.value.unique
 						});
-						a.indexes.push(o);
+						n.indexes.push(a);
 						break;
 					}
 					default: break;
@@ -551,18 +555,18 @@ var b = { REF: "ref" }, x = { TABLE: "table" }, S = (e, t, n) => {
 	}
 	visitCommentstmt(e) {
 		if (e.object_type_any_name()) {
-			let t = e.object_type_any_name().accept(this);
-			if (!t) return;
-			if (t === x.TABLE) {
-				let t = e.any_name().accept(this), n = m(t), r = t.length > 1 ? t[t.length - 2] : void 0, i = S(this.data.tables, r, n);
-				if (!i) return;
-				i.note = { value: C(e.comment_text().accept(this)) };
+			let n = e.object_type_any_name().accept(this);
+			if (!n) return;
+			if (n === S.TABLE) {
+				let n = e.any_name().accept(this), r = t(n), i = n.length > 1 ? n[n.length - 2] : void 0, a = C(this.data.tables, i, r);
+				if (!a) return;
+				a.note = { value: w(e.comment_text().accept(this)) };
 			}
 		}
 		if (e.COLUMN()) {
-			let t = e.any_name().accept(this), n = m(t), r = t.length > 1 ? t[t.length - 2] : void 0, i = t.length > 2 ? t[t.length - 3] : void 0, a = S(this.data.tables, i, r).fields.find((e) => e.name === n);
-			if (!a) return;
-			a.note = { value: C(e.comment_text().accept(this)) };
+			let n = e.any_name().accept(this), r = t(n), i = n.length > 1 ? n[n.length - 2] : void 0, a = n.length > 2 ? n[n.length - 3] : void 0, o = C(this.data.tables, a, i).fields.find((e) => e.name === r);
+			if (!o) return;
+			o.note = { value: w(e.comment_text().accept(this)) };
 		}
 	}
 	visitComment_text(e) {
@@ -576,7 +580,7 @@ var b = { REF: "ref" }, x = { TABLE: "table" }, S = (e, t, n) => {
 		return t.slice(1, t.length - 1);
 	}
 	visitObject_type_any_name(e) {
-		if (e.TABLE()) return x.TABLE;
+		if (e.TABLE()) return S.TABLE;
 	}
 	visitAny_name(e) {
 		let t = [e.colid().accept(this)];
@@ -587,14 +591,14 @@ var b = { REF: "ref" }, x = { TABLE: "table" }, S = (e, t, n) => {
 	}
 	visitDefinestmt(e) {
 		if (e.TYPE_P() && e.enum_val_list_()) {
-			let t = e.any_name(0).accept(this), r = m(t), i = t.length > 1 ? t[t.length - 2] : void 0, a = e.enum_val_list_().accept(this).map((e) => ({ name: e }));
+			let n = e.any_name(0).accept(this), r = t(n), i = n.length > 1 ? n[n.length - 2] : void 0, a = e.enum_val_list_().accept(this).map((e) => ({ name: e }));
 			if (!a) return;
-			let o = new n({
+			let s = new o({
 				name: r,
 				schemaName: i,
 				values: a
 			});
-			this.data.enums.push(o.toJSON());
+			this.data.enums.push(s.toJSON());
 		}
 	}
 	visitEnum_val_list_(e) {
@@ -604,16 +608,16 @@ var b = { REF: "ref" }, x = { TABLE: "table" }, S = (e, t, n) => {
 		return e.sconst().map((e) => e.accept(this));
 	}
 	visitInsertstmt(e) {
-		let t = e.insert_target().accept(this), n = m(t), r = t.length > 1 ? t[t.length - 2] : void 0, { columns: i, values: a } = e.insert_rest().accept(this);
-		if (i.length === 0) {
-			let e = S(this.data.tables, r, n);
-			e && e.fields && (i = e.fields.map((e) => e.name));
+		let n = e.insert_target().accept(this), r = t(n), i = n.length > 1 ? n[n.length - 2] : void 0, { columns: a, values: o } = e.insert_rest().accept(this);
+		if (a.length === 0) {
+			let e = C(this.data.tables, i, r);
+			e && e.fields && (a = e.fields.map((e) => e.name));
 		}
-		let s = new o({
-			schemaName: r,
-			tableName: n,
-			columns: i,
-			values: a
+		let s = new u({
+			schemaName: i,
+			tableName: r,
+			columns: a,
+			values: o
 		});
 		this.data.records.push(s);
 	}
@@ -653,7 +657,7 @@ var b = { REF: "ref" }, x = { TABLE: "table" }, S = (e, t, n) => {
 		return e.values_clause() ? e.values_clause().accept(this) : null;
 	}
 	visitValues_clause(e) {
-		return e.expr_list().map((e) => p(e.accept(this), 21));
+		return e.expr_list().map((e) => n(e.accept(this), 21));
 	}
 	visitA_expr_collate(e) {
 		let [t, n, r = {}] = e.a_expr_typecast().accept(this), { value: i, type: a } = t;
@@ -664,17 +668,17 @@ var b = { REF: "ref" }, x = { TABLE: "table" }, S = (e, t, n) => {
 		};
 	}
 };
-function T(e, n, r, i) {
+function E(e, n, r, i) {
 	if (!n) return;
-	let a = /* @__PURE__ */ new Set(), o = _(i ?? "public", r), s = new v();
-	s.enterEveryRule = (n) => {
-		if (!(n instanceof t.Relation_exprContext)) return;
-		let s = n.accept(e), c = m(s), l = s.length > 1 ? s[s.length - 2] : void 0, u = _(l ?? "public", c);
-		u !== o && (a.has(u) || S(e.data.tables, l, c) && (a.add(u), e.data.deps.push({
+	let o = /* @__PURE__ */ new Set(), s = v(i ?? "public", r), c = new y();
+	c.enterEveryRule = (n) => {
+		if (!(n instanceof a.Relation_exprContext)) return;
+		let c = n.accept(e), l = t(c), u = c.length > 1 ? c[c.length - 2] : void 0, d = v(u ?? "public", l);
+		d !== s && (o.has(d) || C(e.data.tables, u, l) && (o.add(d), e.data.deps.push({
 			edges: [{
 				upstream: {
-					schemaName: l,
-					tableName: c,
+					schemaName: u,
+					tableName: l,
 					fieldNames: []
 				},
 				downstream: {
@@ -686,7 +690,7 @@ function T(e, n, r, i) {
 			note: null,
 			custom: {}
 		})));
-	}, y.DEFAULT.walk(s, n);
+	}, b.DEFAULT.walk(c, n);
 }
 //#endregion
-export { w as default };
+export { T as default };

package/lib/parse/ANTLR/ASTGeneration/snowflake/SnowflakeASTGen.cjs

@@ -1 +1 @@
-require("../../../../_virtual/_rolldown/runtime.cjs");const e=require("../../parsers/snowflake/SnowflakeParserVisitor.cjs"),t=require("../AST.cjs"),n=require("../constants.cjs"),r=require("../helpers.cjs");let i=require("lodash-es"),a=require("@dbml/parse");var o=`public`,s=e=>r.getOriginalText(e).replace(/''/g,`'`).slice(1,-1),c=class extends e.default{constructor(){super(),this.data={schemas:[],tables:[],refs:[],enums:[],tableGroups:[],aliases:[],project:{},records:[]}}findTable(e,t){let n=e||o;return this.data.tables.find(e=>(e.schemaName||o)===n&&e.name===t)}visitSnowflake_file(e){return e.batch()?.accept(this),this.data}visitBatch(e){e.sql_command().forEach(e=>{e.accept(this)})}visitSql_command(e){e.ddl_command()?e.ddl_command().accept(this):e.dml_command()&&e.dml_command().accept(this)}visitDdl_command(e){e.alter_command()?e.alter_command().accept(this):e.create_command()&&e.create_command().accept(this)}visitDml_command(e){e.insert_statement()&&e.insert_statement().accept(this)}visitCreate_command(e){if(e.create_table()){let t=e.create_table().accept(this);this.data.tables.push(t)}else if(e.create_table_like()){let[n,r,i,a]=e.create_table_like().accept(this),o=this.data.tables.reduce((e,t)=>t.name===a&&t.schemaName===i?t:e,null);if(o){let e=new t.Table({name:r,schemaName:n,fields:o.fields});this.data.tables.push(e)}}}visitCreate_database(e){let t=r.getOriginalText(e.object_name());this.data.schemas.push(t)}visitCreate_table(e){let[r,a,o]=e.object_name().accept(this),{definitions:s,tableNote:c}=e.create_table_clause().accept(this),l=e.comment_clause()?.accept(this)||c,[u,d,f,p]=s.reduce((e,t)=>{if(t.kind===n.TABLE_CONSTRAINT_KIND.FIELD)e[0].push(t.value);else if(t.kind===n.TABLE_CONSTRAINT_KIND.UNIQUE)e[1].push(t.value);else if(t.kind===n.TABLE_CONSTRAINT_KIND.FK)e[2].push(t.value);else if(t.kind===n.TABLE_CONSTRAINT_KIND.PK){let n=t.value;n.columns.length>1?e[1].push(t.value):e[3]=n}return e},[[],[],[],null]),m=u.map(e=>{let{field:n,inlineRefs:r}=e;if(n.type.type_name?.toLowerCase()===`enum`){let e=n.type.args.map(e=>({name:e.replace(/'|"|`/g,``).trim()})),r=new t.Enum({name:`${o}_${n.name}_enum`,schemaName:a,values:e});n.type.type_name=r.name,n.type.schemaName=r.schemaName,this.data.enums.push(r)}return r.forEach(e=>{e.endpoints[0].tableName=o,e.endpoints[0].schemaName=a,e.endpoints[0].fieldNames=[n.name]}),r});this.data.refs.push(...(0,i.flatten)(m)),this.data.refs.push(...f.map(e=>(e.endpoints[0].tableName=o,e.endpoints[0].schemaName=a,e)));let h=new t.Table({name:o,schemaName:a,fields:u.map(e=>e.field),indexes:d,note:l});if(p){let e=h.fields.find(e=>e.name===p.columns[0].value);e&&(e.pk=!0)}return h}visitAlter_table(e){if(e.constraint_action()){let[t,r,i]=e.object_name()[0].accept(this),a=e.constraint_action().accept(this);if(a){let e=[],t=[],o=[];a.kind===n.TABLE_CONSTRAINT_KIND.FIELD?e.push(a.value):a.kind===n.TABLE_CONSTRAINT_KIND.UNIQUE?t.push(a.value):a.kind===n.TABLE_CONSTRAINT_KIND.FK&&o.push(a.value),this.data.refs.push(...o.map(e=>(e.endpoints[0].tableName=i,e.endpoints[0].schemaName=r,e)));let s=this.data.tables.reduce((e,t)=>t.name===i&&t.schemaName===r?t:e,null);return s&&(s.fields.push(...e),s.indexes.push(...t)),s}}return null}visitCreate_table_like(e){let[t,n,r]=e.object_name()[0].accept(this),[i,a,o]=e.object_name()[1].accept(this);return[n,r,a,o]}visitObject_name(e){return[e.d?.accept(this),e.s?.accept(this),e.o.accept(this)]}visitId_(e){return e.DOUBLE_QUOTE_ID()?r.getOriginalText(e).slice(1,-1):r.getOriginalText(e)}visitCreate_table_clause(e){return{definitions:e.column_decl_item_list_paren().accept(this),tableNote:e.comment_clause().map(e=>e.accept(this))?.[-1]}}visitComment_clause(e){return s(e.string())}visitColumn_decl_item_list_paren(e){return e.column_decl_item_list().accept(this)}visitColumn_decl_item_list(e){return e.column_decl_item().map(e=>e.accept(this)).filter(e=>e)}visitColumn_decl_item(e){return e.full_col_decl()?{kind:n.TABLE_CONSTRAINT_KIND.FIELD,value:e.full_col_decl().accept(this)}:e.out_of_line_constraint()?e.out_of_line_constraint().accept(this):null}visitFull_col_decl(e){let t=e.col_decl().accept(this),r=[];if(e.inline_constraint()){let a=e.inline_constraint().map(e=>e.accept(this));(0,i.isEmpty)(a)||a.forEach(e=>{e.kind===n.COLUMN_CONSTRAINT_KIND.UNIQUE?t.unique=!0:e.kind===n.COLUMN_CONSTRAINT_KIND.PK?t.pk=!0:e.kind===n.COLUMN_CONSTRAINT_KIND.FK&&r.push(e.value)})}if(e.null_not_null()){let r=e.null_not_null().map(e=>e.accept(this));(0,i.isEmpty)(r)||r.forEach(e=>{e.kind===n.COLUMN_CONSTRAINT_KIND.NOT_NULL&&(t.not_null=e.value)})}return e.default_value()&&e.default_value().map(e=>e.accept(this)).forEach(e=>{e.kind===n.COLUMN_CONSTRAINT_KIND.DEFAULT?t.dbdefault=e.value:e.kind===n.COLUMN_CONSTRAINT_KIND.INCREMENT&&(t.increment=e.value)}),e.COMMENT()&&(t.note=s(e.string())),{field:t,inlineRefs:r}}visitCol_decl(e){return new t.Field({name:e.column_name().accept(this),type:{type_name:r.getOriginalText(e.data_type()),schemaName:null}})}visitColumn_name(e){return e.id_().map(e=>e.accept(this)).join(`.`)}visitInline_constraint(e){if(e.UNIQUE())return{kind:n.COLUMN_CONSTRAINT_KIND.UNIQUE,value:!0};if(e.primary_key())return{kind:n.COLUMN_CONSTRAINT_KIND.PK,value:!0};if(e.foreign_key()&&e.column_name()){let[t,r,i]=e.object_name().accept(this),o=[e.column_name().accept(this)];return{kind:n.COLUMN_CONSTRAINT_KIND.FK,value:{endpoints:[{tableName:null,schemaName:null,fieldNames:null,relation:a.CARDINALITY_MANY},{tableName:i,schemaName:r,fieldNames:o,relation:a.CARDINALITY_MAYBE}]}}}return{kind:null,value:null}}visitOut_of_line_constraint(e){if(e.UNIQUE()){let r=e.id_()?.accept(this),a=new t.Index({unique:!0,columns:(0,i.flatten)(e.column_list_in_parentheses().map(e=>e.accept(this))).map(e=>({value:e,type:n.CONSTRAINT_TYPE.COLUMN}))});return{name:r,kind:n.TABLE_CONSTRAINT_KIND.UNIQUE,value:a}}if(e.primary_key()){let r=e.id_()?.accept(this),a=new t.Index({name:r,pk:!0,columns:(0,i.flatten)(e.column_list_in_parentheses().map(e=>e.accept(this))).map(e=>({value:e,type:n.CONSTRAINT_TYPE.COLUMN}))});return{kind:n.TABLE_CONSTRAINT_KIND.PK,value:a}}if(e.foreign_key()&&e.column_list_in_parentheses().length===2){let[t,r,i]=e.object_name().accept(this),a=e.column_list_in_parentheses()[0].accept(this),o=e.column_list_in_parentheses()[1].accept(this);return{kind:n.TABLE_CONSTRAINT_KIND.FK,value:{endpoints:[{tableName:null,schemaName:null,fieldNames:a,relation:`*`},{tableName:i,schemaName:r,fieldNames:o,relation:`1`}]}}}return null}visitColumn_list_in_parentheses(e){return e.column_list().accept(this)}visitColumn_list(e){return e.column_name().map(e=>e.accept(this))}visitNull_not_null(e){return{kind:n.COLUMN_CONSTRAINT_KIND.NOT_NULL,value:!!e.NOT()}}visitDefault_value(e){return e.DEFAULT()?{kind:n.COLUMN_CONSTRAINT_KIND.DEFAULT,value:e.expr().accept(this)}:e.AUTOINCREMENT()||e.IDENTITY()?{kind:n.COLUMN_CONSTRAINT_KIND.INCREMENT,value:!0}:null}visitExpr(e){return e.primitive_expression()?e.primitive_expression().accept(this):{value:r.getOriginalText(e),type:`expression`}}visitPrimitive_expression(e){return e.NULL_()?{value:null,type:`boolean`}:e.literal()?e.literal().accept(this):{value:r.getOriginalText(e),type:`expression`}}visitLiteral(e){return e.STRING()?{value:s(e),type:`string`}:e.DECIMAL()||e.REAL||e.FLOAT?{value:r.getOriginalText(e),type:`number`}:e.true_false()||e.NULL_()?{value:r.getOriginalText(e),type:`boolean`}:{value:r.getOriginalText(e),type:`expression`}}visitConstraint_action(e){return e.ADD()?e.out_of_line_constraint().accept(this):null}visitInsert_statement(e){let[n,r,i]=e.object_name().accept(this),a=new t.TableRecord({schemaName:r,tableName:i,columns:e.column_list_in_parentheses()?e.column_list_in_parentheses().accept(this):this.findTable(r,i)?.fields.map(e=>e.name)||[],values:e.values_builder()?e.values_builder().accept(this):[]});this.data.records.push(a)}visitValues_builder(e){return e.expr_list().map(e=>(0,i.flattenDepth)(e.accept(this),1))}visitExpr_list(e){return e.expr().map(e=>e.accept(this))}};exports.default=c;
+require("../../../../_virtual/_rolldown/runtime.cjs");const e=require("../../../../node_modules/lodash-es/flatten.cjs"),t=require("../../../../node_modules/lodash-es/flattenDepth.cjs"),n=require("../../../../node_modules/lodash-es/isEmpty.cjs"),r=require("../../../../node_modules/lodash-es/lodash.cjs"),i=require("../../parsers/snowflake/SnowflakeParserVisitor.cjs"),a=require("../AST.cjs"),o=require("../constants.cjs"),s=require("../helpers.cjs");let c=require("@dbml/parse");r.init_lodash();var l=`public`,u=e=>s.getOriginalText(e).replace(/''/g,`'`).slice(1,-1),d=class extends i.default{constructor(){super(),this.data={schemas:[],tables:[],refs:[],enums:[],tableGroups:[],aliases:[],project:{},records:[]}}findTable(e,t){let n=e||l;return this.data.tables.find(e=>(e.schemaName||l)===n&&e.name===t)}visitSnowflake_file(e){return e.batch()?.accept(this),this.data}visitBatch(e){e.sql_command().forEach(e=>{e.accept(this)})}visitSql_command(e){e.ddl_command()?e.ddl_command().accept(this):e.dml_command()&&e.dml_command().accept(this)}visitDdl_command(e){e.alter_command()?e.alter_command().accept(this):e.create_command()&&e.create_command().accept(this)}visitDml_command(e){e.insert_statement()&&e.insert_statement().accept(this)}visitCreate_command(e){if(e.create_table()){let t=e.create_table().accept(this);this.data.tables.push(t)}else if(e.create_table_like()){let[t,n,r,i]=e.create_table_like().accept(this),o=this.data.tables.reduce((e,t)=>t.name===i&&t.schemaName===r?t:e,null);if(o){let e=new a.Table({name:n,schemaName:t,fields:o.fields});this.data.tables.push(e)}}}visitCreate_database(e){let t=s.getOriginalText(e.object_name());this.data.schemas.push(t)}visitCreate_table(t){let[n,r,i]=t.object_name().accept(this),{definitions:s,tableNote:c}=t.create_table_clause().accept(this),l=t.comment_clause()?.accept(this)||c,[u,d,f,p]=s.reduce((e,t)=>{if(t.kind===o.TABLE_CONSTRAINT_KIND.FIELD)e[0].push(t.value);else if(t.kind===o.TABLE_CONSTRAINT_KIND.UNIQUE)e[1].push(t.value);else if(t.kind===o.TABLE_CONSTRAINT_KIND.FK)e[2].push(t.value);else if(t.kind===o.TABLE_CONSTRAINT_KIND.PK){let n=t.value;n.columns.length>1?e[1].push(t.value):e[3]=n}return e},[[],[],[],null]),m=u.map(e=>{let{field:t,inlineRefs:n}=e;if(t.type.type_name?.toLowerCase()===`enum`){let e=t.type.args.map(e=>({name:e.replace(/'|"|`/g,``).trim()})),n=new a.Enum({name:`${i}_${t.name}_enum`,schemaName:r,values:e});t.type.type_name=n.name,t.type.schemaName=n.schemaName,this.data.enums.push(n)}return n.forEach(e=>{e.endpoints[0].tableName=i,e.endpoints[0].schemaName=r,e.endpoints[0].fieldNames=[t.name]}),n});this.data.refs.push(...e.default(m)),this.data.refs.push(...f.map(e=>(e.endpoints[0].tableName=i,e.endpoints[0].schemaName=r,e)));let h=new a.Table({name:i,schemaName:r,fields:u.map(e=>e.field),indexes:d,note:l});if(p){let e=h.fields.find(e=>e.name===p.columns[0].value);e&&(e.pk=!0)}return h}visitAlter_table(e){if(e.constraint_action()){let[t,n,r]=e.object_name()[0].accept(this),i=e.constraint_action().accept(this);if(i){let e=[],t=[],a=[];i.kind===o.TABLE_CONSTRAINT_KIND.FIELD?e.push(i.value):i.kind===o.TABLE_CONSTRAINT_KIND.UNIQUE?t.push(i.value):i.kind===o.TABLE_CONSTRAINT_KIND.FK&&a.push(i.value),this.data.refs.push(...a.map(e=>(e.endpoints[0].tableName=r,e.endpoints[0].schemaName=n,e)));let s=this.data.tables.reduce((e,t)=>t.name===r&&t.schemaName===n?t:e,null);return s&&(s.fields.push(...e),s.indexes.push(...t)),s}}return null}visitCreate_table_like(e){let[t,n,r]=e.object_name()[0].accept(this),[i,a,o]=e.object_name()[1].accept(this);return[n,r,a,o]}visitObject_name(e){return[e.d?.accept(this),e.s?.accept(this),e.o.accept(this)]}visitId_(e){return e.DOUBLE_QUOTE_ID()?s.getOriginalText(e).slice(1,-1):s.getOriginalText(e)}visitCreate_table_clause(e){return{definitions:e.column_decl_item_list_paren().accept(this),tableNote:e.comment_clause().map(e=>e.accept(this))?.[-1]}}visitComment_clause(e){return u(e.string())}visitColumn_decl_item_list_paren(e){return e.column_decl_item_list().accept(this)}visitColumn_decl_item_list(e){return e.column_decl_item().map(e=>e.accept(this)).filter(e=>e)}visitColumn_decl_item(e){return e.full_col_decl()?{kind:o.TABLE_CONSTRAINT_KIND.FIELD,value:e.full_col_decl().accept(this)}:e.out_of_line_constraint()?e.out_of_line_constraint().accept(this):null}visitFull_col_decl(e){let t=e.col_decl().accept(this),r=[];if(e.inline_constraint()){let i=e.inline_constraint().map(e=>e.accept(this));n.default(i)||i.forEach(e=>{e.kind===o.COLUMN_CONSTRAINT_KIND.UNIQUE?t.unique=!0:e.kind===o.COLUMN_CONSTRAINT_KIND.PK?t.pk=!0:e.kind===o.COLUMN_CONSTRAINT_KIND.FK&&r.push(e.value)})}if(e.null_not_null()){let r=e.null_not_null().map(e=>e.accept(this));n.default(r)||r.forEach(e=>{e.kind===o.COLUMN_CONSTRAINT_KIND.NOT_NULL&&(t.not_null=e.value)})}return e.default_value()&&e.default_value().map(e=>e.accept(this)).forEach(e=>{e.kind===o.COLUMN_CONSTRAINT_KIND.DEFAULT?t.dbdefault=e.value:e.kind===o.COLUMN_CONSTRAINT_KIND.INCREMENT&&(t.increment=e.value)}),e.COMMENT()&&(t.note=u(e.string())),{field:t,inlineRefs:r}}visitCol_decl(e){return new a.Field({name:e.column_name().accept(this),type:{type_name:s.getOriginalText(e.data_type()),schemaName:null}})}visitColumn_name(e){return e.id_().map(e=>e.accept(this)).join(`.`)}visitInline_constraint(e){if(e.UNIQUE())return{kind:o.COLUMN_CONSTRAINT_KIND.UNIQUE,value:!0};if(e.primary_key())return{kind:o.COLUMN_CONSTRAINT_KIND.PK,value:!0};if(e.foreign_key()&&e.column_name()){let[t,n,r]=e.object_name().accept(this),i=[e.column_name().accept(this)];return{kind:o.COLUMN_CONSTRAINT_KIND.FK,value:{endpoints:[{tableName:null,schemaName:null,fieldNames:null,relation:c.CARDINALITY_MANY},{tableName:r,schemaName:n,fieldNames:i,relation:c.CARDINALITY_MAYBE}]}}}return{kind:null,value:null}}visitOut_of_line_constraint(t){if(t.UNIQUE()){let n=t.id_()?.accept(this),r=new a.Index({unique:!0,columns:e.default(t.column_list_in_parentheses().map(e=>e.accept(this))).map(e=>({value:e,type:o.CONSTRAINT_TYPE.COLUMN}))});return{name:n,kind:o.TABLE_CONSTRAINT_KIND.UNIQUE,value:r}}if(t.primary_key()){let n=t.id_()?.accept(this),r=new a.Index({name:n,pk:!0,columns:e.default(t.column_list_in_parentheses().map(e=>e.accept(this))).map(e=>({value:e,type:o.CONSTRAINT_TYPE.COLUMN}))});return{kind:o.TABLE_CONSTRAINT_KIND.PK,value:r}}if(t.foreign_key()&&t.column_list_in_parentheses().length===2){let[e,n,r]=t.object_name().accept(this),i=t.column_list_in_parentheses()[0].accept(this),a=t.column_list_in_parentheses()[1].accept(this);return{kind:o.TABLE_CONSTRAINT_KIND.FK,value:{endpoints:[{tableName:null,schemaName:null,fieldNames:i,relation:`*`},{tableName:r,schemaName:n,fieldNames:a,relation:`1`}]}}}return null}visitColumn_list_in_parentheses(e){return e.column_list().accept(this)}visitColumn_list(e){return e.column_name().map(e=>e.accept(this))}visitNull_not_null(e){return{kind:o.COLUMN_CONSTRAINT_KIND.NOT_NULL,value:!!e.NOT()}}visitDefault_value(e){return e.DEFAULT()?{kind:o.COLUMN_CONSTRAINT_KIND.DEFAULT,value:e.expr().accept(this)}:e.AUTOINCREMENT()||e.IDENTITY()?{kind:o.COLUMN_CONSTRAINT_KIND.INCREMENT,value:!0}:null}visitExpr(e){return e.primitive_expression()?e.primitive_expression().accept(this):{value:s.getOriginalText(e),type:`expression`}}visitPrimitive_expression(e){return e.NULL_()?{value:null,type:`boolean`}:e.literal()?e.literal().accept(this):{value:s.getOriginalText(e),type:`expression`}}visitLiteral(e){return e.STRING()?{value:u(e),type:`string`}:e.DECIMAL()||e.REAL||e.FLOAT?{value:s.getOriginalText(e),type:`number`}:e.true_false()||e.NULL_()?{value:s.getOriginalText(e),type:`boolean`}:{value:s.getOriginalText(e),type:`expression`}}visitConstraint_action(e){return e.ADD()?e.out_of_line_constraint().accept(this):null}visitInsert_statement(e){let[t,n,r]=e.object_name().accept(this),i=new a.TableRecord({schemaName:n,tableName:r,columns:e.column_list_in_parentheses()?e.column_list_in_parentheses().accept(this):this.findTable(n,r)?.fields.map(e=>e.name)||[],values:e.values_builder()?e.values_builder().accept(this):[]});this.data.records.push(i)}visitValues_builder(e){return e.expr_list().map(e=>t.default(e.accept(this),1))}visitExpr_list(e){return e.expr().map(e=>e.accept(this))}};exports.default=d;