import { create } from "zustand";
import { immer } from "zustand/middleware/immer";
import type { Server, ServerId, Uuid } from "~/lib/api/types";

type ServerListStore = {
    servers: Record<ServerId, Server>;
    addServers: (newServers: Server[]) => void;
    addServer: (server: Server) => void;
    removeServer: (serverId: Uuid) => void;
};

export const useServerListStore = create<ServerListStore>()(
    immer((set) => ({
        servers: {},
        addServers: (servers: Server[]) =>
            set((state) => {
                for (const server of servers) {
                    state.servers[server.id] = server;
                }
            }),
        addServer: (server: Server) =>
            set((state) => {
                state.servers[server.id] = server;
            }),
        removeServer: (serverId: Uuid) =>
            set((state) => {
                delete state.servers[serverId];
            }),
    })),
);