#pragma once

// --- Includes ---
#include <SDL3/SDL.h>
#include <stdint.h>  // Para uint32_t, uint8_t
#include <stdio.h>   // Para NULL, fseek, fclose, fopen, fread, ftell, FILE, SEEK_END, SEEK_SET
#include <stdlib.h>  // Para free, malloc

#include <iostream>  // Para std::cout
#include <memory>    // Para std::unique_ptr
#include <string>    // Para std::string
#include <vector>    // Para std::vector

#define STB_VORBIS_HEADER_ONLY
#include "external/stb_vorbis.h"  // Para stb_vorbis_open_memory i streaming

// Deleter stateless per a buffers reservats amb `SDL_malloc` / `SDL_LoadWAV*`.
// Compatible amb `std::unique_ptr<Uint8[], SDLFreeDeleter>` — zero size
// overhead gràcies a EBO, igual que un unique_ptr amb default_delete.
struct SDLFreeDeleter {
    void operator()(Uint8* p) const noexcept {
        if (p != nullptr) {
            SDL_free(p);
        }
    }
};

// --- Public Enums ---
enum JA_Channel_state : std::uint8_t {
    JA_CHANNEL_INVALID,
    JA_CHANNEL_FREE,
    JA_CHANNEL_PLAYING,
    JA_CHANNEL_PAUSED,
    JA_SOUND_DISABLED,
};
enum JA_Music_state : std::uint8_t {
    JA_MUSIC_INVALID,
    JA_MUSIC_PLAYING,
    JA_MUSIC_PAUSED,
    JA_MUSIC_STOPPED,
    JA_MUSIC_DISABLED,
};

// --- Struct Definitions ---
enum : std::uint8_t {
    JA_MAX_SIMULTANEOUS_CHANNELS = 20,
    JA_MAX_GROUPS = 2
};

struct JA_Sound_t {
    SDL_AudioSpec spec{SDL_AUDIO_S16, 2, 48000};
    Uint32 length{0};
    // Buffer descomprimit (PCM) propietat del sound. Reservat per SDL_LoadWAV
    // via SDL_malloc; el deleter `SDLFreeDeleter` allibera amb SDL_free.
    std::unique_ptr<Uint8[], SDLFreeDeleter> buffer;
};

struct JA_Channel_t {
    JA_Sound_t* sound{nullptr};
    SDL_AudioStream* stream{nullptr};
    int pos{0};
    int times{0};
    int group{0};
    JA_Channel_state state{JA_CHANNEL_FREE};
};

struct JA_Music_t {
    SDL_AudioSpec spec{SDL_AUDIO_S16, 2, 48000};

    // OGG comprimit en memòria. Propietat nostra; es copia des del buffer
    // d'entrada una sola vegada en JA_LoadMusic i es descomprimix en chunks
    // per streaming. Com que stb_vorbis guarda un punter persistent al
    // `.data()` d'aquest vector, no el podem resize'jar un cop establert
    // (una reallocation invalidaria el punter que el decoder conserva).
    std::vector<Uint8> ogg_data;
    stb_vorbis* vorbis{nullptr};  // handle del decoder, viu tot el cicle del JA_Music_t

    std::string filename;

    int times{0};  // loops restants (-1 = infinit, 0 = un sol play)
    SDL_AudioStream* stream{nullptr};
    JA_Music_state state{JA_MUSIC_INVALID};
};

// --- Internal Global State (inline, C++17) ---

inline JA_Music_t* current_music{nullptr};
inline JA_Channel_t channels[JA_MAX_SIMULTANEOUS_CHANNELS];

inline SDL_AudioSpec JA_audioSpec{SDL_AUDIO_S16, 2, 48000};
inline float JA_musicVolume{1.0F};
inline float JA_soundVolume[JA_MAX_GROUPS];
inline bool JA_musicEnabled{true};
inline bool JA_soundEnabled{true};
inline SDL_AudioDeviceID sdlAudioDevice{0};

