#include <lib_err_code.h>
#include "func_rount.h"
#include "account.h"
#include "role.h"

account_t::account_t(el::lib_mysql_if* db) 
	: el::lib_mysql_table_route1x1(db, g_gen_db_name("USER"), "t_account")
{
}

int account_t::check_pltform_account_existed( uint32_t platform, 
											 const std::string& account_name,
											 USER_ID& uid)
{
	GEN_MYSQL_STRING(account_name_mysql_string, account_name.c_str(), 
		get_valid_len(account_name.c_str(), ACCOUNT_MAX_LEN));

	GEN_SQLSTR(this->sqlstr,"select uid,three from %s where account='%s' and platform=%u",
		this->get_table_name(),
		account_name_mysql_string, platform);

	STD_QUERY_ONE_BEGIN(this->sqlstr, el::ERR_DB::KEY_INEXIST);

	USER_ID t_uid = 0;
	GET_FIELD_UINT64(t_uid);
	uid = t_uid;

	uint32_t t_three = 0;
	GET_FIELD_UINT64(t_three);
	if (0 != t_three){
		return -1;
	}

	STD_QUERY_ONE_END();

	return SUCC;
}

// uint32_t account_t::is_nick_existed( const std::string& nick )
// {
// 	GEN_MYSQL_STRING(nick_mysql_string, nick.c_str(), 
// 		get_valid_len(nick.c_str(), NICK_LEN));
// 
// 	GEN_SQLSTR(this->sqlstr, "select uid from %s where nick = '%s' ",
// 		this->get_table_name(),
// 		nick_mysql_string);
// 
// 	my_ulonglong row_count = 0;
// 	STD_QUERY_AFFECTED_ROWS(this->sqlstr, row_count);
// 	if (row_count > 0){
// 		return el::ERR_DB::KEY_EXIST;
// 	}
// 
// 	return SUCC;
// }

uint32_t account_t::update_uid_nick(uint32_t platform, const std::string& account, USER_ID uid, const std::string& nick )
{
	GEN_MYSQL_STRING(account_mysql_string, account.c_str(), 
		get_valid_len(account.c_str(), ACCOUNT_MAX_LEN));

	GEN_MYSQL_STRING(nick_escape_string, nick.c_str(), 
		get_valid_len(nick.c_str(), NICK_LEN));

	GEN_SQLSTR(this->sqlstr, "update %s set uid=%" PRIu64 ", nick='%s' where platform=%u and account='%s'"
		, this->get_table_name(), 
		uid, nick_escape_string, platform, account_mysql_string
		);

	return this->exec_update_sql(this->sqlstr, el::ERR_DB::KEY_INEXIST);
}

uint32_t account_t::get_uid_by_account( uint32_t platform, const std::string& account, USER_ID& uid )
{
	GEN_MYSQL_STRING(account_escape_string, account.c_str(), 
		get_valid_len(account.c_str(), ACCOUNT_MAX_LEN));

	GEN_SQLSTR(this->sqlstr, "select uid from %s where platform=%u and account='%s'"
		, this->get_table_name(), 
		platform, account_escape_string
		);

	STD_QUERY_ONE_BEGIN(this->sqlstr, el::ERR_DB::KEY_INEXIST);
	USER_ID tmp_uid = 0;
	GET_FIELD_UINT64(tmp_uid);
	uid = tmp_uid;
	STD_QUERY_ONE_END();
}

int account_t::create_account(uint32_t platform, const std::string& account_name)
{
	GEN_MYSQL_STRING(account_mysql_string, account_name.c_str(), 
		get_valid_len(account_name.c_str(), ACCOUNT_MAX_LEN));
	GEN_SQLSTR(this->sqlstr, "insert into %s (platform,account,uid,status,nick,three,u_account)"
		" values(%u, '%s', 0, 0, '%s', 0, '%s')",
		this->get_table_name(),
		platform, account_mysql_string, "", account_mysql_string);

	return this->exec_insert_sql(this->sqlstr, el::ERR_DB::KEY_EXIST);
}

