m_config = config;

	m_prefix = CONFIG_STRING(m_config, "database.prefix");

	mysql_init(&m_conn);

	my_bool my_true = 1;

	mysql_options(&m_conn, MYSQL_OPT_RECONNECT, &my_true);

}

MySQLBackend::~MySQLBackend(){

	disconnect();

}

void MySQLBackend::disconnect() {

	LOG4CXX_INFO(logger, "Disconnecting");

	delete m_setUser;

	delete m_getUser;

	delete m_removeUser;

	delete m_removeUserBuddies;

	delete m_removeUserSettings;

	delete m_removeUserBuddiesSettings;

	delete m_addBuddy;

	delete m_updateBuddy;

	delete m_getBuddies;

	delete m_getBuddiesSettings;

	delete m_getUserSetting;

	delete m_setUserSetting;

	delete m_updateUserSetting;

	delete m_updateBuddySetting;

	delete m_setUserOnline;

	mysql_close(&m_conn);

}

bool MySQLBackend::connect() {

	LOG4CXX_INFO(logger, "Connecting MySQL server " << CONFIG_STRING(m_config, "database.server") << ", user " <<

		CONFIG_STRING(m_config, "database.user") << ", database " << CONFIG_STRING(m_config, "database.database") <<

		", port " << CONFIG_INT(m_config, "database.port")

	);

	if (!mysql_real_connect(&m_conn, CONFIG_STRING(m_config, "database.server").c_str(),

					   CONFIG_STRING(m_config, "database.user").c_str(),

					   CONFIG_STRING(m_config, "database.password").c_str(),

					   CONFIG_STRING(m_config, "database.database").c_str(),

					   CONFIG_INT(m_config, "database.port"), NULL, 0)) {

		LOG4CXX_ERROR(logger, "Can't connect database: " << mysql_error(&m_conn));

		return false;

	}

	createDatabase();

	m_getUser = new Statement(&m_conn, "s|isssssb", "SELECT id, jid, uin, password, encoding, language, vip FROM " + m_prefix + "users WHERE jid=?");

	m_removeUser = new Statement(&m_conn, "i", "DELETE FROM " + m_prefix + "users WHERE id=?");

	m_removeUserBuddies = new Statement(&m_conn, "i", "DELETE FROM " + m_prefix + "buddies WHERE user_id=?");

	m_removeUserSettings = new Statement(&m_conn, "i", "DELETE FROM " + m_prefix + "users_settings WHERE user_id=?");

	m_removeUserBuddiesSettings = new Statement(&m_conn, "i", "DELETE FROM " + m_prefix + "buddies_settings WHERE user_id=?");

	m_addBuddy = new Statement(&m_conn, "issssi", "INSERT INTO " + m_prefix + "buddies (user_id, uin, subscription, groups, nickname, flags) VALUES (?, ?, ?, ?, ?, ?)");

	m_updateBuddy = new Statement(&m_conn, "ssisis", "UPDATE " + m_prefix + "buddies SET groups=?, nickname=?, flags=?, subscription=? WHERE user_id=? AND uin=?");

	m_getBuddies = new Statement(&m_conn, "i|issssi", "SELECT id, uin, subscription, nickname, groups, flags FROM " + m_prefix + "buddies WHERE user_id=? ORDER BY id ASC");

	m_getBuddiesSettings = new Statement(&m_conn, "i|iiss", "SELECT buddy_id, type, var, value FROM " + m_prefix + "buddies_settings WHERE user_id=? ORDER BY buddy_id ASC");

	m_updateBuddySetting = new Statement(&m_conn, "iisiss", "INSERT INTO " + m_prefix + "buddies_settings (user_id, buddy_id, var, type, value) VALUES (?, ?, ?, ?, ?) ON DUPLICATE KEY UPDATE value=?");

	m_getUserSetting = new Statement(&m_conn, "is|is", "SELECT type, value FROM " + m_prefix + "users_settings WHERE user_id=? AND var=?");

	m_setUserSetting = new Statement(&m_conn, "isis", "INSERT INTO " + m_prefix + "users_settings (user_id, var, type, value) VALUES (?,?,?,?)");

	m_updateUserSetting = new Statement(&m_conn, "sis", "UPDATE " + m_prefix + "users_settings SET value=? WHERE user_id=? AND var=?");

	m_setUserOnline = new Statement(&m_conn, "bi", "UPDATE " + m_prefix + "users SET online=?, last_login=NOW()  WHERE id=?");

	return true;

}