// --- Crossfade / Fade State ---
struct JA_FadeState {
    bool active{false};
    Uint64 start_time{0};
    int duration_ms{0};
    float initial_volume{0.0F};
};

struct JA_OutgoingMusic {
    SDL_AudioStream* stream{nullptr};
    JA_FadeState fade;
};

inline JA_OutgoingMusic outgoing_music;
inline JA_FadeState incoming_fade;

// --- Forward Declarations ---
inline void JA_StopMusic();
inline void JA_StopChannel(int channel);
inline auto JA_PlaySoundOnChannel(JA_Sound_t* sound, int channel, int loop = 0, int group = 0) -> int;
inline void JA_CrossfadeMusic(JA_Music_t* music, int crossfade_ms, int loop = -1);

// --- Music streaming internals ---
// Bytes-per-sample per canal (sempre s16)
static constexpr int JA_MUSIC_BYTES_PER_SAMPLE = 2;
// Quants shorts decodifiquem per crida a get_samples_short_interleaved.
// 8192 shorts = 4096 samples/channel en estèreo ≈ 85ms de so a 48kHz.
static constexpr int JA_MUSIC_CHUNK_SHORTS = 8192;
// Umbral d'audio per davant del cursor de reproducció. Mantenim ≥ 0.5 s a
// l'SDL_AudioStream per absorbir jitter de frame i evitar underruns.
static constexpr float JA_MUSIC_LOW_WATER_SECONDS = 0.5F;

// Decodifica un chunk del vorbis i el volca a l'stream. Retorna samples
// decodificats per canal (0 = EOF de l'stream vorbis).
inline auto JA_FeedMusicChunk(JA_Music_t* music) -> int {
    if ((music == nullptr) || (music->vorbis == nullptr) || (music->stream == nullptr)) {
        return 0;
    }

    short chunk[JA_MUSIC_CHUNK_SHORTS];
    const int num_channels = music->spec.channels;
    const int samples_per_channel = stb_vorbis_get_samples_short_interleaved(
        music->vorbis,
        num_channels,
        chunk,
        JA_MUSIC_CHUNK_SHORTS);
    if (samples_per_channel <= 0) {
        return 0;
    }

    const int bytes = samples_per_channel * num_channels * JA_MUSIC_BYTES_PER_SAMPLE;
    SDL_PutAudioStreamData(music->stream, chunk, bytes);
    return samples_per_channel;
}

// Reompli l'stream fins que tinga ≥ JA_MUSIC_LOW_WATER_SECONDS bufferats.
// En arribar a EOF del vorbis, aplica el loop (times) o deixa drenar.
inline void JA_PumpMusic(JA_Music_t* music) {
    if ((music == nullptr) || (music->vorbis == nullptr) || (music->stream == nullptr)) {
        return;
    }

    const int bytes_per_second = music->spec.freq * music->spec.channels * JA_MUSIC_BYTES_PER_SAMPLE;
    const int low_water_bytes = static_cast<int>(JA_MUSIC_LOW_WATER_SECONDS * static_cast<float>(bytes_per_second));

    while (SDL_GetAudioStreamAvailable(music->stream) < low_water_bytes) {
        const int decoded = JA_FeedMusicChunk(music);
        if (decoded > 0) {
            continue;
        }

        // EOF: si queden loops, rebobinar; si no, tallar i deixar drenar.
        if (music->times != 0) {
            stb_vorbis_seek_start(music->vorbis);
            if (music->times > 0) {
                music->times--;
            }
        } else {
            break;
        }
    }
}

