@dbml/core 2.6.1 → 3.1.0

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

@@ -0,0 +1,31 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.PipesAsConcat = exports.NoMode = exports.NoBackslashEscapes = exports.IgnoreSpace = exports.HighNotPrecedence = exports.AnsiQuotes = void 0;
+exports.isSqlModeActive = isSqlModeActive;
+exports.serverVersion = void 0;
+/* eslint-disable no-bitwise */
+
+// SqlMode definition: https://github.com/mysql/mysql-workbench/blob/6e135fb33942123c57f059139cbd787bea4f3f9b/library/parsers/mysql/MySQLRecognizerCommon.h#L51
+var NoMode = 0;
+exports.NoMode = NoMode;
+var AnsiQuotes = 1 << 0;
+exports.AnsiQuotes = AnsiQuotes;
+var HighNotPrecedence = 1 << 1;
+exports.HighNotPrecedence = HighNotPrecedence;
+var PipesAsConcat = 1 << 2;
+exports.PipesAsConcat = PipesAsConcat;
+var IgnoreSpace = 1 << 3;
+exports.IgnoreSpace = IgnoreSpace;
+var NoBackslashEscapes = 1 << 4;
+// These flags affect parsing behavior. https://dev.mysql.com/doc/refman/8.0/en/sql-mode.html
+exports.NoBackslashEscapes = NoBackslashEscapes;
+var sqlMode = NoMode;
+// https://github.com/mysql/mysql-workbench/blob/6e135fb33942123c57f059139cbd787bea4f3f9b/library/parsers/mysql/MySQLRecognizerCommon.cpp#L51
+function isSqlModeActive(mode) {
+  return (sqlMode & mode) !== 0;
+}
+var serverVersion = 80000;
+exports.serverVersion = serverVersion;

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

@@ -0,0 +1,40 @@
+"use strict";
+
+function _typeof(obj) { "@babel/helpers - typeof"; return _typeof = "function" == typeof Symbol && "symbol" == typeof Symbol.iterator ? function (obj) { return typeof obj; } : function (obj) { return obj && "function" == typeof Symbol && obj.constructor === Symbol && obj !== Symbol.prototype ? "symbol" : typeof obj; }, _typeof(obj); }
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports["default"] = void 0;
+var _antlr = _interopRequireDefault(require("antlr4"));
+function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { "default": obj }; }
+function _classCallCheck(instance, Constructor) { if (!(instance instanceof Constructor)) { throw new TypeError("Cannot call a class as a function"); } }
+function _defineProperties(target, props) { for (var i = 0; i < props.length; i++) { var descriptor = props[i]; descriptor.enumerable = descriptor.enumerable || false; descriptor.configurable = true; if ("value" in descriptor) descriptor.writable = true; Object.defineProperty(target, _toPropertyKey(descriptor.key), descriptor); } }
+function _createClass(Constructor, protoProps, staticProps) { if (protoProps) _defineProperties(Constructor.prototype, protoProps); if (staticProps) _defineProperties(Constructor, staticProps); Object.defineProperty(Constructor, "prototype", { writable: false }); return Constructor; }
+function _toPropertyKey(arg) { var key = _toPrimitive(arg, "string"); return _typeof(key) === "symbol" ? key : String(key); }
+function _toPrimitive(input, hint) { if (_typeof(input) !== "object" || input === null) return input; var prim = input[Symbol.toPrimitive]; if (prim !== undefined) { var res = prim.call(input, hint || "default"); if (_typeof(res) !== "object") return res; throw new TypeError("@@toPrimitive must return a primitive value."); } return (hint === "string" ? String : Number)(input); }
+function _inherits(subClass, superClass) { if (typeof superClass !== "function" && superClass !== null) { throw new TypeError("Super expression must either be null or a function"); } subClass.prototype = Object.create(superClass && superClass.prototype, { constructor: { value: subClass, writable: true, configurable: true } }); Object.defineProperty(subClass, "prototype", { writable: false }); if (superClass) _setPrototypeOf(subClass, superClass); }
+function _setPrototypeOf(o, p) { _setPrototypeOf = Object.setPrototypeOf ? Object.setPrototypeOf.bind() : function _setPrototypeOf(o, p) { o.__proto__ = p; return o; }; return _setPrototypeOf(o, p); }
+function _createSuper(Derived) { var hasNativeReflectConstruct = _isNativeReflectConstruct(); return function _createSuperInternal() { var Super = _getPrototypeOf(Derived), result; if (hasNativeReflectConstruct) { var NewTarget = _getPrototypeOf(this).constructor; result = Reflect.construct(Super, arguments, NewTarget); } else { result = Super.apply(this, arguments); } return _possibleConstructorReturn(this, result); }; }
+function _possibleConstructorReturn(self, call) { if (call && (_typeof(call) === "object" || typeof call === "function")) { return call; } else if (call !== void 0) { throw new TypeError("Derived constructors may only return object or undefined"); } return _assertThisInitialized(self); }
+function _assertThisInitialized(self) { if (self === void 0) { throw new ReferenceError("this hasn't been initialised - super() hasn't been called"); } return self; }
+function _isNativeReflectConstruct() { if (typeof Reflect === "undefined" || !Reflect.construct) return false; if (Reflect.construct.sham) return false; if (typeof Proxy === "function") return true; try { Boolean.prototype.valueOf.call(Reflect.construct(Boolean, [], function () {})); return true; } catch (e) { return false; } }
+function _getPrototypeOf(o) { _getPrototypeOf = Object.setPrototypeOf ? Object.getPrototypeOf.bind() : function _getPrototypeOf(o) { return o.__proto__ || Object.getPrototypeOf(o); }; return _getPrototypeOf(o); }
+/* Based on Oracle base class: https://github.com/mysql/mysql-workbench/blob/8.0/library/parsers/mysql/MySQLBaseLexer.h */
+var MySQLBaseLexer = /*#__PURE__*/function (_antlr4$Lexer) {
+  _inherits(MySQLBaseLexer, _antlr4$Lexer);
+  var _super = _createSuper(MySQLBaseLexer);
+  function MySQLBaseLexer() {
+    _classCallCheck(this, MySQLBaseLexer);
+    return _super.apply(this, arguments);
+  }
+  _createClass(MySQLBaseLexer, [{
+    key: "setType",
+    value:
+    // https://www.antlr.org/api/Java/org/antlr/v4/runtime/Lexer.html#setType(int)
+    function setType(val) {
+      this.type = val;
+    }
+  }]);
+  return MySQLBaseLexer;
+}(_antlr["default"].Lexer);
+exports["default"] = MySQLBaseLexer;

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

