Export mssql

2024-07-06 14:52:42 +03:00 · 2024-07-06 14:52:42 +03:00 · a3976bf250
commit a3976bf250
parent 1580641e97
1 changed files with 65 additions and 2 deletions
--- a/src/utils/exportSQL/mssql.js
+++ b/src/utils/exportSQL/mssql.js
@ -1,4 +1,67 @@
+import { dbToTypes } from "../../data/datatypes";
+import { parseDefault } from "./shared";
+
 export function toMSSQL(diagram) {
-  diagram.tables;
-  return "hi from mssql";
+  const typeStatements = diagram.types
+    .map(
+      (type) =>
+        `CREATE TYPE [${type.name}] AS TABLE (\n${type.fields
+          .map((f) => `\t[${f.name}] ${f.type}`)
+          .join("\n")}\n);\n`,
+    )
+    .join("\n");
+
+  return `${typeStatements}${diagram.tables
+    .map(
+      (table) =>
+        `${
+          table.comment === "" ? "" : `/**\n${table.comment}\n*/\n`
+        }CREATE TABLE [${table.name}] (\n${table.fields
+          .map(
+            (field) =>
+              `${field.comment === "" ? "" : `\t-- ${field.comment}\n`}\t[${
+                field.name
+              }] ${field.type}${
+                field.notNull ? " NOT NULL" : ""
+              }${field.increment ? " IDENTITY" : ""}${
+                field.unique ? " UNIQUE" : ""
+              }${
+                field.default !== ""
+                  ? ` DEFAULT ${parseDefault(field, diagram.database)}`
+                  : ""
+              }${
+                field.check === "" ||
+                !dbToTypes[diagram.database][field.type].hasCheck
+                  ? ""
+                  : ` CHECK(${field.check})`
+              }`,
+          )
+          .join(",\n")}${
+          table.fields.filter((f) => f.primary).length > 0
+            ? `,\n\tPRIMARY KEY(${table.fields
+                .filter((f) => f.primary)
+                .map((f) => `[${f.name}]`)
+                .join(", ")})`
+            : ""
+        }\n);\nGO\n${`${table.indices
+          .map(
+            (i) =>
+              `\nCREATE ${i.unique ? "UNIQUE " : ""}INDEX [${
+                i.name
+              }]\nON [${table.name}] (${i.fields
+                .map((f) => `[${f}]`)
+                .join(", ")});\nGO\n`,
+          )
+          .join("")}`}`,
+    )
+    .join("\n")}\n${diagram.references
+    .map(
+      (r) =>
+        `ALTER TABLE [${diagram.tables[r.startTableId].name}]\nADD FOREIGN KEY([${
+          diagram.tables[r.startTableId].fields[r.startFieldId].name
+        }]) REFERENCES [${diagram.tables[r.endTableId].name}]([${
+          diagram.tables[r.endTableId].fields[r.endFieldId].name
+        }])\nON UPDATE ${r.updateConstraint.toUpperCase()} ON DELETE ${r.deleteConstraint.toUpperCase()};\nGO`,
+    )
+    .join("\n")}`;
 }