// Pre-carrega `duration_ms` de so dins l'stream actual abans que l'stream
// siga robat per outgoing_music (crossfade o fade-out). Imprescindible amb
// streaming: l'stream robat no es pot re-alimentar perquè perd la referència
// al seu vorbis decoder. No aplica loop — si el vorbis s'esgota abans, parem.
inline void JA_PreFillOutgoing(JA_Music_t* music, int duration_ms) {
    if ((music == nullptr) || (music->vorbis == nullptr) || (music->stream == nullptr)) {
        return;
    }

    const int bytes_per_second = music->spec.freq * music->spec.channels * JA_MUSIC_BYTES_PER_SAMPLE;
    const int needed_bytes = static_cast<int>((static_cast<int64_t>(duration_ms) * bytes_per_second) / 1000);

    while (SDL_GetAudioStreamAvailable(music->stream) < needed_bytes) {
        const int decoded = JA_FeedMusicChunk(music);
        if (decoded <= 0) {
            break;  // EOF: deixem drenar el que hi haja
        }
    }
}

// --- Core Functions ---

// Fade-out de la música sortint (crossfade o fade-out a silenci). En acabar
// destrueix l'AudioStream sortint.
inline void JA_UpdateOutgoingFade() {
    if ((outgoing_music.stream == nullptr) || !outgoing_music.fade.active) {
        return;
    }
    const Uint64 elapsed = SDL_GetTicks() - outgoing_music.fade.start_time;
    if (elapsed >= static_cast<Uint64>(outgoing_music.fade.duration_ms)) {
        SDL_DestroyAudioStream(outgoing_music.stream);
        outgoing_music.stream = nullptr;
        outgoing_music.fade.active = false;
        return;
    }
    const float percent = static_cast<float>(elapsed) / static_cast<float>(outgoing_music.fade.duration_ms);
    SDL_SetAudioStreamGain(outgoing_music.stream, outgoing_music.fade.initial_volume * (1.0F - percent));
}

// Fade-in de la música entrant (segona meitat d'un crossfade).
inline void JA_UpdateIncomingFade() {
    if (!incoming_fade.active) {
        return;
    }
    const Uint64 elapsed = SDL_GetTicks() - incoming_fade.start_time;
    if (elapsed >= static_cast<Uint64>(incoming_fade.duration_ms)) {
        incoming_fade.active = false;
        SDL_SetAudioStreamGain(current_music->stream, JA_musicVolume);
        return;
    }
    const float percent = static_cast<float>(elapsed) / static_cast<float>(incoming_fade.duration_ms);
    SDL_SetAudioStreamGain(current_music->stream, JA_musicVolume * percent);
}

// Manté l'stream de la música activa alimentat i para la reproducció si
// el vorbis s'ha esgotat i no queden loops.
inline void JA_UpdateCurrentMusic() {
    if (!JA_musicEnabled || current_music == nullptr || current_music->state != JA_MUSIC_PLAYING) {
        return;
    }
    JA_UpdateIncomingFade();
    JA_PumpMusic(current_music);
    if (current_music->times == 0 && SDL_GetAudioStreamAvailable(current_music->stream) == 0) {
        JA_StopMusic();
    }
}

// Avança l'estat d'un sol canal de so: alimenta més samples mentre quedin
// loops; si ja no queden i l'stream s'ha buidat, atura el canal.
inline void JA_UpdateSoundChannel(int channel_index) {
    JA_Channel_t& ch = channels[channel_index];
    if (ch.state != JA_CHANNEL_PLAYING) {
        return;
    }
    if (ch.times == 0) {
        if (SDL_GetAudioStreamAvailable(ch.stream) == 0) {
            JA_StopChannel(channel_index);
        }
        return;
    }
    if (static_cast<Uint32>(SDL_GetAudioStreamAvailable(ch.stream)) < (ch.sound->length / 2)) {
        SDL_PutAudioStreamData(ch.stream, ch.sound->buffer.get(), ch.sound->length);
        if (ch.times > 0) {
            ch.times--;
        }
    }
}

// Avança tots els canals de so actius.
inline void JA_UpdateSoundChannels() {
    if (!JA_soundEnabled) {
        return;
    }
    for (int i = 0; i < JA_MAX_SIMULTANEOUS_CHANNELS; ++i) {
        JA_UpdateSoundChannel(i);
    }
}

