import { type Block, Decoder, type StateAndTagData } from "ebml"
import type { Message } from "../../types/Socket"
import { processAudio } from "./processAudio"

// let channelCount = 2
// let sampleRate = 48000 // Hz
// let audioDelay = 0

let latestIcecastConfig: any = null

export function receiveAudio(_e: Electron.IpcMainEvent, msg: Message) {
    if (msg.channel === "RESET_DECODER") {
        const id = msg.data?.id || "main"
        const dec = ebmlDecoders.get(id)
        if (dec) {
            try {
                dec.removeAllListeners()
            } catch {}
            ebmlDecoders.delete(id)
        }
        return
    }

    if (msg.channel !== "CAPTURE") {
        console.error("Unknown AUDIO channel:", msg.channel)
        return
    }

    const data = msg.data
    const input = toAudioBuffer(data?.buffer)
    if (!input || input.length === 0) return

    latestIcecastConfig = data.icecast

    const decoder = createDecoder(data.id || "main")
    try {
        decoder.write(input)
    } catch (error) {
        console.error("Failed to decode incoming audio chunk", error)
    }
}

function toAudioBuffer(value: unknown): Buffer | null {
    if (!value) return null
    if (Buffer.isBuffer(value)) return value
    if (value instanceof Uint8Array) return Buffer.from(value.buffer, value.byteOffset, value.byteLength)
    if (value instanceof ArrayBuffer) return Buffer.from(value)
    if (ArrayBuffer.isView(value)) return Buffer.from(value.buffer, value.byteOffset, value.byteLength)

    const serialized = value as { type?: unknown; data?: unknown }
    if (serialized?.type === "Buffer" && Array.isArray(serialized.data)) {
        return Buffer.from(serialized.data as number[])
    }

    return null
}

const ebmlDecoders = new Map<string, Decoder>()

function createDecoder(id: string) {
    const existing = ebmlDecoders.get(id)
    if (existing) return existing

    const decoder = new Decoder()
    ebmlDecoders.set(id, decoder)

    decoder.on("data", ([blockType, data]: StateAndTagData) => {
        if (blockType !== "tag" || data.name !== "SimpleBlock" || data.type !== "b") return

        const block = data as Block
        if (!block.payload) return

        processAudio(block.payload, latestIcecastConfig)
    })

    return decoder
}