import { useMutation, useQueryClient } from '@tanstack/react-query';
import { invoke } from '@tauri-apps/api';
import type { Workspace } from '../lib/models';
import { getWorkspace } from '../lib/store';
import { workspacesQueryKey } from './useWorkspaces';

export function useUpdateWorkspace(id: string | null) {
  const queryClient = useQueryClient();
  return useMutation<void, unknown, Partial<Workspace> | ((w: Workspace) => Workspace)>({
    mutationFn: async (v) => {
      const workspace = await getWorkspace(id);
      if (workspace == null) {
        throw new Error("Can't update a null workspace");
      }

      const newWorkspace = typeof v === 'function' ? v(workspace) : { ...workspace, ...v };
      await invoke('update_workspace', { workspace: newWorkspace });
    },
    onMutate: async (v) => {
      const workspace = await getWorkspace(id);
      if (workspace === null) return;

      const newWorkspace = typeof v === 'function' ? v(workspace) : { ...workspace, ...v };
      queryClient.setQueryData<Workspace[]>(workspacesQueryKey(workspace), (workspaces) =>
        (workspaces ?? []).map((w) => (w.id === newWorkspace.id ? newWorkspace : w)),
      );
    },
  });
}