From cbe8cf3a0549adf5dab7b80fd9724846d93f2175 Mon Sep 17 00:00:00 2001
From: Gregory Schier <gschier1990@gmail.com>
Date: Thu, 16 Mar 2023 11:01:30 -0700
Subject: [PATCH] Store appearance in k/v

---
 src-web/components/RequestPane.tsx    |  7 +++++++
 src-web/components/core/Tabs/Tabs.tsx | 17 +++++++++++++----
 src-web/hooks/useKeyValues.ts         |  4 ++--
 src-web/hooks/useTheme.ts             | 25 +++++++++----------------
 src-web/main.tsx                      | 11 +++++++++--
 5 files changed, 40 insertions(+), 24 deletions(-)

diff --git a/src-web/components/RequestPane.tsx b/src-web/components/RequestPane.tsx
index f6da2c14..9fe6ed3c 100644
--- a/src-web/components/RequestPane.tsx
+++ b/src-web/components/RequestPane.tsx
@@ -1,6 +1,7 @@
 import classnames from 'classnames';
 import { useActiveRequest } from '../hooks/useActiveRequest';
 import { useIsResponseLoading } from '../hooks/useIsResponseLoading';
+import { useKeyValues } from '../hooks/useKeyValues';
 import { useSendRequest } from '../hooks/useSendRequest';
 import { useUpdateRequest } from '../hooks/useUpdateRequest';
 import { tryFormatJson } from '../lib/formatters';
@@ -20,6 +21,10 @@ export function RequestPane({ fullHeight, className }: Props) {
   const updateRequest = useUpdateRequest(activeRequest);
   const sendRequest = useSendRequest(activeRequest);
   const responseLoading = useIsResponseLoading();
+  const activeTab = useKeyValues({
+    key: ['active_request_body_tab', activeRequest?.id ?? 'n/a'],
+    initialValue: 'body',
+  });
 
   if (activeRequest === null) return null;
 
@@ -35,6 +40,8 @@ export function RequestPane({ fullHeight, className }: Props) {
         loading={responseLoading}
       />
       <Tabs
+        value={activeTab.value}
+        onChangeValue={activeTab.set}
         tabs={[
           {
             value: 'body',
diff --git a/src-web/components/core/Tabs/Tabs.tsx b/src-web/components/core/Tabs/Tabs.tsx
index 9252b64c..f61078dc 100644
--- a/src-web/components/core/Tabs/Tabs.tsx
+++ b/src-web/components/core/Tabs/Tabs.tsx
@@ -1,7 +1,6 @@
 import * as T from '@radix-ui/react-tabs';
 import classnames from 'classnames';
 import type { ReactNode } from 'react';
-import { useState } from 'react';
 import { Button } from '../Button';
 import type { DropdownMenuRadioItem, DropdownMenuRadioProps } from '../Dropdown';
 import { DropdownMenuRadio, DropdownMenuTrigger } from '../Dropdown';
@@ -13,6 +12,8 @@ import './Tabs.css';
 interface Props {
   defaultValue?: string;
   label: string;
+  onChangeValue: (value: string) => void;
+  value: string;
   tabs: {
     value: string;
     label: string;
@@ -27,12 +28,20 @@ interface Props {
   children: ReactNode;
 }
 
-export function Tabs({ defaultValue, label, children, tabs, className, tabListClassName }: Props) {
-  const [value, setValue] = useState(defaultValue);
+export function Tabs({
+  value,
+  onChangeValue,
+  defaultValue,
+  label,
+  children,
+  tabs,
+  className,
+  tabListClassName,
+}: Props) {
   return (
     <T.Root
       defaultValue={defaultValue}
-      onValueChange={setValue}
+      onValueChange={onChangeValue}
       className={classnames(className, 'h-full grid grid-rows-[auto_minmax(0,1fr)] grid-cols-1')}
     >
       <T.List
diff --git a/src-web/hooks/useKeyValues.ts b/src-web/hooks/useKeyValues.ts
index 14948a29..3513c009 100644
--- a/src-web/hooks/useKeyValues.ts
+++ b/src-web/hooks/useKeyValues.ts
@@ -8,7 +8,7 @@ export function keyValueQueryKey({
   namespace = DEFAULT_NAMESPACE,
   key,
 }: {
-  namespace: string;
+  namespace?: string;
   key: string | string[];
 }) {
   return ['key_value', { namespace, key: buildKey(key) }];
@@ -19,7 +19,7 @@ export function useKeyValues({
   key,
   initialValue,
 }: {
-  namespace: string;
+  namespace?: string;
   key: string | string[];
   initialValue: string;
 }) {
diff --git a/src-web/hooks/useTheme.ts b/src-web/hooks/useTheme.ts
index cd0a8221..8d96fd42 100644
--- a/src-web/hooks/useTheme.ts
+++ b/src-web/hooks/useTheme.ts
@@ -1,38 +1,31 @@
-import { useQuery, useQueryClient } from '@tanstack/react-query';
 import { useEffect } from 'react';
 import type { Appearance } from '../lib/theme/window';
 import {
   getAppearance,
   setAppearance,
   subscribeToPreferredAppearanceChange,
-  toggleAppearance,
 } from '../lib/theme/window';
-
-export function appearanceQueryKey() {
-  return ['theme', 'appearance'];
-}
+import { useKeyValues } from './useKeyValues';
 
 export function useTheme() {
-  const queryClient = useQueryClient();
-  const appearance = useQuery({
-    queryKey: appearanceQueryKey(),
-    queryFn: getAppearance,
-    initialData: getAppearance(),
-  }).data;
+  const appearanceKv = useKeyValues({ key: 'appearance', initialValue: getAppearance() });
 
   const themeChange = (appearance: Appearance) => {
-    setAppearance(appearance);
+    appearanceKv.set(appearance);
   };
 
   const handleToggleAppearance = async () => {
-    const newAppearance = toggleAppearance();
-    await queryClient.setQueryData(appearanceQueryKey(), newAppearance);
+    appearanceKv.set(appearanceKv.value === 'dark' ? 'light' : 'dark');
   };
 
+  // Set appearance when preferred theme changes
   useEffect(() => subscribeToPreferredAppearanceChange(themeChange), []);
 
+  // Sync appearance when k/v changes
+  useEffect(() => setAppearance(appearanceKv.value as Appearance), [appearanceKv.value]);
+
   return {
-    appearance,
+    appearance: appearanceKv.value,
     toggleAppearance: handleToggleAppearance,
   };
 }
diff --git a/src-web/main.tsx b/src-web/main.tsx
index 51eb4622..87f9b0d7 100644
--- a/src-web/main.tsx
+++ b/src-web/main.tsx
@@ -1,10 +1,17 @@
+import { invoke } from '@tauri-apps/api';
 import { StrictMode } from 'react';
 import ReactDOM from 'react-dom/client';
 import { App } from './components/App';
-import { setAppearance } from './lib/theme/window';
+import type { KeyValue } from './lib/models';
+import type { Appearance } from './lib/theme/window';
+import { getPreferredAppearance, setAppearance } from './lib/theme/window';
 import './main.css';
 
-setAppearance();
+const appearance: KeyValue = await invoke('get_key_value', {
+  namespace: 'app',
+  key: 'appearance',
+});
+setAppearance((appearance?.value ?? getPreferredAppearance()) as Appearance);
 
 // root holds our app's root DOM Element:
 ReactDOM.createRoot(document.getElementById('root') as HTMLElement).render(