/*
 * This file is distributed as part of the SkySQL Gateway.  It is free
 * software: you can redistribute it and/or modify it under the terms of the
 * GNU General Public License as published by the Free Software Foundation,
 * version 2.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
 * details.
 *
 * You should have received a copy of the GNU General Public License along with
 * this program; if not, write to the Free Software Foundation, Inc., 51
 * Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Copyright SkySQL Ab 2013
 */

/**
 * @file config.c  - Read the gateway.cnf configuration file
 *
 * @verbatim
 * Revision History
 *
 * Date		Who		Description
 * 21/06/13	Mark Riddoch	Initial implementation
 * 08/07/13	Mark Riddoch	Addition on monitor module support
 * 23/07/13	Mark Riddoch	Addition on default monitor password
 *
 * @endverbatim
 */
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <ini.h>
#include <config.h>
#include <service.h>
#include <server.h>
#include <users.h>
#include <monitor.h>
#include <skygw_utils.h>
#include <log_manager.h>

static	int	process_config_context(CONFIG_CONTEXT	*);
static	int	process_config_update(CONFIG_CONTEXT *);
static	void	free_config_context(CONFIG_CONTEXT	*);
static	char 	*config_get_value(CONFIG_PARAMETER *, const char *);
static	int	handle_global_item(const char *, const char *);
static	void	global_defaults();

static	char		*config_file = NULL;
static	GATEWAY_CONF	gateway;

/**
 * Config item handler for the ini file reader
 *
 * @param userdata	The config context element
 * @param section	The config file section
 * @param name		The Parameter name
 * @param value		The Parameter value
 * @return zero on error
 */
static int
handler(void *userdata, const char *section, const char *name, const char *value)
{
CONFIG_CONTEXT		*cntxt = (CONFIG_CONTEXT *)userdata;
CONFIG_CONTEXT		*ptr = cntxt;
CONFIG_PARAMETER	*param;

	if (strcmp(section, "gateway") == 0 || strcasecmp(section, "MaxScale") == 0)
	{
		return handle_global_item(name, value);
	}
	/*
	 * If we already have some parameters for the object
	 * add the parameters to that object. If not create
	 * a new object.
	 */
	while (ptr && strcmp(ptr->object, section) != 0)
		ptr = ptr->next;
	if (!ptr)
	{
		if ((ptr = (CONFIG_CONTEXT *)malloc(sizeof(CONFIG_CONTEXT))) == NULL)
			return 0;
		ptr->object = strdup(section);
		ptr->parameters = NULL;
		ptr->next = cntxt->next;
		cntxt->next = ptr;
	}
	if ((param = (CONFIG_PARAMETER *)malloc(sizeof(CONFIG_PARAMETER))) == NULL)
		return 0;
	param->name = strdup(name);
	param->value = strdup(value);
	param->next = ptr->parameters;
	ptr->parameters = param;

	return 1;
}

/**
 * Load the configuration file for the gateway
 *
 * @param file	The filename of the configuration file
 */
int
config_load(char *file)
{
CONFIG_CONTEXT	config;
int		rval;

	global_defaults();

	config.object = "";
	config.next = NULL;

	if (ini_parse(file, handler, &config) < 0)
		return 0;

	config_file = file;

	rval = process_config_context(config.next);
	free_config_context(config.next);

	return rval;
}

/**
 * Relooad the configuration file for the gateway
 *
 */
int
config_reload()
{
CONFIG_CONTEXT	config;
int		rval;

	if (!config_file)
		return 0;
	global_defaults();

	config.object = "";
	config.next = NULL;

	if (ini_parse(config_file, handler, &config) < 0)
		return 0;

	rval = process_config_update(config.next);
	free_config_context(config.next);

	return rval;
}

/**
 * Process a configuration context and turn it into the set of object
 * we need.
 *
 * @param context	The configuration data
 */
