drawDB/src/utils/exportSQL/mssql.js
2024-07-06 14:52:42 +03:00

68 lines
2.3 KiB
JavaScript

import { dbToTypes } from "../../data/datatypes";
import { parseDefault } from "./shared";
export function toMSSQL(diagram) {
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")}`;
}