inline void JA_Update() {
    JA_UpdateOutgoingFade();
    JA_UpdateCurrentMusic();
    JA_UpdateSoundChannels();
}

inline void JA_Init(const int freq, const SDL_AudioFormat format, const int num_channels) {
    JA_audioSpec = {.format = format, .channels = num_channels, .freq = freq};
    if (sdlAudioDevice != 0U) {
        SDL_CloseAudioDevice(sdlAudioDevice);
    }
    sdlAudioDevice = SDL_OpenAudioDevice(SDL_AUDIO_DEVICE_DEFAULT_PLAYBACK, &JA_audioSpec);
    if (sdlAudioDevice == 0) {
        std::cout << "Failed to initialize SDL audio!" << '\n';
    }
    for (auto& channel : channels) {
        channel.state = JA_CHANNEL_FREE;
    }
    for (float& i : JA_soundVolume) {
        i = 0.5F;
    }
}

inline void JA_Quit() {
    if (outgoing_music.stream != nullptr) {
        SDL_DestroyAudioStream(outgoing_music.stream);
        outgoing_music.stream = nullptr;
    }
    if (sdlAudioDevice != 0U) {
        SDL_CloseAudioDevice(sdlAudioDevice);
    }
    sdlAudioDevice = 0;
}

// --- Music Functions ---

inline auto JA_LoadMusic(const Uint8* buffer, Uint32 length) -> JA_Music_t* {
    if ((buffer == nullptr) || length == 0) {
        return nullptr;
    }

    // Allocem el JA_Music_t primer per aprofitar el seu `std::vector<Uint8>`
    // com a propietari del OGG comprimit. stb_vorbis guarda un punter
    // persistent al buffer; com que ací no el resize'jem, el .data() és
    // estable durant tot el cicle de vida del music.
    auto* music = new JA_Music_t();
    music->ogg_data.assign(buffer, buffer + length);

    int err = 0;
    music->vorbis = stb_vorbis_open_memory(music->ogg_data.data(),
        static_cast<int>(length),
        &err,
        nullptr);
    if (music->vorbis == nullptr) {
        std::cout << "JA_LoadMusic: stb_vorbis_open_memory failed (error " << err << ")" << '\n';
        delete music;
        return nullptr;
    }

    const stb_vorbis_info info = stb_vorbis_get_info(music->vorbis);
    music->spec.channels = info.channels;
    music->spec.freq = static_cast<int>(info.sample_rate);
    music->spec.format = SDL_AUDIO_S16;
    music->state = JA_MUSIC_STOPPED;

    return music;
}

// Overload amb filename — els callers l'usen per poder comparar la música
// en curs amb JA_GetMusicFilename() i no rearrancar-la si ja és la mateixa.
inline auto JA_LoadMusic(Uint8* buffer, Uint32 length, const char* filename) -> JA_Music_t* {
    JA_Music_t* music = JA_LoadMusic(static_cast<const Uint8*>(buffer), length);
    if ((music != nullptr) && (filename != nullptr)) {
        music->filename = filename;
    }
    return music;
}

inline auto JA_LoadMusic(const char* filename) -> JA_Music_t* {
    // Carreguem primer el arxiu en memòria i després el descomprimim.
    FILE* f = fopen(filename, "rb");
    if (f == nullptr) {
        return nullptr;
    }
    fseek(f, 0, SEEK_END);
    long fsize = ftell(f);
    fseek(f, 0, SEEK_SET);
    auto* buffer = static_cast<Uint8*>(malloc(fsize + 1));
    if (buffer == nullptr) {
        fclose(f);
        return nullptr;
    }
    if (fread(buffer, fsize, 1, f) != 1) {
        fclose(f);
        free(buffer);
        return nullptr;
    }
    fclose(f);

    JA_Music_t* music = JA_LoadMusic(static_cast<const Uint8*>(buffer), static_cast<Uint32>(fsize));
    if (music != nullptr) {
        music->filename = filename;
    }

    free(buffer);

    return music;
}

