OpenPipe-llm/src/state/store.ts

import { type StateCreator, create } from "zustand";
import { immer } from "zustand/middleware/immer";
import { createSelectors } from "./createSelectors";
import {
  type SharedVariantEditorSlice,
  createVariantEditorSlice,
} from "./sharedVariantEditor.slice";
import { type APIClient } from "~/utils/api";

export type State = {
  drawerOpen: boolean;
  openDrawer: () => void;
  closeDrawer: () => void;
  api: APIClient | null;
  setApi: (api: APIClient) => void;
  sharedVariantEditor: SharedVariantEditorSlice;
};

export type SliceCreator<T> = StateCreator<State, [["zustand/immer", never]], [], T>;

export type SetFn = Parameters<SliceCreator<unknown>>[0];
export type GetFn = Parameters<SliceCreator<unknown>>[1];

const useBaseStore = create<State, [["zustand/immer", never]]>(
  immer((set, get, ...rest) => ({
    api: null,
    setApi: (api) =>
      set((state) => {
        state.api = api;
      }),

    drawerOpen: false,
    openDrawer: () =>
      set((state) => {
        state.drawerOpen = true;
      }),
    closeDrawer: () =>
      set((state) => {
        state.drawerOpen = false;
      }),
    sharedVariantEditor: createVariantEditorSlice(set, get, ...rest),
  })),
);

export const useAppStore = createSelectors(useBaseStore);