#include "test.h"

#include <stdio.h>
#include <time.h>

struct kit_tests_list kit_tests_list = { 0 };

static void report(int i, char const *file, int line, int ok) {
  int const n = kit_tests_list.tests[i].assertions++;

  kit_tests_list.tests[i].file[n]   = file;
  kit_tests_list.tests[i].line[n]   = line;
  kit_tests_list.tests[i].status[n] = ok;
}

static long long ns_to_ms(long long ns) {
  return (ns + 500000) / 1000000;
}

static long long sec_to_ms(long long sec) {
  return 1000 * sec;
}

enum code_value { white, yellow, red, green };

static void color_code(int term_color, int c) {
  if (term_color) {
    if (c == white)
      printf("\x1b[37m");
    if (c == yellow)
      printf("\x1b[33m");
    if (c == red)
      printf("\x1b[31m");
    if (c == green)
      printf("\x1b[32m");
  }
}

int kit_run_tests(int argc, char **argv) {
  int   fail_test_count       = 0;
  int   fail_assertion_count  = 0;
  int   total_assertion_count = 0;
  int   status                = 0;
  _Bool term_color            = 1;

  for (int i = 0; i < argc; i++)
    if (strcmp("--no-term-color", argv[i]) == 0)
      term_color = 0;

  for (int i = 0; i < kit_tests_list.size; i++) {
    color_code(term_color, yellow);
    printf("[ RUN... ] %s ", kit_tests_list.tests[i].test_name);
    color_code(term_color, white);

    struct timespec begin, end;
    timespec_get(&begin, TIME_UTC);

    kit_tests_list.tests[i].test_fn(i, report);

    timespec_get(&end, TIME_UTC);
    int duration = (int) (ns_to_ms(end.tv_nsec - begin.tv_nsec) +
                          sec_to_ms(end.tv_sec - begin.tv_sec));

    printf("\r");

    _Bool test_status = 1;

    for (int j = 0; j < kit_tests_list.tests[i].assertions; j++)
      if (kit_tests_list.tests[i].status[j] == 0) {
        fail_assertion_count++;
        test_status = 0;
      }

    total_assertion_count += kit_tests_list.tests[i].assertions;

    if (test_status == 0) {
      color_code(term_color, red);
      printf("[ RUN    ] %s\n", kit_tests_list.tests[i].test_name);
      printf("[ FAILED ] %s - %d ms\n",
             kit_tests_list.tests[i].test_name, duration);
      color_code(term_color, white);
      fail_test_count++;
      status = 1;
    } else {
      color_code(term_color, green);
      printf("[ RUN    ] %s\n", kit_tests_list.tests[i].test_name);
      printf("[     OK ] %s - %d ms\n",
             kit_tests_list.tests[i].test_name, duration);
      color_code(term_color, white);
    }
  }

  printf("\n%d of %d tests passed.\n",
         kit_tests_list.size - fail_test_count, kit_tests_list.size);

  printf("%d of %d assertions passed.\n\n",
         total_assertion_count - fail_assertion_count,
         total_assertion_count);

  if (status != 0) {
    for (int i = 0; i < kit_tests_list.size; i++)
      for (int j = 0; j < kit_tests_list.tests[i].assertions; j++)
        if (!kit_tests_list.tests[i].status[j])
          printf("Assertion on line %d in \"%s\" failed\n",
                 kit_tests_list.tests[i].line[j],
                 kit_tests_list.tests[i].file[j]);

    printf("\n");
  }

  return status;
}