#include "webfused/daemon.h"

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <stddef.h>
#include <stdbool.h>
#include <signal.h>

#include <unistd.h>
#include <getopt.h>

#include <webfuse_adapter.h>
#include <userdb/userdb.h>

#define SERVICE_TIMEOUT (1 * 1000)

struct args
{
	struct wf_server_config * config;
	char * passwd_path;
	bool show_help;
};

static bool shutdown_requested = false;

static void show_help(void)
{
	printf(
		"webfused, Copyright (c) 2019, webfused authors <https://github.com/falk-werner/webfused>\n"
		"Websocket file system daemon\n"
		"\n"
		"Usage: webfused [m <mount_point>] [-d <document_root] [-n <vhost_name>] [-p <port>]\n"
		"            [-c <server_cert_path>] [-k <server_key_path>] [-P <passwd_path>]\n"
		"\n"
		"Options:\n"
		"\t-m, --mount_point      Path of mount point (required)\n"
		"\t-d, --document_root    Path of www directory (default: not set, www disabled)\n"
		"\t-c, --server_cert_path Path of servers own certificate (default: not set, TLS disabled)\n"
		"\t-k, --server_key_path  Path of servers private key (default: not set, TLS disabled)\n"
		"\t-n, --vhost_name       Name of virtual host (default: \"localhost\")\n"
		"\t-p, --port             Number of servers port (default: 8080)\n"
		"\t-P, --passwd_path      Path to password file (default: not set, authentication disabled)\n"
		"\n");
}

static bool authenticate(struct wf_credentials * creds, void * user_data)
{
	bool result = false;
	struct args * args = user_data;

	char const * username = wf_credentials_get(creds, "username");
	char const * password = wf_credentials_get(creds, "password");
	if ((NULL != username) && (NULL != password))
	{
		struct userdb * db = userdb_create("");
		result = userdb_load(db, args->passwd_path); 
		if (result)
		{
			result = userdb_check(db, username, password);
		}

		userdb_dispose(db);
	}

	return result;
}

static int parse_arguments(int argc, char * argv[], struct args * args)
{
	static struct option const options[] =
	{
		{"mount_point", required_argument, NULL, 'm'},
		{"document_root", required_argument, NULL, 'd'},
		{"server_cert_path", required_argument, NULL, 'c'},
		{"server_key_path", required_argument, NULL, 'k'},
		{"vhost_name", required_argument, NULL, 'n'},
		{"port", required_argument, NULL, 'p'},
		{"passwd_path", required_argument, NULL, 'P'},
		{"help", no_argument, NULL, 'h'},
		{NULL, 0, NULL, 0}
	};

	bool result = EXIT_SUCCESS;
	bool finished = false;
	bool has_mountpoint = false;
	while ((!finished) && (EXIT_SUCCESS == result))
	{
		int option_index = 0;
		int const c = getopt_long(argc, argv, "m:d:c:k:n:p:P:h", options, &option_index);

		switch (c)
		{
			case -1:
				finished = true;
				break;
			case 'h':
				args->show_help = true;
				finished = true;
				break;
			case 'm':
				wf_server_config_set_mountpoint(args->config, optarg);
				has_mountpoint = true;
				break;
			case 'd':
				wf_server_config_set_documentroot(args->config, optarg);
				break;
			case 'c':
				wf_server_config_set_certpath(args->config, optarg);
				break;
			case 'k':
				wf_server_config_set_keypath(args->config, optarg);
				break;
			case 'n':
				wf_server_config_set_vhostname(args->config, optarg);
				break;
			case 'p':
				wf_server_config_set_port(args->config, atoi(optarg));
				break;
			case 'P':
				free(args->passwd_path);
				args->passwd_path = strdup(optarg);
				wf_server_config_add_authenticator(args->config, 
					"username",
					&authenticate,
					args);
				break;
			default:
				fprintf(stderr, "error: unknown argument\n");
				result = EXIT_FAILURE;
				break;
		}
	}

	if ((EXIT_SUCCESS == result) && (!args->show_help))
	{
		if (!has_mountpoint)
		{
			fprintf(stderr, "error: missing mount point\n");
			result = EXIT_FAILURE;
		}
	}

	if (EXIT_SUCCESS != result)
	{
		args->show_help = true;
	}

	return result;
}

static void on_interrupt(int signal_id)
{
	(void) signal_id;

	shutdown_requested = true;
}

int wfd_daemon_run(int argc, char * argv[])
{
	struct args args;
	args.config = wf_server_config_create();
	wf_server_config_set_vhostname(args.config, "localhost");
	wf_server_config_set_port(args.config, 8080);
	args.passwd_path = NULL;
	args.show_help = false;

	int result = parse_arguments(argc, argv, &args);
	
	if (!args.show_help)
	{
		signal(SIGINT, on_interrupt);
		struct wf_server * server = wf_server_create(args.config);
		if (NULL != server)
		{
			while (!shutdown_requested)
			{
				wf_server_service(server, SERVICE_TIMEOUT);
			}

			wf_server_dispose(server);			
		}
		else
		{
			fprintf(stderr, "fatal: unable start server\n");
			result = EXIT_FAILURE;
 		}
	}
	else
	{
		show_help();
	}

	free(args.passwd_path);
	wf_server_config_dispose(args.config);
	return result;
}