@@ -0,0 +1,32 @@
+"use strict";
+
+function _typeof(obj) { "@babel/helpers - typeof"; return _typeof = "function" == typeof Symbol && "symbol" == typeof Symbol.iterator ? function (obj) { return typeof obj; } : function (obj) { return obj && "function" == typeof Symbol && obj.constructor === Symbol && obj !== Symbol.prototype ? "symbol" : typeof obj; }, _typeof(obj); }
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports["default"] = void 0;
+var _antlr = _interopRequireDefault(require("antlr4"));
+function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { "default": obj }; }
+function _defineProperties(target, props) { for (var i = 0; i < props.length; i++) { var descriptor = props[i]; descriptor.enumerable = descriptor.enumerable || false; descriptor.configurable = true; if ("value" in descriptor) descriptor.writable = true; Object.defineProperty(target, _toPropertyKey(descriptor.key), descriptor); } }
+function _createClass(Constructor, protoProps, staticProps) { if (protoProps) _defineProperties(Constructor.prototype, protoProps); if (staticProps) _defineProperties(Constructor, staticProps); Object.defineProperty(Constructor, "prototype", { writable: false }); return Constructor; }
+function _toPropertyKey(arg) { var key = _toPrimitive(arg, "string"); return _typeof(key) === "symbol" ? key : String(key); }
+function _toPrimitive(input, hint) { if (_typeof(input) !== "object" || input === null) return input; var prim = input[Symbol.toPrimitive]; if (prim !== undefined) { var res = prim.call(input, hint || "default"); if (_typeof(res) !== "object") return res; throw new TypeError("@@toPrimitive must return a primitive value."); } return (hint === "string" ? String : Number)(input); }
+function _classCallCheck(instance, Constructor) { if (!(instance instanceof Constructor)) { throw new TypeError("Cannot call a class as a function"); } }
+function _inherits(subClass, superClass) { if (typeof superClass !== "function" && superClass !== null) { throw new TypeError("Super expression must either be null or a function"); } subClass.prototype = Object.create(superClass && superClass.prototype, { constructor: { value: subClass, writable: true, configurable: true } }); Object.defineProperty(subClass, "prototype", { writable: false }); if (superClass) _setPrototypeOf(subClass, superClass); }
+function _setPrototypeOf(o, p) { _setPrototypeOf = Object.setPrototypeOf ? Object.setPrototypeOf.bind() : function _setPrototypeOf(o, p) { o.__proto__ = p; return o; }; return _setPrototypeOf(o, p); }
+function _createSuper(Derived) { var hasNativeReflectConstruct = _isNativeReflectConstruct(); return function _createSuperInternal() { var Super = _getPrototypeOf(Derived), result; if (hasNativeReflectConstruct) { var NewTarget = _getPrototypeOf(this).constructor; result = Reflect.construct(Super, arguments, NewTarget); } else { result = Super.apply(this, arguments); } return _possibleConstructorReturn(this, result); }; }
+function _possibleConstructorReturn(self, call) { if (call && (_typeof(call) === "object" || typeof call === "function")) { return call; } else if (call !== void 0) { throw new TypeError("Derived constructors may only return object or undefined"); } return _assertThisInitialized(self); }
+function _assertThisInitialized(self) { if (self === void 0) { throw new ReferenceError("this hasn't been initialised - super() hasn't been called"); } return self; }
+function _isNativeReflectConstruct() { if (typeof Reflect === "undefined" || !Reflect.construct) return false; if (Reflect.construct.sham) return false; if (typeof Proxy === "function") return true; try { Boolean.prototype.valueOf.call(Reflect.construct(Boolean, [], function () {})); return true; } catch (e) { return false; } }
+function _getPrototypeOf(o) { _getPrototypeOf = Object.setPrototypeOf ? Object.getPrototypeOf.bind() : function _getPrototypeOf(o) { return o.__proto__ || Object.getPrototypeOf(o); }; return _getPrototypeOf(o); }
+/* Based on Oracle base class: https://github.com/mysql/mysql-workbench/blob/8.0/library/parsers/mysql/MySQLBaseRecognizer.h */
+var MySQLBaseRecognizer = /*#__PURE__*/function (_antlr4$Parser) {
+  _inherits(MySQLBaseRecognizer, _antlr4$Parser);
+  var _super = _createSuper(MySQLBaseRecognizer);
+  function MySQLBaseRecognizer() {
+    _classCallCheck(this, MySQLBaseRecognizer);
+    return _super.apply(this, arguments);
+  }
+  return _createClass(MySQLBaseRecognizer);
+}(_antlr["default"].Parser);
+exports["default"] = MySQLBaseRecognizer;

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;