From f9f5298a4be9dc1dfbeb8fc9917091212cb92d7a Mon Sep 17 00:00:00 2001
From: MrMelon54 <github@mrmelon54.com>
Date: Tue, 31 Oct 2023 11:06:37 +0000
Subject: [PATCH] Use store for routes and redirects to keep data between
 sidebar page changes

---
 src/stores/target.ts           |  6 ++++++
 src/views/RedirectsView.svelte | 20 ++++++++++----------
 src/views/RoutesView.svelte    | 25 +++++++++++++++----------
 3 files changed, 31 insertions(+), 20 deletions(-)
 create mode 100644 src/stores/target.ts

diff --git a/src/stores/target.ts b/src/stores/target.ts
new file mode 100644
index 0000000..567caba
--- /dev/null
+++ b/src/stores/target.ts
@@ -0,0 +1,6 @@
+import {writable} from "svelte/store";
+import type {CSPair} from "../types/cspair";
+import type {Redirect, Route} from "../types/target";
+
+export const routesTable = writable<{[key: string]: CSPair<Route>}>({});
+export const redirectsTable = writable<{[key: string]: CSPair<Redirect>}>({});
diff --git a/src/views/RedirectsView.svelte b/src/views/RedirectsView.svelte
index 9f0911d..a69ad7f 100644
--- a/src/views/RedirectsView.svelte
+++ b/src/views/RedirectsView.svelte
@@ -5,13 +5,13 @@
   import type {CSPair} from "../types/cspair";
   import {type Redirect, redirectEqual} from "../types/target";
   import {domainOption} from "../stores/domain-option";
+  import {redirectsTable} from "../stores/target";
 
   const apiViolet = import.meta.env.VITE_API_VIOLET;
 
-  let tableData: {[key: string]: CSPair<Redirect>} = {};
   let tableKeys: string[] = [];
 
-  $: tableKeys = Object.entries(tableData)
+  $: tableKeys = Object.entries($redirectsTable)
     .filter(x => x[1].client != null || x[1].server != null)
     .map(x => x[0])
     .filter(x => domainFilter(x, $domainOption))
@@ -26,9 +26,9 @@
     return p.endsWith(domain);
   }
 
-  let promiseForTable: Promise<void> = reloadTable(true);
+  let promiseForTable: Promise<void> = reloadTable();
 
