import { create } from 'zustand'
import { immer } from 'zustand/middleware/immer'
import type { ChannelId, ServerChannel, ServerId } from "~/lib/api/types"

type ServerChannelsStore = {
    channels: Record<ServerId, Record<ChannelId, ServerChannel>>
    addServer: (serverId: ServerId) => void
    addChannel: (channel: ServerChannel) => void
    addChannels: (channels: ServerChannel[]) => void
    removeChannel: (serverId: ServerId, channelId: ChannelId) => void
    removeServer: (serverId: ServerId) => void
}

export const useServerChannelsStore = create<ServerChannelsStore>()(
    immer(
        (set, get) => ({
            channels: {},
            addServer: (serverId) => set((state) => {
                state.channels[serverId] = {}
            }),
            addChannel: (channel) => set((state) => {
                if (state.channels[channel.serverId] === undefined) {
                    state.channels[channel.serverId] = {}
                }

                state.channels[channel.serverId][channel.id] = channel
            }),
            addChannels: (channels) => set((state) => {
                for (const channel of channels) {
                    if (state.channels[channel.serverId] === undefined) {
                        state.channels[channel.serverId] = {}
                    }

                    state.channels[channel.serverId][channel.id] = channel
                }
            }),
            removeChannel: (serverId, channelId) => set((state) => { delete state.channels[serverId][channelId] }),
            removeServer: (serverId) => set((state) => { delete state.channels[serverId] }),
        })
    )
)