diff --git a/src/utils/exportSQL/mssql.js b/src/utils/exportSQL/mssql.js index 11f8555..dfe2bcd 100644 --- 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")}`; }