/*
 * main.c for plistutil
 * right now just prints debug shit
 */

#include "../src/plist.h"
#include "plutil.h"
#include <glib.h>
#include <string.h>
#include <stdio.h>
#include <stdlib.h>


int main(int argc, char *argv[])
{
	struct stat *filestats = (struct stat *) malloc(sizeof(struct stat));
	Options *options = parse_arguments(argc, argv);

	if (!options) {
		print_usage();
		return 0;
	}

	iphone_set_debug(options->debug);

	//read input file
	FILE *iplist = fopen(options->in_file, "r");
	if (!iplist)
		return 1;
	stat(options->in_file, filestats);
	char *plist_entire = (char *) malloc(sizeof(char) * (filestats->st_size + 1));
	fread(plist_entire, sizeof(char), filestats->st_size, iplist);
	fclose(iplist);


	//convert one format to another
	plist_t root_node = NULL;
	char *plist_out = NULL;
	int size = 0;

	if (memcmp(plist_entire, "bplist00", 8) == 0) {
		bin_to_plist(plist_entire, filestats->st_size, &root_node);
		plist_to_xml(root_node, &plist_out, &size);
	} else {
		xml_to_plist(plist_entire, filestats->st_size, &root_node);
		plist_to_bin(root_node, &plist_out, &size);
	}

	if (plist_out) {
		if (options->out_file != NULL) {
			FILE *oplist = fopen(options->out_file, "wb");
			if (!oplist)
				return 1;
			fwrite(plist_out, size, sizeof(char), oplist);
			fclose(oplist);
		}
		//if no output file specified, write to stdout
		else
			fwrite(plist_out, size, sizeof(char), stdout);
	} else
		printf("ERROR\n");
	return 0;
}

Options *parse_arguments(int argc, char *argv[])
{
	int i = 0;

	Options *options = (Options *) malloc(sizeof(Options));
	memset(options, 0, sizeof(Options));

	for (i = 1; i < argc; i++) {
		if (!strcmp(argv[i], "--infile") || !strcmp(argv[i], "-i")) {
			if ((i + 1) == argc) {
				free(options);
				return NULL;
			}
			options->in_file = argv[i + 1];
			i++;
			continue;
		}

		if (!strcmp(argv[i], "--outfile") || !strcmp(argv[i], "-o")) {
			if ((i + 1) == argc) {
				free(options);
				return NULL;
			}
			options->out_file = argv[i + 1];
			i++;
			continue;
		}

		if (!strcmp(argv[i], "--debug") || !strcmp(argv[i], "-d") || !strcmp(argv[i], "-v")) {
			options->debug = 1;
		}

		if (!strcmp(argv[i], "--help") || !strcmp(argv[i], "-h")) {
			free(options);
			return NULL;
		}
	}

	if (!options->in_file /*|| !options->out_file */ ) {
		free(options);
		return NULL;
	}

	return options;
}

void print_usage()
{
	printf("Usage: plistutil -i|--infile in_file.plist -o|--outfile out_file.plist [--debug]\n");
	printf("\n");
	printf("\t-i or --infile: The file to read in.\n");
	printf("\t-o or --outfile: The file to convert to.\n");
	printf("\t-d, -v or --debug: Provide extended debug information.\n\n");
}