static	int
process_config_context(CONFIG_CONTEXT *context)
{
CONFIG_CONTEXT		*obj;

	/**
	 * Process the data and create the services and servers defined
	 * in the data.
	 */
	obj = context;
	while (obj)
	{
		char *type = config_get_value(obj->parameters, "type");
		if (type == NULL)
			skygw_log_write( LOGFILE_ERROR, "Object %s has no type\n", obj->object);
		else if (!strcmp(type, "service"))
		{
			char *router = config_get_value(obj->parameters, "router");
			if (router)
			{
				obj->element = service_alloc(obj->object, router);
				char *user = config_get_value(obj->parameters, "user");
				char *auth = config_get_value(obj->parameters, "auth");
				if (obj->element && user && auth)
					serviceSetUser(obj->element, user, auth);
			}
			else
				skygw_log_write( LOGFILE_ERROR, "No router define for service '%s'\n",
							obj->object);
		}
		else if (!strcmp(type, "server"))
		{
			char *address = config_get_value(obj->parameters, "address");
			char *port = config_get_value(obj->parameters, "port");
			char *protocol = config_get_value(obj->parameters, "protocol");
			char *monuser = config_get_value(obj->parameters, "monitoruser");
			char *monpw = config_get_value(obj->parameters, "monitorpw");
			if (address && port && protocol)
				obj->element = server_alloc(address, protocol, atoi(port));
			if (obj->element && monuser && monpw)
				serverAddMonUser(obj->element, monuser, monpw);
		}

		obj = obj->next;
	}

	/*
	 * Now we have the services we can add the servers to the services
	 * add the protocols to the services
	 */
	obj = context;
	while (obj)
	{
		char *type = config_get_value(obj->parameters, "type");
		if (type == NULL)
			;
		else if (!strcmp(type, "service"))
		{
			char *servers = config_get_value(obj->parameters, "servers");
			char *roptions = config_get_value(obj->parameters, "router_options");
			if (servers && obj->element)
			{
				char *s = strtok(servers, ",");
				while (s)
				{
					CONFIG_CONTEXT *obj1 = context;
					while (obj1)
					{
						if (strcmp(s, obj1->object) == 0 && obj->element && obj1->element)
							serviceAddBackend(obj->element, obj1->element);
						obj1 = obj1->next;
					}
					s = strtok(NULL, ",");
				}
			}
			if (roptions && obj->element)
			{
				char *s = strtok(roptions, ",");
				while (s)
				{
					serviceAddRouterOption(obj->element, s);
					s = strtok(NULL, ",");
				}
			}
		}
		else if (!strcmp(type, "listener"))
		{
			char *service = config_get_value(obj->parameters, "service");
			char *port = config_get_value(obj->parameters, "port");
			char *protocol = config_get_value(obj->parameters, "protocol");
			if (service && port && protocol)
			{
				CONFIG_CONTEXT *ptr = context;
				while (ptr && strcmp(ptr->object, service) != 0)
					ptr = ptr->next;
				if (ptr && ptr->element)
					serviceAddProtocol(ptr->element, protocol, atoi(port));
			}
		}
		else if (!strcmp(type, "monitor"))
		{
			char *module = config_get_value(obj->parameters, "module");
			char *servers = config_get_value(obj->parameters, "servers");
			char *user = config_get_value(obj->parameters, "user");
			char *passwd = config_get_value(obj->parameters, "passwd");
			if (module)
			{
				obj->element = monitor_alloc(obj->object, module);
				if (servers && obj->element)
				{
					char *s = strtok(servers, ",");
					while (s)
					{
						CONFIG_CONTEXT *obj1 = context;
						while (obj1)
						{
							if (strcmp(s, obj1->object) == 0 && obj->element && obj1->element)
								monitorAddServer(obj->element, obj1->element);
							obj1 = obj1->next;
						}
						s = strtok(NULL, ",");
					}
				}
				if (obj->element && user && passwd)
				{
					monitorAddUser(obj->element, user, passwd);
				}
			}
		}

		obj = obj->next;
	}

	return 1;
}

/**
 * Get the value of a config parameter
 *
 * @param params	The linked list of config parameters
 * @param name		The parameter to return
 * @return the parameter value or NULL if not found
 */
static char *
config_get_value(CONFIG_PARAMETER *params, const char *name)
{
	while (params)
	{
		if (!strcmp(params->name, name))
			return params->value;
		params = params->next;
	}
	return NULL;
}

/**
 * Free a config tree
 *
 * @param context	The configuration data
 */
static	void
free_config_context(CONFIG_CONTEXT *context)
{
CONFIG_CONTEXT		*obj;
CONFIG_PARAMETER	*p1, *p2;

	while (context)
	{
		free(context->object);
		p1 = context->parameters;
		while (p1)
		{
			free(p1->name);
			free(p1->value);
			p2 = p1->next;
			free(p1);
			p1 = p2;
		}
		obj = context->next;
		free(context);
		context = obj;
	}
}

/**
 * Return the number of configured threads
 *
 * @return The number of threads configured in the config file
 */
int
config_threadcount()
{
	return gateway.n_threads;
}

