@dbml/core 3.10.2 → 3.11.0-alpha-0

package/lib/model_structure/database.js

@@ -53,7 +53,9 @@ var Database = /*#__PURE__*/function (_Element) {
       _ref$project = _ref.project,
       project = _ref$project === void 0 ? {} : _ref$project,
       _ref$aliases = _ref.aliases,
-      aliases = _ref$aliases === void 0 ? [] : _ref$aliases;
+      aliases = _ref$aliases === void 0 ? [] : _ref$aliases,
+      _ref$records = _ref.records,
+      records = _ref$records === void 0 ? {} : _ref$records;
     _classCallCheck(this, Database);
     _this = _super.call(this);
     _this.dbState = new _dbState["default"]();
@@ -67,6 +69,7 @@ var Database = /*#__PURE__*/function (_Element) {
     _this.name = project.name;
     _this.token = project.token;
     _this.aliases = aliases;
+    _this.records = records;
     _this.processNotes(notes);
     // The process order is important. Do not change !
     _this.processSchemas(schemas);
@@ -291,7 +294,8 @@ var Database = /*#__PURE__*/function (_Element) {
         enumValues: {},
         indexes: {},
         indexColumns: {},
-        fields: {}
+        fields: {},
+        records: this.records
       };
       this.schemas.forEach(function (schema) {
         return schema.normalize(normalizedModel);

package/lib/model_structure/utils.js

@@ -3,6 +3,7 @@
 Object.defineProperty(exports, "__esModule", {
   value: true
 });
+exports.getFullTableName = getFullTableName;
 exports.shouldPrintSchema = shouldPrintSchema;
 exports.shouldPrintSchemaName = shouldPrintSchemaName;
 var _config = require("./config");
@@ -11,4 +12,9 @@ function shouldPrintSchema(schema) {
 }
 function shouldPrintSchemaName(schemaName) {
   return schemaName !== _config.DEFAULT_SCHEMA_NAME;
+}
+
+// TODO: This is an ad hoc function for parsing inserts. It should be replaced with a more robust solution
+function getFullTableName(schemaName, tableName) {
+  return "".concat(schemaName && shouldPrintSchemaName(schemaName) ? "".concat(schemaName, ".") : '').concat(tableName);
 }

package/lib/parse/ANTLR/ASTGeneration/index.js

@@ -15,6 +15,9 @@ var _SnowflakeLexer = _interopRequireDefault(require("../parsers/snowflake/Snowf
 var _SnowflakeParser = _interopRequireDefault(require("../parsers/snowflake/SnowflakeParser"));
 var _SnowflakeASTGen = _interopRequireDefault(require("./snowflake/SnowflakeASTGen"));
 var _ParserErrorListener = _interopRequireDefault(require("./ParserErrorListener"));
+var _TSqlLexer = _interopRequireDefault(require("../parsers/mssql/TSqlLexer"));
+var _TSqlParser = _interopRequireDefault(require("../parsers/mssql/TSqlParser"));
+var _MssqlASTGen = _interopRequireDefault(require("./mssql/MssqlASTGen"));
 function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { "default": obj }; }
 /* eslint-disable import/no-named-as-default-member */
 /* eslint-disable import/no-named-as-default */
@@ -50,16 +53,29 @@ function parse(input, format) {
         if (errorListener.errors.length) throw errorListener.errors;
         break;
       }
-    case 'snowflake':
+    case 'mssql':
       {
-        var _lexer2 = new _SnowflakeLexer["default"](chars);
+        var _lexer2 = new _TSqlLexer["default"](chars);
         var _tokens2 = new _antlr["default"].CommonTokenStream(_lexer2);
-        var _parser2 = new _SnowflakeParser["default"](_tokens2);
+        var _parser2 = new _TSqlParser["default"](_tokens2);
         _parser2.buildParseTrees = true;
         _parser2.removeErrorListeners();
         _parser2.addErrorListener(errorListener);
-        var _parseTree2 = _parser2.snowflake_file();
-        database = _parseTree2.accept(new _SnowflakeASTGen["default"]());
+        var _parseTree2 = _parser2.tsql_file();
+        database = _parseTree2.accept(new _MssqlASTGen["default"]());
+        if (errorListener.errors.length) throw errorListener.errors;
+        break;
+      }
+    case 'snowflake':
+      {
+        var _lexer3 = new _SnowflakeLexer["default"](chars);
+        var _tokens3 = new _antlr["default"].CommonTokenStream(_lexer3);
+        var _parser3 = new _SnowflakeParser["default"](_tokens3);
+        _parser3.buildParseTrees = true;
+        _parser3.removeErrorListeners();
+        _parser3.addErrorListener(errorListener);
+        var _parseTree3 = _parser3.snowflake_file();
+        database = _parseTree3.accept(new _SnowflakeASTGen["default"]());
         if (errorListener.errors.length) throw errorListener.errors;
         break;
       }

package/lib/parse/ANTLR/ASTGeneration/mssql/MssqlASTGen.js

@@ -0,0 +1,393 @@
+"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 _lodash = require("lodash");
+var _utils = require("../../../../model_structure/utils");
+var _TSqlParserVisitor2 = _interopRequireDefault(require("../../parsers/mssql/TSqlParserVisitor"));
+var _constants = require("../constants");
+var _helpers = require("../helpers");
+function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { "default": obj }; }
+function _toConsumableArray(arr) { return _arrayWithoutHoles(arr) || _iterableToArray(arr) || _unsupportedIterableToArray(arr) || _nonIterableSpread(); }
+function _nonIterableSpread() { throw new TypeError("Invalid attempt to spread non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
+function _unsupportedIterableToArray(o, minLen) { if (!o) return; if (typeof o === "string") return _arrayLikeToArray(o, minLen); var n = Object.prototype.toString.call(o).slice(8, -1); if (n === "Object" && o.constructor) n = o.constructor.name; if (n === "Map" || n === "Set") return Array.from(o); if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _arrayLikeToArray(o, minLen); }
+function _iterableToArray(iter) { if (typeof Symbol !== "undefined" && iter[Symbol.iterator] != null || iter["@@iterator"] != null) return Array.from(iter); }
+function _arrayWithoutHoles(arr) { if (Array.isArray(arr)) return _arrayLikeToArray(arr); }
+function _arrayLikeToArray(arr, len) { if (len == null || len > arr.length) len = arr.length; for (var i = 0, arr2 = new Array(len); i < len; i++) arr2[i] = arr[i]; return arr2; }
+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); } /* eslint-disable class-methods-use-this */
+var MssqlASTGen = /*#__PURE__*/function (_TSqlParserVisitor) {
+  _inherits(MssqlASTGen, _TSqlParserVisitor);
+  var _super = _createSuper(MssqlASTGen);
+  function MssqlASTGen() {
+    var _this;
+    _classCallCheck(this, MssqlASTGen);
+    _this = _super.call(this);
+    _this.data = {
+      schemas: [],
+      tables: [],
+      refs: [],
+      enums: [],
+      tableGroups: [],
+      aliases: [],
+      project: {},
+      records: {}
+    };
+    return _this;
+  }
+
+  // tsql_file
+  //   : batch* EOF
+  //   | execute_body_batch go_statement* EOF
+  //   ;
+  _createClass(MssqlASTGen, [{
+    key: "visitTsql_file",
+    value: function visitTsql_file(ctx) {
+      var _this2 = this;
+      ctx.batch().forEach(function (batch) {
+        return batch.accept(_this2);
+      });
+      return this.data;
+    }
+
+    // batch
+    //   : go_statement
+    //   | execute_body_batch? (go_statement | sql_clauses+) go_statement*
+    //   | batch_level_statement go_statement*
+    //   ;
+  }, {
+    key: "visitBatch",
+    value: function visitBatch(ctx) {
+      var _this3 = this;
+      if (ctx.sql_clauses()) {
+        ctx.sql_clauses().forEach(function (sqlClause) {
+          return sqlClause.accept(_this3);
+        });
+      }
+    }
+
+    // sql_clauses
+    //   : dml_clause SEMI?
+    //   | cfl_statement SEMI?
+    //   | another_statement SEMI?
+    //   | ddl_clause SEMI?
+    //   | dbcc_clause SEMI?
+    //   | backup_statement SEMI?
+    //   | SEMI
+    //   ;
+  }, {
+    key: "visitSql_clauses",
+    value: function visitSql_clauses(ctx) {
+      if (ctx.dml_clause()) {
+        ctx.dml_clause().accept(this);
+      }
+    }
+
+    // dml_clause
+    //   : merge_statement
+    //   | delete_statement
+    //   | insert_statement
+    //   | select_statement_standalone
+    //   | update_statement
+    //   ;
+  }, {
+    key: "visitDml_clause",
+    value: function visitDml_clause(ctx) {
+      if (ctx.insert_statement()) {
+        ctx.insert_statement().accept(this);
+      }
+    }
+
+    // insert_statement
+    //   : with_expression? INSERT (TOP '(' expression ')' PERCENT?)? INTO? (
+    //       ddl_object
+    //       | rowset_function_limited
+    //   ) with_table_hints? ('(' insert_column_name_list ')')? output_clause? insert_statement_value for_clause? option_clause? ';'?
+    //   ;
+  }, {
+    key: "visitInsert_statement",
+    value: function visitInsert_statement(ctx) {
+      var _this$data$records$fu;
+      // [ 'users' ]
+      // [ 'test', 'users' ]
+      // [ 'db', 'test', 'users' ]
+      // [ 'server', 'db', 'test', 'users' ]
+      // The last: table name
+      // The second-last: schema name
+      var names = ctx.ddl_object().accept(this);
+      var tableName = (0, _lodash.last)(names);
+      var schemaName = names.length > 1 ? (0, _lodash.nth)(names, -2) : undefined;
+      var columns = ctx.insert_column_name_list() ? ctx.insert_column_name_list().accept(this) : [];
+      var values = ctx.insert_statement_value().accept(this);
+
+      // handle insert into all columns
+      if (columns.length === 0 || values.length === 0) {
+        // temporarily ignore
+        return;
+      }
+      var fullTableName = (0, _utils.getFullTableName)(schemaName, tableName);
+      if (!this.data.records[fullTableName]) {
+        this.data.records[fullTableName] = {
+          schemaName: schemaName,
+          tableName: tableName,
+          columns: columns,
+          values: []
+        };
+      }
+      (_this$data$records$fu = this.data.records[fullTableName].values).push.apply(_this$data$records$fu, _toConsumableArray(values));
+    }
+
+    // ddl_object
+    //   : full_table_name
+    //   | LOCAL_ID
+    //   ;
+  }, {
+    key: "visitDdl_object",
+    value: function visitDdl_object(ctx) {
+      return ctx.full_table_name().accept(this);
+    }
+
+    // full_table_name
+    //   : (
+    //       linkedServer = id_ '.' '.' schema = id_ '.'
+    //       | server = id_ '.' database = id_ '.' schema = id_ '.'
+    //       | database = id_ '.' schema = id_? '.'
+    //       | schema = id_ '.'
+    //   )? table = id_
+    //   ;
+  }, {
+    key: "visitFull_table_name",
+    value: function visitFull_table_name(ctx) {
+      var _this4 = this;
+      return ctx.id_().map(function (id) {
+        return id.accept(_this4);
+      });
+    }
+
+    // id_
+    //   : ID
+    //   | TEMP_ID
+    //   | DOUBLE_QUOTE_ID
+    //   | DOUBLE_QUOTE_BLANK
+    //   | SQUARE_BRACKET_ID
+    //   | keyword
+    //   | RAW
+    //   ;
+  }, {
+    key: "visitId_",
+    value: function visitId_(ctx) {
+      if (ctx.DOUBLE_QUOTE_ID() || ctx.SQUARE_BRACKET_ID()) return (0, _helpers.getOriginalText)(ctx).slice(1, -1);
+      return (0, _helpers.getOriginalText)(ctx);
+    }
+
+    // insert_column_name_list
+    //   : col += insert_column_id (',' col += insert_column_id)*
+    //   ;
+  }, {
+    key: "visitInsert_column_name_list",
+    value: function visitInsert_column_name_list(ctx) {
+      var _this5 = this;
+      var columns = ctx.insert_column_id().map(function (column) {
+        return column.accept(_this5);
+      });
+      return (0, _lodash.flattenDepth)(columns, 1);
+    }
+
+    // insert_column_id
+    //   : (ignore += id_? '.')* id_
+    //   ;
+
+    // insert_statement_value
+    //   : table_value_constructor
+    //   | derived_table
+    //   | execute_statement
+    //   | DEFAULT VALUES
+    //   ;
+  }, {
+    key: "visitInsert_statement_value",
+    value: function visitInsert_statement_value(ctx) {
+      if (!ctx.table_value_constructor()) return [];
+      var rawValues = ctx.table_value_constructor().accept(this);
+      return rawValues;
+    }
+
+    // table_value_constructor
+    //   : VALUES '(' exps += expression_list_ ')' (',' '(' exps += expression_list_ ')')*
+    //   ;
+  }, {
+    key: "visitTable_value_constructor",
+    value: function visitTable_value_constructor(ctx) {
+      var _this6 = this;
+      return ctx.expression_list_().map(function (expression) {
+        return expression.accept(_this6);
+      });
+    }
+
+    // expression_list_
+    //   : exp += expression (',' exp += expression)*
+    //   ;
+  }, {
+    key: "visitExpression_list_",
+    value: function visitExpression_list_(ctx) {
+      var _this7 = this;
+      return ctx.expression().map(function (expression) {
+        return expression.accept(_this7);
+      });
+    }
+
+    // expression
+    //   : primitive_expression
+    //   | function_call
+    //   | expression '.' (value_call | query_call | exist_call | modify_call)
+    //   | expression '.' hierarchyid_call
+    //   | expression COLLATE id_
+    //   | case_expression
+    //   | full_column_name
+    //   | bracket_expression
+    //   | unary_operator_expression
+    //   | expression op = ('*' | '/' | '%') expression
+    //   | expression op = ('+' | '-' | '&' | '^' | '|' | '||') expression
+    //   | expression time_zone
+    //   | over_clause
+    //   | DOLLAR_ACTION
+    //   ;
+  }, {
+    key: "visitExpression",
+    value: function visitExpression(ctx) {
+      if (ctx.primitive_expression()) {
+        return ctx.primitive_expression().accept(this);
+      }
+      if (ctx.function_call()) {
+        return ctx.function_call().accept(this);
+      }
+      if (ctx.unary_operator_expression()) {
+        return ctx.unary_operator_expression().accept(this);
+      }
+
+      // Default case for any other expression type
+      return {
+        value: (0, _helpers.getOriginalText)(ctx),
+        type: _constants.DATA_TYPE.EXPRESSION
+      };
+    }
+
+    // primitive_constant
+    //   : STRING // string, datetime or uniqueidentifier
+    //   | BINARY
+    //   | (DECIMAL | REAL | FLOAT)                    // float or decimal
+    //   | dollar = '$' ('-' | '+')? (DECIMAL | FLOAT) // money
+    //   | parameter
+    //   ;
+  }, {
+    key: "visitPrimitive_constant",
+    value: function visitPrimitive_constant(ctx) {
+      if (ctx.STRING() || ctx.BINARY()) {
+        return {
+          value: ctx.getText(),
+          type: _constants.DATA_TYPE.STRING
+        };
+      }
+      if (ctx.DOLLAR()) {
+        var dollar = (0, _lodash.first)(ctx.children).getText();
+        var sign = ctx.children.length > 2 ? (0, _lodash.nth)(ctx.children, -2) : '';
+        var value = (0, _lodash.last)(ctx.children).getText();
+        return {
+          value: "".concat(dollar).concat(sign).concat(value),
+          type: _constants.DATA_TYPE.STRING
+        };
+      }
+      if (ctx.REAL() || ctx.DECIMAL() || ctx.FLOAT()) {
+        return {
+          value: ctx.getText(),
+          type: _constants.DATA_TYPE.NUMBER
+        };
+      }
+      return {
+        value: ctx.getText(),
+        type: _constants.DATA_TYPE.EXPRESSION
+      };
+    }
+
+    // function_call
+    //   : ranking_windowed_function                      # RANKING_WINDOWED_FUNC
+    //   | aggregate_windowed_function                    # AGGREGATE_WINDOWED_FUNC
+    //   | analytic_windowed_function                     # ANALYTIC_WINDOWED_FUNC
+    //   | built_in_functions                             # BUILT_IN_FUNC
+    //   | scalar_function_name '(' expression_list_? ')' # SCALAR_FUNCTION
+    //   | freetext_function                              # FREE_TEXT
+    //   | partition_function                             # PARTITION_FUNC
+    //   | hierarchyid_static_method                      # HIERARCHYID_METHOD
+    //   ;
+
+    // See packages/dbml-core/src/parse/ANTLR/parsers/mssql/TSqlParser.g4 at line 4338
+  }, {
+    key: "visitBUILT_IN_FUNC",
+    value: function visitBUILT_IN_FUNC(ctx) {
+      return {
+        value: (0, _helpers.getOriginalText)(ctx),
+        type: _constants.DATA_TYPE.EXPRESSION
+      };
+    }
+
+    // unary_operator_expression
+    //   : '~' expression
+    //   | op = ('+' | '-') expression
+    //   ;
+  }, {
+    key: "visitUnary_operator_expression",
+    value: function visitUnary_operator_expression(ctx) {
+      var operator = ctx.children[0].getText();
+      var expression = ctx.expression().accept(this);
+      return {
+        value: "".concat(operator).concat(expression.value),
+        type: expression.type
+      };
+    }
+
+    // data_type
+    //   : scaled = (VARCHAR | NVARCHAR | BINARY_KEYWORD | VARBINARY_KEYWORD | SQUARE_BRACKET_ID) '(' MAX ')'
+    //   | ext_type = id_ '(' scale = DECIMAL ',' prec = DECIMAL ')'
+    //   | ext_type = id_ '(' scale = DECIMAL ')'
+    //   | ext_type = id_ IDENTITY ('(' seed = DECIMAL ',' inc = DECIMAL ')')?
+    //   | double_prec = DOUBLE PRECISION?
+    //   | unscaled_type = id_
+    //   ;
+  }, {
+    key: "visitData_type",
+    value: function visitData_type(ctx) {
+      return ctx.getText();
+    }
+  }, {
+    key: "visitPrimitive_expression",
+    value: function visitPrimitive_expression(ctx) {
+      if (ctx.NULL_()) {
+        return {
+          value: ctx.getText(),
+          type: _constants.DATA_TYPE.EXPRESSION
+        };
+      }
+      if (ctx.primitive_constant()) {
+        return ctx.primitive_constant().accept(this);
+      }
+      return {
+        value: ctx.getText(),
+        type: _constants.DATA_TYPE.EXPRESSION
+      };
+    }
+  }]);
+  return MssqlASTGen;
+}(_TSqlParserVisitor2["default"]);
+exports["default"] = MssqlASTGen;