From 0e041ed91337f99320dda62d870ab74d7ac0aef1 Mon Sep 17 00:00:00 2001
From: 1ilit <lilitmousheghyan@gmail.com>
Date: Sun, 24 Dec 2023 23:26:14 +0200
Subject: [PATCH] Edit templates

---
 src/components/ControlPanel.jsx |   4 ++
 src/data/data.js                |   1 +
 src/pages/Templates.jsx         |   3 +-
 src/pages/editor.jsx            | 100 +++++++++++++++++++++++---------
 4 files changed, 81 insertions(+), 27 deletions(-)

diff --git a/src/components/ControlPanel.jsx b/src/components/ControlPanel.jsx
index 9b3296b..ef12b8e 100644
--- a/src/components/ControlPanel.jsx
+++ b/src/components/ControlPanel.jsx
@@ -1432,6 +1432,7 @@ export default function ControlPanel({
                           }}
                           onDoubleClick={() => {
                             loadDiagram(d.id);
+                            window.name = "d " + d.id;
                             setVisible(MODAL.NONE);
                           }}
                         >
@@ -1704,6 +1705,8 @@ export default function ControlPanel({
         return `Last saved ${lastSaved}`;
       case State.SAVING:
         return "Saving . . .";
+      case State.ERROR:
+        return "Failed to save";
       default:
         return "";
     }
@@ -1729,6 +1732,7 @@ export default function ControlPanel({
                 onMouseLeave={() => setShowEditName(false)}
                 onClick={() => setVisible(MODAL.RENAME)}
               >
+                {window.name.split(" ")[0] === "t" ? "Templates/" : "Diagrams/"}
                 {title}
               </div>
               {(showEditName || visible === MODAL.RENAME) && <IconEdit />}
diff --git a/src/data/data.js b/src/data/data.js
index 2fa1199..d96441e 100644
--- a/src/data/data.js
+++ b/src/data/data.js
@@ -108,6 +108,7 @@ const State = {
   SAVING: 1,
   SAVED: 2,
   LOADING: 3,
+  ERROR: 4,
 };
 
 export {
diff --git a/src/pages/Templates.jsx b/src/pages/Templates.jsx
index 9932c29..fe6ff81 100644
--- a/src/pages/Templates.jsx
+++ b/src/pages/Templates.jsx
@@ -172,7 +172,8 @@ export default function Templates() {
   };
 
   const editTemplate = (id) => {
-    console.log(id);
+    const newWindow = window.open("/editor", "_blank");
+    newWindow.name = "t " + id;
   };
 
   useEffect(() => {
diff --git a/src/pages/editor.jsx b/src/pages/editor.jsx
index 041a368..63b8040 100644
--- a/src/pages/editor.jsx
+++ b/src/pages/editor.jsx
@@ -408,7 +408,7 @@ export default function Editor() {
     );
   };
 
-  const updateNote = (id, values, addToHistory = true) => {
+  const updateNote = (id, values) => {
     setNotes((prev) =>
       prev.map((t) => {
         if (t.id === id) {
@@ -488,45 +488,68 @@ export default function Editor() {
   ]);
 
   useEffect(() => {
-    const save = async () => {
+    const save = async (diagram = true) => {
       if (state !== State.SAVING) {
         return;
       }
-      if (id === 0 && window.name === "") {
-        db.diagrams
-          .add({
-            name: title,
-            lastModified: new Date(),
-            tables: tables,
-            references: relationships,
-            types: types,
-            notes: notes,
-            areas: areas,
-          })
-          .then((id) => {
-            setId(id);
-            window.name = `d ${id}`;
-            setState(State.SAVED);
-            setLastSaved(new Date().toLocaleString());
-          });
+      if (diagram) {
+        if (id === 0 && window.name === "") {
+          db.diagrams
+            .add({
+              name: title,
+              lastModified: new Date(),
+              tables: tables,
+              references: relationships,
+              types: types,
+              notes: notes,
+              areas: areas,
+            })
+            .then((id) => {
+              setId(id);
+              window.name = `d ${id}`;
+              setState(State.SAVED);
+              setLastSaved(new Date().toLocaleString());
+            });
+        } else {
+          db.diagrams
+            .update(id, {
+              name: title,
+              lastModified: new Date(),
+              tables: tables,
+              references: relationships,
+              types: types,
+              notes: notes,
+              areas: areas,
+            })
+            .then(() => {
+              setState(State.SAVED);
+              setLastSaved(new Date().toLocaleString());
+            });
+        }
       } else {
-        db.diagrams
+        db.templates
           .update(id, {
-            name: title,
-            lastModified: new Date(),
+            title: title,
             tables: tables,
-            references: relationships,
+            relationships: relationships,
             types: types,
             notes: notes,
-            areas: areas,
+            subjectAreas: areas,
           })
           .then(() => {
             setState(State.SAVED);
             setLastSaved(new Date().toLocaleString());
+          })
+          .catch(() => {
+            setState(State.ERROR);
           });
       }
     };
-    save();
+    const name = window.name.split(" ");
+    const op = name[0];
+    const diagram = window.name === "" || op === "d";
+
+    save(diagram);
   }, [tables, relationships, notes, areas, types, title, id, state]);
 
   useEffect(() => {
@@ -578,6 +601,27 @@ export default function Editor() {
         });
     };
 
+    const loadTemplate = async (id) => {
+      await db.templates
+        .get(id)
+        .then((diagram) => {
+          if (diagram) {
+            setId(diagram.id);
+            setTitle(diagram.title);
+            setTables(diagram.tables);
+            setTypes(diagram.types);
+            setRelationships(diagram.relationships);
+            setAreas(diagram.subjectAreas);
+            setNotes(diagram.notes);
+            setUndoStack([]);
+            setRedoStack([]);
+          }
+        })
+        .catch((error) => {
+          console.log(error);
+        });
+    };
+
     if (window.name == "") {
       console.log("Loading the latest diagram");
       loadLatestDiagram();
@@ -594,11 +638,15 @@ export default function Editor() {
           console.log("Loading template with id", did);
           break;
         }
+        case "t": {
+          loadTemplate(did);
+          break;
+        }
         default:
           break;
       }
     }
-    
+
     socket.connect();
 
     const onConnect = () => {