#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include "c_proto.h"

void print_int_arr(int* arr, int len)
{
	printf("[%d", arr[0]);
	for(int i = 1; i < len; i++)
		printf(",%d", arr[i]);
	printf("]\n\0");
}

int main(int argc, const char** argv)
{
	if(strcmp(argv[1], "strlen") == 0)
	{
		char* arg0 = (char*)argv[2];
		int ret = strlen_(arg0);

		printf("%d\0", ret);
	}
	else if(strcmp(argv[1], "index_of") == 0)
	{
		char* arg0 = (char*)argv[2];
		char arg1 = argv[3][0];
		int arg2 = atoi(argv[4]);
		int ret = index_of(arg0, arg1, arg2);

		printf("%d\0", ret);
	}
	else if(strcmp(argv[1], "to_lowercase") == 0)
	{
		char* arg0 = (char*)argv[2];
		int ret = to_lowercase(arg0);

		printf("%s\n%d\0", arg0, ret);
	}
	else if(strcmp(argv[1], "generate_ciphertext_alphabet") == 0)
	{
		char* arg0 = (char*)argv[2];
		char* arg1 = (char*)argv[3];
		int ret = generate_ciphertext_alphabet(arg0, arg1);

		printf("%d\n%s\0", ret, arg0);
	}
	else if(strcmp(argv[1], "count_lowercase_letters") == 0)
	{
		int arg0[26];
		char* arg1 = (char*)argv[3];
		int ret = count_lowercase_letters(arg0, arg1);

		printf("%d\n", ret);
		print_int_arr(arg0, 26);
	}
	else if(strcmp(argv[1], "sort_alphabet_by_count") == 0)
	{
		char* arg0 = malloc(26);
		int arg1[26] = {21,17,20,25,21,19,28,26,15,16,21,13,11,16,1,27,24,20,5,23,26,2,29,15,21,8};
		sort_alphabet_by_count(arg0, arg1);

		printf("%s\0", arg0);
	}
	else if(strcmp(argv[1], "generate_plaintext_alphabet") == 0)
	{
		char* arg1 = (char*)argv[2];
		char* arg0 = malloc(62);

		generate_plaintext_alphabet(arg0, arg1);

		printf("%s\0", arg0);
	}
	else if(strcmp(argv[1], "encrypt_letter") == 0)
	{
		char arg0 = argv[2][0];
		int arg1 = atoi(argv[3]);
		char* arg2 = (char*)argv[4];
		char* arg3 = (char*)argv[5];
		int ret = encrypt_letter(arg0, arg1, arg2, arg3);

		if(ret >= 0)
			printf("%c\n\0", ret);
		else
			printf("%d\n\0", ret);
	}
	else if(strcmp(argv[1], "is_alnum") == 0)
	{
		char arg0 = argv[2][0];
		printf("%d\0", is_alnum(arg0));
	}
	else if(strcmp(argv[1], "encrypt") == 0)
	{
		char* plaintext = malloc(68);
		strcpy(plaintext, "Never trust a computer you can't throw out a window. -Steve Wozniak\0");
		char* keyphrase = malloc(82);
		strcpy(keyphrase, "I'll have you know that I stubbed my toe last week and only cried for 20 minutes.\0");
		char* corpus = malloc(407);
		strcpy(corpus, "When in the Course of human events, it becomes necessary for one people to dissolve the political bands which have connected them with another, and to assume among the powers of the earth, the separate and equal station to which the Laws of Nature and of Nature's God entitle them, a decent respect to the opinions of mankind requires that they should declare the causes which impel them to the separation.\0");
		char* ciphertext = malloc(69);
		int* result = encrypt(ciphertext, plaintext, keyphrase, corpus);
		printf("[%d,%d]\n", result[0], result[1]);
		printf("%s\n\n", ciphertext);

		plaintext = malloc(141);
		strcpy(plaintext, "The trouble with having an open mind, of course, is that people will insist on coming along and trying to put things in it. -Terry Pratchett");
		keyphrase = malloc(83);
		strcpy(keyphrase, "What's the difference between ignorance and apathy? I don't know and I don't care.");
		corpus = malloc(306);
		strcpy(corpus, "Call me Ishmael. Some years ago - never mind how long precisely - having little or no money in my purse, and nothing particular to interest me on shore, I thought I would sail about a little and see the watery part of the world. It is a way I have of driving off the spleen and regulating the circulation.");
		ciphertext = malloc(141);
		result = encrypt(ciphertext, plaintext, keyphrase, corpus);
		printf("[%d,%d]\n", result[0], result[1]);
		printf("%s\n\n", ciphertext);

		plaintext = malloc(129);
		strcpy(plaintext, "If debugging is the process of removing software bugs, then programming must be the process of putting them in. -Edsger Dijkstra");
		keyphrase = malloc(58);
		strcpy(keyphrase, "What is the sum of 12 and 37? The answer, CLEARLY, is 49!");
		corpus = malloc(614);
		strcpy(corpus,  "It was the best of times, it was the worst of times, it was the age of wisdom, it was the age of foolishness, it was the epoch of belief, it was the epoch of incredulity, it was the season of Light, it was the season of Darkness, it was the spring of hope, it was the winter of despair, we had everything before us, we had nothing before us, we were all going direct to Heaven, we were all going direct the other way - in short, the period was so far like the present period, that some of its noisiest authorities insisted on its being received, for good or for evil, in the superlative degree of comparison only.");
		ciphertext = malloc(129);
		result = encrypt(ciphertext, plaintext, keyphrase, corpus);
		printf("[%d,%d]\n", result[0], result[1]);
		printf("%s\n", ciphertext);
	}
	else if(strcmp(argv[1], "decrypt") == 0)
	{
		char* plaintext = malloc(68);
		char* ciphertext = malloc(68);
		strcpy(ciphertext, "Dk5wQ 1Q4RW h yLCO4XnQ 8H4 yaE'3 VpQH6 L4V a 6qGoJ6. -R3n5t 6J9GqIA");
		char* keyphrase = malloc(82);
		keyphrase = "I'll have you know that I stubbed my toe last week and only cried for 20 minutes.";
		char* corpus = malloc(407);
		strcpy(corpus, "When in the Course of human events, it becomes necessary for one people to dissolve the political bands which have connected them with another, and to assume among the powers of the earth, the separate and equal station to which the Laws of Nature and of Nature's God entitle them, a decent respect to the opinions of mankind requires that they should declare the causes which impel them to the separation.");
		int* results = decrypt(plaintext, ciphertext, keyphrase, corpus);
		print_int_arr(results, 2);
		printf("\n%s\n", plaintext);

		plaintext = malloc(141);
		ciphertext = malloc(141);
		strcpy(ciphertext, "Xjc 1UN4dDn 6xXj jW5vJk WJ ORcK GmKf, PI iO4TVn, mV 0jW3 RwQREy 6mDE xKVlV1 NJ iMGqLk aDPJk aJf 2U8uHk 2Q R40 2jlJkV xK lZ. -3rUT8 RThYijc23");
		keyphrase = malloc(83);
		keyphrase = "What's the difference between ignorance and apathy? I don't know and I don't care.";
		corpus = malloc(306);
		strcpy(corpus, "Call me Ishmael. Some years ago - never mind how long precisely - having little or no money in my purse, and nothing particular to interest me on shore, I thought I would sail about a little and see the watery part of the world. It is a way I have of driving off the spleen and regulating the circulation.");
		results = decrypt(plaintext, ciphertext, keyphrase, corpus);
		print_int_arr(results, 2);
		printf("\n%s\n", plaintext);

		plaintext = malloc(129);
		ciphertext = malloc(129);
		strcpy(ciphertext, "L7 e1iXTTAkT 9K MCu zFxsnGH x7 F2jxZEkT Gy7P0hDo iXTI, VCfk zFqTDhjj4kT jXKP in Mwm zFqsdKG q7 zXNOEkT VCnj Rk. -2eGTuD eAbcKSDt");
		keyphrase = malloc(58);
		keyphrase = "What is the sum of 12 and 37? The answer, CLEARLY, is 49!";
		corpus = malloc(614);
		strcpy(corpus, "It was the best of times, it was the worst of times, it was the age of wisdom, it was the age of foolishness, it was the epoch of belief, it was the epoch of incredulity, it was the season of Light, it was the season of Darkness, it was the spring of hope, it was the winter of despair, we had everything before us, we had nothing before us, we were all going direct to Heaven, we were all going direct the other way - in short, the period was so far like the present period, that some of its noisiest authorities insisted on its being received, for good or for evil, in the superlative degree of comparison only.");
		results = decrypt(plaintext, ciphertext, keyphrase, corpus);
		print_int_arr(results, 2);
		printf("\n%s\n", plaintext);
	}
	else
	{
		printf("[C PROTO TESTER] Invalid command\n\0");
	}

	return 0;
}