yaak/src-web/hooks/useRecentCookieJars.ts

import { cookieJarsAtom } from "@yaakapp-internal/models";
import { useAtomValue } from "jotai";
import { useEffect, useMemo } from "react";
import { jotaiStore } from "../lib/jotai";
import { getKeyValue, setKeyValue } from "../lib/keyValueStore";
import { activeCookieJarAtom } from "./useActiveCookieJar";
import { useKeyValue } from "./useKeyValue";

const kvKey = (workspaceId: string) => `recent_cookie_jars::${workspaceId}`;
const namespace = "global";
const fallback: string[] = [];

export function useRecentCookieJars() {
  const cookieJars = useAtomValue(cookieJarsAtom);
  const kv = useKeyValue<string[]>({
    key: kvKey(cookieJars[0]?.workspaceId ?? "n/a"),
    namespace,
    fallback,
  });

  const onlyValidIds = useMemo(
    () => kv.value?.filter((id) => cookieJars?.some((e) => e.id === id)) ?? [],
    [kv.value, cookieJars],
  );

  return onlyValidIds;
}

export function useSubscribeRecentCookieJars() {
  useEffect(() => {
    return jotaiStore.sub(activeCookieJarAtom, async () => {
      const activeCookieJar = jotaiStore.get(activeCookieJarAtom);
      if (activeCookieJar == null) return;

      const key = kvKey(activeCookieJar.workspaceId);

      const recentIds = getKeyValue<string[]>({ namespace, key, fallback });
      if (recentIds[0] === activeCookieJar.id) return; // Short-circuit

      const withoutActiveId = recentIds.filter((id) => id !== activeCookieJar.id);
      const value = [activeCookieJar.id, ...withoutActiveId];
      await setKeyValue({ namespace, key, value });
    });
  }, []);
}

export async function getRecentCookieJars(workspaceId: string) {
  return getKeyValue<string[]>({
    namespace,
    key: kvKey(workspaceId),
    fallback,
  });
}