#pragma once

#include <stdbool.h>
#include <stddef.h>
#include <stdint.h>

#include "esp_err.h"
#include "freertos/FreeRTOS.h"
#include "freertos/portmacro.h"
#include "freertos/semphr.h"

#include "audio_receiver.h"

#define AAC_FRAMES_PER_PACKET  352
#define AUDIO_MAX_CHANNELS     2
#define AUDIO_BYTES_PER_SAMPLE 2
#define MAX_SAMPLES_PER_FRAME  4096

typedef struct __attribute__((packed)) {
  uint32_t rtp_timestamp;
  uint16_t samples_per_channel;
  uint8_t channels;
  uint8_t reserved;
} audio_frame_header_t;

#define MAX_RING_BUFFER_FRAMES 1000
#define BYTES_PER_FRAME                                          \
  ((size_t)sizeof(audio_frame_header_t) +                        \
   ((size_t)AAC_FRAMES_PER_PACKET * (size_t)AUDIO_MAX_CHANNELS * \
    (size_t)AUDIO_BYTES_PER_SAMPLE))
#define AUDIO_BUFFER_SIZE (MAX_RING_BUFFER_FRAMES * BYTES_PER_FRAME)

typedef struct {
  uint8_t *pool;                // Pre-allocated frame data in PSRAM
  uint16_t *sorted;             // Slot indices sorted by RTP timestamp
  uint16_t *free_stack;         // Stack of free slot indices
  int count;                    // Frames currently in buffer
  int free_top;                 // Top of free stack (next free slot)
  int capacity;                 // Max frames
  size_t slot_size;             // BYTES_PER_FRAME
  portMUX_TYPE lock;            // Spinlock for count/index manipulation
  SemaphoreHandle_t data_ready; // Counting semaphore (blocks consumer)
  uint8_t *frame_buffer;        // Temp assembly buffer
  int16_t *decode_buffer;       // Decode buffer pointer
  size_t decode_capacity_samples;
} audio_buffer_t;

esp_err_t audio_buffer_init(audio_buffer_t *buffer);
void audio_buffer_deinit(audio_buffer_t *buffer);
void audio_buffer_flush(audio_buffer_t *buffer);
int audio_buffer_get_frame_count(audio_buffer_t *buffer);
bool audio_buffer_is_nearly_full(audio_buffer_t *buffer);
bool audio_buffer_take(audio_buffer_t *buffer, void **item, size_t *item_size,
                       TickType_t ticks);
void audio_buffer_return(audio_buffer_t *buffer, void *item);
int16_t *audio_buffer_get_decode_buffer(audio_buffer_t *buffer,
                                        size_t *capacity_samples);
bool audio_buffer_queue_decoded(audio_buffer_t *buffer, audio_stats_t *stats,
                                uint32_t timestamp, const int16_t *pcm_data,
                                size_t samples, int channels);
/**
 * Peek at the RTP timestamp of the oldest (lowest-timestamp) frame in the
 * buffer without removing it.  Returns false if the buffer is empty.
 */
bool audio_buffer_oldest_timestamp(audio_buffer_t *buffer, uint32_t *timestamp);