# Copyright (c) 2024–2025, Daily
#
# SPDX-License-Identifier: BSD 2-Clause License
"""Stasis RTP frame serializer.

This serializer converts between Pipecat frames and the raw μ-law RTP payload
stream expected by an Stasis *External Media* channel.

The serializer:

* Down-samples PCM to 8-kHz μ-law for **outgoing** audio (:class:`AudioRawFrame`).
* Up-samples μ-law to the pipeline's native rate for **incoming** audio.
"""

from typing import Optional

from loguru import logger
from pipecat.audio.utils import create_default_resampler, pcm_to_ulaw, ulaw_to_pcm
from pipecat.frames.frames import (
    AudioRawFrame,
    Frame,
    InputAudioRawFrame,
    StartFrame,
)
from pipecat.serializers.base_serializer import FrameSerializer, FrameSerializerType
from pydantic import BaseModel


class StasisRTPFrameSerializer(FrameSerializer):
    """Serializer for Asterisk External Media streams (raw μ-law)."""

    class InputParams(BaseModel):
        """Configuration parameters.

        Attributes:
        ----------
        stasis_sample_rate : int, default 8000
            The sample-rate used by Stasis when sending μ-law (PCMU).
        sample_rate : Optional[int]
            Override for the pipeline's *input* sample-rate.  When omitted the
            value from the :class:`StartFrame` is used.
        """

        stasis_sample_rate: int = 8000
        sample_rate: Optional[int] = None

    def __init__(self, params: Optional[InputParams] = None):
        """Initialize Stasis RTP frame serializer.

        Args:
            params: Optional configuration parameters for the serializer.
        """
        self._params = params or self.InputParams()

        # Wire / pipeline rates
        self._stasis_sample_rate = self._params.stasis_sample_rate
        self._sample_rate = 0  # pipeline rate, filled in *setup*

        # Resampler shared between encode / decode paths
        self._resampler = create_default_resampler()

    @property
    def type(self) -> FrameSerializerType:
        """Stasis uses raw bytes → BINARY."""
        return FrameSerializerType.BINARY

    async def setup(self, frame: StartFrame):
        """Remember pipeline configuration."""
        self._sample_rate = self._params.sample_rate or frame.audio_in_sample_rate

    async def serialize(self, frame: Frame) -> bytes | str | None:
        """Convert a Pipecat frame to a wire payload.

        Only :class:`AudioRawFrame` instances are translated all other frame
        types are silently ignored, allowing higher-level transports to deal
        with them as needed.
        """
        if isinstance(frame, AudioRawFrame):
            try:
                # Pipeline PCM → 8-kHz μ-law
                encoded = await pcm_to_ulaw(
                    frame.audio,
                    frame.sample_rate,
                    self._stasis_sample_rate,
                    self._resampler,
                )
                return encoded  # raw bytes
            except Exception as exc:  # pragma: no cover – robustness
                logger.error(
                    f"StasisRTPFrameSerializer.serialize: encode failed: {exc}"
                )
                return None

        # Non-audio frames are not transmitted on the media path
        return None

    async def deserialize(self, data: bytes | str) -> Frame | None:
        """Convert wire payloads to Pipecat frames.

        The Stasis media socket delivers bare μ-law bytes, therefore *data*
        must be *bytes*.  Any *str* is ignored.
        """
        if not isinstance(data, (bytes, bytearray)):
            return None

        try:
            pcm = await ulaw_to_pcm(
                bytes(data),
                self._stasis_sample_rate,
                self._sample_rate,
                self._resampler,
            )
            return InputAudioRawFrame(
                audio=pcm,
                sample_rate=self._sample_rate,
                num_channels=1,
            )
        except Exception as exc:  # pragma: no cover
            logger.error(f"StasisRTPFrameSerializer.deserialize: decode failed: {exc}")
            return None