int account_t::change_nick(USER_ID uid, const std::string& nick)
{
// 	this->ret = is_nick_existed(nick);
// 	if (SUCC != this->ret){
// 		return el::ERR_DB::KEY_EXIST;
// 	}

	GEN_MYSQL_STRING(str_nick, nick.c_str(), get_valid_len(nick.c_str(), NICK_LEN));
	GEN_SQLSTR(this->sqlstr,"update %s set nick='%s' where uid=%" PRIu64 "", this->get_table_name(), str_nick, uid);
	this->ret = this->exec_update_sql(this->sqlstr, el::ERR_DB::KEY_INEXIST);
	if (SUCC != this->ret){
		return el::ERR_DB::KEY_EXIST;
	}

	this->ret = g_route_func->role.change_nick(uid, nick);
	if (SUCC != this->ret) {
		this->ret = el::ERR_DB::KEY_EXIST;
	}

	return this->ret;
}

int account_t::check_pltform_wx_unionid_existed( uint32_t platform, const std::string& wx_unionid, USER_ID& uid )
{
	GEN_MYSQL_STRING(wx_unionid_mysql_string, wx_unionid.c_str(), 
		get_valid_len(wx_unionid.c_str(), ACCOUNT_MAX_LEN));

	GEN_SQLSTR(this->sqlstr,"select uid,three from %s where account='%s' and platform=%u",
		this->get_table_name(),
		wx_unionid_mysql_string, platform);

	STD_QUERY_ONE_BEGIN(this->sqlstr, el::ERR_DB::KEY_INEXIST);

	USER_ID t_uid = 0;
	GET_FIELD_UINT64(t_uid);
	uid = t_uid;

	uint32_t t_three = 0;
	GET_FIELD_UINT64(t_three);
	if (1 != t_three){
		return -1;
	}

	STD_QUERY_ONE_END();

	return SUCC;
}

int account_t::update_wx_account( uint32_t platform, USER_ID uid, const std::string& wx_unionid, const std::string& wx_nick )
{
	GEN_MYSQL_STRING(wx_unionid_escape_string, wx_unionid.c_str(), 
		get_valid_len(wx_unionid.c_str(), ACCOUNT_MAX_LEN));

	GEN_MYSQL_STRING(wx_nick_escape_string, wx_nick.c_str(), 
		get_valid_len(wx_nick.c_str(), NICK_LEN));

	GEN_SQLSTR(this->sqlstr, "update %s set account='%s', nick='%s', three=1 where uid=%" PRIu64,
		this->get_table_name(), wx_unionid_escape_string, wx_nick_escape_string, uid);

	return this->exec_update_sql(this->sqlstr, el::ERR_DB::KEY_INEXIST);
}

int account_t::create_wx_account( uint32_t platform, const std::string& wx_unionid, const std::string& wx_nick, const std::string& account )
{
	GEN_MYSQL_STRING(wx_unionid_mysql_string, wx_unionid.c_str(), 
		get_valid_len(wx_unionid.c_str(), ACCOUNT_MAX_LEN));
	GEN_MYSQL_STRING(wx_nick_escape_string, wx_nick.c_str(), 
		get_valid_len(wx_nick.c_str(), NICK_LEN));
	GEN_MYSQL_STRING(account_mysql_string, account.c_str(), 
		get_valid_len(account.c_str(), ACCOUNT_MAX_LEN));
	GEN_SQLSTR(this->sqlstr, "insert into %s (platform,account,uid,status,nick,three,u_account)"
		" values(%u, '%s', 0, 0, '%s', 1, '%s')",
		this->get_table_name(),
		platform, wx_unionid_mysql_string, wx_nick_escape_string, account_mysql_string);

	//////////////////////////////////////////////////////////////////////////
	
//	return this->exec_insert_sql(this->sqlstr, el::ERR_DB::KEY_EXIST);

	//////////////////////////////////////////////////////////////////////////
	
	this->ret = this->exec_insert_sql(this->sqlstr, el::ERR_DB::KEY_EXIST);

	if (el::ERR_DB::SYS == this->ret){
		GEN_SQLSTR(this->sqlstr, "insert into %s (platform,account,uid,status,nick,three,u_account)"
			" values(%u, '%s', 0, 0, 'dcyly', 1, '%s')",
			this->get_table_name(),
			platform, wx_unionid_mysql_string,account_mysql_string);
		this->ret = this->exec_insert_sql(this->sqlstr, el::ERR_DB::KEY_EXIST);
	}

	return this->ret;
}

