import { create } from "zustand";

import type { Drop } from "@/lib/types";

import type { StreamStatus } from "@/lib/sse";

const BUFFER_CAP = 200;

interface UiState {

  sidebarCollapsed: boolean;

  toggleSidebar: () => void;

  setSidebar: (v: boolean) => void;

  commandOpen: boolean;

  setCommandOpen: (v: boolean) => void;

  streamStatus: StreamStatus;

  setStreamStatus: (s: StreamStatus) => void;

  liveDrops: Drop[];

  freshIds: Set<number>;

  pushDrop: (drop: Drop) => void;

  seedDrops: (drops: Drop[]) => void;

  clearFresh: (id: number) => void;

}

export const useUiStore = create<UiState>((set) => ({

  sidebarCollapsed: false,

  toggleSidebar: () => set((s) => ({ sidebarCollapsed: !s.sidebarCollapsed })),

  setSidebar: (v) => set({ sidebarCollapsed: v }),

  commandOpen: false,

  setCommandOpen: (v) => set({ commandOpen: v }),

  streamStatus: "connecting",

  setStreamStatus: (s) => set({ streamStatus: s }),

  liveDrops: [],

  freshIds: new Set<number>(),

  pushDrop: (drop) =>

    set((s) => {

      if (s.liveDrops.some((d) => d.id === drop.id)) return s;

      const next = [drop, ...s.liveDrops].slice(0, BUFFER_CAP);

      const fresh = new Set(s.freshIds);

      fresh.add(drop.id);

      return { liveDrops: next, freshIds: fresh };

    }),

  seedDrops: (drops) =>

    set(() => ({ liveDrops: drops.slice(0, BUFFER_CAP), freshIds: new Set<number>() })),

  clearFresh: (id) =>

    set((s) => {

      if (!s.freshIds.has(id)) return s;

      const fresh = new Set(s.freshIds);

      fresh.delete(id);

      return { freshIds: fresh };

    }),

}));