@dbml/core 3.11.0-alpha-0 → 3.12.0-alpha-0

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

@@ -10,13 +10,12 @@ var _utils = require("../../../../model_structure/utils");
 var _TSqlParserVisitor2 = _interopRequireDefault(require("../../parsers/mssql/TSqlParserVisitor"));
 var _constants = require("../constants");
 var _helpers = require("../helpers");
+var _AST = require("../AST");
 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; }
@@ -28,7 +27,91 @@ function _createSuper(Derived) { var hasNativeReflectConstruct = _isNativeReflec
 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 */
+function _getPrototypeOf(o) { _getPrototypeOf = Object.setPrototypeOf ? Object.getPrototypeOf.bind() : function _getPrototypeOf(o) { return o.__proto__ || Object.getPrototypeOf(o); }; return _getPrototypeOf(o); }
+function _slicedToArray(arr, i) { return _arrayWithHoles(arr) || _iterableToArrayLimit(arr, i) || _unsupportedIterableToArray(arr, i) || _nonIterableRest(); }
+function _nonIterableRest() { throw new TypeError("Invalid attempt to destructure 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 _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 _iterableToArrayLimit(arr, i) { var _i = null == arr ? null : "undefined" != typeof Symbol && arr[Symbol.iterator] || arr["@@iterator"]; if (null != _i) { var _s, _e, _x, _r, _arr = [], _n = !0, _d = !1; try { if (_x = (_i = _i.call(arr)).next, 0 === i) { if (Object(_i) !== _i) return; _n = !1; } else for (; !(_n = (_s = _x.call(_i)).done) && (_arr.push(_s.value), _arr.length !== i); _n = !0); } catch (err) { _d = !0, _e = err; } finally { try { if (!_n && null != _i["return"] && (_r = _i["return"](), Object(_r) !== _r)) return; } finally { if (_d) throw _e; } } return _arr; } }
+function _arrayWithHoles(arr) { if (Array.isArray(arr)) return arr; } /* eslint-disable class-methods-use-this */
+var ADD_DESCRIPTION_FUNCTION_NAME = 'sp_addextendedproperty';
+var getSchemaAndTableName = function getSchemaAndTableName(names) {
+  var tableName = (0, _lodash.last)(names);
+  var schemaName = names.length > 1 ? (0, _lodash.nth)(names, -2) : undefined;
+  return {
+    schemaName: schemaName,
+    tableName: tableName
+  };
+};
+var getStringFromRawString = function getStringFromRawString(rawString) {
+  if (rawString.startsWith("N'")) {
+    return rawString;
+  }
+  return rawString.slice(1, rawString.length - 1);
+};
+
+/**
+ * @param {any[]} columnDefTableConstraints
+ * @returns {{
+ *  fieldsData: any[],
+ *  indexes: any[],
+ *  tableRefs: any[],
+ * }}
+ */
+var splitColumnDefTableConstraints = function splitColumnDefTableConstraints(columnDefTableConstraints) {
+  var _columnDefTableConstr = columnDefTableConstraints.reduce(function (acc, columnDefTableConstraint) {
+      switch (columnDefTableConstraint.kind) {
+        case _constants.TABLE_CONSTRAINT_KIND.FIELD:
+          acc[0].push(columnDefTableConstraint.value);
+          break;
+        case _constants.TABLE_CONSTRAINT_KIND.INDEX:
+        case _constants.TABLE_CONSTRAINT_KIND.PK:
+        case _constants.TABLE_CONSTRAINT_KIND.UNIQUE:
+          acc[1].push(columnDefTableConstraint.value);
+          break;
+        case _constants.TABLE_CONSTRAINT_KIND.FK:
+          acc[2].push(columnDefTableConstraint.value);
+          break;
+        case _constants.TABLE_CONSTRAINT_KIND.DEFAULT:
+          acc[3].push(columnDefTableConstraint.value);
+          break;
+        default:
+          break;
+      }
+      return acc;
+    }, [[], [], [], []]),
+    _columnDefTableConstr2 = _slicedToArray(_columnDefTableConstr, 4),
+    fieldsData = _columnDefTableConstr2[0],
+    indexes = _columnDefTableConstr2[1],
+    tableRefs = _columnDefTableConstr2[2],
+    columnDefaults = _columnDefTableConstr2[3];
+  return {
+    fieldsData: fieldsData,
+    indexes: indexes,
+    tableRefs: tableRefs,
+    columnDefaults: columnDefaults
+  };
+};
+var parseFieldsAndInlineRefsFromFieldsData = function parseFieldsAndInlineRefsFromFieldsData(fieldsData, tableName, schemaName) {
+  var _fieldsData$reduce = fieldsData.reduce(function (acc, fieldData) {
+      var inlineRefs = fieldData.inline_refs.map(function (inlineRef) {
+        inlineRef.endpoints[0].tableName = tableName;
+        inlineRef.endpoints[0].schemaName = schemaName;
+        inlineRef.endpoints[0].fieldNames = [fieldData.field.name];
+        return inlineRef;
+      });
+      acc[0].push(inlineRefs);
+      acc[1].push(fieldData.field);
+      return acc;
+    }, [[], []]),
+    _fieldsData$reduce2 = _slicedToArray(_fieldsData$reduce, 2),
+    resInlineRefs = _fieldsData$reduce2[0],
+    fields = _fieldsData$reduce2[1];
+  return {
+    inlineRefs: resInlineRefs,
+    fields: fields
+  };
+};
 var MssqlASTGen = /*#__PURE__*/function (_TSqlParserVisitor) {
   _inherits(MssqlASTGen, _TSqlParserVisitor);
   var _super = _createSuper(MssqlASTGen);
@@ -44,7 +127,7 @@ var MssqlASTGen = /*#__PURE__*/function (_TSqlParserVisitor) {
       tableGroups: [],
       aliases: [],
       project: {},
-      records: {}
+      records: []
     };
     return _this;
   }
@@ -91,8 +174,16 @@ var MssqlASTGen = /*#__PURE__*/function (_TSqlParserVisitor) {
   }, {
     key: "visitSql_clauses",
     value: function visitSql_clauses(ctx) {
+      if (ctx.ddl_clause()) {
+        ctx.ddl_clause().accept(this);
+        return;
+      }
       if (ctx.dml_clause()) {
         ctx.dml_clause().accept(this);
+        return;
+      }
+      if (ctx.another_statement()) {
+        ctx.another_statement().accept(this);
       }
     }
 
@@ -120,7 +211,6 @@ var MssqlASTGen = /*#__PURE__*/function (_TSqlParserVisitor) {
   }, {
     key: "visitInsert_statement",
     value: function visitInsert_statement(ctx) {
-      var _this$data$records$fu;
       // [ 'users' ]
       // [ 'test', 'users' ]
       // [ 'db', 'test', 'users' ]
@@ -132,22 +222,13 @@ var MssqlASTGen = /*#__PURE__*/function (_TSqlParserVisitor) {
       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));
+      var record = {
+        schemaName: schemaName,
+        tableName: tableName,
+        columns: columns,
+        values: values
+      };
+      this.data.records.push(record);
     }
 
     // ddl_object
@@ -276,6 +357,9 @@ var MssqlASTGen = /*#__PURE__*/function (_TSqlParserVisitor) {
       if (ctx.unary_operator_expression()) {
         return ctx.unary_operator_expression().accept(this);
       }
+      if (ctx.bracket_expression()) {
+        return ctx.bracket_expression().accept(this);
+      }
 
       // Default case for any other expression type
       return {
@@ -284,6 +368,16 @@ var MssqlASTGen = /*#__PURE__*/function (_TSqlParserVisitor) {
       };
     }
 
+    // bracket_expression
+    //   : '(' expression ')'
+    //   | '(' subquery ')'
+    //   ;
+  }, {
+    key: "visitBracket_expression",
+    value: function visitBracket_expression(ctx) {
+      return ctx.expression() ? ctx.expression().accept(this) : null;
+    }
+
     // primitive_constant
     //   : STRING // string, datetime or uniqueidentifier
     //   | BINARY
@@ -295,17 +389,18 @@ var MssqlASTGen = /*#__PURE__*/function (_TSqlParserVisitor) {
     key: "visitPrimitive_constant",
     value: function visitPrimitive_constant(ctx) {
       if (ctx.STRING() || ctx.BINARY()) {
+        var value = getStringFromRawString(ctx.getText());
         return {
-          value: ctx.getText(),
-          type: _constants.DATA_TYPE.STRING
+          value: value,
+          type: value.startsWith("N'") ? _constants.DATA_TYPE.EXPRESSION : _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();
+        var _value = (0, _lodash.last)(ctx.children).getText();
         return {
-          value: "".concat(dollar).concat(sign).concat(value),
+          value: "".concat(dollar).concat(sign).concat(_value),
           type: _constants.DATA_TYPE.STRING
         };
       }
@@ -316,7 +411,7 @@ var MssqlASTGen = /*#__PURE__*/function (_TSqlParserVisitor) {
         };
       }
       return {
-        value: ctx.getText(),
+        value: (0, _helpers.getOriginalText)(ctx),
         type: _constants.DATA_TYPE.EXPRESSION
       };
     }
@@ -341,6 +436,14 @@ var MssqlASTGen = /*#__PURE__*/function (_TSqlParserVisitor) {
         type: _constants.DATA_TYPE.EXPRESSION
       };
     }
+  }, {
+    key: "visitSCALAR_FUNCTION",
+    value: function visitSCALAR_FUNCTION(ctx) {
+      return {
+        value: (0, _helpers.getOriginalText)(ctx),
+        type: _constants.DATA_TYPE.EXPRESSION
+      };
+    }
 
     // unary_operator_expression
     //   : '~' expression
@@ -368,7 +471,32 @@ var MssqlASTGen = /*#__PURE__*/function (_TSqlParserVisitor) {
   }, {
     key: "visitData_type",
     value: function visitData_type(ctx) {
-      return ctx.getText();
+      var id = ctx.id_().accept(this);
+      if (ctx.MAX()) {
+        return "".concat(id, "(MAX)");
+      }
+      if (ctx.IDENTITY()) {
+        if (ctx.DECIMAL().length) {
+          var seedAndInc = ctx.DECIMAL().map(function (decimal) {
+            return decimal.getText();
+          });
+          return "".concat(id, " IDENTITY(").concat(seedAndInc, ")");
+        }
+        return "".concat(id, " IDENTITY");
+      }
+      if (ctx.DOUBLE()) {
+        if (ctx.PRECISION()) {
+          return "".concat(id, "(").concat(ctx.PRECISION().getText(), ")");
+        }
+        return id;
+      }
+      if (ctx.DECIMAL().length) {
+        var scaleAndPrec = ctx.DECIMAL().map(function (decimal) {
+          return decimal.getText();
+        });
+        return "".concat(id, "(").concat(scaleAndPrec.join(','), ")");
+      }
+      return id;
     }
   }, {
     key: "visitPrimitive_expression",
@@ -376,7 +504,7 @@ var MssqlASTGen = /*#__PURE__*/function (_TSqlParserVisitor) {
       if (ctx.NULL_()) {
         return {
           value: ctx.getText(),
-          type: _constants.DATA_TYPE.EXPRESSION
+          type: _constants.DATA_TYPE.BOOLEAN
         };
       }
       if (ctx.primitive_constant()) {
@@ -387,6 +515,819 @@ var MssqlASTGen = /*#__PURE__*/function (_TSqlParserVisitor) {
         type: _constants.DATA_TYPE.EXPRESSION
       };
     }
+
+    // ddl_clause
+    //   | alter_table
+    //   | create_index
+    //   | create_table
+    //   ;
+    // more details at: packages/dbml-core/src/parse/ANTLR/parsers/mssql/TSqlParser.g4 line 73
+  }, {
+    key: "visitDdl_clause",
+    value: function visitDdl_clause(ctx) {
+      if (ctx.create_table()) {
+        ctx.create_table().accept(this);
+        return;
+      }
+      if (ctx.alter_table()) {
+        ctx.alter_table().accept(this);
+        return;
+      }
+      if (ctx.create_index()) {
+        ctx.create_index().accept(this);
+      }
+    }
+
+    // create_table
+    //   : CREATE TABLE table_name '(' column_def_table_constraints (','? table_indices)* ','? ')' (
+    //       LOCK simple_id
+    //   )? table_options* (ON id_ | DEFAULT | on_partition_or_filegroup)? (TEXTIMAGE_ON id_ | DEFAULT)? ';'?
+    //   ;
+  }, {
+    key: "visitCreate_table",
+    value: function visitCreate_table(ctx) {
+      var _this8 = this,
+        _this$data$refs,
+        _this$data$refs2;
+      var _getSchemaAndTableNam = getSchemaAndTableName(ctx.table_name().accept(this)),
+        schemaName = _getSchemaAndTableNam.schemaName,
+        tableName = _getSchemaAndTableNam.tableName;
+      var columnDefTableConstraints = ctx.column_def_table_constraints().accept(this);
+      var tableIndices = ctx.table_indices().map(function (tableIndex) {
+        return tableIndex.accept(_this8);
+      });
+      var _splitColumnDefTableC = splitColumnDefTableConstraints(columnDefTableConstraints),
+        fieldsData = _splitColumnDefTableC.fieldsData,
+        indexes = _splitColumnDefTableC.indexes,
+        tableRefs = _splitColumnDefTableC.tableRefs;
+      var _parseFieldsAndInline = parseFieldsAndInlineRefsFromFieldsData(fieldsData, tableName, schemaName),
+        inlineRefs = _parseFieldsAndInline.inlineRefs,
+        fields = _parseFieldsAndInline.fields;
+      (_this$data$refs = this.data.refs).push.apply(_this$data$refs, _toConsumableArray((0, _lodash.flatten)(inlineRefs)));
+      (_this$data$refs2 = this.data.refs).push.apply(_this$data$refs2, _toConsumableArray(tableRefs.map(function (tableRef) {
+        tableRef.endpoints[0].tableName = tableName;
+        tableRef.endpoints[0].schemaName = schemaName;
+        return tableRef;
+      })));
+      var table = new _AST.Table({
+        name: tableName,
+        schemaName: schemaName,
+        fields: fields,
+        indexes: tableIndices.concat(indexes)
+      });
+      this.data.tables.push(table);
+    }
+
+    // table_name
+    //   : (database = id_ '.' schema = id_? '.' | schema = id_ '.')? (
+    //       table = id_
+    //       | blocking_hierarchy = BLOCKING_HIERARCHY
+    //   )
+    //   ;
+  }, {
+    key: "visitTable_name",
+    value: function visitTable_name(ctx) {
+      var _this9 = this;
+      return ctx.id_().map(function (id) {
+        return id.accept(_this9);
+      });
+    }
+
+    // column_def_table_constraints
+    //   : column_def_table_constraint (','? column_def_table_constraint)*
+    //   ;
+  }, {
+    key: "visitColumn_def_table_constraints",
+    value: function visitColumn_def_table_constraints(ctx) {
+      var _this10 = this;
+      return ctx.column_def_table_constraint().map(function (columnDef) {
+        return columnDef.accept(_this10);
+      }).filter(function (columnDef) {
+        return columnDef;
+      });
+    }
+
+    // column_def_table_constraint
+    //   : column_definition
+    //   | materialized_column_definition
+    //   | table_constraint
+    //   ;
+  }, {
+    key: "visitColumn_def_table_constraint",
+    value: function visitColumn_def_table_constraint(ctx) {
+      if (ctx.column_definition()) {
+        return ctx.column_definition().accept(this);
+      }
+      if (ctx.table_constraint()) {
+        return ctx.table_constraint().accept(this);
+      }
+      return null;
+    }
+
+    // column_definition
+    // : id_ (data_type | AS expression PERSISTED?) column_definition_element* column_index?
+    // ;
+  }, {
+    key: "visitColumn_definition",
+    value: function visitColumn_definition(ctx) {
+      var _this11 = this;
+      var columnName = ctx.id_().accept(this);
+      var type = '';
+      if (ctx.data_type()) {
+        type = ctx.data_type().accept(this);
+      } else if (ctx.expression()) {
+        // { value: "(first_name + ' ' + last_name)", type: 'expression' }
+        var expression = ctx.expression().accept(this);
+        var as = ctx.AS().getText();
+        var persisted = ctx.PERSISTED() ? " ".concat(ctx.PERSISTED().getText()) : '';
+        type = "".concat(as, " ").concat(expression.value).concat(persisted);
+      }
+      var field = new _AST.Field({
+        name: columnName,
+        type: {
+          type_name: type,
+          schemaName: null
+        }
+      });
+      var definition = {
+        kind: _constants.TABLE_CONSTRAINT_KIND.FIELD,
+        value: {
+          field: field,
+          inline_refs: []
+        }
+      };
+      var columnDefinitions = ctx.column_definition_element().map(function (columnDef) {
+        return columnDef.accept(_this11);
+      });
+
+      // e.g.
+      // [
+      //   { kind: 'not_null', value: true },
+      //   { kind: 'dbdefault', value: { value: 'GETDATE()', type: 'expression' } }
+      // ]
+      columnDefinitions.filter(function (columnDef) {
+        return columnDef;
+      }).forEach(function (columnDef) {
+        switch (columnDef.kind) {
+          case _constants.COLUMN_CONSTRAINT_KIND.DEFAULT:
+            field.dbdefault = columnDef.value;
+            break;
+          case _constants.COLUMN_CONSTRAINT_KIND.INCREMENT:
+            field.increment = columnDef.value;
+            break;
+          case _constants.COLUMN_CONSTRAINT_KIND.NOT_NULL:
+            field.not_null = columnDef.value;
+            break;
+          case _constants.COLUMN_CONSTRAINT_KIND.PK:
+            field.pk = columnDef.value;
+            break;
+          case _constants.COLUMN_CONSTRAINT_KIND.UNIQUE:
+            field.unique = columnDef.value;
+            break;
+          case _constants.COLUMN_CONSTRAINT_KIND.INLINE_REF:
+            definition.value.inline_refs.push(columnDef.value);
+            break;
+          case _constants.COLUMN_CONSTRAINT_KIND.CHECK:
+            field.type.type_name = "".concat(field.type.type_name, " ").concat(columnDef.value);
+            break;
+          default:
+            break;
+        }
+      });
+
+      // skip column index since it is just the name of the index that a column belongs to
+
+      return definition;
+    }
+
+    // column_definition_element
+    //   : FILESTREAM
+    //   | COLLATE collation_name = id_
+    //   | SPARSE
+    //   | MASKED WITH '(' FUNCTION '=' mask_function = STRING ')'
+    //   | (CONSTRAINT constraint = id_)? DEFAULT constant_expr = expression
+    //   | IDENTITY ('(' seed = DECIMAL ',' increment = DECIMAL ')')?
+    //   | NOT FOR REPLICATION
+    //   | GENERATED ALWAYS AS (ROW | TRANSACTION_ID | SEQUENCE_NUMBER) (START | END) HIDDEN_KEYWORD?
+    //   // NULL / NOT NULL is a constraint
+    //   | ROWGUIDCOL
+    //   | ENCRYPTED WITH '(' COLUMN_ENCRYPTION_KEY '=' key_name = STRING ',' ENCRYPTION_TYPE '=' (
+    //       DETERMINISTIC
+    //       | RANDOMIZED
+    //   ) ',' ALGORITHM '=' algo = STRING ')'
+    //   | column_constraint
+    //   ;
+  }, {
+    key: "visitColumn_definition_element",
+    value: function visitColumn_definition_element(ctx) {
+      if (ctx.DEFAULT()) {
+        return {
+          kind: _constants.COLUMN_CONSTRAINT_KIND.DEFAULT,
+          value: ctx.expression().accept(this)
+        };
+      }
+      if (ctx.IDENTITY()) {
+        return {
+          kind: _constants.COLUMN_CONSTRAINT_KIND.INCREMENT,
+          value: true
+        };
+      }
+      if (ctx.column_constraint()) {
+        return ctx.column_constraint().accept(this);
+      }
+      return null;
+    }
+
+    // column_constraint
+    //   : (CONSTRAINT constraint = id_)? (
+    //       null_notnull
+    //       | ( (PRIMARY KEY | UNIQUE) clustered? primary_key_options)
+    //       | ( (FOREIGN KEY)? foreign_key_options)
+    //       | check_constraint
+    //   )
+    //   ;
+  }, {
+    key: "visitColumn_constraint",
+    value: function visitColumn_constraint(ctx) {
+      if (ctx.null_notnull()) {
+        var notNull = false;
+        var nullNotnull = ctx.null_notnull().accept(this);
+        if (nullNotnull.includes('NOT')) notNull = true;
+        return {
+          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.foreign_key_options()) {
+        var _ctx$foreign_key_opti = ctx.foreign_key_options().accept(this),
+          refTableName = _ctx$foreign_key_opti.refTableName,
+          refSchemaName = _ctx$foreign_key_opti.refSchemaName,
+          fieldNames = _ctx$foreign_key_opti.fieldNames,
+          onDelete = _ctx$foreign_key_opti.onDelete,
+          onUpdate = _ctx$foreign_key_opti.onUpdate;
+        return {
+          kind: _constants.COLUMN_CONSTRAINT_KIND.INLINE_REF,
+          value: {
+            endpoints: [{
+              tableName: null,
+              schemaName: null,
+              fieldNames: null,
+              relation: '*'
+            }, {
+              tableName: refTableName,
+              schemaName: refSchemaName,
+              fieldNames: fieldNames,
+              relation: '1'
+            }],
+            onDelete: onDelete,
+            onUpdate: onUpdate
+          }
+        };
+      }
+
+      // we do not handle check constraint since it is complicated and hard to extract enum from it
+      if (ctx.check_constraint()) {
+        return {
+          kind: _constants.COLUMN_CONSTRAINT_KIND.CHECK,
+          value: ctx.check_constraint().accept(this)
+        };
+      }
+      return null;
+    }
+
+    // check_constraint
+    //   : CHECK (NOT FOR REPLICATION)? '(' search_condition ')'
+    //   ;
+  }, {
+    key: "visitCheck_constraint",
+    value: function visitCheck_constraint(ctx) {
+      return (0, _helpers.getOriginalText)(ctx);
+    }
+  }, {
+    key: "visitNull_notnull",
+    value: function visitNull_notnull(ctx) {
+      return (0, _helpers.getOriginalText)(ctx);
+    }
+
+    // foreign_key_options
+    //   : REFERENCES table_name '(' pk = column_name_list ')' (on_delete | on_update)* (
+    //       NOT FOR REPLICATION
+    //   )?
+    //   ;
+  }, {
+    key: "visitForeign_key_options",
+    value: function visitForeign_key_options(ctx) {
+      var _this12 = this;
+      var names = ctx.table_name().accept(this);
+      var _getSchemaAndTableNam2 = getSchemaAndTableName(names),
+        refSchemaName = _getSchemaAndTableNam2.schemaName,
+        refTableName = _getSchemaAndTableNam2.tableName;
+      var fieldNames = ctx.column_name_list().accept(this);
+      var onDelete = ctx.on_delete() ? ctx.on_delete().map(function (action) {
+        return action.accept(_this12);
+      })[0] : null;
+      var onUpdate = ctx.on_update() ? ctx.on_update().map(function (action) {
+        return action.accept(_this12);
+      })[0] : null;
+      return {
+        refTableName: refTableName,
+        refSchemaName: refSchemaName,
+        onDelete: onDelete,
+        onUpdate: onUpdate,
+        fieldNames: fieldNames
+      };
+    }
+
+    // column_name_list
+    //   : col += id_ (',' col += id_)*
+    //   ;
+  }, {
+    key: "visitColumn_name_list",
+    value: function visitColumn_name_list(ctx) {
+      var _this13 = this;
+      return ctx.id_().map(function (id) {
+        return id.accept(_this13);
+      });
+    }
+
+    // on_delete
+    //   : ON DELETE (NO ACTION | CASCADE | SET NULL_ | SET DEFAULT)
+    //   ;
+  }, {
+    key: "visitOn_delete",
+    value: function visitOn_delete(ctx) {
+      if (ctx.NO()) {
+        return 'NO ACTION';
+      }
+      if (ctx.CASCADE()) {
+        return 'CASCADE';
+      }
+      if (ctx.NULL_()) {
+        return 'SET NULL';
+      }
+      return 'SET DEFAULT';
+    }
+
+    // on_update
+    //   : ON UPDATE (NO ACTION | CASCADE | SET NULL_ | SET DEFAULT)
+    //   ;
+  }, {
+    key: "visitOn_update",
+    value: function visitOn_update(ctx) {
+      if (ctx.NO()) {
+        return 'NO ACTION';
+      }
+      if (ctx.CASCADE()) {
+        return 'CASCADE';
+      }
+      if (ctx.NULL_()) {
+        return 'SET NULL';
+      }
+      return 'SET DEFAULT';
+    }
+
+    // table_constraint
+    //   : (CONSTRAINT constraint = id_)? (
+    //       ((PRIMARY KEY | UNIQUE) clustered? '(' column_name_list_with_order ')' primary_key_options)
+    //       | ( FOREIGN KEY '(' fk = column_name_list ')' foreign_key_options)
+    //       | ( CONNECTION '(' connection_node ( ',' connection_node)* ')')
+    //       | ( DEFAULT constant_expr = expression FOR column = id_ (WITH VALUES)?)
+    //       | check_constraint
+    //   )
+    //   ;
+  }, {
+    key: "visitTable_constraint",
+    value: function visitTable_constraint(ctx) {
+      var _this14 = this;
+      var name = ctx.id_() ? ctx.id_().map(function (id) {
+        return id.accept(_this14);
+      })[0] : '';
+      if (ctx.PRIMARY() || ctx.UNIQUE()) {
+        var columns = ctx.column_name_list_with_order().accept(this);
+        var index = new _AST.Index({
+          name: name,
+          columns: columns
+        });
+        if (ctx.PRIMARY()) {
+          index.pk = true;
+        }
+        if (ctx.UNIQUE()) {
+          index.unique = true;
+        }
+        return {
+          kind: ctx.PRIMARY() ? _constants.TABLE_CONSTRAINT_KIND.PK : _constants.TABLE_CONSTRAINT_KIND.UNIQUE,
+          value: index
+        };
+      }
+      if (ctx.FOREIGN()) {
+        var _columns = ctx.column_name_list().accept(this);
+        var _ctx$foreign_key_opti2 = ctx.foreign_key_options().accept(this),
+          refTableName = _ctx$foreign_key_opti2.refTableName,
+          refSchemaName = _ctx$foreign_key_opti2.refSchemaName,
+          fieldNames = _ctx$foreign_key_opti2.fieldNames,
+          onDelete = _ctx$foreign_key_opti2.onDelete,
+          onUpdate = _ctx$foreign_key_opti2.onUpdate;
+        return {
+          kind: _constants.TABLE_CONSTRAINT_KIND.FK,
+          value: {
+            name: name,
+            endpoints: [{
+              tableName: null,
+              schemaName: null,
+              fieldNames: _columns,
+              relation: '*'
+            }, {
+              tableName: refTableName,
+              schemaName: refSchemaName,
+              fieldNames: fieldNames,
+              relation: '1'
+            }],
+            onDelete: onDelete,
+            onUpdate: onUpdate
+          }
+        };
+      }
+      if (ctx.DEFAULT()) {
+        var column = ctx.column.accept(this);
+        var expression = ctx.expression().accept(this);
+        return {
+          kind: _constants.TABLE_CONSTRAINT_KIND.DEFAULT,
+          value: {
+            column: column,
+            defaultValue: expression
+          }
+        };
+      }
+      return null;
+    }
+
+    // column_name_list_with_order
+    //   : id_ (ASC | DESC)? (',' id_ (ASC | DESC)?)*
+    //   ;
+  }, {
+    key: "visitColumn_name_list_with_order",
+    value: function visitColumn_name_list_with_order(ctx) {
+      var _this15 = this;
+      return ctx.id_().map(function (id) {
+        return {
+          value: id.accept(_this15),
+          type: 'column'
+        };
+      });
+    }
+
+    // table_indices
+    //   : INDEX id_ UNIQUE? clustered? '(' column_name_list_with_order ')'
+    //   | INDEX id_ CLUSTERED COLUMNSTORE
+    //   | INDEX id_ NONCLUSTERED? COLUMNSTORE '(' column_name_list ')' create_table_index_options? (
+    //       ON id_
+    //   )?
+    //   ;
+  }, {
+    key: "visitTable_indices",
+    value: function visitTable_indices(ctx) {
+      var _this16 = this;
+      var index = new _AST.Index({
+        name: ctx.id_().map(function (id) {
+          return id.accept(_this16);
+        })[0],
+        unique: !!ctx.UNIQUE(),
+        columns: ctx.column_name_list_with_order().accept(this)
+      });
+      return index;
+    }
+
+    // alter_table
+    //   : ALTER TABLE table_name (
+    //       SET '(' LOCK_ESCALATION '=' (AUTO | TABLE | DISABLE) ')'
+    //       | ADD column_def_table_constraints
+    //       | ALTER COLUMN (column_definition | column_modifier)
+    //       | DROP COLUMN id_ (',' id_)*
+    //       | DROP CONSTRAINT constraint = id_
+    //       | WITH (CHECK | NOCHECK) ADD (CONSTRAINT constraint = id_)? (
+    //           FOREIGN KEY '(' fk = column_name_list ')' REFERENCES table_name (
+    //               '(' pk = column_name_list ')'
+    //           )? (on_delete | on_update)*
+    //           | CHECK '(' search_condition ')'
+    //       )
+    //       | (NOCHECK | CHECK) CONSTRAINT constraint = id_
+    //       | (ENABLE | DISABLE) TRIGGER id_?
+    //       | REBUILD table_options
+    //       | SWITCH switch_partition
+    //   ) ';'?
+    //   ;
+  }, {
+    key: "visitAlter_table",
+    value: function visitAlter_table(ctx) {
+      var _this$data$refs3, _this$data$refs4, _table$fields, _table$indexes;
+      // table_name() returns an array because there are multiple table_name in the clause (REFERENCES table_name ...)
+      var names = ctx.table_name()[0].accept(this);
+      var _getSchemaAndTableNam3 = getSchemaAndTableName(names),
+        schemaName = _getSchemaAndTableNam3.schemaName,
+        tableName = _getSchemaAndTableNam3.tableName;
+      var table = this.data.tables.find(function (t) {
+        return t.name === tableName && t.schemaName === schemaName;
+      });
+      if (!table) return; // ALTER TABLE should appear after CREATE TABLE, so skip if table is not created yet
+
+      var columnDefTableConstraints = ctx.column_def_table_constraints() ? ctx.column_def_table_constraints().accept(this) : [];
+      var _splitColumnDefTableC2 = splitColumnDefTableConstraints(columnDefTableConstraints),
+        fieldsData = _splitColumnDefTableC2.fieldsData,
+        indexes = _splitColumnDefTableC2.indexes,
+        tableRefs = _splitColumnDefTableC2.tableRefs,
+        columnDefaults = _splitColumnDefTableC2.columnDefaults;
+      var _parseFieldsAndInline2 = parseFieldsAndInlineRefsFromFieldsData(fieldsData, tableName, schemaName),
+        inlineRefs = _parseFieldsAndInline2.inlineRefs,
+        fields = _parseFieldsAndInline2.fields;
+      (_this$data$refs3 = this.data.refs).push.apply(_this$data$refs3, _toConsumableArray((0, _lodash.flatten)(inlineRefs)));
+      (_this$data$refs4 = this.data.refs).push.apply(_this$data$refs4, _toConsumableArray(tableRefs.map(function (tableRef) {
+        tableRef.endpoints[0].tableName = tableName;
+        tableRef.endpoints[0].schemaName = schemaName;
+        return tableRef;
+      })));
+      (_table$fields = table.fields).push.apply(_table$fields, _toConsumableArray(fields));
+      (_table$indexes = table.indexes).push.apply(_table$indexes, _toConsumableArray(indexes));
+      columnDefaults.forEach(function (columnDefault) {
+        var field = table.fields.find(function (f) {
+          return f.name === columnDefault.column;
+        });
+        if (!field) return;
+        field.dbdefault = columnDefault.defaultValue;
+      });
+    }
+
+    // create_index
+    //   : CREATE UNIQUE? clustered? INDEX id_ ON table_name '(' column_name_list_with_order ')' (
+    //       INCLUDE '(' column_name_list ')'
+    //   )? (WHERE where = search_condition)? (create_index_options)? (ON id_)? ';'?
+    //   ;
+  }, {
+    key: "visitCreate_index",
+    value: function visitCreate_index(ctx) {
+      var _getSchemaAndTableNam4 = getSchemaAndTableName(ctx.table_name().accept(this)),
+        schemaName = _getSchemaAndTableNam4.schemaName,
+        tableName = _getSchemaAndTableNam4.tableName;
+      var table = this.data.tables.find(function (t) {
+        return t.name === tableName && t.schemaName === schemaName;
+      });
+      if (!table) return; // ALTER TABLE should appear after CREATE TABLE, so skip if table is not created yet
+
+      var index = new _AST.Index({
+        name: ctx.id_()[0].accept(this),
+        unique: !!ctx.UNIQUE(),
+        columns: ctx.column_name_list_with_order().accept(this)
+      });
+      table.indexes.push(index);
+    }
+
+    // another_statement
+    //   : alter_queue
+    //   | checkpoint_statement
+    //   | conversation_statement
+    //   | create_contract
+    //   | create_queue
+    //   | cursor_statement
+    //   | declare_statement
+    //   | execute_statement
+    //   | kill_statement
+    //   | message_statement
+    //   | reconfigure_statement
+    //   | security_statement
+    //   | set_statement
+    //   | setuser_statement
+    //   | shutdown_statement
+    //   | transaction_statement
+    //   | use_statement
+    //   ;
+  }, {
+    key: "visitAnother_statement",
+    value: function visitAnother_statement(ctx) {
+      if (ctx.execute_statement()) {
+        ctx.execute_statement().accept(this);
+      }
+    }
+
+    // execute_statement
+    //   : EXECUTE execute_body ';'?
+    //   ;
+  }, {
+    key: "visitExecute_statement",
+    value: function visitExecute_statement(ctx) {
+      ctx.execute_body().accept(this);
+    }
+
+    // execute_body
+    //   : (return_status = LOCAL_ID '=')? (func_proc_name_server_database_schema | execute_var_string) execute_statement_arg?
+    //   | '(' execute_var_string (',' execute_var_string)* ')' (AS (LOGIN | USER) '=' STRING)? (
+    //       AT_KEYWORD linkedServer = id_
+    //   )?
+    //   | AS ( (LOGIN | USER) '=' STRING | CALLER)
+    //   ;
+  }, {
+    key: "visitExecute_body",
+    value: function visitExecute_body(ctx) {
+      var funcNames = ctx.func_proc_name_server_database_schema() ? ctx.func_proc_name_server_database_schema().accept(this) : [];
+      var funcName = (0, _lodash.last)(funcNames);
+      if (funcName !== ADD_DESCRIPTION_FUNCTION_NAME) {
+        return;
+      }
+      if (ctx.execute_statement_arg()) {
+        // [
+        //   { name: '@name', value: "N'Column_Description'", type: 'expression' },
+        //   { name: '@value', value: '$-1', type: 'string' },
+        //   { name: '@level0type', value: "N'Schema'", type: 'expression' },
+        //   { name: '@level0name', value: 'dbo', type: 'string' },
+        //   { name: '@level1type', value: "N'Table'", type: 'expression' },
+        //   { name: '@level1name', value: 'orders', type: 'string' },
+        //   { name: '@level2type', value: "N'Column'", type: 'expression' },
+        //   { name: '@level2name', value: 'status', type: 'string' }
+        // ]
+        var args = ctx.execute_statement_arg().accept(this);
+
+        // {
+        //   name: "N'Table_Description'",
+        //   value: 'This is a note in table "orders"',
+        //   level0type: "N'Schema'",
+        //   level0name: 'dbo',
+        //   level1type: "N'Table'",
+        //   level1name: 'orders'
+        // }
+        var argsObj = args.reduce(function (acc, arg) {
+          var name = arg.name.slice(1);
+          acc[name] = arg.value;
+          return acc;
+        }, {});
+        if (!argsObj['name'].includes('Description')) {
+          return;
+        }
+
+        // https://learn.microsoft.com/en-us/sql/relational-databases/system-stored-procedures/sp-addextendedproperty-transact-sql?view=sql-server-ver16#----level0type
+        var level0Type = argsObj['level0type'].toLowerCase();
+        if (!level0Type.includes('schema')) return;
+        var schemaName = argsObj['level0name'] !== 'dbo' ? argsObj['level0name'] : undefined;
+        var level1Type = argsObj['level1type'].toLowerCase();
+        var tableName = level1Type.includes('table') ? argsObj['level1name'] : null;
+        var table = this.data.tables.find(function (t) {
+          return t.name === tableName && t.schemaName === schemaName;
+        });
+        if (!table) return;
+        if (!argsObj['level2type']) {
+          table.note = {
+            value: argsObj.value
+          };
+          return;
+        }
+        var level2Type = argsObj['level2type'].toLowerCase();
+        var columnName = level2Type.includes('column') ? argsObj['level2name'] : null;
+        var field = table.fields.find(function (f) {
+          return f.name === columnName;
+        });
+        if (!field) return;
+        field.note = {
+          value: argsObj.value
+        };
+      }
+    }
+
+    // func_proc_name_server_database_schema
+    //   : server = id_? '.' database = id_? '.' schema = id_? '.' procedure = id_
+    //   | func_proc_name_database_schema
+    //   ;
+  }, {
+    key: "visitFunc_proc_name_server_database_schema",
+    value: function visitFunc_proc_name_server_database_schema(ctx) {
+      var _this17 = this;
+      if (ctx.func_proc_name_database_schema()) {
+        return ctx.func_proc_name_database_schema().accept(this);
+      } else {
+        return ctx.id_().map(function (id) {
+          return id.accept(_this17);
+        });
+      }
+    }
+
+    // func_proc_name_database_schema
+    //   : database = id_? '.' schema = id_? '.' procedure = id_
+    //   | func_proc_name_schema
+    //   ;
+  }, {
+    key: "visitFunc_proc_name_database_schema",
+    value: function visitFunc_proc_name_database_schema(ctx) {
+      var _this18 = this;
+      if (ctx.func_proc_name_schema()) {
+        return ctx.func_proc_name_schema().accept(this);
+      } else {
+        return ctx.id_().map(function (id) {
+          return id.accept(_this18);
+        });
+      }
+    }
+
+    // func_proc_name_schema
+    //   : ((schema = id_) '.')? procedure = id_
+    //   ;
+  }, {
+    key: "visitFunc_proc_name_schema",
+    value: function visitFunc_proc_name_schema(ctx) {
+      var _this19 = this;
+      return ctx.id_().map(function (id) {
+        return id.accept(_this19);
+      });
+    }
+
+    // execute_statement_arg
+    //   : execute_statement_arg_unnamed (',' execute_statement_arg)*     //Unnamed params can continue unnamed
+    //   | execute_statement_arg_named (',' execute_statement_arg_named)* //Named can only be continued by unnamed
+    //   ;
+  }, {
+    key: "visitExecute_statement_arg",
+    value: function visitExecute_statement_arg(ctx) {
+      var _this20 = this;
+      if (ctx.execute_statement_arg_unnamed()) {
+        return ctx.execute_statement_arg_unnamed().map(function (item) {
+          return item.accept(_this20);
+        });
+      }
+      return ctx.execute_statement_arg_named().map(function (item) {
+        return item.accept(_this20);
+      });
+    }
+
+    // execute_statement_arg_named
+    //   : name = LOCAL_ID '=' value = execute_parameter
+    //   ;
+  }, {
+    key: "visitExecute_statement_arg_named",
+    value: function visitExecute_statement_arg_named(ctx) {
+      var _ctx$execute_paramete = ctx.execute_parameter().accept(this),
+        value = _ctx$execute_paramete.value,
+        type = _ctx$execute_paramete.type;
+      return {
+        name: ctx.LOCAL_ID().getText(),
+        value: value,
+        type: type
+      };
+    }
+
+    // execute_parameter
+    //   : (constant | LOCAL_ID (OUTPUT | OUT)? | id_ | DEFAULT | NULL_)
+    //   ;
+  }, {
+    key: "visitExecute_parameter",
+    value: function visitExecute_parameter(ctx) {
+      if (ctx.constant()) {
+        return ctx.constant().accept(this);
+      }
+      return ctx.getText();
+    }
+
+    // constant
+    //   : STRING // string, datetime or uniqueidentifier
+    //   | BINARY
+    //   | '-'? (DECIMAL | REAL | FLOAT)                    // float or decimal
+    //   | '-'? dollar = '$' ('-' | '+')? (DECIMAL | FLOAT) // money
+    //   | parameter
+    //   ;
+  }, {
+    key: "visitConstant",
+    value: function visitConstant(ctx) {
+      if (ctx.STRING() || ctx.BINARY()) {
+        var value = getStringFromRawString(ctx.getText());
+        return {
+          value: value,
+          type: value.startsWith("N'") ? _constants.DATA_TYPE.EXPRESSION : _constants.DATA_TYPE.STRING
+        };
+      }
+      if (ctx.DOLLAR()) {
+        var _value2 = ctx.getText();
+        return {
+          value: _value2,
+          type: _constants.DATA_TYPE.STRING
+        };
+      }
+      if (ctx.REAL() || ctx.DECIMAL() || ctx.FLOAT()) {
+        return {
+          value: ctx.getText(),
+          type: _constants.DATA_TYPE.NUMBER
+        };
+      }
+      return {
+        value: (0, _helpers.getOriginalText)(ctx),
+        type: _constants.DATA_TYPE.EXPRESSION
+      };
+    }
   }]);
   return MssqlASTGen;
 }(_TSqlParserVisitor2["default"]);