#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <time.h>
#include "dns.h"
#include "servers.h"
#include "slist.h"

#define NUM_TESTS 10

struct timespec test_server(char *dns_addr, struct hosts_list *hosts);

int main(int argc, char** argv)
{
	int option, type = T_A;
	char* server = "1.1.1.1", *hostname = "google.com";
	struct hosts_list *servers_hosts = NULL;
	struct dns_list *servers_dns = NULL;
	while((option = getopt(argc, argv, "s:h:t:")) != -1) {
		switch (option) {
		case 's': //server to use
			add_dns_server(&servers_dns, optarg);
			break;
		case 'h': //hostname to search
			add_hosts_server(&servers_hosts, optarg);
			break;
		case 't':
			if (!strcmp(optarg, "A")) {
				type = T_A;
			} else if (!strcmp(optarg, "AAAA")) {
				type = T_AAAA;
			} else if (!strcmp(optarg, "NS")) {
				type = T_NS;
			} else if (!strcmp(optarg, "CNAME")) {
				type = T_CNAME;
			} else if (!strcmp(optarg, "SOA")) {
				type = T_SOA;
			} else if (!strcmp(optarg, "PTR")) {
				type = T_PTR;
			} else if (!strcmp(optarg, "MX")) {
				type = T_MX;
			} else if (!strcmp(optarg, "TXT")) {
				type = T_TXT;
			} else {
				printf("Error: %s is not a valid DNS record type\n", optarg);
				free_hosts_list(&servers_hosts);
				free_dns_list(&servers_dns);
				exit(1);
			}
			break;
		case '?':
		default:
			printf("Error: invalid option -%c\n", optopt);
			exit(1);
		}
	}
	struct timespec total;
	total = test_server(server, servers_hosts);
	printf("Test took %ld.%09lds\n", total.tv_sec, total.tv_nsec);
	free_dns_list(&servers_dns);
	free_hosts_list(&servers_hosts);
	return 0;
}

struct timespec test_server(char* dns_addr, struct hosts_list *hosts)
{
	struct timespec total;
	total.tv_sec = 0; total.tv_nsec = 0;
	for (int i = 0; i < NUM_TESTS; i++) {
		struct hosts_list *curr = hosts;
		while (curr) {
			struct timespec run;
			unsigned char buf[65536];
			run = resolve(buf, curr->server, dns_addr, T_A);
			total.tv_sec += run.tv_sec;
			total.tv_nsec += run.tv_nsec;
			if (total.tv_nsec >= 1000000000) {
				total.tv_sec += 1;
				total.tv_nsec -= 1000000000;
			}
			curr = curr->next;
		}
	}
	return total;
}