fix(import-database): update database type after importing into an existing generic diagra (#456)

Co-authored-by: Jonathan Fishner <jonathanfishner11@gmail.com>
This commit is contained in:
Guy Ben-Aharon
2024-11-27 16:59:07 +02:00
committed by GitHub
parent ddeef3b134
commit a8fe491c1b

View File

@@ -1,6 +1,6 @@
import { Dialog, DialogContent } from '@/components/dialog/dialog';
import { useDialog } from '@/hooks/use-dialog';
import type { DatabaseType } from '@/lib/domain/database-type';
import { DatabaseType } from '@/lib/domain/database-type';
import React, { useCallback, useEffect, useState } from 'react';
import { ImportDatabase } from '../common/import-database/import-database';
import type { DatabaseEdition } from '@/lib/domain/database-edition';
@@ -30,6 +30,8 @@ export const ImportDatabaseDialog: React.FC<ImportDatabaseDialogProps> = ({
addTables,
addRelationships,
diagramName,
databaseType: currentDatabaseType,
updateDatabaseType,
} = useChartDB();
const [scriptResult, setScriptResult] = useState('');
const { resetRedoStack, resetUndoStack } = useRedoUndoStack();
@@ -282,6 +284,10 @@ export const ImportDatabaseDialog: React.FC<ImportDatabaseDialogProps> = ({
}),
]);
if (currentDatabaseType === DatabaseType.GENERIC) {
await updateDatabaseType(databaseType);
}
setNodes((nodes) =>
nodes.map((node) => ({
...node,
@@ -297,6 +303,8 @@ export const ImportDatabaseDialog: React.FC<ImportDatabaseDialogProps> = ({
closeImportDatabaseDialog();
}, [
databaseEdition,
currentDatabaseType,
updateDatabaseType,
databaseType,
scriptResult,
tables,