@dbml/core 5.4.0 → 5.5.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.0",
+  "version": "5.5.0",
   "description": "> TODO: description",
   "author": "Holistics <dev@holistics.io>",
   "license": "Apache-2.0",
@@ -25,7 +25,8 @@
     "access": "public"
   },
   "scripts": {
-    "test": "jest",
+    "test": "jest --coverage=false",
+    "coverage": "jest --coverage",
     "benchmark": "jest --projects jest-bench.config.json",
     "build:lib": "babel src --out-dir lib --copy-files",
     "build:parser": "npx babel-node src/parse/buildParser.js",
@@ -35,7 +36,7 @@
     "lint:fix": "eslint --fix ."
   },
   "dependencies": {
-    "@dbml/parse": "^5.4.0",
+    "@dbml/parse": "^5.5.0",
     "antlr4": "^4.13.1",
     "lodash": "^4.17.15",
     "parsimmon": "^1.13.0",
@@ -58,22 +59,11 @@
     "jest": "^29.5.0",
     "jest-bench": "^29.4.1",
     "pegjs-require-import": "0.0.6",
+    "ts-jest": "^29.4.5",
     "typescript": "^5.9.3",
     "typescript-eslint": "^8.46.3"
   },
-  "jest": {
-    "setupFiles": [
-      "./jestHelpers.js"
-    ],
-    "transform": {
-      "^.+\\.js$": "babel-jest",
-      "\\.(?!json$)[^.]*$": "@glen/jest-raw-loader"
-    },
-    "moduleNameMapper": {
-      "^lodash-es$": "lodash"
-    }
-  },
-  "gitHead": "e7837a6418602e2b9d5a350335b669155d5c7ad4",
+  "gitHead": "dc9d32e973037784935ec6d92d22e751945bb194",
   "engines": {
     "node": ">=16"
   }

package/types/export/ModelExporter.d.ts

@@ -1,6 +1,6 @@
 import Database, { NormalizedDatabase } from '../model_structure/database';
 
-export type ExportFormatOption = 'dbml' | 'mysql' | 'postgres' | 'json' | 'mssql' | 'oracle';
+export declare type ExportFormatOption = 'dbml' | 'mysql' | 'postgres' | 'json' | 'mssql' | 'oracle';
 declare class ModelExporter {
     static export(model: Database | NormalizedDatabase, format: ExportFormatOption, isNormalized?: boolean): string;
 }

package/types/parse/Parser.d.ts

@@ -1,7 +1,7 @@
 import { Compiler } from '@dbml/parse';
 import Database, { RawDatabase } from '../model_structure/database';
 
-declare type ParseFormat = 'json'
+export declare type ParseFormat = 'json'
     | 'mysql' | 'mysqlLegacy'
     | 'postgres' | 'postgresLegacy'
     | 'dbml' | 'dbmlv2'