#pragma once

#include "PlaybackRecorder.h"
#include "kiss_fftr.h"
#include <pulse/pulseaudio.h>

#define WAVE_BUFFER_SIZE 1024

namespace WallpaperEngine::Audio::Drivers::Recorders {
class PlaybackRecorder;

class PulseAudioPlaybackRecorder final : public PlaybackRecorder {
public:
    /**
     * Struct that contains all the required data for the PulseAudio callbacks
     */
    struct PulseAudioData {
	kiss_fftr_cfg kisscfg;
	uint8_t* audioBuffer;
	uint8_t* audioBufferTmp;
	size_t currentWritePointer;
	bool fullFrameReady;
	pa_stream* captureStream;
    };

    PulseAudioPlaybackRecorder ();
    ~PulseAudioPlaybackRecorder () override;

    void update () override;

private:
    pa_mainloop* m_mainloop;
    pa_mainloop_api* m_mainloopApi;
    pa_context* m_context;
    PulseAudioData m_captureData;

    float m_audioFFTbuffer[WAVE_BUFFER_SIZE] = { 0.0f };
    kiss_fft_cpx m_FFTinfo[WAVE_BUFFER_SIZE / 2 + 1] = { { .r = 0.0f, .i = 0.0f } };
    float m_FFTdestination64[64] = { 0 };
    float m_FFTdestination32[32] = { 0 };
    float m_FFTdestination16[16] = { 0 };
};
} // namespace WallpaperEngine::Audio::Drivers::Recorders