@dbml/core 3.14.1 → 4.0.0

package/lib/parse/ANTLR/ASTGeneration/mysql/MySQLASTGen.js

@@ -9,6 +9,7 @@ var _lodash = require("lodash");
 var _MySqlParserVisitor = _interopRequireDefault(require("../../parsers/mysql/MySqlParserVisitor"));
 var _AST = require("../AST");
 var _constants = require("../constants");
+var _helpers = require("../helpers");
 function _interopRequireDefault(e) { return e && e.__esModule ? e : { "default": e }; }
 function ownKeys(e, r) { var t = Object.keys(e); if (Object.getOwnPropertySymbols) { var o = Object.getOwnPropertySymbols(e); r && (o = o.filter(function (r) { return Object.getOwnPropertyDescriptor(e, r).enumerable; })), t.push.apply(t, o); } return t; }
 function _objectSpread(e) { for (var r = 1; r < arguments.length; r++) { var t = null != arguments[r] ? arguments[r] : {}; r % 2 ? ownKeys(Object(t), !0).forEach(function (r) { _defineProperty(e, r, t[r]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(e, Object.getOwnPropertyDescriptors(t)) : ownKeys(Object(t)).forEach(function (r) { Object.defineProperty(e, r, Object.getOwnPropertyDescriptor(t, r)); }); } return e; }
@@ -40,7 +41,8 @@ var TABLE_OPTIONS_KIND = {
 };
 var ALTER_KIND = {
   ADD_PK: 'add_pk',
-  ADD_FK: 'add_fk'
+  ADD_FK: 'add_fk',
+  ADD_CHECK: 'add_check'
 };
 var INDEX_OPTION_KIND = {
   TYPE: 'type'
@@ -143,18 +145,19 @@ var MySQLASTGen = exports["default"] = /*#__PURE__*/function (_MySQLParserVisito
           definitions = createTableResult.definitions,
           options = createTableResult.options;
         var _definitions$reduce = definitions.reduce(function (acc, ele) {
-            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.UNIQUE) 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.PK) {
+            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.UNIQUE) 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.CHECK) acc[4].push(ele.value);else if (ele.kind === _constants.TABLE_CONSTRAINT_KIND.PK) {
               /** @type {Index} */
               var index = ele.value;
               if (index.columns.length > 1) acc[1].push(ele.value);else acc[3] = index;
             }
             return acc;
-          }, [[], [], [], null]),
-          _definitions$reduce2 = _slicedToArray(_definitions$reduce, 4),
+          }, [[], [], [], null, []]),
+          _definitions$reduce2 = _slicedToArray(_definitions$reduce, 5),
           fieldsData = _definitions$reduce2[0],
           indexes = _definitions$reduce2[1],
           tableRefs = _definitions$reduce2[2],