int account_t::load( USER_ID uid, std::string& nick )
{
	GEN_SQLSTR(this->sqlstr, "select nick from %s where uid=%" PRIu64, this->get_table_name(), uid);
	STD_QUERY_ONE_BEGIN(this->sqlstr, SUCC);

	static char buf[NICK_LEN];
	::memset(buf, 0, sizeof(buf));
	GET_NEXT_FIELD_BIN(buf, sizeof(buf));
	nick = buf;

	STD_QUERY_ONE_END_WITHOUT_RETURN();

	return SUCC;
}

int account_t::check_uid_existed( USER_ID uid )
{
	GEN_SQLSTR(this->sqlstr, "select uid from %s where uid=%" PRIu64,
		this->get_table_name(), uid);

	STD_QUERY_ONE_BEGIN(this->sqlstr, el::ERR_DB::KEY_INEXIST);
	STD_QUERY_ONE_END();

	return SUCC;
}

int account_t::load_account( USER_ID uid, std::string& account )
{
	GEN_SQLSTR(this->sqlstr, "select account from %s where uid=%" PRIu64, this->get_table_name(), uid);
	STD_QUERY_ONE_BEGIN(this->sqlstr, SUCC);

	static char buf[ACCOUNT_MAX_LEN];
	::memset(buf, 0, sizeof(buf));
	GET_NEXT_FIELD_BIN(buf, sizeof(buf));
	account = buf;

	STD_QUERY_ONE_END_WITHOUT_RETURN();

	return SUCC;
}

uint32_t account_t::gen_gateway_id( uint32_t platform, std::string account, uint32_t gateway_cnt )
{
	std::string str_gateway_key;
	str_gateway_key += "platform=" + el::convert_to_string(platform);
	str_gateway_key += "account=" + account;
	return el::g_hash_str(str_gateway_key.c_str())%gateway_cnt + 1;
}

int account_t::get_random( db_msg::random_friend_msg_res* out )
{
	GEN_SQLSTR(this->sqlstr,"select uid from %s  ORDER BY RAND() LIMIT 10", this->get_table_name());

	PB_STD_QUERY_WHILE_BEGIN_NO_ADD(this->sqlstr);

	USER_ID t_id = 0;

	GET_FIELD_UINT64(t_id);
	out->add_uid(t_id);

	PB_STD_QUERY_WHILE_END();
}


int Cfunc_route::on_change_nick_msg(USER_ID uid, google::protobuf::Message* msg)
{
	auto in = (db_msg::change_nick_msg*)msg;
	int change_ret = this->account.change_nick(uid, in->nick());

	db_msg::change_nick_msg_res out;
	if (el::ERR_DB::KEY_EXIST == change_ret){
		out.set_res(1);
	}
	out.set_nick(in->nick());
	this->encode_msg2sendbuf(&out);
	return this->ret;
}

int Cfunc_route::on_login_msg(USER_ID uid, google::protobuf::Message* msg)
{
	auto in = (db_msg::login_msg*)msg;
	db_msg::login_msg_res out;

	this->ret = this->account.check_pltform_account_existed(in->platform(), in->account(), uid);
	if (-1 == this->ret){
		return this->ret;
	}
	if (el::ERR_DB::KEY_INEXIST == this->ret){
		this->ret = this->account.create_account(in->platform(), in->account());
		if (SUCC != this->ret){
			return this->ret;
		}
	} else if (SUCC != this->ret){
		return this->ret;
	}

	this->head.id = uid;

	this->encode_msg2sendbuf(&out);
	return this->ret;
}

