#include <cassert>
#include <map>
#include <vector>
#include <ctime>
#include <string>
#include "tree_sitter/runtime.h"
#include "helpers/load_language.h"
#include "helpers/stderr_logger.h"
#include "helpers/read_test_entries.h"

using std::map;
using std::vector;
using std::string;

vector<string> language_names({
  "c",
  "cpp",
  "javascript",
  "python",
});

int main(int argc, char *arg[]) {
  map<string, vector<ExampleEntry>> example_entries_by_language_name;

  auto document = ts_document_new();

  if (getenv("TREE_SITTER_BENCHMARK_SVG")) {
    ts_document_print_debugging_graphs(document, true);
  } else if (getenv("TREE_SITTER_BENCHMARK_LOG")) {
    ts_document_set_logger(document, stderr_logger_new(false));
  }

  auto language_filter = getenv("TREE_SITTER_BENCHMARK_LANGUAGE");
  auto file_name_filter = getenv("TREE_SITTER_BENCHMARK_FILE_NAME");

  for (auto &language_name : language_names) {
    example_entries_by_language_name[language_name] = examples_for_language(language_name);
  }

  for (auto &language_name : language_names) {
    if (language_filter && language_name != language_filter) continue;

    ts_document_set_language(document, load_real_language(language_name));

    printf("%s\n", language_name.c_str());

    for (auto &example : example_entries_by_language_name[language_name]) {
      if (file_name_filter && example.file_name != file_name_filter) continue;

      ts_document_invalidate(document);
      ts_document_set_input_string(document, example.input.c_str());

      clock_t start_time = clock();
      ts_document_parse(document);
      clock_t end_time = clock();
      unsigned duration = (end_time - start_time) * 1000 / CLOCKS_PER_SEC;
      assert(!ts_node_has_error(ts_document_root_node(document)));
      printf("  %-30s\t%u\n", example.file_name.c_str(), duration);
    }

    for (auto &other_language_name : language_names) {
      if (other_language_name == language_name) continue;

      for (auto &example : example_entries_by_language_name[other_language_name]) {
        if (file_name_filter && example.file_name != file_name_filter) continue;

        ts_document_invalidate(document);
        ts_document_set_input_string(document, example.input.c_str());

        clock_t start_time = clock();
        ts_document_parse(document);
        clock_t end_time = clock();
        unsigned duration = (end_time - start_time) * 1000 / CLOCKS_PER_SEC;
        printf("  %-30s\t%u\n", example.file_name.c_str(), duration);
      }
    }

    puts("");
  }

  return 0;
}