#include "mapreduce.h"

int main(int argc, char *argv[]) {
	
	if(argc < 4) {
		printf("Less number of arguments.\n");
		printf("./mapreduce #mappers #reducers inputFile\n");
		exit(0);
	}

	// ###### DO NOT REMOVE ######
	int nMappers 	= strtol(argv[1], NULL, 10);
	int nReducers 	= strtol(argv[2], NULL, 10);
	char *inputFile = argv[3];

	// ###### DO NOT REMOVE ######
	bookeepingCode();

	// ###### DO NOT REMOVE ######
	pid_t pid = fork();
	if(pid == 0){
		//send chunks of data to the mappers in RR fashion
		sendChunkData(inputFile, nMappers);
		exit(0);
	}
	sleep(1);


	pid_t mapperPid;
	for (int i = 1; i <= nMappers; i++) {
		mapperPid = fork();
		if (mapperPid == 0) {
			//TODO: exec here
			char numMap[10];
			sprintf(numMap, "%d", i);
			execl("./mapper", "./mapper", &numMap, NULL);
			exit(0);
		}
	}
	
    for (int i = 0; i <= nMappers; i++) {
		//Hopefully this works
		wait(NULL);
	}


	// ###### DO NOT REMOVE ######
    // shuffle sends the word.txt files generated by mapper 
    // to reducer based on a hash function
	// pid = fork();
	// if(pid == 0){
	// 	shuffle(nMappers, nReducers);
	// 	exit(0);
	// }
	// sleep(1);



	// pid_t reducerPid;
	// for (int i = 1; i <= nReducers; i++) {
	// 	reducerPid = fork();
	// 	if (reducerPid == 0) {
	// 		char numRed[10];
	// 		sprintf(numRed, "%d", i);
	// 		execl("./reducer", "./reducer", &numRed, NULL);
	// 	}
	// }

	// for (int i = 1; i <= nReducers; i++) {
	// 	//Maybe this works too?
	// 	wait(NULL);
	// }

	return 0;
}