/**
 * Configuration handler for items in the global [gateway] section
 *
 * @param name	The item name
 * @param value	The item value
 * @return 0 on error
 */
static	int
handle_global_item(const char *name, const char *value)
{
	if (strcmp(name, "threads") == 0) {
		gateway.n_threads = atoi(value);
        } else {
                return 0;
        }
	return 1;
}

/**
 * Set the defaults for the global configuration options
 */
static void
global_defaults()
{
	gateway.n_threads = 1;
}

/**
 * Process a configuration context update and turn it into the set of object
 * we need.
 *
 * @param context	The configuration data
 */
static	int
process_config_update(CONFIG_CONTEXT *context)
{
CONFIG_CONTEXT		*obj;
SERVICE			*service;
SERVER			*server;

	/**
	 * Process the data and create the services and servers defined
	 * in the data.
	 */
	obj = context;
	while (obj)
	{
		char *type = config_get_value(obj->parameters, "type");
		if (type == NULL)
			skygw_log_write( LOGFILE_ERROR, "Object %s has no type\n", obj->object);
		else if (!strcmp(type, "service"))
		{
			char *router = config_get_value(obj->parameters, "router");
			if (router)
			{
				if ((service = service_find(obj->object)) != NULL)
				{
					char *user = config_get_value(obj->parameters, "user");
					char *auth = config_get_value(obj->parameters, "auth");
					if (user && auth)
						service_update(service, router, user, auth);
					obj->element = service;
				}
				else
				{
					obj->element = service_alloc(obj->object, router);
					char *user = config_get_value(obj->parameters, "user");
					char *auth = config_get_value(obj->parameters, "auth");
					if (obj->element && user && auth)
						serviceSetUser(obj->element, user, auth);
				}
			}
			else
				skygw_log_write( LOGFILE_ERROR, "No router defined for service '%s'\n",
							obj->object);
		}
		else if (!strcmp(type, "server"))
		{
			char *address = config_get_value(obj->parameters, "address");
			char *port = config_get_value(obj->parameters, "port");
			char *protocol = config_get_value(obj->parameters, "protocol");
			char *monuser = config_get_value(obj->parameters, "monitoruser");
			char *monpw = config_get_value(obj->parameters, "monitorpw");
			if (address && port && protocol)
			{
				if ((server = server_find(address, atoi(port))) != NULL)
				{
					server_update(server, protocol, monuser, monpw);
					obj->element = server;
				}
				else
				{
					obj->element = server_alloc(address, protocol, atoi(port));
					if (obj->element && monuser && monpw)
						serverAddMonUser(obj->element, monuser, monpw);
				}
			}
		}

		obj = obj->next;
	}

	/*
	 * Now we have the services we can add the servers to the services
	 * add the protocols to the services
	 */
	obj = context;
	while (obj)
	{
		char *type = config_get_value(obj->parameters, "type");
		if (type == NULL)
			;
		else if (!strcmp(type, "service"))
		{
			char *servers = config_get_value(obj->parameters, "servers");
			char *roptions = config_get_value(obj->parameters, "router_options");
			if (servers && obj->element)
			{
				char *s = strtok(servers, ",");
				while (s)
				{
					CONFIG_CONTEXT *obj1 = context;
					while (obj1)
					{
						if (strcmp(s, obj1->object) == 0 && obj->element && obj1->element)
							if (!serviceHasBackend(obj->element, obj1->element))
								serviceAddBackend(obj->element, obj1->element);
						obj1 = obj1->next;
					}
					s = strtok(NULL, ",");
				}
			}
			if (roptions && obj->element)
			{
				char *s = strtok(roptions, ",");
				serviceClearRouterOptions(obj->element);
				while (s)
				{
					serviceAddRouterOption(obj->element, s);
					s = strtok(NULL, ",");
				}
			}
		}
		else if (!strcmp(type, "listener"))
		{
			char *service = config_get_value(obj->parameters, "service");
			char *port = config_get_value(obj->parameters, "port");
			char *protocol = config_get_value(obj->parameters, "protocol");
			if (service && port && protocol)
			{
				CONFIG_CONTEXT *ptr = context;
				while (ptr && strcmp(ptr->object, service) != 0)
					ptr = ptr->next;
				if (ptr && ptr->element && serviceHasProtocol(ptr->element, protocol, atoi(port)) == 0)
				{
					serviceAddProtocol(ptr->element, protocol, atoi(port));
					serviceStartProtocol(ptr->element, protocol, atoi(port));
				}
			}
		}

		obj = obj->next;
	}

	return 1;
}