import { parseStreamKey, Payload, WebSocket } from "@spacebar/gateway";
import { emitEvent, Stream, StreamDeleteEvent, VoiceState, VoiceStateUpdateEvent } from "@spacebar/util";
import { check } from "./instanceOf";
import { StreamDeleteSchema } from "@spacebar/schemas";

export async function onStreamDelete(this: WebSocket, data: Payload) {
    const startTime = Date.now();
    check.call(this, StreamDeleteSchema, data.d);
    const body = data.d as StreamDeleteSchema;

    let parsedKey: {
        type: "guild" | "call";
        channelId: string;
        guildId?: string;
        userId: string;
    };

    try {
        parsedKey = parseStreamKey(body.stream_key);
    } catch (e) {
        return this.close(4000, "Invalid stream key");
    }

    const { userId, channelId, guildId, type } = parsedKey;

    // when a user selects to stop watching another user stream, this event gets triggered
    // just disconnect user without actually deleting stream
    if (this.user_id !== userId) {
        await emitEvent({
            event: "STREAM_DELETE",
            data: {
                stream_key: body.stream_key,
            },
            user_id: this.user_id,
        } as StreamDeleteEvent);
        return;
    }

    const stream = await Stream.findOne({
        where: { channel_id: channelId, owner_id: userId },
    });

    if (!stream) return;

    await stream.remove();

    const voiceState = await VoiceState.findOne({
        where: { user_id: this.user_id },
    });

    if (voiceState) {
        voiceState.self_stream = false;
        await voiceState.save();

        await emitEvent({
            event: "VOICE_STATE_UPDATE",
            data: voiceState.toPublicVoiceState(),
            guild_id: guildId,
            channel_id: channelId,
        } as VoiceStateUpdateEvent);
    }

    await emitEvent({
        event: "STREAM_DELETE",
        data: {
            stream_key: body.stream_key,
        },
        guild_id: guildId,
        channel_id: channelId,
    } as StreamDeleteEvent);

    console.log(`[Gateway/${this.user_id}] STREAM_DELETE for user ${this.user_id} in channel ${channelId} with stream key ${body.stream_key} in ${Date.now() - startTime}ms`);
}