titor/drawDB
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Suppost postgres

Browse Source
This commit is contained in:
1ilit 2023-09-19 15:51:26 +03:00
parent 69aabaefa9
commit 36ad44f0db
4 changed files with 142 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/components/control_panel.jsx
View File

@ -37,7 +37,8 @@ import {
exitFullscreen,
ddbDiagramIsValid,
dataURItoBlob,
jsonToSQL,
jsonToMySQL,
jsonToPostgreSQL,
} from "../utils";
import {
AreaContext,
@ -728,7 +729,21 @@ export default function ControlPanel(props) {
{
MySQL: () => {
setVisible(MODAL.CODE);
const src = jsonToSQL({
const src = jsonToMySQL({
tables: tables,
references: relationships,
});
setExportData((prev) => ({
...prev,
data: src,
extension: "sql",
}));
},
},
{
PostgreSQL: () => {
setVisible(MODAL.CODE);
const src = jsonToPostgreSQL({
tables: tables,
references: relationships,
});
@ -739,7 +754,6 @@ export default function ControlPanel(props) {
}));
},
},
{ PostgreSQL: () => {} },
{ DBML: () => {} },
],
function: () => {},

2
src/components/table.jsx
View File

@ -382,7 +382,7 @@ export default function Table(props) {
updateField(props.tableData.id, j, {
type: value,
default: "",
values: [],
values: f.values ? [...f.values] : [],
increment: incr,
});
} else if (isSized(value) || hasPrecision(value)) {

2
src/components/table_overview.jsx
View File

@ -228,7 +228,7 @@ export default function TableOverview(props) {
updateField(i, j, {
type: value,
default: "",
values: [],
values: f.values ? [...f.values] : [],
increment: incr,
});
} else if (isSized(value) || hasPrecision(value)) {

114
src/utils/index.js
View File

@ -47,7 +47,8 @@ function dataURItoBlob(dataUrl) {
return new Blob([intArray], { type: mimeString });
}
function getTypeString(field) {
function getTypeString(field, dbms = "mysql") {
if (dbms === "mysql") {
if (field.type === "UUID") {
return `VARCHAR(36)`;
}
@ -61,6 +62,42 @@ function getTypeString(field) {
return `${field.type}(${field.values.map((v) => `"${v}"`).join(", ")})`;
}
return field.type;
} else if (dbms === "postgres") {
if (field.type === "SMALLINT" && field.increment) {
return "smallserial";
}
if (field.type === "INT" && field.increment) {
return "serial";
}
if (field.type === "BIGINT" && field.increment) {
return "bigserial";
}
if (field.type === "ENUM") {
return `${field.name}_t`;
}
if (field.type === "SET") {
return `${field.name}_t[]`;
}
if (field.type === "TIMESTAMP") {
return "TIMESTAMPTZ";
}
if (field.type === "DATETIME") {
return `timestamp`;
}
if (isSized(field.type)) {
const type =
field.type === "BINARY"
? "bit"
: field.type === "VARBINARY"
? "bit varying"
: field.type.toLowerCase();
return `${type}(${field.size})`;
}
if (hasPrecision(field.type) && field.size !== "") {
return `${field.type}${field.size}`;
}
return field.type.toLowerCase();
}
}
function hasQuotes(type) {
@ -77,7 +114,7 @@ function hasQuotes(type) {
].includes(type);
}
function jsonToSQL(obj) {
function jsonToMySQL(obj) {
return `${obj.tables
.map(
(table) =>
@ -139,6 +176,76 @@ function jsonToSQL(obj) {
.join("\n")}`;
}
function jsonToPostgreSQL(obj) {
return `${obj.tables
.map(
(table) =>
`${table.comment === "" ? "" : `/**\n${table.comment}\n*/\n`}${
table.fields.filter((f) => f.type === "ENUM" || f.type === "SET")
.length > 0
? `${table.fields
.filter((f) => f.type === "ENUM" || f.type === "SET")
.map(
(f) =>
`CREATE TYPE "${f.name}_t" AS ENUM (${f.values
.map((v) => `'${v}'`)
.join(", ")});\n\n`
)}`
: ""
}CREATE TABLE "${table.name}" (\n${table.fields
.map(
(field) =>
`${field.comment === "" ? "" : `\t-- ${field.comment}\n`}\t"${
field.name
}" ${getTypeString(field, "postgres")}${
field.notNull ? " NOT NULL" : ""
}${
field.default !== ""
? ` DEFAULT ${
hasQuotes(field.type) &&
field.default.toLowerCase() !== "null"
? `'${field.default}'`
: `${field.default}`
}`
: ""
}${
field.check === "" || !hasCheck(field.type)
? ""
: ` 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);\n${
table.indices.length > 0
? `${table.indices.map(
(i) =>
`\nCREATE ${i.unique ? "UNIQUE " : ""}INDEX "${
i.name
}"\nON "${table.name}" (${i.fields
.map((f) => `"${f}"`)
.join(", ")});`
)}`
: ""
}`
)
.join("\n")}\n${obj.references
.map(
(r) =>
`ALTER TABLE "${obj.tables[r.startTableId].name}"\nADD FOREIGN KEY("${
obj.tables[r.startTableId].fields[r.startFieldId].name
}") REFERENCES "${obj.tables[r.endTableId].name}"("${
obj.tables[r.endTableId].fields[r.endFieldId].name
}")\nON UPDATE ${r.updateConstraint.toUpperCase()} ON DELETE ${r.deleteConstraint.toUpperCase()};`
)
.join("\n")}`;
}
function arrayIsEqual(arr1, arr2) {
return JSON.stringify(arr1) === JSON.stringify(arr2);
}
@ -383,7 +490,8 @@ export {
jsonDiagramIsValid,
ddbDiagramIsValid,
dataURItoBlob,
jsonToSQL,
jsonToMySQL,
jsonToPostgreSQL,
validateDiagram,
arrayIsEqual,
isSized,