Split codebase (#455)

apps/yaak-client/hooks/useActiveCookieJar.ts

@@ -0,0 +1,58 @@
+import { useSearch } from "@tanstack/react-router";
+import type { CookieJar } from "@yaakapp-internal/models";
+import { cookieJarsAtom } from "@yaakapp-internal/models";
+import { atom, useAtomValue } from "jotai";
+import { useEffect } from "react";
+import { jotaiStore } from "../lib/jotai";
+import { setWorkspaceSearchParams } from "../lib/setWorkspaceSearchParams";
+
+export const activeCookieJarAtom = atom<CookieJar | null>(null);
+
+export function useActiveCookieJar() {
+  return useAtomValue(activeCookieJarAtom);
+}
+
+export function useSubscribeActiveCookieJarId() {
+  const search = useSearch({ strict: false });
+  const cookieJarId = search.cookie_jar_id;
+  const cookieJars = useAtomValue(cookieJarsAtom);
+
+  useEffect(() => {
+    if (search == null) return; // Happens during Vite hot reload
+    const activeCookieJar = cookieJars?.find((j) => j.id === cookieJarId) ?? null;
+    jotaiStore.set(activeCookieJarAtom, activeCookieJar);
+  }, [cookieJarId, cookieJars, search]);
+}
+
+export function getActiveCookieJar() {
+  return jotaiStore.get(activeCookieJarAtom);
+}
+
+export function useEnsureActiveCookieJar() {
+  const cookieJars = useAtomValue(cookieJarsAtom);
+  const { cookie_jar_id: activeCookieJarId } = useSearch({ from: "/workspaces/$workspaceId/" });
+
+  // Set the active cookie jar to the first one, if none set
+  // NOTE: We only run this on cookieJars to prevent data races when switching workspaces since a lot of
+  //  things change when switching workspaces, and we don't currently have a good way to ensure that all
+  //  stores have updated.
+  // TODO: Create a global data store that can handle this case
+  // oxlint-disable-next-line react-hooks/exhaustive-deps
+  useEffect(() => {
+    if (cookieJars == null) return; // Hasn't loaded yet
+
+    if (cookieJars.find((j) => j.id === activeCookieJarId)) {
+      return; // There's an active jar
+    }
+
+    const firstJar = cookieJars[0];
+    if (firstJar == null) {
+      console.log(`Workspace doesn't have any cookie jars to activate`);
+      return;
+    }
+
+    // There's no active jar, so set it to the first one
+    console.log("Defaulting active cookie jar to first jar", firstJar);
+    setWorkspaceSearchParams({ cookie_jar_id: firstJar.id });
+  }, [cookieJars]);
+}