#define getcwd _getcwd

#include <windows.h>

#define PATH_MAX MAX_PATH

#endif

using namespace boost::program_options;

namespace Transport {

int getRandomPort(const std::string &s) {

	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.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.")

		("service.reuse_old_backends", value<bool>()->default_value(true), "True if Spectrum should use old backends which were full in the past.")

		("service.idle_reconnect_time", value<int>()->default_value(0), "Time in seconds after which idle users are reconnected to let their backend die.")

		("service.memory_collector_time", value<int>()->default_value(0), "Time in seconds after which backend with most memory is set to die.")

		("service.more_resources", value<bool>()->default_value(false), "Allow more resources to be connected in server mode at the same time.")

		("service.enable_privacy_lists", value<bool>()->default_value(true), "")

		("vhosts.vhost", value<std::vector<std::string> >()->multitoken(), "")

		("identity.name", value<std::string>()->default_value("Spectrum 2 Transport"), "Name showed in service discovery.")

		("identity.category", value<std::string>()->default_value("gateway"), "Disco#info identity category. 'gateway' by default.")

		("identity.type", value<std::string>()->default_value(""), "Type of transport ('icq','msn','gg','irc', ...)")

		("registration.enable_public_registration", value<bool>()->default_value(true), "True if users should be able to register.")

		("registration.language", value<std::string>()->default_value("en"), "Default language for registration form")

		("registration.instructions", value<std::string>()->default_value("Enter your legacy network username and password."), "Instructions showed to user in registration form")

		("registration.username_label", value<std::string>()->default_value("Legacy network username:"), "Label for username field")

		("registration.username_mask", value<std::string>()->default_value(""), "Username mask")

		("registration.auto_register", value<bool>()->default_value(false), "Register new user automatically when the presence arrives.")

		("registration.encoding", value<std::string>()->default_value("utf8"), "Default encoding in registration form")

		("registration.require_local_account", value<bool>()->default_value(false), "True if users have to have a local account to register to this transport from remote servers.")

		("registration.local_username_label", value<std::string>()->default_value("Local username:"), "Label for local usernme field")

		("registration.local_account_server", value<std::string>()->default_value("localhost"), "The server on which the local accounts will be checked for validity")

		("registration.local_account_server_timeout", value<int>()->default_value(10000), "Timeout when checking local user on local_account_server (msecs)")

		("gateway_responder.prompt", value<std::string>()->default_value("Contact ID"), "Value of <prompt> </promt> field")

		("gateway_responder.label", value<std::string>()->default_value("Enter legacy network contact ID."), "Label for add contact ID field")

		("database.type", value<std::string>()->default_value("none"), "Database type.")

		("database.database", value<std::string>()->default_value("/var/lib/spectrum2/$jid/database.sql"), "Database used to store data")

		("database.server", value<std::string>()->default_value("localhost"), "Database server.")

		("database.user", value<std::string>()->default_value(""), "Database user.")

		("database.password", value<std::string>()->default_value(""), "Database Password.")

		("database.port", value<int>()->default_value(0), "Database port.")

		("database.prefix", value<std::string>()->default_value(""), "Prefix of tables in database")

		("database.encryption_key", value<std::string>()->default_value(""), "Encryption key.")

		("logging.config", value<std::string>()->default_value(""), "Path to log4cxx config file which is used for Spectrum 2 instance")

		("logging.backend_config", value<std::string>()->default_value(""), "Path to log4cxx config file which is used for backends")

		("backend.default_avatar", value<std::string>()->default_value(""), "Full path to default avatar")

		("backend.avatars_directory", value<std::string>()->default_value(""), "Path to directory with avatars")

		("backend.no_vcard_fetch", value<bool>()->default_value(false), "True if VCards for buddies should not be fetched. Only avatars will be forwarded.")

	;

	parsed_options parsed = parse_config_file(ifs, opts, true);

	bool found_working = false;

		}

	}

	// Remove user from database

	m_storageBackend->removeUser(userInfo.id);

	// Disconnect the user

	User *user = m_userManager->getUser(barejid);

	if (user) {

		m_userManager->removeUser(user);

	}

	if (remoteRosterNotSupported) {

		Swift::Presence::ref response;

		response = Swift::Presence::create();

		response->setTo(Swift::JID(barejid));

		response->setFrom(m_component->getJID());

		response->setType(Swift::Presence::Unsubscribe);

		m_component->getStanzaChannel()->sendPresence(response);

		response = Swift::Presence::create();

		response->setTo(Swift::JID(barejid));

		response->setFrom(m_component->getJID());

		response->setType(Swift::Presence::Unsubscribed);

		m_component->getStanzaChannel()->sendPresence(response);

	}

	else {

		Swift::RosterPayload::ref payload = Swift::RosterPayload::ref(new Swift::RosterPayload());

		Swift::RosterItemPayload item;

		item.setJID(m_component->getJID());

		item.setSubscription(Swift::RosterItemPayload::Remove);

		payload->addItem(item);

		Swift::SetRosterRequest::ref request = Swift::SetRosterRequest::create(payload, barejid, m_component->getIQRouter());

		request->send();

	}

}