int Cfunc_route::on_login_wx_msg(USER_ID uid, google::protobuf::Message* msg)
{
	auto in = (db_msg::login_wx_msg*)msg;
	db_msg::login_wx_msg_res out;

	this->ret = this->account.check_pltform_wx_unionid_existed(in->platform(), in->wx_unionid(), uid);
	if (-1 == this->ret){
		return this->ret;
	}

	if (el::ERR_DB::KEY_INEXIST == this->ret){
		// 新开户微信 用account去查找用户,并标记微信,替换account 	
		int r1 = this->account.check_pltform_account_existed(in->platform(), in->account(), uid);
		if (-1 == r1){
			return r1;
		}
		if (el::ERR_DB::KEY_INEXIST == r1){
			int r2 = this->account.create_wx_account(in->platform(), in->wx_unionid(), in->wx_nick(), in->account());
			if (SUCC != r2){
				return r2;
			}
			this->ret = 0;
		} else if (SUCC != r1){
			return r1;
		} else {
			int r3 = this->account.update_wx_account(in->platform(), uid, in->wx_unionid(), in->wx_nick());
			if (SUCC != r3){
				return r3;
			}
			this->ret = 0;
		}
	} else if (SUCC != this->ret){
		return this->ret;
	}

	this->head.id = uid;

	this->encode_msg2sendbuf(&out);
	return this->ret;
}

int Cfunc_route::on_create_role_msg(USER_ID uid, google::protobuf::Message* msg)
{
	auto in = (db_msg::create_role_msg*)msg;
	db_msg::create_role_msg_res out;

	USER_ID user_id = 0;	
	this->ret = this->account.get_uid_by_account(in->platform(), in->account(), user_id);
	if (this->ret != SUCC){
		return share_msg::EC_INEXISTENT_ACCOUNT;
	}

	if (user_id != 0){
		WARN_LOG("role exist %s", in->account().c_str());
		return share_msg::EC_EXISTENT_ROLE;
	}

	this->ret = this->gen_uid(user_id);
	if (this->ret != SUCC){
		return this->ret;
	}

	std::string str_nick = in->nick();//el::convert_to_string(user_id);
	if (str_nick.empty()){
		char c_nick[NICK_LEN];
		::memset(c_nick,0,NICK_LEN);
		::sprintf(c_nick, "%s%06u", "豆豆", (uint32_t)user_id%1000000);
		str_nick = c_nick;
	}
	
// 	this->ret = this->account.is_nick_existed(str_nick);
// 	if (this->ret != SUCC){
// 		return this->ret;
// 	}

	this->ret = this->account.update_uid_nick(in->platform(), in->account(), user_id, str_nick);
	if (this->ret != SUCC){
		return this->ret;
	}

	this->ret = this->role.create(user_id, in->head(), str_nick);
	if (this->ret != SUCC){
		return this->ret;
	}

	{
		FOREACH_PB(in->add_items, idx){
			const share_msg::item_t& item = in->add_items(idx);
			this->ret = this->item.update(user_id, item.id(), item.cnt());
			if (this->ret != SUCC){
				return this->ret;
			}
		}
	}
	
	{
		FOREACH_PB(in->add_events, idx){
			const common_msg::event_t& event = in->add_events(idx);
			this->ret = this->user_event.update(user_id, event.type(), event.id(), event.data(),
				event.time(), event.str_data(), event.bin_data());
			if (this->ret != SUCC){
				return this->ret;
			}
		}
	}

	this->head.id = user_id;

	this->encode_msg2sendbuf(&out);
	return this->ret;
}

int Cfunc_route::on_random_friend_msg(USER_ID uid, google::protobuf::Message* msg)
{
	db_msg::random_friend_msg_res out;
	this->account.get_random(&out);

	this->encode_msg2sendbuf(&out);
	
	return this->ret;
}

int Cfunc_route::on_check_uid_msg(USER_ID uid, google::protobuf::Message* msg)
{
	db_msg::check_uid_msg_res out;

	auto in = (db_msg::check_uid_msg*)msg;
	USER_ID peer_uid = in->uid();
	this->ret = this->account.check_uid_existed(peer_uid);
	if (SUCC != this->ret){
		out.set_uid(0);
	} else {
		out.set_uid(peer_uid);
	}

	this->encode_msg2sendbuf(&out);

	return 0;
}