inline void JA_PlayMusic(JA_Music_t* music, const int loop = -1) {
    if (!JA_musicEnabled || (music == nullptr) || (music->vorbis == nullptr)) {
        return;
    }

    JA_StopMusic();

    current_music = music;
    current_music->state = JA_MUSIC_PLAYING;
    current_music->times = loop;

    // Rebobinem l'stream de vorbis al principi. Cobreix tant play-per-primera-
    // vegada com replays/canvis de track que tornen a la mateixa pista.
    stb_vorbis_seek_start(current_music->vorbis);

    current_music->stream = SDL_CreateAudioStream(&current_music->spec, &JA_audioSpec);
    if (current_music->stream == nullptr) {
        std::cout << "Failed to create audio stream!" << '\n';
        current_music->state = JA_MUSIC_STOPPED;
        return;
    }
    SDL_SetAudioStreamGain(current_music->stream, JA_musicVolume);

    // Pre-cargem el buffer abans de bindejar per evitar un underrun inicial.
    JA_PumpMusic(current_music);

    if (!SDL_BindAudioStream(sdlAudioDevice, current_music->stream)) {
        std::cout << "[ERROR] SDL_BindAudioStream failed!" << '\n';
    }
}

inline auto JA_GetMusicFilename(const JA_Music_t* music = nullptr) -> const char* {
    if (music == nullptr) {
        music = current_music;
    }
    if ((music == nullptr) || music->filename.empty()) {
        return nullptr;
    }
    return music->filename.c_str();
}

inline void JA_PauseMusic() {
    if (!JA_musicEnabled) {
        return;
    }
    if ((current_music == nullptr) || current_music->state != JA_MUSIC_PLAYING) {
        return;
    }

    current_music->state = JA_MUSIC_PAUSED;
    SDL_UnbindAudioStream(current_music->stream);
}

inline void JA_ResumeMusic() {
    if (!JA_musicEnabled) {
        return;
    }
    if ((current_music == nullptr) || current_music->state != JA_MUSIC_PAUSED) {
        return;
    }

    current_music->state = JA_MUSIC_PLAYING;
    SDL_BindAudioStream(sdlAudioDevice, current_music->stream);
}

inline void JA_StopMusic() {
    // Limpiar outgoing crossfade si existe
    if (outgoing_music.stream != nullptr) {
        SDL_DestroyAudioStream(outgoing_music.stream);
        outgoing_music.stream = nullptr;
        outgoing_music.fade.active = false;
    }
    incoming_fade.active = false;

    if ((current_music == nullptr) || current_music->state == JA_MUSIC_INVALID || current_music->state == JA_MUSIC_STOPPED) {
        return;
    }

    current_music->state = JA_MUSIC_STOPPED;
    if (current_music->stream != nullptr) {
        SDL_DestroyAudioStream(current_music->stream);
        current_music->stream = nullptr;
    }
    // Deixem el handle de vorbis viu — es tanca en JA_DeleteMusic.
    // Rebobinem perquè un futur JA_PlayMusic comence des del principi.
    if (current_music->vorbis != nullptr) {
        stb_vorbis_seek_start(current_music->vorbis);
    }
}

