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

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

@@ -1 +1 @@
-require("../../../../_virtual/_rolldown/runtime.cjs"),require("../../../../model_structure/config.cjs");const e=require("../../parsers/postgresql/PostgreSQLParserVisitor.cjs"),t=require("../../parsers/postgresql/PostgreSQLParser.cjs"),n=require("../AST.cjs"),r=require("../constants.cjs"),i=require("../helpers.cjs");let a=require("lodash-es"),o=require("@dbml/parse"),s=require("antlr4");var c={REF:`ref`},l={TABLE:`table`},u=(e,t,n)=>{let r=t??`public`;return e.find(e=>(e.schemaName??`public`)===r&&e.name===n)},d=e=>e&&e.replaceAll(`''`,`'`),f=class extends e.default{constructor(){super(),this.data={schemas:[],tables:[],refs:[],deps:[],enums:[],tableGroups:[],aliases:[],project:{},records:[]}}visitViewstmt(e){let t=e.qualified_name().accept(this),n=(0,a.last)(t),r=t.length>1?t[t.length-2]:void 0,o=i.getOriginalText(e);this.data.tables.push({name:n,schemaName:r,fields:[],indexes:[],checks:[],note:null,headerColor:null,custom:{query:o,kind:`view`}}),p(this,e.selectstmt(),n,r)}visitCreatematviewstmt(e){let t=e.create_mv_target();if(!t)return;let n=t.qualified_name().accept(this),r=(0,a.last)(n),o=n.length>1?n[n.length-2]:void 0,s=i.getOriginalText(e);this.data.tables.push({name:r,schemaName:o,fields:[],indexes:[],checks:[],note:null,headerColor:null,custom:{query:s,kind:`materialized_view`,materialized:!0}}),p(this,e.selectstmt(),r,o)}visitRoot(e){return e.stmtblock().accept(this),this.data}visitStmtblock(e){e.stmtmulti().accept(this)}visitStmtmulti(e){e.stmt().map(e=>e.accept(this))}visitStmt(e){if(e.createstmt()){let t=e.createstmt().accept(this);t&&this.data.tables.push(t);return}if(e.indexstmt()){let t=e.indexstmt().accept(this),{tableName:n,schemaName:r}=t.pathName,i=u(this.data.tables,r,n);i&&i.indexes.push(t.index);return}if(e.altertablestmt()){e.altertablestmt().accept(this);return}if(e.commentstmt()){e.commentstmt().accept(this);return}if(e.definestmt()){e.definestmt().accept(this);return}e.insertstmt()&&e.insertstmt().accept(this)}visitCreatestmt(e){let t=e.qualified_name(0).accept(this),i=(0,a.last)(t),o=t.length>1?t[t.length-2]:void 0;if(!e.opttableelementlist())return;let[s,c,l,u]=e.opttableelementlist().accept(this).filter(e=>e).reduce((e,t)=>(t.kind===r.TABLE_CONSTRAINT_KIND.FIELD?e[0].push(t.value):t.kind===r.TABLE_CONSTRAINT_KIND.INDEX?e[1].push(t.value):t.kind===r.TABLE_CONSTRAINT_KIND.FK?e[2].push(t.value):t.kind===r.TABLE_CONSTRAINT_KIND.UNIQUE?e[1].push(t.value):t.kind===r.TABLE_CONSTRAINT_KIND.CHECK&&e[3].push(t.value),e),[[],[],[],[]]);return this.data.refs.push(...(0,a.flatten)(s.map(e=>e.inline_refs.map(t=>(t.endpoints[0].tableName=i,t.endpoints[0].schemaName=o,t.endpoints[0].fieldNames=[e.field.name],t))))),this.data.refs.push(...l.map(e=>(e.endpoints[0].tableName=i,e.endpoints[0].schemaName=o,e))),new n.Table({name:i,schemaName:o,fields:s.map(e=>e.field),indexes:c,checks:u})}visitOpttableelementlist(e){return e.tableelementlist()?e.tableelementlist().accept(this):[]}visitTableelementlist(e){return e.tableelement().map(e=>e.accept(this))}visitTableelement(e){if(e.columnDef())return e.columnDef().accept(this);if(e.tableconstraint())return e.tableconstraint().accept(this)}visitTableconstraint(e){let t=e.name()?.accept(this),n=e.constraintelem().accept(this);if(n)return n.value.name=t,n}visitConstraintelem(e){if(e.PRIMARY())return{kind:r.TABLE_CONSTRAINT_KIND.INDEX,value:new n.Index({pk:!0,columns:e.columnlist().accept(this)})};if(e.FOREIGN()){let t=e.qualified_name().accept(this),n=(0,a.last)(t),i=t.length>1?t[t.length-2]:void 0,s=e.columnlist().accept(this).map(e=>e.value),c=e.column_list_()?.accept(this)?.map(e=>e.value),l=e.key_actions(),u=l?l.accept(this):{onDelete:null,onUpdate:null};return{kind:r.TABLE_CONSTRAINT_KIND.FK,value:{endpoints:[{tableName:null,schemaName:null,fieldNames:s,relation:o.CARDINALITY_MANY},{tableName:n,schemaName:i,fieldNames:c,relation:o.CARDINALITY_MAYBE}],onDelete:u.onDelete,onUpdate:u.onUpdate}}}if(e.UNIQUE())return{kind:r.TABLE_CONSTRAINT_KIND.UNIQUE,value:new n.Index({unique:!0,columns:e.columnlist().accept(this)})};if(e.CHECK()){let t=i.getOriginalText(e.a_expr());return{kind:r.TABLE_CONSTRAINT_KIND.CHECK,value:{expression:t}}}}visitColumn_list_(e){return e.columnlist().accept(this)}visitColumnlist(e){return e.columnElem().map(e=>e.accept(this))}visitColumnElem(e){return{value:e.colid().accept(this),type:r.CONSTRAINT_TYPE.COLUMN}}visitColumnDef(e){let t=e.colid().accept(this),i=e.typename().accept(this),a=e.colquallist().accept(this),o=new Set([`serial`,`smallserial`,`bigserial`]),s=i.type_name.toLowerCase();return o.has(s)&&(a.increment=!0),{kind:r.TABLE_CONSTRAINT_KIND.FIELD,value:{field:new n.Field({name:t,type:i,...a}),inline_refs:a.inline_refs}}}visitColquallist(e){let t={inline_refs:[],checks:[]};return e.colconstraint().forEach(e=>{let n=e.accept(this);if(n){if(n.kind===r.COLUMN_CONSTRAINT_KIND.INLINE_REF){t.inline_refs.push(n.value);return}if(n.kind===r.COLUMN_CONSTRAINT_KIND.CHECK){t.checks.push(n.value);return}t[n.kind]=n.value}}),t}visitColconstraint(e){if(e.colconstraintelem())return e.colconstraintelem().accept(this)}visitColconstraintelem(e){if(e.NULL_P()){let t=!1;return e.NOT()&&(t=!0),{kind:r.COLUMN_CONSTRAINT_KIND.NOT_NULL,value:t}}if(e.UNIQUE())return{kind:r.COLUMN_CONSTRAINT_KIND.UNIQUE,value:!0};if(e.PRIMARY())return{kind:r.COLUMN_CONSTRAINT_KIND.PK,value:!0};if(e.DEFAULT())return{kind:r.COLUMN_CONSTRAINT_KIND.DEFAULT,value:e.b_expr().accept(this)};if(e.IDENTITY_P())return{kind:r.COLUMN_CONSTRAINT_KIND.INCREMENT,value:!0};if(e.CHECK()){let t=i.getOriginalText(e.a_expr());return{kind:r.COLUMN_CONSTRAINT_KIND.CHECK,value:{expression:t}}}if(e.REFERENCES()){let t=e.qualified_name().accept(this),n=(0,a.last)(t),i=t.length>1?t[t.length-2]:void 0,s=e.column_list_()?.accept(this)?.map(e=>e.value),c=e.key_actions(),l=c?c.accept(this):{onDelete:null,onUpdate:null};return{kind:r.COLUMN_CONSTRAINT_KIND.INLINE_REF,value:{endpoints:[{tableName:null,schemaName:null,fieldNames:null,relation:o.CARDINALITY_MANY},{tableName:n,schemaName:i,fieldNames:s,relation:o.CARDINALITY_MAYBE}],onDelete:l.onDelete,onUpdate:l.onUpdate}}}}visitB_expr(e){return e.c_expr()?e.c_expr().accept(this):{value:e.getText(),type:r.DATA_TYPE.EXPRESSION}}visitC_expr_expr(e){return e.aexprconst()?e.aexprconst().accept(this):e.a_expr()?{value:e.a_expr().getText(),type:r.DATA_TYPE.EXPRESSION}:{value:e.getText(),type:r.DATA_TYPE.EXPRESSION}}visitC_expr_exists(e){return{value:e.getText(),type:r.DATA_TYPE.EXPRESSION}}visitC_expr_case(e){return{value:e.getText(),type:r.DATA_TYPE.EXPRESSION}}visitAexprconst(e){return e.sconst()&&e.getChildCount()===1?{value:e.sconst().accept(this),type:r.DATA_TYPE.STRING}:e.TRUE_P()||e.FALSE_P()||e.NULL_P()?{value:e.getText().toLowerCase(),type:r.DATA_TYPE.BOOLEAN}:e.iconst()?{value:e.iconst().accept(this),type:r.DATA_TYPE.NUMBER}:e.fconst()?{value:e.getText(),type:r.DATA_TYPE.NUMBER}:{value:e.getText(),type:r.DATA_TYPE.EXPRESSION}}visitKey_actions(e){let[t,n]=[null,null];return e.key_update()&&(n=e.key_update().accept(this)),e.key_delete()&&(t=e.key_delete().accept(this)),{onDelete:t,onUpdate:n}}visitKey_update(e){return e.key_action().accept(this)}visitKey_delete(e){return e.key_action().accept(this)}visitKey_action(e){let t=[...Array(e.getChildCount()).keys()].reduce((t,n)=>(t+=`${e.getChild(n).getText()} `,t),``);return t.slice(0,t.length-1)}visitQualified_name(e){let t=[e.colid().accept(this)];return e.indirection()&&t.push(...e.indirection().accept(this)),t}visitColid(e){return e.getChild(0).accept(this)}visitUnreserved_keyword(e){return e.getText()}visitIdentifier(e){if(e.Identifier())return e.Identifier().getText();if(e.QuotedIdentifier()){let t=e.QuotedIdentifier().getText();return t.slice(1,t.length-1)}if(e.UnicodeQuotedIdentifier()){let t=e.UnicodeQuotedIdentifier().getText();return t.slice(1,t.length-1)}return e.getChild(0).getText()}visitCol_name_keyword(e){return e.getChild(0).getText()}visitIndirection(e){return e.indirection_el().map(e=>e.accept(this))}visitIndirection_el(e){if(e.attr_name())return e.attr_name().accept(this)}visitAttr_name(e){return e.colLabel().accept(this)}visitColLabel(e){return e.EXIT()?e.EXIT().getText():e.getChild(0).accept(this)}visitReserved_keyword(e){return e.getText()}visitTypename(e){if(e.simpletypename()){let t=``,n=e.simpletypename().accept(this);return e.opt_array_bounds()&&(t=e.opt_array_bounds().accept(this)),e.ARRAY()&&(t=`[${e.iconst()?e.iconst().accept(this):``}]`),{type_name:n.type+t,schemaName:n.schemaName}}}visitSimpletypename(e){return e.generictype()?e.generictype().accept(this):e.character()?{type:e.character().accept(this),schemaName:null}:e.numeric()?{type:e.numeric().accept(this),schemaName:null}:e.constdatetime()?{type:e.constdatetime().accept(this),schemaName:null}:{type:e.getText(),schemaName:null}}visitConstdatetime(e){return`${e.getChild(0).getText()}${e.iconst()?`(${e.iconst().accept(this)})`:``}`}visitNumeric(e){return e.getText()}visitGenerictype(e){if(e.attrs()){let t=[e.getChild(0).getText(),...e.attrs().accept(this)];return{type:(0,a.last)(t),schemaName:t.length>1?t[t.length-2]:void 0}}let t=``;t=e.type_function_name()?e.type_function_name().accept(this):e.getText();let n=e.type_modifiers_();return{type:t+(n?n.getText():``),schemaName:null}}visitCharacter(e){let t=e.character_c().accept(this);return e.getChildCount()>1&&(t+=`(${e.iconst().accept(this)})`),t}visitCharacter_c(e){let t=[...Array(e.getChildCount()).keys()].reduce((t,n)=>(t+=`${e.getChild(n).getText()} `,t),``);return t.slice(0,t.length-1)}visitType_function_name(e){return e.getChild(0).accept(this)}visitType_func_name_keyword(e){return e.getText()}visitOpt_array_bounds(e){return e.getText()}visitIconst(e){return Number(e.getText()).toString()}visitIndexstmt(e){let t=e.relation_expr().accept(this),r=(0,a.last)(t),i=t.length>1?t[t.length-2]:void 0,o=null;e.index_name_()?o=e.index_name_().accept(this):e.name()&&(o=e.name().accept(this));let s=e.access_method_clause();return{pathName:{tableName:r,schemaName:i},index:new n.Index({name:o,unique:e.unique_()?.accept(this)??!1,type:s?s.accept(this):void 0,columns:e.index_params().accept(this)})}}visitIndex_name_(e){return e.name()?.accept(this)}visitName(e){return e.colid().accept(this)}visitAccess_method_clause(e){if(e.name())return e.name().accept(this)}visitRelation_expr(e){return e.qualified_name().accept(this)}visitIndex_params(e){return e.index_elem().map(e=>e.accept(this)).filter(e=>!!e)}visitIndex_elem(e){return e.colid()?{value:e.colid().accept(this),type:r.CONSTRAINT_TYPE.STRING}:{value:e.getText(),type:r.CONSTRAINT_TYPE.EXPRESSION}}visitUnique_(e){return!!e.UNIQUE()}visitAltertablestmt(e){if(e.TABLE()&&e.relation_expr()&&e.alter_table_cmds()){let t=e.relation_expr().accept(this),i=(0,a.last)(t),o=t.length>1?t[t.length-2]:void 0;return e.alter_table_cmds().accept(this).map(e=>{if(!e)return null;let t=null;switch(e.kind){case r.TABLE_CONSTRAINT_KIND.FK:t=c.REF,e.value.endpoints[0].tableName=i,e.value.endpoints[0].schemaName=o,this.data.refs.push(e.value);break;case r.TABLE_CONSTRAINT_KIND.CHECK:{let t=u(this.data.tables,o,i);if(!t)break;t.checks||=[],t.checks.push(e.value);break}case r.TABLE_CONSTRAINT_KIND.UNIQUE:case r.TABLE_CONSTRAINT_KIND.PK:case r.TABLE_CONSTRAINT_KIND.INDEX:{if(e.value.columns.length===0||!(e.value.pk||e.value.unique))break;t=e.kind;let r=u(this.data.tables,o,i);if(!r)break;if(e.value.columns.length===1&&(e.value.unique||e.value.pk)){let t=e.value.columns[0].value,n=r.fields.find(e=>e.name===t);if(n){n[e.value.unique?`unique`:`pk`]=!0;break}}let a=new n.Index({name:e.value.name,columns:e.value.columns,note:e.value.note,pk:e.value.pk,type:e.value.type,unique:e.value.unique});r.indexes.push(a);break}default:break}return{kind:t,value:e.value}}).filter(e=>e)}}visitAlter_table_cmds(e){return e.alter_table_cmd().map(e=>e.accept(this))}visitAlter_table_cmd(e){if(e.tableconstraint())return e.tableconstraint().accept(this)}visitCommentstmt(e){if(e.object_type_any_name()){let t=e.object_type_any_name().accept(this);if(!t)return;if(t===l.TABLE){let t=e.any_name().accept(this),n=(0,a.last)(t),r=t.length>1?t[t.length-2]:void 0,i=u(this.data.tables,r,n);if(!i)return;i.note={value:d(e.comment_text().accept(this))}}}if(e.COLUMN()){let t=e.any_name().accept(this),n=(0,a.last)(t),r=t.length>1?t[t.length-2]:void 0,i=t.length>2?t[t.length-3]:void 0,o=u(this.data.tables,i,r).fields.find(e=>e.name===n);if(!o)return;o.note={value:d(e.comment_text().accept(this))}}}visitComment_text(e){return e.NULL_P()?null:e.sconst().accept(this)}visitSconst(e){return e.anysconst().accept(this)}visitAnysconst(e){let t=e.getChild(0).getText();return t.slice(1,t.length-1)}visitObject_type_any_name(e){if(e.TABLE())return l.TABLE}visitAny_name(e){let t=[e.colid().accept(this)];return e.attrs()&&t.push(...e.attrs().accept(this)),t}visitAttrs(e){return e.attr_name().map(e=>e.accept(this))}visitDefinestmt(e){if(e.TYPE_P()&&e.enum_val_list_()){let t=e.any_name(0).accept(this),r=(0,a.last)(t),i=t.length>1?t[t.length-2]:void 0,o=e.enum_val_list_().accept(this).map(e=>({name:e}));if(!o)return;let s=new n.Enum({name:r,schemaName:i,values:o});this.data.enums.push(s.toJSON())}}visitEnum_val_list_(e){return e.enum_val_list()?.accept(this)}visitEnum_val_list(e){return e.sconst().map(e=>e.accept(this))}visitInsertstmt(e){let t=e.insert_target().accept(this),r=(0,a.last)(t),i=t.length>1?t[t.length-2]:void 0,{columns:o,values:s}=e.insert_rest().accept(this);if(o.length===0){let e=u(this.data.tables,i,r);e&&e.fields&&(o=e.fields.map(e=>e.name))}let c=new n.TableRecord({schemaName:i,tableName:r,columns:o,values:s});this.data.records.push(c)}visitInsert_target(e){return e.qualified_name().accept(this)}visitInsert_rest(e){return{columns:e.insert_column_list()?e.insert_column_list().accept(this):[],values:(e.selectstmt().accept(this)||[]).filter(e=>Array.isArray(e)).map(e=>e.filter(e=>e).map(({value:e,type:t})=>({value:e,type:t})))}}visitInsert_column_list(e){return e.insert_column_item().map(e=>e.accept(this))}visitInsert_column_item(e){return e.colid().accept(this)}visitSelectstmt(e){return e.select_no_parens()?e.select_no_parens().accept(this):null}visitSelect_no_parens(e){return e.select_clause().accept(this)}visitSelect_clause(e){let t=e.simple_select_intersect();return!t||t.length===0?null:t[0].accept(this)}visitSimple_select_intersect(e){let t=e.simple_select_pramary();return!t||t.length===0?null:t[0].accept(this)}visitSimple_select_pramary(e){return e.values_clause()?e.values_clause().accept(this):null}visitValues_clause(e){return e.expr_list().map(e=>(0,a.flattenDepth)(e.accept(this),21))}visitA_expr_collate(e){let[t,n,r={}]=e.a_expr_typecast().accept(this),{value:i,type:a}=t;return{value:i,type:a,...r}}};function p(e,n,r,i){if(!n)return;let c=new Set,l=(0,o.normalizeQualifiedName)(i??`public`,r),d=new s.ParseTreeListener;d.enterEveryRule=n=>{if(!(n instanceof t.default.Relation_exprContext))return;let s=n.accept(e),d=(0,a.last)(s),f=s.length>1?s[s.length-2]:void 0,p=(0,o.normalizeQualifiedName)(f??`public`,d);p!==l&&(c.has(p)||u(e.data.tables,f,d)&&(c.add(p),e.data.deps.push({edges:[{upstream:{schemaName:f,tableName:d,fieldNames:[]},downstream:{schemaName:i,tableName:r,fieldNames:[]}}],note:null,custom:{}})))},s.ParseTreeWalker.DEFAULT.walk(d,n)}exports.default=f;
+require("../../../../_virtual/_rolldown/runtime.cjs");const e=require("../../../../node_modules/lodash-es/flatten.cjs"),t=require("../../../../node_modules/lodash-es/last.cjs"),n=require("../../../../node_modules/lodash-es/flattenDepth.cjs"),r=require("../../../../node_modules/lodash-es/lodash.cjs");require("../../../../model_structure/config.cjs");const i=require("../../parsers/postgresql/PostgreSQLParserVisitor.cjs"),a=require("../../parsers/postgresql/PostgreSQLParser.cjs"),o=require("../AST.cjs"),s=require("../constants.cjs"),c=require("../helpers.cjs");let l=require("@dbml/parse"),u=require("antlr4");r.init_lodash();var d={REF:`ref`},f={TABLE:`table`},p=(e,t,n)=>{let r=t??`public`;return e.find(e=>(e.schemaName??`public`)===r&&e.name===n)},m=e=>e&&e.replaceAll(`''`,`'`),h=class extends i.default{constructor(){super(),this.data={schemas:[],tables:[],refs:[],deps:[],enums:[],tableGroups:[],aliases:[],project:{},records:[]}}visitViewstmt(e){let n=e.qualified_name().accept(this),r=t.default(n),i=n.length>1?n[n.length-2]:void 0,a=c.getOriginalText(e);this.data.tables.push({name:r,schemaName:i,fields:[],indexes:[],checks:[],note:null,headerColor:null,custom:{query:a,kind:`view`}}),g(this,e.selectstmt(),r,i)}visitCreatematviewstmt(e){let n=e.create_mv_target();if(!n)return;let r=n.qualified_name().accept(this),i=t.default(r),a=r.length>1?r[r.length-2]:void 0,o=c.getOriginalText(e);this.data.tables.push({name:i,schemaName:a,fields:[],indexes:[],checks:[],note:null,headerColor:null,custom:{query:o,kind:`materialized_view`,materialized:!0}}),g(this,e.selectstmt(),i,a)}visitRoot(e){return e.stmtblock().accept(this),this.data}visitStmtblock(e){e.stmtmulti().accept(this)}visitStmtmulti(e){e.stmt().map(e=>e.accept(this))}visitStmt(e){if(e.createstmt()){let t=e.createstmt().accept(this);t&&this.data.tables.push(t);return}if(e.indexstmt()){let t=e.indexstmt().accept(this),{tableName:n,schemaName:r}=t.pathName,i=p(this.data.tables,r,n);i&&i.indexes.push(t.index);return}if(e.altertablestmt()){e.altertablestmt().accept(this);return}if(e.commentstmt()){e.commentstmt().accept(this);return}if(e.definestmt()){e.definestmt().accept(this);return}e.insertstmt()&&e.insertstmt().accept(this)}visitCreatestmt(n){let r=n.qualified_name(0).accept(this),i=t.default(r),a=r.length>1?r[r.length-2]:void 0;if(!n.opttableelementlist())return;let[c,l,u,d]=n.opttableelementlist().accept(this).filter(e=>e).reduce((e,t)=>(t.kind===s.TABLE_CONSTRAINT_KIND.FIELD?e[0].push(t.value):t.kind===s.TABLE_CONSTRAINT_KIND.INDEX?e[1].push(t.value):t.kind===s.TABLE_CONSTRAINT_KIND.FK?e[2].push(t.value):t.kind===s.TABLE_CONSTRAINT_KIND.UNIQUE?e[1].push(t.value):t.kind===s.TABLE_CONSTRAINT_KIND.CHECK&&e[3].push(t.value),e),[[],[],[],[]]);return this.data.refs.push(...e.default(c.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(...u.map(e=>(e.endpoints[0].tableName=i,e.endpoints[0].schemaName=a,e))),new o.Table({name:i,schemaName:a,fields:c.map(e=>e.field),indexes:l,checks:d})}visitOpttableelementlist(e){return e.tableelementlist()?e.tableelementlist().accept(this):[]}visitTableelementlist(e){return e.tableelement().map(e=>e.accept(this))}visitTableelement(e){if(e.columnDef())return e.columnDef().accept(this);if(e.tableconstraint())return e.tableconstraint().accept(this)}visitTableconstraint(e){let t=e.name()?.accept(this),n=e.constraintelem().accept(this);if(n)return n.value.name=t,n}visitConstraintelem(e){if(e.PRIMARY())return{kind:s.TABLE_CONSTRAINT_KIND.INDEX,value:new o.Index({pk:!0,columns:e.columnlist().accept(this)})};if(e.FOREIGN()){let n=e.qualified_name().accept(this),r=t.default(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),c=e.key_actions(),u=c?c.accept(this):{onDelete:null,onUpdate:null};return{kind:s.TABLE_CONSTRAINT_KIND.FK,value:{endpoints:[{tableName:null,schemaName:null,fieldNames:a,relation:l.CARDINALITY_MANY},{tableName:r,schemaName:i,fieldNames:o,relation:l.CARDINALITY_MAYBE}],onDelete:u.onDelete,onUpdate:u.onUpdate}}}if(e.UNIQUE())return{kind:s.TABLE_CONSTRAINT_KIND.UNIQUE,value:new o.Index({unique:!0,columns:e.columnlist().accept(this)})};if(e.CHECK()){let t=c.getOriginalText(e.a_expr());return{kind:s.TABLE_CONSTRAINT_KIND.CHECK,value:{expression:t}}}}visitColumn_list_(e){return e.columnlist().accept(this)}visitColumnlist(e){return e.columnElem().map(e=>e.accept(this))}visitColumnElem(e){return{value:e.colid().accept(this),type:s.CONSTRAINT_TYPE.COLUMN}}visitColumnDef(e){let t=e.colid().accept(this),n=e.typename().accept(this),r=e.colquallist().accept(this),i=new Set([`serial`,`smallserial`,`bigserial`]),a=n.type_name.toLowerCase();return i.has(a)&&(r.increment=!0),{kind:s.TABLE_CONSTRAINT_KIND.FIELD,value:{field:new o.Field({name:t,type:n,...r}),inline_refs:r.inline_refs}}}visitColquallist(e){let t={inline_refs:[],checks:[]};return e.colconstraint().forEach(e=>{let n=e.accept(this);if(n){if(n.kind===s.COLUMN_CONSTRAINT_KIND.INLINE_REF){t.inline_refs.push(n.value);return}if(n.kind===s.COLUMN_CONSTRAINT_KIND.CHECK){t.checks.push(n.value);return}t[n.kind]=n.value}}),t}visitColconstraint(e){if(e.colconstraintelem())return e.colconstraintelem().accept(this)}visitColconstraintelem(e){if(e.NULL_P()){let t=!1;return e.NOT()&&(t=!0),{kind:s.COLUMN_CONSTRAINT_KIND.NOT_NULL,value:t}}if(e.UNIQUE())return{kind:s.COLUMN_CONSTRAINT_KIND.UNIQUE,value:!0};if(e.PRIMARY())return{kind:s.COLUMN_CONSTRAINT_KIND.PK,value:!0};if(e.DEFAULT())return{kind:s.COLUMN_CONSTRAINT_KIND.DEFAULT,value:e.b_expr().accept(this)};if(e.IDENTITY_P())return{kind:s.COLUMN_CONSTRAINT_KIND.INCREMENT,value:!0};if(e.CHECK()){let t=c.getOriginalText(e.a_expr());return{kind:s.COLUMN_CONSTRAINT_KIND.CHECK,value:{expression:t}}}if(e.REFERENCES()){let n=e.qualified_name().accept(this),r=t.default(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(),c=o?o.accept(this):{onDelete:null,onUpdate:null};return{kind:s.COLUMN_CONSTRAINT_KIND.INLINE_REF,value:{endpoints:[{tableName:null,schemaName:null,fieldNames:null,relation:l.CARDINALITY_MANY},{tableName:r,schemaName:i,fieldNames:a,relation:l.CARDINALITY_MAYBE}],onDelete:c.onDelete,onUpdate:c.onUpdate}}}}visitB_expr(e){return e.c_expr()?e.c_expr().accept(this):{value:e.getText(),type:s.DATA_TYPE.EXPRESSION}}visitC_expr_expr(e){return e.aexprconst()?e.aexprconst().accept(this):e.a_expr()?{value:e.a_expr().getText(),type:s.DATA_TYPE.EXPRESSION}:{value:e.getText(),type:s.DATA_TYPE.EXPRESSION}}visitC_expr_exists(e){return{value:e.getText(),type:s.DATA_TYPE.EXPRESSION}}visitC_expr_case(e){return{value:e.getText(),type:s.DATA_TYPE.EXPRESSION}}visitAexprconst(e){return e.sconst()&&e.getChildCount()===1?{value:e.sconst().accept(this),type:s.DATA_TYPE.STRING}:e.TRUE_P()||e.FALSE_P()||e.NULL_P()?{value:e.getText().toLowerCase(),type:s.DATA_TYPE.BOOLEAN}:e.iconst()?{value:e.iconst().accept(this),type:s.DATA_TYPE.NUMBER}:e.fconst()?{value:e.getText(),type:s.DATA_TYPE.NUMBER}:{value:e.getText(),type:s.DATA_TYPE.EXPRESSION}}visitKey_actions(e){let[t,n]=[null,null];return e.key_update()&&(n=e.key_update().accept(this)),e.key_delete()&&(t=e.key_delete().accept(this)),{onDelete:t,onUpdate:n}}visitKey_update(e){return e.key_action().accept(this)}visitKey_delete(e){return e.key_action().accept(this)}visitKey_action(e){let t=[...Array(e.getChildCount()).keys()].reduce((t,n)=>(t+=`${e.getChild(n).getText()} `,t),``);return t.slice(0,t.length-1)}visitQualified_name(e){let t=[e.colid().accept(this)];return e.indirection()&&t.push(...e.indirection().accept(this)),t}visitColid(e){return e.getChild(0).accept(this)}visitUnreserved_keyword(e){return e.getText()}visitIdentifier(e){if(e.Identifier())return e.Identifier().getText();if(e.QuotedIdentifier()){let t=e.QuotedIdentifier().getText();return t.slice(1,t.length-1)}if(e.UnicodeQuotedIdentifier()){let t=e.UnicodeQuotedIdentifier().getText();return t.slice(1,t.length-1)}return e.getChild(0).getText()}visitCol_name_keyword(e){return e.getChild(0).getText()}visitIndirection(e){return e.indirection_el().map(e=>e.accept(this))}visitIndirection_el(e){if(e.attr_name())return e.attr_name().accept(this)}visitAttr_name(e){return e.colLabel().accept(this)}visitColLabel(e){return e.EXIT()?e.EXIT().getText():e.getChild(0).accept(this)}visitReserved_keyword(e){return e.getText()}visitTypename(e){if(e.simpletypename()){let t=``,n=e.simpletypename().accept(this);return e.opt_array_bounds()&&(t=e.opt_array_bounds().accept(this)),e.ARRAY()&&(t=`[${e.iconst()?e.iconst().accept(this):``}]`),{type_name:n.type+t,schemaName:n.schemaName}}}visitSimpletypename(e){return e.generictype()?e.generictype().accept(this):e.character()?{type:e.character().accept(this),schemaName:null}:e.numeric()?{type:e.numeric().accept(this),schemaName:null}:e.constdatetime()?{type:e.constdatetime().accept(this),schemaName:null}:{type:e.getText(),schemaName:null}}visitConstdatetime(e){return`${e.getChild(0).getText()}${e.iconst()?`(${e.iconst().accept(this)})`:``}`}visitNumeric(e){return e.getText()}visitGenerictype(e){if(e.attrs()){let n=[e.getChild(0).getText(),...e.attrs().accept(this)];return{type:t.default(n),schemaName:n.length>1?n[n.length-2]:void 0}}let n=``;n=e.type_function_name()?e.type_function_name().accept(this):e.getText();let r=e.type_modifiers_();return{type:n+(r?r.getText():``),schemaName:null}}visitCharacter(e){let t=e.character_c().accept(this);return e.getChildCount()>1&&(t+=`(${e.iconst().accept(this)})`),t}visitCharacter_c(e){let t=[...Array(e.getChildCount()).keys()].reduce((t,n)=>(t+=`${e.getChild(n).getText()} `,t),``);return t.slice(0,t.length-1)}visitType_function_name(e){return e.getChild(0).accept(this)}visitType_func_name_keyword(e){return e.getText()}visitOpt_array_bounds(e){return e.getText()}visitIconst(e){return Number(e.getText()).toString()}visitIndexstmt(e){let n=e.relation_expr().accept(this),r=t.default(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 s=e.access_method_clause();return{pathName:{tableName:r,schemaName:i},index:new o.Index({name:a,unique:e.unique_()?.accept(this)??!1,type:s?s.accept(this):void 0,columns:e.index_params().accept(this)})}}visitIndex_name_(e){return e.name()?.accept(this)}visitName(e){return e.colid().accept(this)}visitAccess_method_clause(e){if(e.name())return e.name().accept(this)}visitRelation_expr(e){return e.qualified_name().accept(this)}visitIndex_params(e){return e.index_elem().map(e=>e.accept(this)).filter(e=>!!e)}visitIndex_elem(e){return e.colid()?{value:e.colid().accept(this),type:s.CONSTRAINT_TYPE.STRING}:{value:e.getText(),type:s.CONSTRAINT_TYPE.EXPRESSION}}visitUnique_(e){return!!e.UNIQUE()}visitAltertablestmt(e){if(e.TABLE()&&e.relation_expr()&&e.alter_table_cmds()){let n=e.relation_expr().accept(this),r=t.default(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 s.TABLE_CONSTRAINT_KIND.FK:t=d.REF,e.value.endpoints[0].tableName=r,e.value.endpoints[0].schemaName=i,this.data.refs.push(e.value);break;case s.TABLE_CONSTRAINT_KIND.CHECK:{let t=p(this.data.tables,i,r);if(!t)break;t.checks||=[],t.checks.push(e.value);break}case s.TABLE_CONSTRAINT_KIND.UNIQUE:case s.TABLE_CONSTRAINT_KIND.PK:case s.TABLE_CONSTRAINT_KIND.INDEX:{if(e.value.columns.length===0||!(e.value.pk||e.value.unique))break;t=e.kind;let n=p(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,r=n.fields.find(e=>e.name===t);if(r){r[e.value.unique?`unique`:`pk`]=!0;break}}let a=new o.Index({name:e.value.name,columns:e.value.columns,note:e.value.note,pk:e.value.pk,type:e.value.type,unique:e.value.unique});n.indexes.push(a);break}default:break}return{kind:t,value:e.value}}).filter(e=>e)}}visitAlter_table_cmds(e){return e.alter_table_cmd().map(e=>e.accept(this))}visitAlter_table_cmd(e){if(e.tableconstraint())return e.tableconstraint().accept(this)}visitCommentstmt(e){if(e.object_type_any_name()){let n=e.object_type_any_name().accept(this);if(!n)return;if(n===f.TABLE){let n=e.any_name().accept(this),r=t.default(n),i=n.length>1?n[n.length-2]:void 0,a=p(this.data.tables,i,r);if(!a)return;a.note={value:m(e.comment_text().accept(this))}}}if(e.COLUMN()){let n=e.any_name().accept(this),r=t.default(n),i=n.length>1?n[n.length-2]:void 0,a=n.length>2?n[n.length-3]:void 0,o=p(this.data.tables,a,i).fields.find(e=>e.name===r);if(!o)return;o.note={value:m(e.comment_text().accept(this))}}}visitComment_text(e){return e.NULL_P()?null:e.sconst().accept(this)}visitSconst(e){return e.anysconst().accept(this)}visitAnysconst(e){let t=e.getChild(0).getText();return t.slice(1,t.length-1)}visitObject_type_any_name(e){if(e.TABLE())return f.TABLE}visitAny_name(e){let t=[e.colid().accept(this)];return e.attrs()&&t.push(...e.attrs().accept(this)),t}visitAttrs(e){return e.attr_name().map(e=>e.accept(this))}visitDefinestmt(e){if(e.TYPE_P()&&e.enum_val_list_()){let n=e.any_name(0).accept(this),r=t.default(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 s=new o.Enum({name:r,schemaName:i,values:a});this.data.enums.push(s.toJSON())}}visitEnum_val_list_(e){return e.enum_val_list()?.accept(this)}visitEnum_val_list(e){return e.sconst().map(e=>e.accept(this))}visitInsertstmt(e){let n=e.insert_target().accept(this),r=t.default(n),i=n.length>1?n[n.length-2]:void 0,{columns:a,values:s}=e.insert_rest().accept(this);if(a.length===0){let e=p(this.data.tables,i,r);e&&e.fields&&(a=e.fields.map(e=>e.name))}let c=new o.TableRecord({schemaName:i,tableName:r,columns:a,values:s});this.data.records.push(c)}visitInsert_target(e){return e.qualified_name().accept(this)}visitInsert_rest(e){return{columns:e.insert_column_list()?e.insert_column_list().accept(this):[],values:(e.selectstmt().accept(this)||[]).filter(e=>Array.isArray(e)).map(e=>e.filter(e=>e).map(({value:e,type:t})=>({value:e,type:t})))}}visitInsert_column_list(e){return e.insert_column_item().map(e=>e.accept(this))}visitInsert_column_item(e){return e.colid().accept(this)}visitSelectstmt(e){return e.select_no_parens()?e.select_no_parens().accept(this):null}visitSelect_no_parens(e){return e.select_clause().accept(this)}visitSelect_clause(e){let t=e.simple_select_intersect();return!t||t.length===0?null:t[0].accept(this)}visitSimple_select_intersect(e){let t=e.simple_select_pramary();return!t||t.length===0?null:t[0].accept(this)}visitSimple_select_pramary(e){return e.values_clause()?e.values_clause().accept(this):null}visitValues_clause(e){return e.expr_list().map(e=>n.default(e.accept(this),21))}visitA_expr_collate(e){let[t,n,r={}]=e.a_expr_typecast().accept(this),{value:i,type:a}=t;return{value:i,type:a,...r}}};function g(e,n,r,i){if(!n)return;let o=new Set,s=(0,l.normalizeQualifiedName)(i??`public`,r),c=new u.ParseTreeListener;c.enterEveryRule=n=>{if(!(n instanceof a.default.Relation_exprContext))return;let c=n.accept(e),u=t.default(c),d=c.length>1?c[c.length-2]:void 0,f=(0,l.normalizeQualifiedName)(d??`public`,u);f!==s&&(o.has(f)||p(e.data.tables,d,u)&&(o.add(f),e.data.deps.push({edges:[{upstream:{schemaName:d,tableName:u,fieldNames:[]},downstream:{schemaName:i,tableName:r,fieldNames:[]}}],note:null,custom:{}})))},u.ParseTreeWalker.DEFAULT.walk(c,n)}exports.default=h;