AFFiNE-Mirror/apps/web/src/hooks/current/use-current-workspace.ts

import { atom, useAtom, useAtomValue, useSetAtom } from 'jotai';
import { atomWithStorage } from 'jotai/utils';
import { useCallback } from 'react';

import {
  currentPageIdAtom,
  currentWorkspaceIdAtom,
  workspacesAtom,
} from '../../atoms';
import type { RemWorkspace } from '../../shared';

export const currentWorkspaceAtom = atom<Promise<RemWorkspace | null>>(
  async get => {
    const id = get(currentWorkspaceIdAtom);
    const workspaces = await get(workspacesAtom);
    return workspaces.find(workspace => workspace.id === id) ?? null;
  }
);

export const lastWorkspaceIdAtom = atomWithStorage<string | null>(
  'last_workspace_id',
  null
);

export function useCurrentWorkspace(): [
  RemWorkspace | null,
  (id: string | null) => void
] {
  const currentWorkspace = useAtomValue(currentWorkspaceAtom);
  const [, setId] = useAtom(currentWorkspaceIdAtom);
  const [, setPageId] = useAtom(currentPageIdAtom);
  const setLast = useSetAtom(lastWorkspaceIdAtom);
  return [
    currentWorkspace,
    useCallback(
      (id: string | null) => {
        setPageId(null);
        setLast(id);
        setId(id);
      },
      [setId, setLast, setPageId]
    ),
  ];
}