import io
import struct
from typing import Callable, Any


def decode_opus_from_file_stream(input_file, callback: Callable[[Any], Any]):
    """
    从p3文件中解码 Opus 数据，由 callback 处理 Opus 数据包。
    """
    with open(input_file, 'rb') as f:
        while True:
            # 读取头部（4字节）：[1字节类型，1字节保留，2字节长度]
            header = f.read(4)
            if not header:
                break

            # 解包头部信息
            _, _, data_len = struct.unpack('>BBH', header)

            # 根据头部指定的长度读取 Opus 数据
            opus_data = f.read(data_len)
            if len(opus_data) != data_len:
                raise ValueError(f"Data length({len(opus_data)}) mismatch({data_len}) in the file.")

            callback(opus_data)


def decode_opus_from_bytes_stream(input_bytes, callback: Callable[[Any], Any]):
    """
    从p3二进制数据中解码 Opus 数据，由 callback 处理 Opus 数据包。
    """

    f = io.BytesIO(input_bytes)
    while True:
        header = f.read(4)
        if not header:
            break
        _, _, data_len = struct.unpack('>BBH', header)
        opus_data = f.read(data_len)
        if len(opus_data) != data_len:
            raise ValueError(f"Data length({len(opus_data)}) mismatch({data_len}) in the bytes.")
        callback(opus_data)