#include "subtitles.h"

#include <algorithm>
#include <regex>

#include "common/util/FileUtil.h"

#include "third-party/fmt/core.h"
#include "third-party/fmt/ranges.h"
#include "third-party/json.hpp"

bool write_subtitle_db_to_files(const GameSubtitleDB& db) {
  // Write the subtitles out
  std::vector<int> completed_banks = {};
  for (const auto& [id, bank] : db.m_banks) {
    // If we've done the bank before, skip it
    auto it = find(completed_banks.begin(), completed_banks.end(), bank->m_lang_id);
    if (it != completed_banks.end()) {
      continue;
    }
    // Check to see if this bank is shared by any other, if so do it at the same time
    // and skip it
    // This is basically just to deal with US/UK english in a not so hacky way
    std::vector<int> banks = {};
    for (const auto& [_id, _bank] : db.m_banks) {
      if (_bank->file_path == bank->file_path) {
        banks.push_back(_bank->m_lang_id);
        completed_banks.push_back(_bank->m_lang_id);
      }
    }

    std::string file_contents = "";
    file_contents += fmt::format("(language-id {})\n", fmt::join(banks, " "));

    for (const auto& group_name : db.m_subtitle_groups->m_group_order) {
      file_contents +=
          fmt::format("\n;; -----------------\n;; {}\n;; -----------------\n", group_name);
      for (const auto& [scene_name, scene_info] : bank->m_scenes) {
        if (scene_info.m_sorting_group != group_name) {
          continue;
        }
        file_contents += fmt::format("\n(\"{}\"", scene_name);
        if (scene_info.m_kind == SubtitleSceneKind::Hint) {
          file_contents += " :hint #x0";
        } else if (scene_info.m_kind == SubtitleSceneKind::HintNamed) {
          file_contents += fmt::format(" :hint #x{0:x}", scene_info.m_id);
        }
        file_contents += "\n";
        if (scene_info.m_lines.empty()) {
          file_contents += fmt::format("  ()\n");
        } else {
          for (const auto& line : scene_info.m_lines) {
            // Clear screen entries
            if (line.line_utf8.empty()) {
              file_contents += fmt::format("  ({})\n", line.frame);
            } else {
              file_contents += fmt::format("  ({}", line.frame);
              if (line.offscreen && scene_info.m_kind == SubtitleSceneKind::Movie) {
                file_contents += " :offscreen";
              }
              file_contents += fmt::format(" \"{}\"", line.speaker_utf8);
              // escape quotes
              std::string temp = line.line_utf8;
              temp = std::regex_replace(temp, std::regex("\""), "\\\"");
              file_contents += fmt::format(" \"{}\")\n", temp);
            }
          }
        }
        file_contents += "  )\n";
      }
    }

    // Commit it to the file
    std::string full_path =
        (file_util::get_jak_project_dir() / std::filesystem::path(bank->file_path)).string();
    file_util::write_text_file(full_path, file_contents);
  }

  // Write the subtitle group info out
  nlohmann::json json(db.m_subtitle_groups->m_groups);
  json[db.m_subtitle_groups->group_order_key] = nlohmann::json(db.m_subtitle_groups->m_group_order);
  std::string file_path = (file_util::get_jak_project_dir() / "game" / "assets" / "jak1" /
                           "subtitle" / "subtitle-groups.json")
                              .string();
  file_util::write_text_file(file_path, json.dump(2));

  return true;
}