From 7dcecf3c1fd6021e8305b6b8a880bc64c50dd2a2 Mon Sep 17 00:00:00 2001
From: 1ilit <1ilit@proton.me>
Date: Thu, 29 Aug 2024 17:05:47 +0400
Subject: [PATCH] Create and update gist on click

---
 src/components/EditorHeader/ControlPanel.jsx | 12 +--
 src/components/EditorHeader/Modal/Share.jsx  | 51 +---------
 src/components/EditorHeader/ShareButton.jsx  | 97 ++++++++++++++++++++
 src/components/Workspace.jsx                 | 34 +++++--
 4 files changed, 125 insertions(+), 69 deletions(-)
 create mode 100644 src/components/EditorHeader/ShareButton.jsx

diff --git a/src/components/EditorHeader/ControlPanel.jsx b/src/components/EditorHeader/ControlPanel.jsx
index 042561d..1961046 100644
--- a/src/components/EditorHeader/ControlPanel.jsx
+++ b/src/components/EditorHeader/ControlPanel.jsx
@@ -9,7 +9,6 @@ import {
   IconUndo,
   IconRedo,
   IconEdit,
-  IconShareStroked,
 } from "@douyinfe/semi-icons";
 import { Link, useNavigate } from "react-router-dom";
 import icon from "../../assets/icon_dark_64.png";
@@ -71,6 +70,7 @@ import { exportSQL } from "../../utils/exportSQL";
 import { databases } from "../../data/databases";
 import { jsonToMermaid } from "../../utils/exportAs/mermaid";
 import { isRtl } from "../../i18n/utils/rtl";
