@dbml/core 5.4.1 → 5.5.1-alpha.0

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

@@ -1157,6 +1157,40 @@ var MssqlASTGen = exports["default"] = /*#__PURE__*/function (_TSqlParserVisitor
       });
       if (!table) return; // ALTER TABLE should appear after CREATE TABLE, so skip if table is not created yet
 
+      // Handle WITH CHECK/NOCHECK ADD CONSTRAINT FK
+      if (ctx.WITH() && (ctx.CHECK() || ctx.NOCHECK()) && ctx.FOREIGN()) {
+        var constraintName = ctx.constraint ? ctx.constraint.accept(this) : undefined;
+        var localColumns = ctx.fk.accept(this);
+
+        // table_name()[1] is the referenced table (table_name()[0] is the table being altered)
+        var refTableNames = ctx.table_name()[1].accept(this);
+        var _getSchemaAndTableNam4 = getSchemaAndTableName(refTableNames),
+          refSchemaName = _getSchemaAndTableNam4.schemaName,
+          refTableName = _getSchemaAndTableNam4.tableName;
+
+        // pk is optional - if not specified, assume same column names as fk
+        var refColumns = ctx.pk ? ctx.pk.accept(this) : localColumns;
+        var onDelete = ctx.on_delete().length > 0 ? ctx.on_delete()[0].accept(this) : null;
+        var onUpdate = ctx.on_update().length > 0 ? ctx.on_update()[0].accept(this) : null;
+        var ref = {
+          name: constraintName,
+          endpoints: [{
+            tableName: tableName,
+            schemaName: schemaName,
+            fieldNames: localColumns,
+            relation: '*'
+          }, {
+            tableName: refTableName,
+            schemaName: refSchemaName,
+            fieldNames: refColumns,
+            relation: '1'
+          }],
+          onDelete: onDelete,
+          onUpdate: onUpdate
+        };
+        this.data.refs.push(ref);
+        return;
+      }
       var columnDefTableConstraints = ctx.column_def_table_constraints() ? ctx.column_def_table_constraints().accept(this) : [];
       var _splitColumnDefTableC2 = splitColumnDefTableConstraints(columnDefTableConstraints),
         fieldsData = _splitColumnDefTableC2.fieldsData,
@@ -1201,9 +1235,9 @@ var MssqlASTGen = exports["default"] = /*#__PURE__*/function (_TSqlParserVisitor
   }, {
     key: "visitCreate_index",
     value: function visitCreate_index(ctx) {
-      var _getSchemaAndTableNam4 = getSchemaAndTableName(ctx.table_name().accept(this)),
-        schemaName = _getSchemaAndTableNam4.schemaName,
-        tableName = _getSchemaAndTableNam4.tableName;
+      var _getSchemaAndTableNam5 = getSchemaAndTableName(ctx.table_name().accept(this)),
+        schemaName = _getSchemaAndTableNam5.schemaName,
+        tableName = _getSchemaAndTableNam5.tableName;
       var table = this.data.tables.find(function (t) {
         return t.name === tableName && t.schemaName === schemaName;
       });

package/package.json

@@ -1,7 +1,7 @@
 {
   "$schema": "https://json.schemastore.org/package",
   "name": "@dbml/core",
-  "version": "5.4.1",
+  "version": "5.5.1-alpha.0",
   "description": "> TODO: description",
   "author": "Holistics <dev@holistics.io>",
   "license": "Apache-2.0",
@@ -36,7 +36,7 @@
     "lint:fix": "eslint --fix ."
   },
   "dependencies": {
-    "@dbml/parse": "^5.4.1",
+    "@dbml/parse": "^5.5.1-alpha.0",
     "antlr4": "^4.13.1",
     "lodash": "^4.17.15",
     "parsimmon": "^1.13.0",
@@ -63,7 +63,7 @@
     "typescript": "^5.9.3",
     "typescript-eslint": "^8.46.3"
   },
-  "gitHead": "f5c8bf0c9d95948e3874cfb6cc777f65c4180891",
+  "gitHead": "355cbecd2d809b9750aea2854fbc85bf469db257",
   "engines": {
     "node": ">=16"
   }