From 0908d040e05474eab7bf6a0aa5f121cac4933f01 Mon Sep 17 00:00:00 2001
From: 1ilit <lilitmousheghyan@gmail.com>
Date: Sun, 9 Jun 2024 20:18:12 +0300
Subject: [PATCH] Load database type on open

---
 src/components/EditorHeader/Modal/Modal.jsx | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/src/components/EditorHeader/Modal/Modal.jsx b/src/components/EditorHeader/Modal/Modal.jsx
index d92ed19..f4af4f2 100644
--- a/src/components/EditorHeader/Modal/Modal.jsx
+++ b/src/components/EditorHeader/Modal/Modal.jsx
@@ -5,7 +5,7 @@ import {
   Toast,
   Modal as SemiUIModal,
 } from "@douyinfe/semi-ui";
-import { MODAL, STATUS } from "../../../data/constants";
+import { DB, MODAL, STATUS } from "../../../data/constants";
 import { useState } from "react";
 import { db } from "../../../data/db";
 import {
@@ -52,7 +52,7 @@ export default function Modal({
   setExportData,
 }) {
   const { t } = useTranslation();
-  const { setTables, setRelationships } = useTables();
+  const { setTables, setRelationships, setDatabase } = useTables();
   const { setNotes } = useNotes();
   const { setAreas } = useAreas();
   const { setTypes } = useTypes();
@@ -88,6 +88,11 @@ export default function Modal({
       .get(id)
       .then((diagram) => {
         if (diagram) {
+          if (diagram.database) {
+            setDatabase(diagram.database);
+          } else {
+            setDatabase(DB.GENERIC);
+          }
           setDiagramId(diagram.id);
           setTitle(diagram.name);
           setTables(diagram.tables);