inline void JA_FadeOutMusic(const int milliseconds) {
    if (!JA_musicEnabled) {
        return;
    }
    if ((current_music == nullptr) || current_music->state != JA_MUSIC_PLAYING) {
        return;
    }

    // Destruir outgoing anterior si existe
    if (outgoing_music.stream != nullptr) {
        SDL_DestroyAudioStream(outgoing_music.stream);
        outgoing_music.stream = nullptr;
    }

    // Pre-omplim l'stream amb `milliseconds` de so: un cop robat, ja no
    // tindrà accés al vorbis decoder i només podrà drenar el que tinga.
    JA_PreFillOutgoing(current_music, milliseconds);

    // Robar el stream del current_music al outgoing
    outgoing_music.stream = current_music->stream;
    outgoing_music.fade = {.active = true, .start_time = SDL_GetTicks(), .duration_ms = milliseconds, .initial_volume = JA_musicVolume};

    // Dejar current_music sin stream (ya lo tiene outgoing)
    current_music->stream = nullptr;
    current_music->state = JA_MUSIC_STOPPED;
    if (current_music->vorbis != nullptr) {
        stb_vorbis_seek_start(current_music->vorbis);
    }
    incoming_fade.active = false;
}

inline void JA_CrossfadeMusic(JA_Music_t* music, const int crossfade_ms, const int loop) {
    if (!JA_musicEnabled || (music == nullptr) || (music->vorbis == nullptr)) {
        return;
    }

    // Destruir outgoing anterior si existe (crossfade durante crossfade)
    if (outgoing_music.stream != nullptr) {
        SDL_DestroyAudioStream(outgoing_music.stream);
        outgoing_music.stream = nullptr;
        outgoing_music.fade.active = false;
    }

    // Robar el stream de la musica actual al outgoing para el fade-out.
    // Pre-omplim amb `crossfade_ms` de so perquè no es quede en silenci
    // abans d'acabar el fade (l'stream robat ja no pot alimentar-se).
    if ((current_music != nullptr) && current_music->state == JA_MUSIC_PLAYING && (current_music->stream != nullptr)) {
        JA_PreFillOutgoing(current_music, crossfade_ms);
        outgoing_music.stream = current_music->stream;
        outgoing_music.fade = {.active = true, .start_time = SDL_GetTicks(), .duration_ms = crossfade_ms, .initial_volume = JA_musicVolume};
        current_music->stream = nullptr;
        current_music->state = JA_MUSIC_STOPPED;
        if (current_music->vorbis != nullptr) {
            stb_vorbis_seek_start(current_music->vorbis);
        }
    }

    // Iniciar la nueva pista con gain=0 (el fade-in la sube gradualmente)
    current_music = music;
    current_music->state = JA_MUSIC_PLAYING;
    current_music->times = loop;

    stb_vorbis_seek_start(current_music->vorbis);
    current_music->stream = SDL_CreateAudioStream(&current_music->spec, &JA_audioSpec);
    if (current_music->stream == nullptr) {
        std::cout << "Failed to create audio stream for crossfade!" << '\n';
        current_music->state = JA_MUSIC_STOPPED;
        return;
    }
    SDL_SetAudioStreamGain(current_music->stream, 0.0F);
    JA_PumpMusic(current_music);  // pre-carrega abans de bindejar
    SDL_BindAudioStream(sdlAudioDevice, current_music->stream);

    // Configurar fade-in
    incoming_fade = {.active = true, .start_time = SDL_GetTicks(), .duration_ms = crossfade_ms, .initial_volume = 0.0F};
}

inline auto JA_GetMusicState() -> JA_Music_state {
    if (!JA_musicEnabled) {
        return JA_MUSIC_DISABLED;
    }
    if (current_music == nullptr) {
        return JA_MUSIC_INVALID;
    }

    return current_music->state;
}

inline void JA_DeleteMusic(JA_Music_t* music) {
    if (music == nullptr) {
        return;
    }
    if (current_music == music) {
        JA_StopMusic();
        current_music = nullptr;
    }
    if (music->stream != nullptr) {
        SDL_DestroyAudioStream(music->stream);
    }
    if (music->vorbis != nullptr) {
        stb_vorbis_close(music->vorbis);
    }
    // ogg_data (std::vector) i filename (std::string) s'alliberen sols
    // al destructor de JA_Music_t.
    delete music;
}

