/**

 * libtransport -- C++ library for easy XMPP Transports development

 *

 * Copyright (C) 2011, Jan Kaluza <hanzz.k@gmail.com>

 *

 * This program 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; either version 2 of the License, or

 * (at your option) any later version.

 *

 * 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  02111-1301  USA

 */

#include "transport/config.h"

#include <fstream>

#ifdef _MSC_VER

#include <direct.h>

#define getcwd _getcwd

#include <windows.h>

#define PATH_MAX MAX_PATH

#endif

using namespace boost::program_options;

namespace Transport {

	unsigned long r = 0;

	BOOST_FOREACH(char c, s) {

		r += (int) c;

	}

	srand(time(NULL) + r);

	return 30000 + rand() % 10000;

}

bool Config::load(const std::string &configfile, boost::program_options::options_description &opts, const std::string &jid) {

	std::ifstream ifs(configfile.c_str());

	if (!ifs.is_open())

		return false;

	m_file = configfile;

	bool ret = load(ifs, opts, jid);

	ifs.close();

	char path[PATH_MAX] = "";

	if (m_file.find_first_of("/") != 0) {

		getcwd(path, PATH_MAX);

		m_file = std::string(path) + "/" + m_file;

	}

	return ret;

}

bool Config::load(std::istream &ifs, boost::program_options::options_description &opts, const std::string &_jid) {

	m_unregistered.clear();

	opts.add_options()

		("service.jid", value<std::string>()->default_value(""), "Transport Jabber ID")

		("service.server", value<std::string>()->default_value(""), "Server to connect to")

		("service.password", value<std::string>()->default_value(""), "Password used to auth the server")

		("service.port", value<int>()->default_value(0), "Port the server is listening on")

		("service.user", value<std::string>()->default_value(""), "The name of user Spectrum runs as.")

		("service.group", value<std::string>()->default_value(""), "The name of group Spectrum runs as.")

		("service.backend", value<std::string>()->default_value("libpurple_backend"), "Backend")

		("service.protocol", value<std::string>()->default_value(""), "Protocol")

		("service.pidfile", value<std::string>()->default_value("/var/run/spectrum2/$jid.pid"), "Full path to pid file")

		("service.working_dir", value<std::string>()->default_value("/var/lib/spectrum2/$jid"), "Working dir")

		("service.allowed_servers", value<std::vector<std::string> >()->multitoken(), "Only users from these servers can connect")

		("service.server_mode", value<bool>()->default_value(false), "True if Spectrum should behave as server")

		("service.users_per_backend", value<int>()->default_value(100), "Number of users per one legacy network backend")

		("service.backend_host", value<std::string>()->default_value("localhost"), "Host to bind backend server to")

		("service.backend_port", value<std::string>()->default_value("0"), "Port to bind backend server to")

		("service.cert", value<std::string>()->default_value(""), "PKCS#12 Certificate.")

		("service.cert_password", value<std::string>()->default_value(""), "PKCS#12 Certificate password.")

		("service.admin_jid", value<std::vector<std::string> >()->multitoken(), "Administrator jid.")

		("service.admin_password", value<std::string>()->default_value(""), "Administrator password.")

		std::vector<std::string> value;

		value.push_back("/var/lib/spectrum2/$jid");

		parsed.options.push_back(boost::program_options::basic_option<char>("service.working_dir", value));

	}

	if (!found_pidfile) {

		std::vector<std::string> value;

		value.push_back("/var/run/spectrum2/$jid.pid");

		parsed.options.push_back(boost::program_options::basic_option<char>("service.pidfile", value));

	}

	if (!found_backend_port) {

		std::vector<std::string> value;

		std::string p = boost::lexical_cast<std::string>(getRandomPort(_jid.empty() ? jid : _jid));

		value.push_back(p);

		parsed.options.push_back(boost::program_options::basic_option<char>("service.backend_port", value));

	}

	if (!found_database) {

		std::vector<std::string> value;

		value.push_back("/var/lib/spectrum2/$jid/database.sql");

		parsed.options.push_back(boost::program_options::basic_option<char>("database.database", value));

	}

	BOOST_FOREACH(option &opt, parsed.options) {

		if (opt.unregistered) {

			m_unregistered[opt.string_key] = opt.value[0];

		}

		else if (opt.value[0].find("$jid") != std::string::npos) {

			boost::replace_all(opt.value[0], "$jid", jid);

		}

	}

	store(parsed, m_variables);

	notify(m_variables);

	onConfigReloaded();

	return true;

}

bool Config::load(std::istream &ifs) {

	options_description opts("Transport options");

	return load(ifs, opts);

}

bool Config::load(const std::string &configfile, const std::string &jid) {

	try {

		options_description opts("Transport options");

		return load(configfile, opts, jid);

	} catch ( const boost::program_options::multiple_occurrences& e ) {

		std::cerr << configfile << " parsing error: " << e.what() << " from option: " << e.get_option_name() << std::endl;

		return false;

	}

}

bool Config::reload() {

	if (m_file.empty()) {

		return false;

	}

	return load(m_file);

}

}