#include "include/utils.h"
#include "include/video.h"
#include "libs/json.h"
#include <getopt.h>
#include <json-c/json.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#define VERSION "1.0"
#define SEARCHLINK "https://www.youtube.com/results?search_query="

void processOptions (int *options, char **args, int argc);

int
main (int argc, char **argv)
{
	int o = 0, opts[] = { 0, 0, 0, 0, 0 };
	char **args = (char **)calloc (3, sizeof (char *));
	for (int i = 0; i < 3; i++)
	{
		args[i] = (char *)calloc (1, sizeof (char *));
	}

	static struct option long_options[]
		= { { "format", required_argument, 0, 'f' },
			{ "help", no_argument, 0, 'h' },
			{ "max", required_argument, 0, 'm' },
			{ "search", required_argument, 0, 's' },
			{ "version", no_argument, 0, 'v' },
			{ 0, 0, 0, 0 } };

	while ((o = getopt_long (argc, argv, "f:hm:s:v", long_options, NULL)) != -1)
	{
		switch (o)
		{
			case 'f':
				opts[0]++;
				args[0] = (char *)realloc (args[0], strlen (optarg) + 1);
				strcpy (args[0], optarg);
				break;
			case 'h':
				opts[1]++;
				break;
			case 'm':
				opts[2]++;
				args[1] = (char *)realloc (args[1], strlen (optarg) + 1);
				strcpy (args[1], optarg);
				break;
			case 's':
				opts[3]++;
				args[2] = (char *)realloc (args[2], strlen (optarg) + 1);
				strcpy (args[2], optarg);
				break;
			case 'v':
				opts[4]++;
		}
	}

	processOptions (opts, args, argc);

	for (int i = 0; i < 3; i++)
	{
		free (args[i]);
	}
	free (args);

	return 0;
}

void
processOptions (int *options, char **args, int argc)
{
	if (!options[1] && options[3] && !options[4])
	{
		char *query = queryNormalizer (args[2]);
		if (query != NULL)
		{
			char *link = (char *)calloc (
					strlen (SEARCHLINK) + strlen (query) + 1, sizeof (char));
			strcpy (link, SEARCHLINK);
			strcat (link, query);

			char *json = extractQueryJSON (link);
			free (link);
			free (query);

			if (json != NULL && strlen (json) > 0)
			{
				json_object *queryRoot = jsonParseString (json);
				free (json);
				if (queryRoot != NULL
						&& json_object_get_type (queryRoot) != json_type_null)
				{
					json_object *handler = navigateToVideos (queryRoot);
					int videoCount = videoCounter (handler);

					if (videoCount > 0)
					{
						Video *videos = generateVideos (handler);
						json_object_put (queryRoot);
						int printedVideos
							= (options[2] && checkNumber (args[1]) > 0)
							? checkNumber (args[1])
							: videoCount;
						char *format = (options[0]) ? args[0] : "";
						printVideo (format, videos, printedVideos);

						for (int i = 0; i < videoCount; i++)
						{
							freeVideo (&videos[i]);
						}

						free (videos);
					}
					else
					{
						json_object_put (queryRoot);
					}
				}
				else
				{
					json_object_put (queryRoot);
				}
			}
			else
			{
				free (json);
			}
		}
	}
	else
	{
		if (options[1] || (options[3] && options[4]) || argc == 1)
		{
			printf (
					"Usage: queryt -s [SEARCH]... [OPTIONS]...\n"
					"API Keyless YouTube querying tool.\n\n"
					"  -f, --format		uses  format  modifiers  to  "
					"customize  the  output\n"
					"		a 	author;\n"
					"		d	duration;\n"
					"		i	video id;\n"
					"		t	title;\n"
					"		R	return;\n"
					"		T	tab;\n"
					"  -h, --help		display this help and exit\n"
					"  -m, --max		defines the max number of displayed "
					"results\n"
					"  -s, --search		search content on youtube\n"
					"  -v, --version		output version information and "
					"exit\n\n"
					"Examples:\n"
					"  queryt -s \"Never Gonna Give You Up\" -m 5 -f iTaTdRt\n\n");
		}
		else
		{
			if (options[4])
			{
				printf ("queryt-%s\n", VERSION);
			}
			else
			{
				puts ("-s is required!");
			}
		}
	}
}