bool UserRegistration::handleGetRequest(const Swift::JID& from, const Swift::JID& to, const std::string& id, boost::shared_ptr<Swift::InBandRegistrationPayload> payload) {

	// TODO: backend should say itself if registration is needed or not...

	if (CONFIG_STRING(m_config, "service.protocol") == "irc") {

		sendError(from, id, ErrorPayload::BadRequest, ErrorPayload::Modify);

		return true;

	}

	std::string barejid = from.toBare().toString();

	if (!CONFIG_BOOL(m_config,"registration.enable_public_registration")) {

		if (std::find(x.begin(), x.end(), from.getDomain()) == x.end()) {

			LOG4CXX_INFO(logger, barejid << ": This user has no permissions to register an account")

			sendError(from, id, ErrorPayload::BadRequest, ErrorPayload::Modify);

			return true;

		}

	}

	boost::shared_ptr<InBandRegistrationPayload> reg(new InBandRegistrationPayload());

	UserInfo res;

	bool registered = m_storageBackend->getUser(barejid, res);

	std::string instructions = CONFIG_STRING(m_config, "registration.instructions");

	std::string usernameField = CONFIG_STRING(m_config, "registration.username_label");

	// normal jabber:iq:register

	reg->setInstructions(instructions);

	reg->setRegistered(registered);

	reg->setUsername(res.uin);

	if (CONFIG_STRING(m_config, "service.protocol") != "twitter" && CONFIG_STRING(m_config, "service.protocol") != "bonjour")

		reg->setPassword(res.password);

	// form

	Form::ref form(new Form(Form::FormType));

	form->setTitle((("Registration")));

	form->setInstructions((instructions));

	HiddenFormField::ref type = HiddenFormField::create();

	type->setName("FORM_TYPE");

	type->setValue("jabber:iq:register");

	form->addField(type);

	TextSingleFormField::ref username = TextSingleFormField::create();

	username->setName("username");

	username->setLabel((usernameField));

	username->setValue(res.uin);

	username->setRequired(true);

	form->addField(username);

	if (CONFIG_STRING(m_config, "service.protocol") != "twitter" && CONFIG_STRING(m_config, "service.protocol") != "bonjour") {

		TextPrivateFormField::ref password = TextPrivateFormField::create();

		password->setName("password");

		password->setLabel((("Password")));

		password->setRequired(true);

		form->addField(password);

	}

//	TextSingleFormField::ref encoding = TextSingleFormField::create();

//	encoding->setName("encoding");

//	encoding->setLabel((("Encoding")));

//	if (registered)

//		encoding->setValue(res.encoding);

//	else

//		encoding->setValue(CONFIG_STRING(m_config, "registration.encoding"));

//	form->addField(encoding);

	if (registered) {

		BooleanFormField::ref boolean = BooleanFormField::create();

		boolean->setName("unregister");

		boolean->setLabel((("Remove your registration")));

		boolean->setValue(0);

		form->addField(boolean);

	} else {

		if (CONFIG_BOOL(m_config,"registration.require_local_account")) {

			std::string localUsernameField = CONFIG_STRING(m_config, "registration.local_username_label");

			TextSingleFormField::ref local_username = TextSingleFormField::create();

			local_username->setName("local_username");

			local_username->setLabel((localUsernameField));

			local_username->setRequired(true);

			form->addField(local_username);

			TextPrivateFormField::ref local_password = TextPrivateFormField::create();

			local_password->setName("local_password");

			local_password->setLabel((("Local Password")));

			local_password->setRequired(true);

			form->addField(local_password);

		}

	}

	reg->setForm(form);

	sendResponse(from, id, reg);

	return true;

}

bool UserRegistration::handleSetRequest(const Swift::JID& from, const Swift::JID& to, const std::string& id, boost::shared_ptr<Swift::InBandRegistrationPayload> payload) {

	// TODO: backend should say itself if registration is needed or not...

	if (CONFIG_STRING(m_config, "service.protocol") == "irc") {

		sendError(from, id, ErrorPayload::BadRequest, ErrorPayload::Modify);

		return true;

	}

	std::string barejid = from.toBare().toString();

	if (!CONFIG_BOOL(m_config,"registration.enable_public_registration")) {

		if (std::find(x.begin(), x.end(), from.getDomain()) == x.end()) {

			LOG4CXX_INFO(logger, barejid << ": This user has no permissions to register an account")

			sendError(from, id, ErrorPayload::BadRequest, ErrorPayload::Modify);

			return true;

		}

	}

	UserInfo res;

	bool registered = m_storageBackend->getUser(barejid, res);

	std::string encoding;

	std::string language;

	std::string local_username("");

	std::string local_password("");

	Form::ref form = payload->getForm();

	if (form) {

		const std::vector<FormField::ref> fields = form->getFields();

		for (std::vector<FormField::ref>::const_iterator it = fields.begin(); it != fields.end(); it++) {

			TextSingleFormField::ref textSingle = boost::dynamic_pointer_cast<TextSingleFormField>(*it);

			if (textSingle) {

				if (textSingle->getName() == "username") {

					payload->setUsername(textSingle->getValue());

				}

				else if (textSingle->getName() == "encoding") {

					encoding = textSingle->getValue();

				}

				// Pidgin sends it as textSingle, not sure why...

				else if (textSingle->getName() == "password") {

					payload->setPassword(textSingle->getValue());

				}

				else if (textSingle->getName() == "local_username") {

					local_username = textSingle->getValue();

				}

				// Pidgin sends it as textSingle, not sure why...

				else if (textSingle->getName() == "local_password") {

					local_password = textSingle->getValue();

				}

				continue;

			}

			TextPrivateFormField::ref textPrivate = boost::dynamic_pointer_cast<TextPrivateFormField>(*it);

			if (textPrivate) {

				if (textPrivate->getName() == "password") {

					payload->setPassword(textPrivate->getValue());

				}

				else if (textPrivate->getName() == "local_password") {

					local_password = textPrivate->getValue();