import { create } from "zustand";
import { WebRTCClient } from "~/lib/websocket/voice/client";
import { ConnectionState } from "~/lib/websocket/voice/types";
import { useVoiceStateStore } from "./voice-state-store";

const VOICE_GATEWAY_URL = "ws://localhost:12345/voice/ws";

interface WebRTCState {
    client: WebRTCClient | null;
    status: ConnectionState;
    remoteStream: MediaStream | null;
    error: string | null;
    connect: (token: string) => Promise<void>;
    disconnect: () => void;
    createOffer: (localStream: MediaStream) => Promise<void>;
}

export const useWebRTCStore = create<WebRTCState>()((set, get) => {
    const client = new WebRTCClient(
        VOICE_GATEWAY_URL,
        (state) => set({ status: state }),
        (error) => {
            set({
                status: ConnectionState.ERROR,
                error: error.message,
            });
            useVoiceStateStore.getState().setError(error.message);
        },
        (stream) => set({ remoteStream: stream }),
    );

    return {
        client,
        status: ConnectionState.DISCONNECTED,
        remoteStream: null,
        error: null,

        connect: async (token) => {
            await client.connect(token);
        },

        disconnect: () => {
            client.disconnect();
            set({
                status: ConnectionState.DISCONNECTED,
                remoteStream: null,
            });
        },

        createOffer: async (localStream) => {
            await client.createOffer(localStream);
        },
    };
});