/*
 * Copyright 2023 jacqueline <me@jacqueline.id.au>
 *
 * SPDX-License-Identifier: GPL-3.0-only
 */

#include "fatfs_audio_input.hpp"

#include <algorithm>
#include <climits>
#include <cstddef>
#include <cstdint>
#include <functional>
#include <future>
#include <memory>
#include <mutex>
#include <string>
#include <variant>

#include "esp_heap_caps.h"
#include "esp_log.h"
#include "ff.h"
#include "freertos/portmacro.h"
#include "freertos/projdefs.h"
#include "idf_additions.h"
#include "span.hpp"

#include "audio_events.hpp"
#include "audio_fsm.hpp"
#include "audio_source.hpp"
#include "codec.hpp"
#include "event_queue.hpp"
#include "fatfs_source.hpp"
#include "future_fetcher.hpp"
#include "tag_parser.hpp"
#include "tasks.hpp"
#include "types.hpp"

static const char* kTag = "SRC";

namespace audio {

FatfsAudioInput::FatfsAudioInput(database::ITagParser& tag_parser)
    : IAudioSource(),
      tag_parser_(tag_parser),
      new_stream_mutex_(),
      new_stream_(),
      has_new_stream_(xSemaphoreCreateBinary()),
      pending_path_() {}

FatfsAudioInput::~FatfsAudioInput() {
  vSemaphoreDelete(has_new_stream_);
}

auto FatfsAudioInput::SetPath(std::future<std::optional<std::string>> fut)
    -> void {
  std::lock_guard<std::mutex> guard{new_stream_mutex_};
  pending_path_.reset(
      new database::FutureFetcher<std::optional<std::string>>(std::move(fut)));

  xSemaphoreGive(has_new_stream_);
}

auto FatfsAudioInput::SetPath(const std::string& path) -> void {
  std::lock_guard<std::mutex> guard{new_stream_mutex_};
  if (OpenFile(path)) {
    xSemaphoreGive(has_new_stream_);
  }
}

auto FatfsAudioInput::SetPath() -> void {
  std::lock_guard<std::mutex> guard{new_stream_mutex_};
  new_stream_.reset();
  xSemaphoreGive(has_new_stream_);
}

auto FatfsAudioInput::HasNewStream() -> bool {
  bool res = xSemaphoreTake(has_new_stream_, 0);
  if (res) {
    xSemaphoreGive(has_new_stream_);
  }
  return res;
}

auto FatfsAudioInput::NextStream() -> std::shared_ptr<codecs::IStream> {
  while (true) {
    xSemaphoreTake(has_new_stream_, portMAX_DELAY);

    {
      std::lock_guard<std::mutex> guard{new_stream_mutex_};
      // If the path is a future, then wait for it to complete.
      // TODO(jacqueline): We should really make some kind of
      // FreeRTOS-integrated way to block a task whilst awaiting a future.
      if (pending_path_) {
        while (!pending_path_->Finished()) {
          vTaskDelay(pdMS_TO_TICKS(100));
        }
        auto res = pending_path_->Result();
        pending_path_.reset();

        if (res && *res) {
          OpenFile(**res);
        }
      }

      if (new_stream_ == nullptr) {
        continue;
      }

      auto stream = new_stream_;
      new_stream_ = nullptr;
      return stream;
    }
  }
}

auto FatfsAudioInput::OpenFile(const std::string& path) -> bool {
  ESP_LOGI(kTag, "opening file %s", path.c_str());

  database::TrackTags tags;
  if (!tag_parser_.ReadAndParseTags(path, &tags)) {
    ESP_LOGE(kTag, "failed to read tags");
    return false;
  }

  auto stream_type = ContainerToStreamType(tags.encoding());
  if (!stream_type.has_value()) {
    ESP_LOGE(kTag, "couldn't match container to stream");
    return false;
  }

  std::unique_ptr<FIL> file = std::make_unique<FIL>();
  FRESULT res = f_open(file.get(), path.c_str(), FA_READ);
  if (res != FR_OK) {
    ESP_LOGE(kTag, "failed to open file! res: %i", res);
    return false;
  }

  new_stream_.reset(new FatfsSource(stream_type.value(), std::move(file)));
  return true;
}

auto FatfsAudioInput::ContainerToStreamType(database::Container enc)
    -> std::optional<codecs::StreamType> {
  switch (enc) {
    case database::Container::kMp3:
      return codecs::StreamType::kMp3;
    case database::Container::kWav:
      return codecs::StreamType::kPcm;
    case database::Container::kOgg:
      return codecs::StreamType::kVorbis;
    case database::Container::kFlac:
      return codecs::StreamType::kFlac;
    case database::Container::kOpus:
      return codecs::StreamType::kOpus;
    case database::Container::kUnsupported:
    default:
      return {};
  }
}

}  // namespace audio