#pragma once

#include <cstddef>

#include "ff.h"

#include "audio_element.hpp"
#include "codec.hpp"

namespace audio {

/*
 * An audio element that accepts various kinds of encoded audio streams as
 * input, and converts them to uncompressed PCM output.
 */
class AudioDecoder : public IAudioElement {
 public:
  AudioDecoder();
  ~AudioDecoder();

  auto SetInputBuffer(MessageBufferHandle_t*) -> void;
  auto SetOutputBuffer(MessageBufferHandle_t*) -> void;

  auto ProcessStreamInfo(StreamInfo& info)
      -> cpp::result<void, AudioProcessingError>;
  auto ProcessChunk(uint8_t* data, std::size_t length)
      -> cpp::result<size_t, AudioProcessingError>;
  auto ProcessIdle() -> cpp::result<void, AudioProcessingError>;

  AudioDecoder(const AudioDecoder&) = delete;
  AudioDecoder& operator=(const AudioDecoder&) = delete;

 private:
  std::unique_ptr<codecs::ICodec> current_codec_;
  std::optional<StreamInfo> stream_info_;

  uint8_t* chunk_buffer_;
};

}  // namespace audio