#include <stdio.h>
#include <stdarg.h>

static int passed;
static int failed;

typedef enum {
	DEFAULT = 39,
	RED = 31,
	GREEN = 32,
	YELLOW = 33,
	BLUE = 34,
	MAGENTA = 35
} Color;

static void set_output_color(Color c)
{
	printf("\033[%dm", c);
}

static void print_result(int success, const char *format, ...)
{
	va_list ap;
	Color color = GREEN;
	char indicator = '+';

	if (success) {
		set_output_color(GREEN);
		printf("+ ");
		passed++;
	} else {
		set_output_color(RED);
		printf("- ");
		failed++;
	}

	set_output_color(color);
	printf("%c ", indicator);

	va_start(ap, format);
	vprintf(format, ap);
	va_end(ap);

	printf("%c", success ? '\r' : '\n');
	set_output_color(DEFAULT);
}

void testing_header(const char *header)
{
	printf("Testing header ");
	set_output_color(MAGENTA);
	printf("<%s>", header);
	set_output_color(DEFAULT);
	printf("\n");
}

void testing_end(void)
{
	printf("%d tests passed, %d tests failed\n", passed, failed);

	passed = 0;
	failed = 0;
}

void testing_comment(const char *comment)
{
	printf("- %s\n", comment);
}

void test_int_equals_imp(const char *expression, int result, int expected)
{
	print_result(result == expected, "%s == %d", expression, expected);
}

void test_void_imp(const char *expression)
{
	printf("? %s\n", expression);
}

void test_bool_imp(const char *expression, int result, int expected)
{
	int success = (result && expected) || (!result && !expected);
	print_result(success, "%s%s", expected ? "" : "!", expression);
}

void test_string_imp(const char *expression, const char *totest, const char *tocompare)
{
	int success = 1;
	int i;
	for (i = 0; totest[i] != '\0'; i++) {
		if (totest[i] != tocompare[i]) {
			success = 0;
		}
	}
	if (tocompare[i] != '\0') {
		success = 0;
	}
	print_result(success, "%s == \"%s\"", expression, tocompare);
}