@dbml/core 3.0.0 → 3.1.1

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

@@ -8,6 +8,7 @@ exports["default"] = void 0;
 var _lodash = require("lodash");
 var _PostgreSQLParserVisitor = _interopRequireDefault(require("../../parsers/postgresql/PostgreSQLParserVisitor"));
 var _AST = require("../AST");
+var _constants = require("../constants");
 function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { "default": obj }; }
 function ownKeys(object, enumerableOnly) { var keys = Object.keys(object); if (Object.getOwnPropertySymbols) { var symbols = Object.getOwnPropertySymbols(object); enumerableOnly && (symbols = symbols.filter(function (sym) { return Object.getOwnPropertyDescriptor(object, sym).enumerable; })), keys.push.apply(keys, symbols); } return keys; }
 function _objectSpread(target) { for (var i = 1; i < arguments.length; i++) { var source = null != arguments[i] ? arguments[i] : {}; i % 2 ? ownKeys(Object(source), !0).forEach(function (key) { _defineProperty(target, key, source[key]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(target, Object.getOwnPropertyDescriptors(source)) : ownKeys(Object(source)).forEach(function (key) { Object.defineProperty(target, key, Object.getOwnPropertyDescriptor(source, key)); }); } return target; }
@@ -37,22 +38,9 @@ function _getPrototypeOf(o) { _getPrototypeOf = Object.setPrototypeOf ? Object.g
 var COMMAND_KIND = {
   REF: 'ref'
 };
-var TABLE_CONSTRAINT_KIND = {
-  FIELD: 'field',
-  INDEX: 'index',
-  FK: 'fk',
-  UNIQUE: 'unique'
-};
 var COMMENT_OBJECT_TYPE = {
   TABLE: 'table'
 };
-
-// legacy - for compatibility with model_structure
-var CONSTRAINT_TYPE = {
-  COLUMN: 'column',
-  STRING: 'string',
-  EXPRESSION: 'expression'
-};
 var findTable = function findTable(tables, schemaName, tableName) {
   var realSchemaName = schemaName || 'public';
   var table = tables.find(function (table) {
@@ -157,7 +145,7 @@ var PostgresASTGen = /*#__PURE__*/function (_PostgreSQLParserVisi) {
         return e;
       });
       var _tableElements$reduce = tableElements.reduce(function (acc, ele) {
-          if (ele.kind === TABLE_CONSTRAINT_KIND.FIELD) acc[0].push(ele.value);else if (ele.kind === TABLE_CONSTRAINT_KIND.INDEX) acc[1].push(ele.value);else if (ele.kind === TABLE_CONSTRAINT_KIND.FK) acc[2].push(ele.value);else if (ele.kind === TABLE_CONSTRAINT_KIND.UNIQUE) acc[1].push(ele.value);
+          if (ele.kind === _constants.TABLE_CONSTRAINT_KIND.FIELD) acc[0].push(ele.value);else if (ele.kind === _constants.TABLE_CONSTRAINT_KIND.INDEX) acc[1].push(ele.value);else if (ele.kind === _constants.TABLE_CONSTRAINT_KIND.FK) acc[2].push(ele.value);else if (ele.kind === _constants.TABLE_CONSTRAINT_KIND.UNIQUE) acc[1].push(ele.value);
           return acc;
         }, [[], [], []]),
         _tableElements$reduce2 = _slicedToArray(_tableElements$reduce, 3),
@@ -237,7 +225,7 @@ var PostgresASTGen = /*#__PURE__*/function (_PostgreSQLParserVisi) {
     value: function visitConstraintelem(ctx) {
       if (ctx.PRIMARY()) {
         return {
-          kind: TABLE_CONSTRAINT_KIND.INDEX,
+          kind: _constants.TABLE_CONSTRAINT_KIND.INDEX,
           value: new _AST.Index({
             pk: true,
             columns: ctx.columnlist().accept(this)
@@ -262,7 +250,7 @@ var PostgresASTGen = /*#__PURE__*/function (_PostgreSQLParserVisi) {
         });
         var actions = ctx.key_actions().accept(this);
         return {
-          kind: TABLE_CONSTRAINT_KIND.FK,
+          kind: _constants.TABLE_CONSTRAINT_KIND.FK,
           value: {
             endpoints: [{
               tableName: null,
@@ -282,7 +270,7 @@ var PostgresASTGen = /*#__PURE__*/function (_PostgreSQLParserVisi) {
       }
       if (ctx.UNIQUE()) {
         return {
-          kind: TABLE_CONSTRAINT_KIND.UNIQUE,
+          kind: _constants.TABLE_CONSTRAINT_KIND.UNIQUE,
           value: new _AST.Index({
             unique: true,
             columns: ctx.columnlist().accept(this)
@@ -315,7 +303,7 @@ var PostgresASTGen = /*#__PURE__*/function (_PostgreSQLParserVisi) {
     value: function visitColumnElem(ctx) {
       return {
         value: ctx.colid().accept(this),
-        type: CONSTRAINT_TYPE.COLUMN
+        type: _constants.CONSTRAINT_TYPE.COLUMN
       };
     }
 
@@ -330,7 +318,7 @@ var PostgresASTGen = /*#__PURE__*/function (_PostgreSQLParserVisi) {
       var columnTypeName = type.type_name.toLowerCase();
       if (serialIncrementType.has(columnTypeName)) contraints.increment = true;
       return {
-        kind: TABLE_CONSTRAINT_KIND.FIELD,
+        kind: _constants.TABLE_CONSTRAINT_KIND.FIELD,
         value: {
           field: new _AST.Field(_objectSpread({
             name: name,
@@ -352,7 +340,7 @@ var PostgresASTGen = /*#__PURE__*/function (_PostgreSQLParserVisi) {
       ctx.colconstraint().forEach(function (c) {
         var constraint = c.accept(_this5);
         if (!constraint) return;
-        if (constraint.kind === 'inline_ref') {
+        if (constraint.kind === _constants.COLUMN_CONSTRAINT_KIND.INLINE_REF) {
           r.inline_refs.push(constraint.value);
           return;
         }
@@ -389,29 +377,37 @@ var PostgresASTGen = /*#__PURE__*/function (_PostgreSQLParserVisi) {
     key: "visitColconstraintelem",
     value: function visitColconstraintelem(ctx) {
       if (ctx.NULL_P()) {
-        var not_null = false;
-        if (ctx.NOT()) not_null = true;
+        var notNull = false;
+        if (ctx.NOT()) notNull = true;
         return {
-          kind: 'not_null',
-          value: not_null
+          kind: _constants.COLUMN_CONSTRAINT_KIND.NOT_NULL,
+          value: notNull
+        };
+      }
+      if (ctx.UNIQUE()) {
+        return {
+          kind: _constants.COLUMN_CONSTRAINT_KIND.UNIQUE,
+          value: true
+        };
+      }
+      if (ctx.PRIMARY()) {
+        return {
+          kind: _constants.COLUMN_CONSTRAINT_KIND.PK,
+          value: true
+        };
+      }
+      if (ctx.DEFAULT()) {
+        return {
+          kind: _constants.COLUMN_CONSTRAINT_KIND.DEFAULT,
+          value: ctx.b_expr().accept(this)
+        };
+      }
+      if (ctx.IDENTITY_P()) {
+        return {
+          kind: _constants.COLUMN_CONSTRAINT_KIND.INCREMENT,
+          value: true
         };
       }
-      if (ctx.UNIQUE()) return {
-        kind: 'unique',
-        value: true
-      };
-      if (ctx.PRIMARY()) return {
-        kind: 'pk',
-        value: true
-      };
-      if (ctx.DEFAULT()) return {
-        kind: 'dbdefault',
-        value: ctx.b_expr().accept(this)
-      };
-      if (ctx.IDENTITY_P()) return {
-        kind: 'increment',
-        value: true
-      };
       if (ctx.REFERENCES()) {
         var _ctx$opt_column_list$2;
         var names = ctx.qualified_name().accept(this);
@@ -422,7 +418,7 @@ var PostgresASTGen = /*#__PURE__*/function (_PostgreSQLParserVisi) {
         });
         var actions = ctx.key_actions().accept(this);
         return {
-          kind: 'inline_ref',
+          kind: _constants.COLUMN_CONSTRAINT_KIND.INLINE_REF,
           value: {
             endpoints: [{
               tableName: null,
@@ -449,7 +445,7 @@ var PostgresASTGen = /*#__PURE__*/function (_PostgreSQLParserVisi) {
       if (ctx.c_expr()) return ctx.c_expr().accept(this);
       return {
         value: ctx.getText(),
-        type: 'expression'
+        type: _constants.DATA_TYPE.EXPRESSION
       };
     }
 
@@ -460,11 +456,11 @@ var PostgresASTGen = /*#__PURE__*/function (_PostgreSQLParserVisi) {
       if (ctx.aexprconst()) return ctx.aexprconst().accept(this);
       if (ctx.a_expr()) return {
         value: ctx.a_expr().getText(),
-        type: 'expression'
+        type: _constants.DATA_TYPE.EXPRESSION
       };
       return {
         value: ctx.getText(),
-        type: 'expression'
+        type: _constants.DATA_TYPE.EXPRESSION
       };
     }
   }, {
@@ -472,7 +468,7 @@ var PostgresASTGen = /*#__PURE__*/function (_PostgreSQLParserVisi) {
     value: function visitC_expr_exists(ctx) {
       return {
         value: ctx.getText(),
-        type: 'expression'
+        type: _constants.DATA_TYPE.EXPRESSION
       };
     }
   }, {
@@ -480,7 +476,7 @@ var PostgresASTGen = /*#__PURE__*/function (_PostgreSQLParserVisi) {
     value: function visitC_expr_case(ctx) {
       return {
         value: ctx.getText(),
-        type: 'expression'
+        type: _constants.DATA_TYPE.EXPRESSION
       };
     }
 
@@ -488,21 +484,27 @@ var PostgresASTGen = /*#__PURE__*/function (_PostgreSQLParserVisi) {
   }, {
     key: "visitAexprconst",
     value: function visitAexprconst(ctx) {
-      if (ctx.sconst() && ctx.getChildCount() === 1) return {
-        value: ctx.sconst().accept(this),
-        type: 'string'
-      };
-      if (ctx.TRUE_P() || ctx.FALSE_P() || ctx.NULL_P()) return {
-        value: ctx.getText(),
-        type: 'boolean'
-      };
-      if (ctx.iconst() || ctx.fconst) return {
-        value: ctx.getText(),
-        type: 'number'
-      };
+      if (ctx.sconst() && ctx.getChildCount() === 1) {
+        return {
+          value: ctx.sconst().accept(this),
+          type: _constants.DATA_TYPE.STRING
+        };
+      }
+      if (ctx.TRUE_P() || ctx.FALSE_P() || ctx.NULL_P()) {
+        return {
+          value: ctx.getText(),
+          type: _constants.DATA_TYPE.BOOLEAN
+        };
+      }
+      if (ctx.iconst() || ctx.fconst) {
+        return {
+          value: ctx.getText(),
+          type: _constants.DATA_TYPE.NUMBER
+        };
+      }
       return {
         value: ctx.getText(),
-        type: 'expression'
+        type: _constants.DATA_TYPE.EXPRESSION
       };
     }
 
@@ -875,12 +877,12 @@ var PostgresASTGen = /*#__PURE__*/function (_PostgreSQLParserVisi) {
       if (ctx.colid()) {
         return {
           value: ctx.colid().accept(this),
-          type: CONSTRAINT_TYPE.STRING
+          type: _constants.CONSTRAINT_TYPE.STRING
         };
       }
       return {
         value: ctx.getText(),
-        type: CONSTRAINT_TYPE.EXPRESSION
+        type: _constants.CONSTRAINT_TYPE.EXPRESSION
       };
     }
 
@@ -915,7 +917,7 @@ var PostgresASTGen = /*#__PURE__*/function (_PostgreSQLParserVisi) {
           if (!cmd) return;
           var kind = null;
           switch (cmd.kind) {
-            case TABLE_CONSTRAINT_KIND.FK:
+            case _constants.TABLE_CONSTRAINT_KIND.FK:
               kind = COMMAND_KIND.REF;
               cmd.value.endpoints[0].tableName = tableName;
               cmd.value.endpoints[0].schemaName = schemaName;