#pragma once

#include "multiaddr/multiaddr.h"

int test_new_from_string() {
	struct MultiAddress* a = multiaddress_new_from_string("/ip4/127.0.0.1/tcp/8080/");
	printf("Number of Bytes: %lu, Bytes: ", a->bsize);
	for(int i = 0; i < a->bsize; i++) {
		printf("%02x ", a->bytes[i]);
	}
	printf(" End of bytes\n");
	multiaddress_free(a);
	return 1;
}

int test_full() {
	char addrstr[100];
	strcpy(addrstr,"/ip4/192.168.1.1/");
	printf("INITIAL: %s\n",addrstr);
	struct MultiAddress* a;
	a= multiaddress_new_from_string(addrstr);
	printf("TEST BYTES: %s\n",Var_To_Hex(a->bsize, a->bytes));

	//Remember, Decapsulation happens from right to left, never in reverse!

	printf("A STRING:%s\n",a->string);
	multiaddress_encapsulate(a,"/udp/3333/");
	printf("A STRING ENCAPSULATED:%s\n",a->string);
	printf("TEST BYTES: %s\n",Var_To_Hex(a->bsize, a->bytes));
	multiaddress_decapsulate(a,"udp");
	printf("A STRING DECAPSULATED UDP:%s\n",a->string);
	printf("TEST BYTES: %s\n",Var_To_Hex(a->bsize, a->bytes));
	multiaddress_encapsulate(a,"/udp/3333/");
	printf("A STRING ENCAPSULATED UDP: %s\n",a->string);
	multiaddress_encapsulate(a,"/ipfs/QmYwAPJzv5CZsnA625s3Xf2nemtYgPpHdWEz79ojWnPbdG");
	printf("A STRING ENCAPSULATED IPFS:%s\n",a->string);
	printf("TEST BYTES: %s\n",Var_To_Hex(a->bsize, a->bytes));
	printf("TEST BYTE SIZE: %lu\n",a->bsize);

	struct MultiAddress* beta;
	beta = multiaddress_new_from_bytes(a->bytes,a->bsize);
	printf("B STRING: %s\n",beta->string);

	multiaddress_free(a);
	multiaddress_free(beta);
	return 1;
}

int test_hex_to_var() {
	size_t d;
	unsigned char* result = Hex_To_Var("04", &d);
	if (d != 1)
		return 0;
	if (result[0] != 4)
		return 0;

	if (result != NULL)
		free(result);
	return 1;
}

int test_int_to_hex() {
	int val = 2555351;
	char* result = Int_To_Hex(val);
	int retVal = Hex_To_Int(result);
	if (retVal != val)
		return 0;
	return 1;
}

int test_multiaddr_utils() {
	struct MultiAddress* addr = multiaddress_new_from_string("/ip4/127.0.0.1/tcp/4001");
	if (!multiaddress_is_ip(addr)) {
		fprintf(stderr, "The address should be an IP\n");
		return 0;
	}
	char* ip = NULL;
	multiaddress_get_ip_address(addr, &ip);
	if (ip == NULL) {
		fprintf(stderr, "get_ip_address returned NULL\n");
		return 0;
	}
	if(strcmp(ip, "127.0.0.1") != 0) {
		fprintf(stderr, "ip addresses are not equal\n");
		return 0;
	}
	int port = multiaddress_get_ip_port(addr);
	if (port != 4001) {
		fprintf(stderr, "port incorrect. %d was returned instead of %d\n", port, 4001);
		return 0;
	}
	return 1;
}

int test_multiaddr_peer_id() {
	char* orig_address = "QmKhhKHkjhkjhKjhiuhKJh";
	char full_string[255];
	char* result = NULL;
	int retVal = 0;
	struct MultiAddress* addr;

	sprintf(full_string, "/ip4/127.0.0.1/tcp/4001/ipfs/%s", orig_address);

	addr = multiaddress_new_from_string(full_string);

	result = multiaddress_get_peer_id(addr);

	if (result == NULL || strncmp(result, orig_address, strlen(orig_address)) != 0)
		goto exit;

	retVal = 1;
	exit:
	if (addr != NULL)
		multiaddress_free(addr);
	return retVal;
}