import {useParams} from '@tanstack/react-router';
import type {Workspace} from '@yaakapp-internal/models';
import {atom, useAtomValue} from 'jotai/index';
import {useEffect} from 'react';
import {jotaiStore} from '../lib/jotai';
import {workspacesAtom} from './useWorkspaces';

export const activeWorkspaceIdAtom = atom<string>();

export const activeWorkspaceAtom = atom<Workspace | null>((get) => {
    const activeWorkspaceId = get(activeWorkspaceIdAtom);
    const workspaces = get(workspacesAtom);
    return workspaces.find((w) => w.id === activeWorkspaceId) ?? null;
});

export function useActiveWorkspace(): Workspace | null {
    return useAtomValue(activeWorkspaceAtom);
}

export function getActiveWorkspaceId() {
    return jotaiStore.get(activeWorkspaceIdAtom) ?? null;
}

export function getActiveWorkspace() {
    return jotaiStore.get(activeWorkspaceAtom) ?? null;
}

export function useSubscribeActiveWorkspaceId() {
    const {workspaceId} = useParams({strict: false});
    useEffect(() => jotaiStore.set(activeWorkspaceIdAtom, workspaceId), [workspaceId]);
}