#include "hoshidicts/lookup.hpp"

#include <algorithm>
#include <chrono>
#include <format>
#include <fstream>
#include <iostream>
#include <numeric>
#include <string_view>
#include <vector>

#include "hoshidicts/deinflector.hpp"
#include "hoshidicts/query.hpp"

namespace {
std::vector<std::string> read_word_list(const std::string& path) {
  std::vector<std::string> words;
  std::ifstream file(path);
  std::string line;
  while (std::getline(file, line)) {
    if (line.ends_with('\r')) {
      line.pop_back();
    }
    std::string word(line, 0, line.find(','));
    if (!word.empty() && word != "Word") {
      words.push_back(std::move(word));
    }
  }
  return words;
}
}

int main(int argc, char** argv) {
  if (argc < 5) {
    std::cout << std::format(
        "{} <csv_path> <iterations> --term <dict_path>... [--freq <dict_path>...] [--pitch <dict_path>...]\n", argv[0]);
    return 1;
  }

  const std::string csv_path = argv[1];
  const int iterations = std::stoi(argv[2]);
  const std::vector<std::string> words = read_word_list(csv_path);

  std::vector<std::string> term_paths;
  std::vector<std::string> freq_paths;
  std::vector<std::string> pitch_paths;
  std::vector<std::string>* current = &term_paths;
  for (int i = 3; i < argc; ++i) {
    const std::string_view arg = argv[i];
    if (arg == "--term") {
      current = &term_paths;
    } else if (arg == "--freq") {
      current = &freq_paths;
    } else if (arg == "--pitch") {
      current = &pitch_paths;
    } else {
      current->emplace_back(arg);
    }
  }

  DictionaryQuery query;
  for (const auto& path : term_paths) {
    query.add_term_dict(path);
  }
  for (const auto& path : freq_paths) {
    query.add_freq_dict(path);
  }
  for (const auto& path : pitch_paths) {
    query.add_pitch_dict(path);
  }

  Deinflector deinflector;
  Lookup lookup(query, deinflector);

  std::vector<double> durations;
  durations.reserve(static_cast<size_t>(iterations) * words.size());
  for (int i = 0; i < iterations; ++i) {
    for (const auto& word : words) {
      const auto start = std::chrono::high_resolution_clock::now();
      const auto results = lookup.lookup(word);
      const auto end = std::chrono::high_resolution_clock::now();

      const std::chrono::duration<double, std::milli> elapsed = end - start;
      durations.push_back(elapsed.count());
    }
  }

  if (durations.empty()) {
    return 1;
  }

  const auto [min, max] = std::ranges::minmax_element(durations);
  const double total = std::accumulate(durations.begin(), durations.end(), 0.0);
  const double average = total / durations.size();

  std::cout << std::format("words: {} ({}) iterations: {}\n", csv_path, words.size(), iterations);
  std::cout << std::format("total: {:.2f}ms\n", total);
  std::cout << std::format("avg: {:.2f}ms\n", average);
  std::cout << std::format("min: {:.2f}ms\n", *min);
  std::cout << std::format("max: {:.2f}ms\n", *max);

  return 0;
}