inline auto JA_SetMusicVolume(float volume) -> float {
    JA_musicVolume = SDL_clamp(volume, 0.0F, 1.0F);
    if ((current_music != nullptr) && (current_music->stream != nullptr)) {
        SDL_SetAudioStreamGain(current_music->stream, JA_musicVolume);
    }
    return JA_musicVolume;
}

inline void JA_SetMusicPosition(float /*value*/) {
    // No implementat amb el backend de streaming.
}

inline auto JA_GetMusicPosition() -> float {
    return 0.0F;
}

inline void JA_EnableMusic(const bool value) {
    if (!value && (current_music != nullptr) && (current_music->state == JA_MUSIC_PLAYING)) {
        JA_StopMusic();
    }
    JA_musicEnabled = value;
}

// --- Sound Functions ---

inline auto JA_LoadSound(uint8_t* buffer, uint32_t size) -> JA_Sound_t* {
    auto sound = std::make_unique<JA_Sound_t>();
    Uint8* raw = nullptr;
    if (!SDL_LoadWAV_IO(SDL_IOFromMem(buffer, size), true, &sound->spec, &raw, &sound->length)) {
        std::cout << "Failed to load WAV from memory: " << SDL_GetError() << '\n';
        return nullptr;
    }
    sound->buffer.reset(raw);  // adopta el SDL_malloc'd buffer
    return sound.release();
}

inline auto JA_LoadSound(const char* filename) -> JA_Sound_t* {
    auto sound = std::make_unique<JA_Sound_t>();
    Uint8* raw = nullptr;
    if (!SDL_LoadWAV(filename, &sound->spec, &raw, &sound->length)) {
        std::cout << "Failed to load WAV file: " << SDL_GetError() << '\n';
        return nullptr;
    }
    sound->buffer.reset(raw);  // adopta el SDL_malloc'd buffer
    return sound.release();
}

inline auto JA_PlaySound(JA_Sound_t* sound, const int loop = 0, const int group = 0) -> int {
    if (!JA_soundEnabled || (sound == nullptr)) {
        return -1;
    }

    int channel = 0;
    while (channel < JA_MAX_SIMULTANEOUS_CHANNELS && channels[channel].state != JA_CHANNEL_FREE) { channel++; }
    if (channel == JA_MAX_SIMULTANEOUS_CHANNELS) {
        // No hay canal libre, reemplazamos el primero
        channel = 0;
    }

    return JA_PlaySoundOnChannel(sound, channel, loop, group);
}

inline auto JA_PlaySoundOnChannel(JA_Sound_t* sound, const int channel, const int loop, const int group) -> int {
    if (!JA_soundEnabled || (sound == nullptr)) {
        return -1;
    }
    if (channel < 0 || channel >= JA_MAX_SIMULTANEOUS_CHANNELS) {
        return -1;
    }

    JA_StopChannel(channel);

    channels[channel].sound = sound;
    channels[channel].times = loop;
    channels[channel].pos = 0;
    channels[channel].group = group;
    channels[channel].state = JA_CHANNEL_PLAYING;
    channels[channel].stream = SDL_CreateAudioStream(&channels[channel].sound->spec, &JA_audioSpec);

    if (channels[channel].stream == nullptr) {
        std::cout << "Failed to create audio stream for sound!" << '\n';
        channels[channel].state = JA_CHANNEL_FREE;
        return -1;
    }

    SDL_PutAudioStreamData(channels[channel].stream, channels[channel].sound->buffer.get(), channels[channel].sound->length);
    SDL_SetAudioStreamGain(channels[channel].stream, JA_soundVolume[group]);
    SDL_BindAudioStream(sdlAudioDevice, channels[channel].stream);

    return channel;
}

inline void JA_DeleteSound(JA_Sound_t* sound) {
    if (sound == nullptr) {
        return;
    }
    for (int i = 0; i < JA_MAX_SIMULTANEOUS_CHANNELS; i++) {
        if (channels[i].sound == sound) {
            JA_StopChannel(i);
        }
    }
    // buffer es destrueix automàticament via RAII (SDLFreeDeleter).
    delete sound;
}