-          singlePkIndex = _definitions$reduce2[3];
+          singlePkIndex = _definitions$reduce2[3],
+          tableConstraints = _definitions$reduce2[4];
         var inlineRefsOfFields = fieldsData.map(function (fieldData) {
           var _field$type$type_name;
           var field = fieldData.field,
@@ -198,7 +201,8 @@ var MySQLASTGen = exports["default"] = /*#__PURE__*/function (_MySQLParserVisito
           fields: fieldsData.map(function (fd) {
             return fd.field;
           }),
-          indexes: indexes
+          indexes: indexes,
+          constraints: tableConstraints
         }, tableOptions));
         if (singlePkIndex) {
           var field = table.fields.find(function (f) {
@@ -408,7 +412,8 @@ var MySQLASTGen = exports["default"] = /*#__PURE__*/function (_MySQLParserVisito
       var _this9 = this;
       var type = ctx.dataType().accept(this);
       var constraints = {
-        inlineRefs: []
+        inlineRefs: [],
+        constraints: []
       };
       ctx.columnConstraint().forEach(function (c) {
         var constraint = c.accept(_this9);
@@ -417,6 +422,10 @@ var MySQLASTGen = exports["default"] = /*#__PURE__*/function (_MySQLParserVisito
           constraints.inlineRefs.push(constraint.value);
           return;
         }
+        if (constraint.kind === _constants.COLUMN_CONSTRAINT_KIND.CHECK) {
+          constraints.constraints.push(constraint.value);
+          return;
+        }
         constraints[constraint.kind] = constraint.value;
       });
       return {
@@ -664,6 +673,22 @@ var MySQLASTGen = exports["default"] = /*#__PURE__*/function (_MySQLParserVisito
       };
     }
 
+    // checkColumnConstraint: (CONSTRAINT name = uid?)? CHECK '(' expression ')'
+  }, {
+    key: "visitCheckColumnConstraint",
+    value: function visitCheckColumnConstraint(ctx) {
+      var _ctx$name;
+      var name = (_ctx$name = ctx.name) === null || _ctx$name === void 0 ? void 0 : _ctx$name.accept(this);
+      var expression = (0, _helpers.getOriginalText)(ctx.expression());
+      return {
+        kind: _constants.COLUMN_CONSTRAINT_KIND.CHECK,
+        value: {
+          expression: expression,
+          name: name
+        }
+      };
+    }
+
     // NULL_LITERAL | CAST '(' expression AS convertedDataType ')' | unaryOperator? constant
     // | currentTimestamp (ON UPDATE currentTimestamp)? | '(' expression ')' | '(' fullId ')'
   }, {
@@ -863,12 +888,12 @@ var MySQLASTGen = exports["default"] = /*#__PURE__*/function (_MySQLParserVisito
   }, {
     key: "visitPrimaryKeyTableConstraint",
     value: function visitPrimaryKeyTableConstraint(ctx) {
-      var _ctx$name,
+      var _ctx$name2,
         _ctx$index,
         _ctx$indexType,
         _ctx$indexOption,
         _this1 = this;
-      var name = ((_ctx$name = ctx.name) === null || _ctx$name === void 0 ? void 0 : _ctx$name.accept(this)) || ((_ctx$index = ctx.index) === null || _ctx$index === void 0 ? void 0 : _ctx$index.accept(this));
+      var name = ((_ctx$name2 = ctx.name) === null || _ctx$name2 === void 0 ? void 0 : _ctx$name2.accept(this)) || ((_ctx$index = ctx.index) === null || _ctx$index === void 0 ? void 0 : _ctx$index.accept(this));
       var type = (_ctx$indexType = ctx.indexType()) === null || _ctx$indexType === void 0 ? void 0 : _ctx$indexType.accept(this);
       if ((_ctx$indexOption = ctx.indexOption()) !== null && _ctx$indexOption !== void 0 && _ctx$indexOption.length) {
         var indexOptions = ctx.indexOption().map(function (io) {
@@ -893,12 +918,12 @@ var MySQLASTGen = exports["default"] = /*#__PURE__*/function (_MySQLParserVisito
   }, {
     key: "visitUniqueKeyTableConstraint",
     value: function visitUniqueKeyTableConstraint(ctx) {
-      var _ctx$name2,
+      var _ctx$name3,
         _ctx$index2,
         _ctx$indexType2,
         _ctx$indexOption2,
         _this10 = this;
-      var name = ((_ctx$name2 = ctx.name) === null || _ctx$name2 === void 0 ? void 0 : _ctx$name2.accept(this)) || ((_ctx$index2 = ctx.index) === null || _ctx$index2 === void 0 ? void 0 : _ctx$index2.accept(this));
+      var name = ((_ctx$name3 = ctx.name) === null || _ctx$name3 === void 0 ? void 0 : _ctx$name3.accept(this)) || ((_ctx$index2 = ctx.index) === null || _ctx$index2 === void 0 ? void 0 : _ctx$index2.accept(this));
       var type = (_ctx$indexType2 = ctx.indexType()) === null || _ctx$indexType2 === void 0 ? void 0 : _ctx$indexType2.accept(this);
       if ((_ctx$indexOption2 = ctx.indexOption()) !== null && _ctx$indexOption2 !== void 0 && _ctx$indexOption2.length) {
         var indexOptions = ctx.indexOption().map(function (io) {
@@ -923,9 +948,9 @@ var MySQLASTGen = exports["default"] = /*#__PURE__*/function (_MySQLParserVisito
   }, {
     key: "visitForeignKeyTableConstraint",
     value: function visitForeignKeyTableConstraint(ctx) {
-      var _ctx$name3, _ctx$index3;
+      var _ctx$name4, _ctx$index3;
       var ref = ctx.referenceDefinition().accept(this);
-      ref.name = ((_ctx$name3 = ctx.name) === null || _ctx$name3 === void 0 ? void 0 : _ctx$name3.accept(this)) || ((_ctx$index3 = ctx.index) === null || _ctx$index3 === void 0 ? void 0 : _ctx$index3.accept(this));
+      ref.name = ((_ctx$name4 = ctx.name) === null || _ctx$name4 === void 0 ? void 0 : _ctx$name4.accept(this)) || ((_ctx$index3 = ctx.index) === null || _ctx$index3 === void 0 ? void 0 : _ctx$index3.accept(this));
       ref.endpoints[0].fieldNames = ctx.indexColumnNames().accept(this).map(function (icn) {
         return icn.value;
       });
@@ -934,10 +959,21 @@ var MySQLASTGen = exports["default"] = /*#__PURE__*/function (_MySQLParserVisito
         value: ref
       };
     }
+
+    // checkTableConstraint: (CONSTRAINT name = uid?)? CHECK '(' expression ')'
   }, {
     key: "visitCheckTableConstraint",
-    value: function visitCheckTableConstraint() {
-      // ignored
+    value: function visitCheckTableConstraint(ctx) {
+      var _ctx$name5;
+      var name = (_ctx$name5 = ctx.name) === null || _ctx$name5 === void 0 ? void 0 : _ctx$name5.accept(this);
+      var expression = (0, _helpers.getOriginalText)(ctx.expression());
+      return {
+        kind: _constants.TABLE_CONSTRAINT_KIND.CHECK,
+        value: {
+          expression: expression,
+          name: name
+        }
+      };
     }
 
     // USING (BTREE | HASH)
@@ -1054,6 +1090,8 @@ var MySQLASTGen = exports["default"] = /*#__PURE__*/function (_MySQLParserVisito
           ref.endpoints[0].schemaName = schemaName;
           ref.endpoints[0].tableName = tableName;
           _this13.data.refs.push(ref);
+        } else if (alter.kind === ALTER_KIND.ADD_CHECK) {
+          table.constraints.push(alter.value);
         }
         return null;
       });
@@ -1067,12 +1105,12 @@ var MySQLASTGen = exports["default"] = /*#__PURE__*/function (_MySQLParserVisito
   }, {
     key: "visitAlterByAddPrimaryKey",
     value: function visitAlterByAddPrimaryKey(ctx) {
-      var _ctx$name4,
+      var _ctx$name6,
         _ctx$index4,
         _ctx$indexType4,
         _ctx$indexOption5,
         _this14 = this;
-      var name = ((_ctx$name4 = ctx.name) === null || _ctx$name4 === void 0 ? void 0 : _ctx$name4.accept(this)) || ((_ctx$index4 = ctx.index) === null || _ctx$index4 === void 0 ? void 0 : _ctx$index4.accept(this));
+      var name = ((_ctx$name6 = ctx.name) === null || _ctx$name6 === void 0 ? void 0 : _ctx$name6.accept(this)) || ((_ctx$index4 = ctx.index) === null || _ctx$index4 === void 0 ? void 0 : _ctx$index4.accept(this));
       var type = (_ctx$indexType4 = ctx.indexType()) === null || _ctx$indexType4 === void 0 ? void 0 : _ctx$indexType4.accept(this);
       if ((_ctx$indexOption5 = ctx.indexOption()) !== null && _ctx$indexOption5 !== void 0 && _ctx$indexOption5.length) {
         var indexOptions = ctx.indexOption().map(function (io) {
@@ -1099,9 +1137,9 @@ var MySQLASTGen = exports["default"] = /*#__PURE__*/function (_MySQLParserVisito
   }, {
     key: "visitAlterByAddForeignKey",
     value: function visitAlterByAddForeignKey(ctx) {
-      var _ctx$name5, _ctx$index5;
+      var _ctx$name7, _ctx$index5;
       var ref = ctx.referenceDefinition().accept(this);
-      ref.name = ((_ctx$name5 = ctx.name) === null || _ctx$name5 === void 0 ? void 0 : _ctx$name5.accept(this)) || ((_ctx$index5 = ctx.index) === null || _ctx$index5 === void 0 ? void 0 : _ctx$index5.accept(this));
+      ref.name = ((_ctx$name7 = ctx.name) === null || _ctx$name7 === void 0 ? void 0 : _ctx$name7.accept(this)) || ((_ctx$index5 = ctx.index) === null || _ctx$index5 === void 0 ? void 0 : _ctx$index5.accept(this));
       ref.endpoints[0].fieldNames = ctx.indexColumnNames().accept(this).map(function (icn) {
         return icn.value;
       });
@@ -1111,6 +1149,22 @@ var MySQLASTGen = exports["default"] = /*#__PURE__*/function (_MySQLParserVisito
       };
     }
 
+    // ADD (CONSTRAINT name = uid?)? CHECK '(' expression ')'
+  }, {
+    key: "visitAlterByAddCheckTableConstraint",
+    value: function visitAlterByAddCheckTableConstraint(ctx) {
+      var _ctx$name8;
+      var name = (_ctx$name8 = ctx.name) === null || _ctx$name8 === void 0 ? void 0 : _ctx$name8.accept(this);
+      var expression = (0, _helpers.getOriginalText)(ctx.expression());
+      return {
+        kind: ALTER_KIND.ADD_CHECK,
+        value: {
+          expression: expression,
+          name: name
+        }
+      };
+    }
+
     // CREATE intimeAction = (ONLINE | OFFLINE)? indexCategory = (UNIQUE | FULLTEXT | SPATIAL)? INDEX uid indexType? ON tableName indexColumnNames
     // indexOption* (ALGORITHM EQUAL_SYMBOL? algType = (DEFAULT | INPLACE | COPY) | LOCK EQUAL_SYMBOL? lockType = (DEFAULT | NONE | SHARED | EXCLUSIVE))*
   }, {

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

@@ -9,6 +9,7 @@ var _lodash = require("lodash");
 var _PostgreSQLParserVisitor = _interopRequireDefault(require("../../parsers/postgresql/PostgreSQLParserVisitor"));
 var _AST = require("../AST");
 var _constants = require("../constants");
+var _helpers = require("../helpers");
 function _interopRequireDefault(e) { return e && e.__esModule ? e : { "default": e }; }
 function ownKeys(e, r) { var t = Object.keys(e); if (Object.getOwnPropertySymbols) { var o = Object.getOwnPropertySymbols(e); r && (o = o.filter(function (r) { return Object.getOwnPropertyDescriptor(e, r).enumerable; })), t.push.apply(t, o); } return t; }
 function _objectSpread(e) { for (var r = 1; r < arguments.length; r++) { var t = null != arguments[r] ? arguments[r] : {}; r % 2 ? ownKeys(Object(t), !0).forEach(function (r) { _defineProperty(e, r, t[r]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(e, Object.getOwnPropertyDescriptors(t)) : ownKeys(Object(t)).forEach(function (r) { Object.defineProperty(e, r, Object.getOwnPropertyDescriptor(t, r)); }); } return e; }
@@ -157,13 +158,14 @@ var PostgresASTGen = exports["default"] = /*#__PURE__*/function (_PostgreSQLPars
         return e;
       });
       var _tableElements$reduce = tableElements.reduce(function (acc, ele) {
-          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);
+          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);else if (ele.kind === _constants.TABLE_CONSTRAINT_KIND.CHECK) acc[3].push(ele.value);
           return acc;
-        }, [[], [], []]),
-        _tableElements$reduce2 = _slicedToArray(_tableElements$reduce, 3),
+        }, [[], [], [], []]),
+        _tableElements$reduce2 = _slicedToArray(_tableElements$reduce, 4),
         fieldsData = _tableElements$reduce2[0],
         indexes = _tableElements$reduce2[1],
-        tableRefs = _tableElements$reduce2[2];
+        tableRefs = _tableElements$reduce2[2],
+        tableConstraints = _tableElements$reduce2[3];
       (_this$data$refs = this.data.refs).push.apply(_this$data$refs, _toConsumableArray((0, _lodash.flatten)(fieldsData.map(function (fieldData) {
         return fieldData.inline_refs.map(function (inlineRef) {
           inlineRef.endpoints[0].tableName = tableName;
@@ -183,7 +185,8 @@ var PostgresASTGen = exports["default"] = /*#__PURE__*/function (_PostgreSQLPars
         fields: fieldsData.map(function (fd) {
           return fd.field;
         }),
-        indexes: indexes
+        indexes: indexes,
+        constraints: tableConstraints
       });
     }
 
@@ -288,6 +291,15 @@ var PostgresASTGen = exports["default"] = /*#__PURE__*/function (_PostgreSQLPars
           })
         };
       }
+      if (ctx.CHECK()) {
+        var expression = (0, _helpers.getOriginalText)(ctx.a_expr());
+        return {
+          kind: _constants.TABLE_CONSTRAINT_KIND.CHECK,
+          value: {
+            expression: expression
+          }
+        };
+      }
     }
 
     // OPEN_PAREN columnlist CLOSE_PAREN |
@@ -324,18 +336,18 @@ var PostgresASTGen = exports["default"] = /*#__PURE__*/function (_PostgreSQLPars
     value: function visitColumnDef(ctx) {
       var name = ctx.colid().accept(this);
       var type = ctx.typename().accept(this);
-      var contraints = ctx.colquallist().accept(this);
+      var constraints = ctx.colquallist().accept(this);
       var serialIncrementType = new Set(['serial', 'smallserial', 'bigserial']);
       var columnTypeName = type.type_name.toLowerCase();
-      if (serialIncrementType.has(columnTypeName)) contraints.increment = true;
+      if (serialIncrementType.has(columnTypeName)) constraints.increment = true;
       return {
         kind: _constants.TABLE_CONSTRAINT_KIND.FIELD,
         value: {
           field: new _AST.Field(_objectSpread({
             name: name,
             type: type
-          }, contraints)),
-          inline_refs: contraints.inline_refs
+          }, constraints)),
+          inline_refs: constraints.inline_refs
         }
       };
     }
@@ -346,7 +358,8 @@ var PostgresASTGen = exports["default"] = /*#__PURE__*/function (_PostgreSQLPars
     value: function visitColquallist(ctx) {
       var _this5 = this;
       var r = {
-        inline_refs: []
+        inline_refs: [],
+        constraints: []
       };
       ctx.colconstraint().forEach(function (c) {
         var constraint = c.accept(_this5);
@@ -355,6 +368,10 @@ var PostgresASTGen = exports["default"] = /*#__PURE__*/function (_PostgreSQLPars
           r.inline_refs.push(constraint.value);
           return;
         }
+        if (constraint.kind === _constants.COLUMN_CONSTRAINT_KIND.CHECK) {
+          r.constraints.push(constraint.value);
+          return;
+        }
         r[constraint.kind] = constraint.value;
       });
       return r;
@@ -419,6 +436,15 @@ var PostgresASTGen = exports["default"] = /*#__PURE__*/function (_PostgreSQLPars
           value: true
         };
       }
+      if (ctx.CHECK()) {
+        var expression = (0, _helpers.getOriginalText)(ctx.a_expr());
+        return {
+          kind: _constants.COLUMN_CONSTRAINT_KIND.CHECK,
+          value: {
+            expression: expression
+          }
+        };
+      }
       if (ctx.REFERENCES()) {
         var _ctx$opt_column_list$2;
         var names = ctx.qualified_name().accept(this);
@@ -942,6 +968,14 @@ var PostgresASTGen = exports["default"] = /*#__PURE__*/function (_PostgreSQLPars
               cmd.value.endpoints[0].schemaName = schemaName;
               _this8.data.refs.push(cmd.value);
               break;
+            case _constants.TABLE_CONSTRAINT_KIND.CHECK:
+              {
+                var table = findTable(_this8.data.tables, schemaName, tableName);
+                if (!table) break;
+                if (!table.constraints) table.constraints = [];
+                table.constraints.push(cmd.value);
+                break;
+              }
             case _constants.TABLE_CONSTRAINT_KIND.UNIQUE:
             case _constants.TABLE_CONSTRAINT_KIND.PK:
             case _constants.TABLE_CONSTRAINT_KIND.INDEX:
@@ -950,11 +984,11 @@ var PostgresASTGen = exports["default"] = /*#__PURE__*/function (_PostgreSQLPars
                 if (!(cmd.value.pk || cmd.value.unique)) break;
                 // eslint-disable-next-line prefer-destructuring
                 kind = cmd.kind;
-                var table = findTable(_this8.data.tables, schemaName, tableName);
-                if (!table) break;
+                var _table2 = findTable(_this8.data.tables, schemaName, tableName);
+                if (!_table2) break;
                 if (cmd.value.columns.length === 1 && (cmd.value.unique || cmd.value.pk)) {
                   var key = cmd.value.columns[0].value;
-                  var fieldToUpdate = table.fields.find(function (f) {
+                  var fieldToUpdate = _table2.fields.find(function (f) {
                     return f.name === key;
                   });
                   if (fieldToUpdate) {
@@ -972,7 +1006,7 @@ var PostgresASTGen = exports["default"] = /*#__PURE__*/function (_PostgreSQLPars
                   type: cmd.value.type,
                   unique: cmd.value.unique
                 });
-                table.indexes.push(index);
+                _table2.indexes.push(index);
                 break;
               }
             default:
@@ -1051,8 +1085,8 @@ var PostgresASTGen = exports["default"] = /*#__PURE__*/function (_PostgreSQLPars
         var fieldName = (0, _lodash.last)(_names);
         var _tableName = _names.length > 1 ? _names[_names.length - 2] : undefined;
         var _schemaName = _names.length > 2 ? _names[_names.length - 3] : undefined;
-        var _table2 = findTable(this.data.tables, _schemaName, _tableName);
-        var field = _table2.fields.find(function (field) {
+        var _table3 = findTable(this.data.tables, _schemaName, _tableName);
+        var field = _table3.fields.find(function (field) {
           return field.name === fieldName;
         });
         if (!field) return;

package/lib/parse/ANTLR/parsers/postgresql/PostgreSQLParser.g4

@@ -5401,7 +5401,7 @@ plsql_unreserved_keyword
    //| COLUMN_NAME
    | COMMIT
    | CONSTANT
-   | CONSTRAINT
+   //| CONSTRAINT // TDD 72
    //| CONSTRAINT_NAME
    | CONTINUE_P
    | CURRENT_P