bool MySQLBackend::createDatabase() {

	int not_exist = exec("CREATE TABLE IF NOT EXISTS `" + m_prefix + "buddies` ("

							"`id` int(10) unsigned NOT NULL auto_increment,"

							"`user_id` int(10) unsigned NOT NULL,"

							"`uin` varchar(255) collate utf8_bin NOT NULL,"

							"`subscription` enum('to','from','both','ask','none') collate utf8_bin NOT NULL,"

							"`nickname` varchar(255) collate utf8_bin NOT NULL,"

							"`groups` varchar(255) collate utf8_bin NOT NULL,"

							"`flags` smallint(4) NOT NULL DEFAULT '0',"

							"PRIMARY KEY (`id`),"

							"UNIQUE KEY `user_id` (`user_id`,`uin`)"

						") ENGINE=InnoDB DEFAULT CHARSET=utf8 COLLATE=utf8_bin;");

	if (not_exist) {

		exec("CREATE TABLE IF NOT EXISTS `" + m_prefix + "buddies_settings` ("

				"`user_id` int(10) unsigned NOT NULL,"

				"`buddy_id` int(10) unsigned NOT NULL,"

				"`var` varchar(50) collate utf8_bin NOT NULL,"

				"`type` smallint(4) unsigned NOT NULL,"

				"`value` varchar(255) collate utf8_bin NOT NULL,"

				"PRIMARY KEY (`buddy_id`,`var`),"

				"KEY `user_id` (`user_id`)"

			") ENGINE=InnoDB DEFAULT CHARSET=utf8 COLLATE=utf8_bin;");

		exec("CREATE TABLE IF NOT EXISTS `" + m_prefix + "users` ("

				"`id` int(10) unsigned NOT NULL auto_increment,"

				"`jid` varchar(255) collate utf8_bin NOT NULL,"

				"`uin` varchar(4095) collate utf8_bin NOT NULL,"

				"`password` varchar(255) collate utf8_bin NOT NULL,"

				"`language` varchar(25) collate utf8_bin NOT NULL,"

				"`encoding` varchar(50) collate utf8_bin NOT NULL default 'utf8',"

				"`last_login` datetime,"

				"`vip` tinyint(1) NOT NULL  default '0',"

				"`online` tinyint(1) NOT NULL  default '0',"

				"PRIMARY KEY (`id`),"

				"UNIQUE KEY `jid` (`jid`)"

			") ENGINE=InnoDB DEFAULT CHARSET=utf8 COLLATE=utf8_bin;");

		exec("CREATE TABLE IF NOT EXISTS `" + m_prefix + "users_settings` ("

				"`user_id` int(10) unsigned NOT NULL,"

				"`var` varchar(50) collate utf8_bin NOT NULL,"

				"`type` smallint(4) unsigned NOT NULL,"

				"`value` varchar(255) collate utf8_bin NOT NULL,"

				"PRIMARY KEY (`user_id`,`var`),"

				"KEY `user_id` (`user_id`)"

			") ENGINE=InnoDB DEFAULT CHARSET=utf8 COLLATE=utf8_bin;");

		exec("CREATE TABLE IF NOT EXISTS `" + m_prefix + "db_version` ("

				"`ver` int(10) unsigned NOT NULL default '1',"

				"UNIQUE KEY `ver` (`ver`)"

			") ENGINE=InnoDB DEFAULT CHARSET=utf8 COLLATE=utf8_bin;");

		exec("INSERT IGNORE INTO db_version (ver) VALUES ('2');");

	}

	return true;

}

bool MySQLBackend::exec(const std::string &query) {

	if (mysql_query(&m_conn, query.c_str())) {

		LOG4CXX_ERROR(logger, query << " " << mysql_error(&m_conn));

		return false;

	}

	return true;

}

void MySQLBackend::setUser(const UserInfo &user) {

	std::string encrypted = user.password;

	if (!CONFIG_STRING(m_config, "database.encryption_key").empty()) {

		encrypted = Util::encryptPassword(encrypted, CONFIG_STRING(m_config, "database.encryption_key"));

	}

	EXEC(m_setUser, setUser(user));

}

bool MySQLBackend::getUser(const std::string &barejid, UserInfo &user) {

	*m_getUser << barejid;

	EXEC(m_getUser, getUser(barejid, user));

	if (!exec_ok)

		return false;

	int ret = false;

	while (m_getUser->fetch() == 0) {

		ret = true;

		*m_getUser >> user.id >> user.jid >> user.uin >> user.password >> user.encoding >> user.language >> user.vip;

		if (!CONFIG_STRING(m_config, "database.encryption_key").empty()) {

			user.password = Util::decryptPassword(user.password, CONFIG_STRING(m_config, "database.encryption_key"));

		}

	}

	return ret;

}

void MySQLBackend::setUserOnline(long id, bool online) {

	*m_setUserOnline << online << id;

	EXEC(m_setUserOnline, setUserOnline(id, online));

}

long MySQLBackend::addBuddy(long userId, const BuddyInfo &buddyInfo) {

// 	"INSERT INTO " + m_prefix + "buddies (user_id, uin, subscription, groups, nickname, flags) VALUES (?, ?, ?, ?, ?, ?)"

	*m_addBuddy << userId << buddyInfo.legacyName << buddyInfo.subscription;

	*m_addBuddy << (buddyInfo.groups.size() == 0 ? "" : buddyInfo.groups[0]);

	*m_addBuddy << buddyInfo.alias << buddyInfo.flags;

	EXEC(m_addBuddy, addBuddy(userId, buddyInfo));

	long id = (long) mysql_insert_id(&m_conn);

// 	INSERT OR REPLACE INTO " + m_prefix + "buddies_settings (user_id, buddy_id, var, type, value) VALUES (?, ?, ?, ?, ?)

	if (!buddyInfo.settings.find("icon_hash")->second.s.empty()) {

		*m_updateBuddySetting << userId << id << buddyInfo.settings.find("icon_hash")->first << (int) TYPE_STRING << buddyInfo.settings.find("icon_hash")->second.s << buddyInfo.settings.find("icon_hash")->second.s;

		EXEC(m_updateBuddySetting, addBuddy(userId, buddyInfo));

	}

	return id;

}

void MySQLBackend::updateBuddy(long userId, const BuddyInfo &buddyInfo) {

// 	"UPDATE " + m_prefix + "buddies SET groups=?, nickname=?, flags=?, subscription=? WHERE user_id=? AND uin=?"