@dbml/core 3.11.0 → 3.12.0-alpha-0

package/lib/model_structure/database.js

@@ -7,7 +7,7 @@ Object.defineProperty(exports, "__esModule", {
 exports["default"] = void 0;
 var _lodash = _interopRequireWildcard(require("lodash"));
 var _schema = _interopRequireDefault(require("./schema"));
-var _ref2 = _interopRequireDefault(require("./ref"));
+var _ref3 = _interopRequireDefault(require("./ref"));
 var _enum2 = _interopRequireDefault(require("./enum"));
 var _tableGroup = _interopRequireDefault(require("./tableGroup"));
 var _table = _interopRequireDefault(require("./table"));
@@ -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,7 +69,9 @@ var Database = /*#__PURE__*/function (_Element) {
     _this.name = project.name;
     _this.token = project.token;
     _this.aliases = aliases;
+    _this.records = [];
     _this.processNotes(notes);
+    _this.processRecords(records);
     // The process order is important. Do not change !
     _this.processSchemas(schemas);
     _this.processSchemaElements(enums, _config.ENUM);
@@ -92,6 +96,24 @@ var Database = /*#__PURE__*/function (_Element) {
         })));
       });
     }
+  }, {
+    key: "processRecords",
+    value: function processRecords(rawRecords) {
+      var _this3 = this;
+      rawRecords.forEach(function (_ref2) {
+        var schemaName = _ref2.schemaName,
+          tableName = _ref2.tableName,
+          columns = _ref2.columns,
+          values = _ref2.values;
+        _this3.records.push({
+          id: _this3.dbState.generateId('recordId'),
+          schemaName: schemaName,
+          tableName: tableName,
+          columns: columns,
+          values: values
+        });
+      });
+    }
   }, {
     key: "pushNote",
     value: function pushNote(note) {
@@ -110,10 +132,10 @@ var Database = /*#__PURE__*/function (_Element) {
   }, {
     key: "processSchemas",
     value: function processSchemas(rawSchemas) {
-      var _this3 = this;
+      var _this4 = this;
       rawSchemas.forEach(function (schema) {
-        _this3.pushSchema(new _schema["default"](_objectSpread(_objectSpread({}, schema), {}, {
-          database: _this3
+        _this4.pushSchema(new _schema["default"](_objectSpread(_objectSpread({}, schema), {}, {
+          database: _this4
         })));
       });
     }
@@ -135,16 +157,16 @@ var Database = /*#__PURE__*/function (_Element) {
   }, {
     key: "processSchemaElements",
     value: function processSchemaElements(elements, elementType) {
-      var _this4 = this;
+      var _this5 = this;
       var schema;
       elements.forEach(function (element) {
         if (element.schemaName) {
-          schema = _this4.findOrCreateSchema(element.schemaName);
+          schema = _this5.findOrCreateSchema(element.schemaName);
           if (element.schemaName === _config.DEFAULT_SCHEMA_NAME) {
             // this.hasDefaultSchema = true;
           }
         } else {
-          schema = _this4.findOrCreateSchema(_config.DEFAULT_SCHEMA_NAME);
+          schema = _this5.findOrCreateSchema(_config.DEFAULT_SCHEMA_NAME);
         }
         switch (elementType) {
           case _config.TABLE:
@@ -163,7 +185,7 @@ var Database = /*#__PURE__*/function (_Element) {
             })));
             break;
           case _config.REF:
-            schema.pushRef(new _ref2["default"](_objectSpread(_objectSpread({}, element), {}, {
+            schema.pushRef(new _ref3["default"](_objectSpread(_objectSpread({}, element), {}, {
               schema: schema
             })));
             break;
@@ -259,6 +281,9 @@ var Database = /*#__PURE__*/function (_Element) {
         }),
         notes: this.notes.map(function (n) {
           return n["export"]();
+        }),
+        records: this.records.map(function (r) {
+          return _objectSpread({}, r);
         })
       };
     }
@@ -291,7 +316,8 @@ var Database = /*#__PURE__*/function (_Element) {
         enumValues: {},
         indexes: {},
         indexColumns: {},
-        fields: {}
+        fields: {},
+        records: {}
       };
       this.schemas.forEach(function (schema) {
         return schema.normalize(normalizedModel);
@@ -299,6 +325,9 @@ var Database = /*#__PURE__*/function (_Element) {
       this.notes.forEach(function (note) {
         return note.normalize(normalizedModel);
       });
+      this.records.forEach(function (record) {
+        return normalizedModel.records[record.id] = _objectSpread({}, record);
+      });
       return normalizedModel;
     }
   }]);

package/lib/model_structure/dbState.js

@@ -25,6 +25,7 @@ var DbState = /*#__PURE__*/function () {
     this.indexId = 1;
     this.fieldId = 1;
     this.indexColumnId = 1;
+    this.recordId = 1;
   }
   _createClass(DbState, [{
     key: "generateId",

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/constants.js

@@ -9,7 +9,8 @@ var TABLE_CONSTRAINT_KIND = {
   INDEX: 'index',
   FK: 'fk',
   UNIQUE: 'unique',
-  PK: 'pk'
+  PK: 'pk',
+  DEFAULT: 'default'
 };
 exports.TABLE_CONSTRAINT_KIND = TABLE_CONSTRAINT_KIND;
 var COLUMN_CONSTRAINT_KIND = {
@@ -19,7 +20,8 @@ var COLUMN_CONSTRAINT_KIND = {
   DEFAULT: 'dbdefault',
   INCREMENT: 'increment',
   INLINE_REF: 'inline_ref',
-  NOTE: 'note'
+  NOTE: 'note',
+  CHECK: 'check'
 };
 exports.COLUMN_CONSTRAINT_KIND = COLUMN_CONSTRAINT_KIND;
 var DATA_TYPE = {

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;
       }