Return types after import

This commit is contained in:
1ilit
2024-06-27 11:12:09 +03:00
parent 9e2684e7a9
commit d79b0c205d
5 changed files with 24 additions and 13 deletions

View File

@@ -10,7 +10,7 @@ import { fromPostgres } from "./postgres";
import { fromSQLite } from "./sqlite";
export function importSQL(ast, toDb = DB.MYSQL, diagramDb = DB.GENERIC) {
let diagram = { tables: [], relationships: [] };
let diagram;
switch (toDb) {
case DB.SQLITE:
diagram = fromSQLite(ast, diagramDb);

View File

@@ -20,6 +20,7 @@ const affinity = {
export function fromPostgres(ast, diagramDb = DB.GENERIC) {
const tables = [];
const relationships = [];
const types = [];
ast.forEach((e) => {
if (e.type === "create") {
@@ -241,5 +242,5 @@ export function fromPostgres(ast, diagramDb = DB.GENERIC) {
relationships.forEach((r, i) => (r.id = i));
return { tables, relationships };
return { tables, relationships, types };
}