yaak/src-web/hooks/useActiveCookieJar.ts

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
  // biome-ignore lint/correctness/useExhaustiveDependencies: none
  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]);
}