+import ShareButton from "./ShareButton";
 
 export default function ControlPanel({
   diagramId,
@@ -1363,15 +1363,7 @@ export default function ControlPanel({
         {layout.header && (
           <div className="flex justify-between items-center me-7">
             {header()}
-            <Button
-              type="primary"
-              className="text-base me-2 pe-6 ps-5 py-[18px] rounded-md"
-              size="default"
-              icon={<IconShareStroked />}
-              onClick={() => setModal(MODAL.SHARE)}
-            >
-              {t("share")}
-            </Button>
+            <ShareButton setModal={setModal} />
           </div>
         )}
         {layout.toolbar && toolbar()}
diff --git a/src/components/EditorHeader/Modal/Share.jsx b/src/components/EditorHeader/Modal/Share.jsx
index f8af1b9..bc4896d 100644
--- a/src/components/EditorHeader/Modal/Share.jsx
+++ b/src/components/EditorHeader/Modal/Share.jsx
@@ -1,43 +1,7 @@
-import { Banner, Button, Spin } from "@douyinfe/semi-ui";
-import { IconLink } from "@douyinfe/semi-icons";
-import { useTranslation } from "react-i18next";
-import { Octokit } from "octokit";
-import { useState } from "react";
+import { Banner } from "@douyinfe/semi-ui";
 import { MODAL } from "../../../data/constants";
 
 export default function Share({ setModal }) {
-  const { t } = useTranslation();
-  const [loading, setLoading] = useState(false);
-
-  const generateLink = async () => {
-    setLoading(true);
-    const userToken = localStorage.getItem("github_token");
-
-    const octokit = new Octokit({
-      auth:
-        userToken ?? import.meta.env.VITE_GITHUB_ACCESS_TOKEN,
-    });
-
-    try {
-      const res = await octokit.request("POST /gists", {
-        description: "Hello world",
-        public: false,
-        files: {
-          "test.json": {
-            content: '{"Hello":"WORLD"}',
-          },
-        },
-        headers: {
-          "X-GitHub-Api-Version": "2022-11-28",
-        },
-      });
-      console.log(res);
-    } catch (e) {
-      console.error(e);
-    } finally {
-      setLoading(false);
-    }
-  };
 
   return (
     <div id="share" className="space-y-4">
@@ -68,18 +32,7 @@ export default function Share({ setModal }) {
           </ul>
         }
       />
-      <div className="text-center">
-        <Button
-          type="primary"
-          theme="solid"
-          className="text-base me-2 pe-6 ps-5 py-[18px] rounded-md"
-          size="default"
-          icon={loading ? <Spin /> : <IconLink />}
-          onClick={generateLink}
-        >
-          {t("generate_link")}
-        </Button>
-      </div>
+     
     </div>
   );
 }
diff --git a/src/components/EditorHeader/ShareButton.jsx b/src/components/EditorHeader/ShareButton.jsx
new file mode 100644
index 0000000..826a2b0
--- /dev/null
+++ b/src/components/EditorHeader/ShareButton.jsx
@@ -0,0 +1,97 @@
+import { Button, Spin } from "@douyinfe/semi-ui";
+import { IconShareStroked } from "@douyinfe/semi-icons";
+import { useTranslation } from "react-i18next";
+import { Octokit } from "octokit";
+import { MODAL } from "../../data/constants";
+import { useContext, useState } from "react";
+import { IdContext } from "../Workspace";
+
+export default function ShareButton({ setModal }) {
+  const { t } = useTranslation();
+  const { gistId, setGistId } = useContext(IdContext);
+  const [loading, setLoading] = useState(false);
+
+  const updateGist = async () => {
+    setLoading(true);
+    const userToken = localStorage.getItem("github_token");
+
+    const octokit = new Octokit({
+      auth: userToken ?? import.meta.env.VITE_GITHUB_ACCESS_TOKEN,
+    });
+
+    try {
+      const res = await octokit.request(`PATCH /gists/${gistId}`, {
+        gist_id: gistId,
+        description: "drawDB diagram",
+        files: {
+          "test.json": {
+            content: '{"Hello":"SEAMAN"}',
+          },
+        },
+        headers: {
+          "X-GitHub-Api-Version": "2022-11-28",
+        },
+      });
+      console.log(res);
+    } catch (e) {
+      console.error(e);
+    } finally {
+      setLoading(false);
+    }
+  };
+
+  const generateLink = async () => {
+    setLoading(true);
+    const userToken = localStorage.getItem("github_token");
+
+    const octokit = new Octokit({
+      auth: userToken ?? import.meta.env.VITE_GITHUB_ACCESS_TOKEN,
+    });
+
+    try {
+      const res = await octokit.request("POST /gists", {
+        description: "drawDB diagram",
+        public: false,
+        files: {
+          "test.json": {
+            content: '{"Hello":"WORLD"}',
+          },
+        },
+        headers: {
+          "X-GitHub-Api-Version": "2022-11-28",
+        },
+      });
+      setGistId(res.data.id);
+    } catch (e) {
+      console.error(e);
+    } finally {
+      setLoading(false);
+    }
+  };
+
+  const onShare = async () => {
+    try {
+      if (!gistId || gistId === "") {
+        await generateLink();
+      } else {
+        await updateGist();
+      }
+    } catch (e) {
+      console.error(e);
+    } finally {
+      setModal(MODAL.SHARE);
+    }
+  };
+
+  return (
+    <Button
+      type="primary"
+      className="text-base me-2 pe-6 ps-5 py-[18px] rounded-md"
+      size="default"
+      icon={loading ? <Spin /> : <IconShareStroked />}
+      onClick={onShare}
+    >
+      {t("share")}
+    </Button>
+  );
+}
diff --git a/src/components/Workspace.jsx b/src/components/Workspace.jsx
index 4d8c803..b0e861f 100644
--- a/src/components/Workspace.jsx
+++ b/src/components/Workspace.jsx
@@ -1,4 +1,4 @@
-import { useState, useEffect, useCallback } from "react";
+import { useState, useEffect, useCallback, createContext } from "react";
 import ControlPanel from "./EditorHeader/ControlPanel";
 import Canvas from "./EditorCanvas/Canvas";
 import { CanvasContextProvider } from "../context/CanvasContext";
@@ -24,8 +24,11 @@ import { useTranslation } from "react-i18next";
 import { databases } from "../data/databases";
 import { isRtl } from "../i18n/utils/rtl";
 
+export const IdContext = createContext({ gistId: "" });
+
 export default function WorkSpace() {
   const [id, setId] = useState(0);
+  const [gistId, setGistId] = useState("");
   const [title, setTitle] = useState("Untitled Diagram");
   const [resize, setResize] = useState(false);
   const [width, setWidth] = useState(340);
@@ -72,6 +75,7 @@ export default function WorkSpace() {
           .add({
             database: database,
             name: title,
+            gistId: gistId ?? "",
             lastModified: new Date(),
             tables: tables,
             references: relationships,
@@ -100,6 +104,7 @@ export default function WorkSpace() {
             notes: notes,
             areas: areas,
             todos: tasks,
+            gistId: gistId ?? "",
             pan: transform.pan,
             zoom: transform.zoom,
             ...(databases[database].hasEnums && { enums: enums }),
@@ -146,6 +151,7 @@ export default function WorkSpace() {
     setSaveState,
     database,
     enums,
+    gistId,
   ]);
 
   const load = useCallback(async () => {
@@ -161,6 +167,7 @@ export default function WorkSpace() {
               setDatabase(DB.GENERIC);
             }
             setId(d.id);
+            setGistId(d.gistId);
             setTitle(d.name);
             setTables(d.tables);
             setRelationships(d.references);
@@ -196,6 +203,7 @@ export default function WorkSpace() {
               setDatabase(DB.GENERIC);
             }
             setId(diagram.id);
+            setGistId(diagram.gistId);
             setTitle(diagram.name);
             setTables(diagram.tables);
             setRelationships(diagram.references);
@@ -327,11 +335,15 @@ export default function WorkSpace() {
     setSaveState,
   ]);
 
+  useEffect(() => {
+    setSaveState(State.SAVING);
+  }, [gistId, setSaveState]);
+
   useEffect(() => {
     if (saveState !== State.SAVING) return;
 
     save();
-  }, [id, saveState, save]);
+  }, [id, gistId, saveState, save]);
 
   useEffect(() => {
     document.title = "Editor | drawDB";
@@ -341,14 +353,16 @@ export default function WorkSpace() {
 
   return (
     <div className="h-full flex flex-col overflow-hidden theme">
-      <ControlPanel
-        diagramId={id}
-        setDiagramId={setId}
-        title={title}
-        setTitle={setTitle}
-        lastSaved={lastSaved}
-        setLastSaved={setLastSaved}
-      />
+      <IdContext.Provider value={{ gistId, setGistId }}>
+        <ControlPanel
+          diagramId={id}
+          setDiagramId={setId}
+          title={title}
+          setTitle={setTitle}
+          lastSaved={lastSaved}
+          setLastSaved={setLastSaved}
+        />
+      </IdContext.Provider>
       <div
         className="flex h-full overflow-y-auto"
         onPointerUp={(e) => e.isPrimary && setResize(false)}