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

Fix parsing alter statements

Browse Source
This commit is contained in:
1ilit 2024-04-23 13:48:31 +03:00
parent 11e83bb12b
commit ca8ee61ee9
1 changed files with 101 additions and 132 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

233
src/utils/astToDiagram.js
View File

@ -8,7 +8,6 @@ import {
export function astToDiagram(ast) { export function astToDiagram(ast) {
const tables = []; const tables = [];
const relationships = []; const relationships = [];
const inlineForeignKeys = [];
ast.forEach((e) => { ast.forEach((e) => {
if (e.type === "create") { if (e.type === "create") {
@ -19,6 +18,7 @@ export function astToDiagram(ast) {
table.color = "#175e7a"; table.color = "#175e7a";
table.fields = []; table.fields = [];
table.indices = []; table.indices = [];
table.id = tables.length;
e.create_definitions.forEach((d) => { e.create_definitions.forEach((d) => {
if (d.resource === "column") { if (d.resource === "column") {
const field = {}; const field = {};
@ -112,14 +112,57 @@ export function astToDiagram(ast) {
}); });
}); });
} else if (d.constraint_type === "FOREIGN KEY") { } else if (d.constraint_type === "FOREIGN KEY") {
inlineForeignKeys.push({ ...d, startTable: e.table[0].table }); const relationship = {};
const startTableId = table.id;
const startTable = e.table[0].table;
const startField = d.definition[0].column;
const endTable = d.reference_definition.table[0].table;
const endField = d.reference_definition.definition[0].column;
const endTableId = tables.findIndex((t) => t.name === endTable);
if (endTableId === -1) return;
const endFieldId = tables[endTableId].fields.findIndex(
(f) => f.name === endField,
);
if (endField === -1) return;
const startFieldId = table.fields.findIndex(
(f) => f.name === startField,
);
if (startFieldId === -1) return;
relationship.name = startTable + "_" + startField + "_fk";
relationship.startTableId = startTableId;
relationship.endTableId = endTableId;
relationship.endFieldId = endFieldId;
relationship.startFieldId = startFieldId;
let updateConstraint = "No action";
let deleteConstraint = "No action";
d.reference_definition.on_action.forEach((c) => {
if (c.type === "on update") {
updateConstraint = c.value.value;
updateConstraint =
updateConstraint[0].toUpperCase() +
updateConstraint.substring(1);
} else if (c.type === "on delete") {
deleteConstraint = c.value.value;
deleteConstraint =
deleteConstraint[0].toUpperCase() +
deleteConstraint.substring(1);
}
});
relationship.updateConstraint = updateConstraint;
relationship.deleteConstraint = deleteConstraint;
relationship.cardinality = Cardinality.ONE_TO_ONE;
relationships.push(relationship);
} }
} }
}); });
table.fields.forEach((f, j) => { table.fields.forEach((f, j) => {
f.id = j; f.id = j;
}); });
table.id = tables.length;
tables.push(table); tables.push(table);
} else if (e.keyword === "index") { } else if (e.keyword === "index") {
const index = {}; const index = {};
@ -141,144 +184,70 @@ export function astToDiagram(ast) {
if (found !== -1) tables[found].indices.forEach((i, j) => (i.id = j)); if (found !== -1) tables[found].indices.forEach((i, j) => (i.id = j));
} }
} else if (e.type === "alter") { } else if (e.type === "alter") {
if ( e.expr.forEach((expr) => {
e.expr[0].action === "add" && if (
e.expr[0].create_definitions.constraint_type === "FOREIGN KEY" expr.action === "add" &&
) { expr.create_definitions.constraint_type === "FOREIGN KEY"
const relationship = {}; ) {
const startTable = e.table[0].table; console.log(e);
const startField = e.expr[0].create_definitions.definition[0].column; const relationship = {};
const endTable = const startTable = e.table[0].table;
e.expr[0].create_definitions.reference_definition.table[0].table; const startField = expr.create_definitions.definition[0].column;
const endField = const endTable =
e.expr[0].create_definitions.reference_definition.definition[0] expr.create_definitions.reference_definition.table[0].table;
.column; const endField =
let updateConstraint = "No action"; expr.create_definitions.reference_definition.definition[0].column;
let deleteConstraint = "No action"; let updateConstraint = "No action";
e.expr[0].create_definitions.reference_definition.on_action.forEach( let deleteConstraint = "No action";
(c) => { expr.create_definitions.reference_definition.on_action.forEach(
if (c.type === "on update") { (c) => {
updateConstraint = c.value.value; if (c.type === "on update") {
updateConstraint = updateConstraint = c.value.value;
updateConstraint[0].toUpperCase() + updateConstraint =
updateConstraint.substring(1); updateConstraint[0].toUpperCase() +
} else if (c.type === "on delete") { updateConstraint.substring(1);
deleteConstraint = c.value.value; } else if (c.type === "on delete") {
deleteConstraint = deleteConstraint = c.value.value;
deleteConstraint[0].toUpperCase() + deleteConstraint =
deleteConstraint.substring(1); deleteConstraint[0].toUpperCase() +
} deleteConstraint.substring(1);
}, }
); },
);
let startTableId = -1; const startTableId = tables.findIndex((t) => t.name === startTable);
let startFieldId = -1; if (startTable === -1) return;
let endTableId = -1;
let endFieldId = -1;
tables.forEach((t) => { const endTableId = tables.findIndex((t) => t.name === endTable);
if (t.name === startTable) { if (endTableId === -1) return;
startTableId = t.id;
return;
}
if (t.name === endTable) { const endFieldId = tables[endTableId].fields.findIndex(
endTableId = t.id; (f) => f.name === endField,
} );
}); if (endField === -1) return;
if (startTableId === -1 || endTableId === -1) return; const startFieldId = tables[startTableId].fields.findIndex(
(f) => f.name === startField,
);
if (startFieldId === -1) return;
tables[startTableId].fields.forEach((f) => { relationship.name = startTable + "_" + startField + "_fk";
if (f.name === startField) { relationship.startTableId = startTableId;
startFieldId = f.id; relationship.startFieldId = startFieldId;
return; relationship.endTableId = endTableId;
} relationship.endFieldId = endFieldId;
relationship.updateConstraint = updateConstraint;
relationship.deleteConstraint = deleteConstraint;
relationship.cardinality = Cardinality.ONE_TO_ONE;
relationships.push(relationship);
console.log(relationship);
if (f.name === endField) { relationships.forEach((r, i) => (r.id = i));
endFieldId = f.id; }
} });
});
if (startFieldId === -1 || endFieldId === -1) return;
relationship.name = startTable + "_" + startField + "_fk";
relationship.startTableId = startTableId;
relationship.startFieldId = startFieldId;
relationship.endTableId = endTableId;
relationship.endFieldId = endFieldId;
relationship.updateConstraint = updateConstraint;
relationship.deleteConstraint = deleteConstraint;
relationship.cardinality = Cardinality.ONE_TO_ONE;
relationships.push(relationship);
relationships.forEach((r, i) => (r.id = i));
}
} }
}); });
inlineForeignKeys.forEach((fk) => {
const relationship = {};
const startTable = fk.startTable;
const startField = fk.definition[0].column;
const endTable = fk.reference_definition.table[0].table;
const endField = fk.reference_definition.definition[0].column;
let updateConstraint = "No action";
let deleteConstraint = "No action";
fk.reference_definition.on_action.forEach((c) => {
if (c.type === "on update") {
updateConstraint = c.value.value;
updateConstraint =
updateConstraint[0].toUpperCase() + updateConstraint.substring(1);
} else if (c.type === "on delete") {
deleteConstraint = c.value.value;
deleteConstraint =
deleteConstraint[0].toUpperCase() + deleteConstraint.substring(1);
}
});
let startTableId = -1;
let startFieldId = -1;
let endTableId = -1;
let endFieldId = -1;
tables.forEach((t) => {
if (t.name === startTable) {
startTableId = t.id;
return;
}
if (t.name === endTable) {
endTableId = t.id;
}
});
if (startTableId === -1 || endTableId === -1) return;
tables[startTableId].fields.forEach((f) => {
if (f.name === startField) {
startFieldId = f.id;
return;
}
if (f.name === endField) {
endFieldId = f.id;
}
});
if (startFieldId === -1 || endFieldId === -1) return;
relationship.name = startTable + "_" + startField + "_fk";
relationship.startTableId = startTableId;
relationship.startFieldId = startFieldId;
relationship.endTableId = endTableId;
relationship.endFieldId = endFieldId;
relationship.updateConstraint = updateConstraint;
relationship.deleteConstraint = deleteConstraint;
relationship.cardinality = Cardinality.ONE_TO_ONE;
relationships.push(relationship);
});
relationships.forEach((r, i) => (r.id = i)); relationships.forEach((r, i) => (r.id = i));
let maxHeight = -1; let maxHeight = -1;