inline void JA_PauseChannel(const int channel) {
    if (!JA_soundEnabled) {
        return;
    }

    if (channel == -1) {
        for (auto& channel : channels) {
            if (channel.state == JA_CHANNEL_PLAYING) {
                channel.state = JA_CHANNEL_PAUSED;
                SDL_UnbindAudioStream(channel.stream);
            }
        }
    } else if (channel >= 0 && channel < JA_MAX_SIMULTANEOUS_CHANNELS) {
        if (channels[channel].state == JA_CHANNEL_PLAYING) {
            channels[channel].state = JA_CHANNEL_PAUSED;
            SDL_UnbindAudioStream(channels[channel].stream);
        }
    }
}

inline void JA_ResumeChannel(const int channel) {
    if (!JA_soundEnabled) {
        return;
    }

    if (channel == -1) {
        for (auto& channel : channels) {
            if (channel.state == JA_CHANNEL_PAUSED) {
                channel.state = JA_CHANNEL_PLAYING;
                SDL_BindAudioStream(sdlAudioDevice, channel.stream);
            }
        }
    } else if (channel >= 0 && channel < JA_MAX_SIMULTANEOUS_CHANNELS) {
        if (channels[channel].state == JA_CHANNEL_PAUSED) {
            channels[channel].state = JA_CHANNEL_PLAYING;
            SDL_BindAudioStream(sdlAudioDevice, channels[channel].stream);
        }
    }
}

inline void JA_StopChannel(const int channel) {
    if (channel == -1) {
        for (auto& channel : channels) {
            if (channel.state != JA_CHANNEL_FREE) {
                if (channel.stream != nullptr) {
                    SDL_DestroyAudioStream(channel.stream);
                }
                channel.stream = nullptr;
                channel.state = JA_CHANNEL_FREE;
                channel.pos = 0;
                channel.sound = nullptr;
            }
        }
    } else if (channel >= 0 && channel < JA_MAX_SIMULTANEOUS_CHANNELS) {
        if (channels[channel].state != JA_CHANNEL_FREE) {
            if (channels[channel].stream != nullptr) {
                SDL_DestroyAudioStream(channels[channel].stream);
            }
            channels[channel].stream = nullptr;
            channels[channel].state = JA_CHANNEL_FREE;
            channels[channel].pos = 0;
            channels[channel].sound = nullptr;
        }
    }
}

inline auto JA_GetChannelState(const int channel) -> JA_Channel_state {
    if (!JA_soundEnabled) {
        return JA_SOUND_DISABLED;
    }
    if (channel < 0 || channel >= JA_MAX_SIMULTANEOUS_CHANNELS) {
        return JA_CHANNEL_INVALID;
    }

    return channels[channel].state;
}

inline auto JA_SetSoundVolume(float volume, const int group = -1) -> float {
    const float v = SDL_clamp(volume, 0.0F, 1.0F);

    if (group == -1) {
        for (float& i : JA_soundVolume) {
            i = v;
        }
    } else if (group >= 0 && group < JA_MAX_GROUPS) {
        JA_soundVolume[group] = v;
    } else {
        return v;
    }

    // Aplicar volum als canals actius.
    for (auto& channel : channels) {
        if ((channel.state == JA_CHANNEL_PLAYING) || (channel.state == JA_CHANNEL_PAUSED)) {
            if (group == -1 || channel.group == group) {
                if (channel.stream != nullptr) {
                    SDL_SetAudioStreamGain(channel.stream, JA_soundVolume[channel.group]);
                }
            }
        }
    }
    return v;
}

inline void JA_EnableSound(const bool value) {
    if (!value) {
        JA_StopChannel(-1);
    }
    JA_soundEnabled = value;
}

inline auto JA_SetVolume(float volume) -> float {
    float v = JA_SetMusicVolume(volume);
    JA_SetSoundVolume(v, -1);
    return v;
}