-  function reloadTable(firstLoad: boolean = false): Promise<void> {
+  function reloadTable(): Promise<void> {
     return new Promise<void>((res, rej) => {
       fetch(apiViolet + "/redirect", {headers: {Authorization: getBearer()}})
         .then(x => {
@@ -38,8 +38,8 @@
         .then(x => {
           let rows = x as Redirect[];
           rows.forEach(x => {
-            tableData[x.src] = {
-              client: firstLoad || !tableData[x.src] ? JSON.parse(JSON.stringify(x)) : tableData[x.src]?.client,
+            $redirectsTable[x.src] = {
+              client: !$redirectsTable[x.src] ? JSON.parse(JSON.stringify(x)) : $redirectsTable[x.src]?.client,
               server: x,
               p: Promise.resolve(),
             };
@@ -57,7 +57,7 @@
 
   function saveChanges() {
     let tableProm = tableKeys
-      .map(x => tableData[x])
+      .map(x => $redirectsTable[x])
       .filter(x => x.client != null || x.server != null)
       .filter(x => !redirectEqual(x.client, x.server))
       .map((x: CSPair<Redirect>): Savable<Redirect> => {
@@ -109,7 +109,7 @@
           <RedirectCreator
             on:make={e => {
               const x = e.detail;
-              tableData[x.src] = {client: x, server: tableData[x.src]?.server, p: Promise.resolve()};
+              $redirectsTable[x.src] = {client: x, server: $redirectsTable[x.src]?.server, p: Promise.resolve()};
               tableKeys.push(x.src);
               tableKeys = tableKeys;
             }}
@@ -117,10 +117,10 @@
         </thead>
         <tbody>
           {#each tableKeys as src (src)}
-            {#await tableData[src].p}
+            {#await $redirectsTable[src].p}
               <tr><td colspan="5">Loading...</td></tr>
             {:then _}
-              <RedirectRow bind:value={tableData[src]} />
+              <RedirectRow bind:value={$redirectsTable[src]} />
             {:catch err}
               <tr><td colspan="5">Error loading row for {src}: {err}</td></tr>
             {/await}
diff --git a/src/views/RoutesView.svelte b/src/views/RoutesView.svelte
index e3b2858..bf17b65 100644
--- a/src/views/RoutesView.svelte
+++ b/src/views/RoutesView.svelte
@@ -5,13 +5,13 @@
   import type {CSPair} from "../types/cspair";
   import {type Route, routeEqual} from "../types/target";
   import {domainOption} from "../stores/domain-option";
+  import {routesTable} from "../stores/target";
 
   const apiViolet = import.meta.env.VITE_API_VIOLET;
 
-  let tableData: {[key: string]: CSPair<Route>} = {};
   let tableKeys: string[] = [];
 
-  $: tableKeys = Object.entries(tableData)
+  $: tableKeys = Object.entries($routesTable)
     .filter(x => x[1].client != null || x[1].server != null)
     .map(x => x[0])
     .filter(x => domainFilter(x, $domainOption))
@@ -26,9 +26,9 @@
     return p.endsWith(domain);
   }
 
-  let promiseForTable: Promise<void> = reloadTable(true);
+  let promiseForTable: Promise<void> = reloadTable();
 
-  function reloadTable(firstLoad: boolean = false): Promise<void> {
+  function reloadTable(): Promise<void> {
     return new Promise<void>((res, rej) => {
       fetch(apiViolet + "/route", {headers: {Authorization: getBearer()}})
         .then(x => {
@@ -37,12 +37,17 @@
         })
         .then(x => {
           let rows = x as Route[];
+          let srcs = new Set(Object.keys($routesTable));
           rows.forEach(x => {
-            tableData[x.src] = {
-              client: firstLoad || !tableData[x.src] ? JSON.parse(JSON.stringify(x)) : tableData[x.src]?.client,
+            $routesTable[x.src] = {
+              client: !$routesTable[x.src] ? JSON.parse(JSON.stringify(x)) : $routesTable[x.src]?.client,
               server: x,
               p: Promise.resolve(),
             };
+            srcs.delete(x.src);
+          });
+          srcs.forEach(x => {
+            $routesTable[x].server = null;
           });
           res();
         })
@@ -57,7 +62,7 @@
 
   function saveChanges() {
     let tableProm = tableKeys
-      .map(x => tableData[x])
+      .map(x => $routesTable[x])
       .filter(x => x.client != null || x.server != null)
       .filter(x => !routeEqual(x.client, x.server))
       .map((x: CSPair<Route>): Savable<Route> => {
@@ -108,7 +113,7 @@
           <RouteCreator
             on:make={e => {
               const x = e.detail;
-              tableData[x.src] = {client: x, server: tableData[x.src]?.server, p: Promise.resolve()};
+              $routesTable[x.src] = {client: x, server: $routesTable[x.src]?.server, p: Promise.resolve()};
               tableKeys.push(x.src);
               tableKeys = tableKeys;
             }}
@@ -116,10 +121,10 @@
         </thead>
         <tbody>
           {#each tableKeys as src (src)}
-            {#await tableData[src].p}
+            {#await $routesTable[src].p}
               <tr><td colspan="5">Loading...</td></tr>
             {:then _}
-              <RouteRow bind:value={tableData[src]} />
+              <RouteRow bind:value={$routesTable[src]} />
             {:catch err}
               <tr><td colspan="5">Error loading row for {src}: {err}</td></tr>
             {/await}