Cookie Support (#19)

src-web/hooks/useUpdateCookieJar.ts

@@ -0,0 +1,30 @@
+import { useMutation, useQueryClient } from '@tanstack/react-query';
+import { invoke } from '@tauri-apps/api';
+import type { CookieJar } from '../lib/models';
+import { getCookieJar } from '../lib/store';
+import { cookieJarsQueryKey } from './useCookieJars';
+
+export function useUpdateCookieJar(id: string | null) {
+  const queryClient = useQueryClient();
+  return useMutation<void, unknown, Partial<CookieJar> | ((j: CookieJar) => CookieJar)>({
+    mutationFn: async (v) => {
+      const cookieJar = await getCookieJar(id);
+      if (cookieJar == null) {
+        throw new Error("Can't update a null workspace");
+      }
+
+      const newCookieJar = typeof v === 'function' ? v(cookieJar) : { ...cookieJar, ...v };
+      console.log('NEW COOKIE JAR', newCookieJar.cookies.length);
+      await invoke('update_cookie_jar', { cookieJar: newCookieJar });
+    },
+    onMutate: async (v) => {
+      const cookieJar = await getCookieJar(id);
+      if (cookieJar === null) return;
+
+      const newCookieJar = typeof v === 'function' ? v(cookieJar) : { ...cookieJar, ...v };
+      queryClient.setQueryData<CookieJar[]>(cookieJarsQueryKey(cookieJar), (cookieJars) =>
+        (cookieJars ?? []).map((j) => (j.id === newCookieJar.id ? newCookieJar : j)),
+      );
+    },
+  });
+}