Backup and Data Retention for PWV

Author: AttAditya

src/defs/PyWebview.ts

@@ -10,7 +10,7 @@ export interface PyWebview {
     load_file: () => Promise<string>;
     load_project_id: () => Promise<string | null>;
     backup_project: (content: string) => Promise<boolean>;
-    restore_project: () => Promise<void>;
+    restore_project: () => Promise<string>;
   }
 }
 

src/utils/appData.tsx

@@ -16,6 +16,9 @@ import { usePromptContext } from "@utils/prompt";
 import { usePalleteContext } from "@utils/pallete";
 import { useVariablesContext } from "@utils/variables";
 import { loadProject, saveProject } from "@utils/engineTools";
+import { packProject } from "@utils/packerTools";
+import { backupProject, getRestoredProject } from "@utils/desktopTools";
+import { saveData } from "@utils/persistentTools";
 import {
   getBreadcrumb,
   getWindowTools,
@@ -155,6 +158,8 @@ const createAppDataContext = () => {
     useEffect(() => {
       (async () => {
         const existingId = await loadExistingId();
+        const restoredProject = await getRestoredProject();
+        saveData(`p:${existingId}`, restoredProject);
         setProjectId(existingId);
 
         loadProject(
@@ -209,6 +214,32 @@ const createAppDataContext = () => {
       statesList,
       customComponents,
     ]);
+
+    useEffect(() => {
+      const backupInterval = setInterval(() => {
+        const project = packProject(
+          projectId!,
+          nodeSystem,
+          entries,
+          {
+            states: statesList,
+            customComponents,
+          }
+        );
+
+        backupProject(JSON.stringify(project, null, 2));
+      }, 5000);
+
+      return () => {
+        clearInterval(backupInterval);
+      }
+    }, [
+      customComponents,
+      entries,
+      nodeSystem,
+      projectId,
+      statesList,
+    ]);
 
     const exposed = {
       projectId,

src/utils/desktopTools.ts

@@ -99,13 +99,27 @@ const setupDesktopTools = () => {
     return (await getApi()!.load_project_id()) || null;
   }
 
+  const backupProject = async (content: string): Promise<boolean> => {
+    if (!isDesktop()) return false;
+    await waitForPWV();
+    return getApi()!.backup_project(content);
+  }
+
+  const getRestoredProject = async (): Promise<string> => {
+    if (!isDesktop()) return "";
+    await waitForPWV();
+    return getApi()!.restore_project();
+  }
+
   return {
     isDesktop,
     closeWindow,
     toggleWindowFullscreen,
     saveFile,
     loadFile,
     loadProjectId,
+    backupProject,
+    getRestoredProject
   }
 };
 
@@ -116,5 +130,7 @@ export const {
   saveFile,
   loadFile,
   loadProjectId,
+  backupProject,
+  getRestoredProject,
 } = setupDesktopTools();