/*
Copyright © 2016-22 Miranda NG team

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, see <http://www.gnu.org/licenses/>.
*/

#include "stdafx.h"

int StrToStatus(const CMStringW &str)
{
	if (str == L"idle")
		return ID_STATUS_AWAY;
	if (str == L"dnd")
		return ID_STATUS_DND;
	if (str == L"online")
		return ID_STATUS_ONLINE;
	if (str == L"offline")
		return ID_STATUS_OFFLINE;
	return 0;
}

/////////////////////////////////////////////////////////////////////////////////////////

time_t StringToDate(const CMStringW &str)
{
	struct tm T = { 0 };
	int boo;
	if (swscanf(str, L"%04d-%02d-%02dT%02d:%02d:%02d.%d", &T.tm_year, &T.tm_mon, &T.tm_mday, &T.tm_hour, &T.tm_min, &T.tm_sec, &boo) != 7)
		return time(0);

	T.tm_year -= 1900;
	T.tm_mon--;
	time_t t = mktime(&T);

	_tzset();
	t -= _timezone;
	return (t >= 0) ? t : 0;
}

/////////////////////////////////////////////////////////////////////////////////////////

static LONG volatile g_counter = 1;

int SerialNext()
{
	return InterlockedIncrement(&g_counter);
}

/////////////////////////////////////////////////////////////////////////////////////////

CMStringW getName(const JSONNode &pNode)
{
	CMStringW wszNick = pNode["global_name"].as_mstring();
	if (wszNick.IsEmpty())
		wszNick = pNode["username"].as_mstring();
	return wszNick;
}

CMStringW getNick(const JSONNode &pNode)
{
	CMStringW name = pNode["username"].as_mstring(), discriminator = pNode["discriminator"].as_mstring();
	if (discriminator == L"0")
		return name;

	return name + L"#" + discriminator;
}

/////////////////////////////////////////////////////////////////////////////////////////

SnowFlake getId(const JSONNode &pNode)
{
	return _wtoi64(pNode.as_mstring());
}

SnowFlake CDiscordProto::getId(const char *szSetting)
{
	DBVARIANT dbv;
	dbv.type = DBVT_BLOB;
	if (db_get(0, m_szModuleName, szSetting, &dbv))
		return 0;
	
	SnowFlake result = (dbv.cpbVal == sizeof(SnowFlake)) ? *(SnowFlake*)dbv.pbVal : 0;
	db_free(&dbv);
	return result;
}

SnowFlake CDiscordProto::getId(MCONTACT hContact, const char *szSetting)
{
	DBVARIANT dbv;
	dbv.type = DBVT_BLOB;
	if (db_get(hContact, m_szModuleName, szSetting, &dbv))
		return 0;

	SnowFlake result;
	switch (dbv.type) {
	case DBVT_BLOB:
		result = (dbv.cpbVal == sizeof(SnowFlake)) ? *(SnowFlake *)dbv.pbVal : 0;
		break;
	case DBVT_ASCIIZ:
	case DBVT_UTF8:
		result = _atoi64(dbv.pszVal);
		break;
	case DBVT_WCHAR:
		result = _wtoi64(dbv.pwszVal);
		break;
	default:
		result = 0;
	}
	db_free(&dbv);
	return result;
}

void CDiscordProto::setId(const char *szSetting, SnowFlake iValue)
{
	SnowFlake oldVal = getId(szSetting);
	if (oldVal != iValue)
		db_set_blob(0, m_szModuleName, szSetting, &iValue, sizeof(iValue));
}

void CDiscordProto::setId(MCONTACT hContact, const char *szSetting, SnowFlake iValue)
{
	SnowFlake oldVal = getId(hContact, szSetting);
	if (oldVal != iValue)
		db_set_blob(hContact, m_szModuleName, szSetting, &iValue, sizeof(iValue));
}

bool CDiscordProto::surelyGetBool(MCONTACT hContact, const char *szSetting)
{
	int iValue = getDword(hContact, szSetting, -1);
	if (iValue == -1)
		setByte(hContact, szSetting, iValue = 1);
	
	return iValue != 0;
}

/////////////////////////////////////////////////////////////////////////////////////////

void CopyId(const CMStringW &nick)
{
	if (!OpenClipboard(nullptr))
		return;

	EmptyClipboard();

	int length = nick.GetLength() + 1;
	if (HGLOBAL hMemory = GlobalAlloc(GMEM_FIXED, length * sizeof(wchar_t))) {
		mir_wstrncpy((wchar_t*)GlobalLock(hMemory), nick, length);
		GlobalUnlock(hMemory);
		SetClipboardData(CF_UNICODETEXT, hMemory);
	}
	CloseClipboard();
}

/////////////////////////////////////////////////////////////////////////////////////////

static CDiscordUser *g_myUser = new CDiscordUser(0);

CDiscordUser* CDiscordProto::FindUser(SnowFlake id)
{
	return arUsers.find((CDiscordUser*)&id);
}

CDiscordUser* CDiscordProto::FindUser(const wchar_t *pwszUsername, int iDiscriminator)
{
	for (auto &p : arUsers)
		if (p->wszUsername == pwszUsername && p->iDiscriminator == iDiscriminator)
			return p;

	return nullptr;
}

CDiscordUser* CDiscordProto::FindUserByChannel(SnowFlake channelId)
{
	for (auto &p : arUsers)
		if (p->channelId == channelId)
			return p;

	return nullptr;
}

/////////////////////////////////////////////////////////////////////////////////////////
// Common JSON processing routines

void CDiscordProto::PreparePrivateChannel(const JSONNode &root)
{
	CDiscordUser *pUser = nullptr;

	CMStringW wszChannelId = root["id"].as_mstring();
	SnowFlake channelId = _wtoi64(wszChannelId);

	int type = root["type"].as_int();
	switch (type) {
	case 1: // single channel
		for (auto &it : root["recipients"])
			pUser = PrepareUser(it);
		if (pUser == nullptr) {
			debugLogA("Invalid recipients list, exiting");
			return;
		}
		break;

	case 3: // private groupchat
		if ((pUser = FindUserByChannel(channelId)) == nullptr) {
			pUser = new CDiscordUser(channelId);
			arUsers.insert(pUser);
		}
		pUser->bIsGroup = true;
		pUser->wszUsername = wszChannelId;
		pUser->wszChannelName = root["name"].as_mstring();
		if (pUser->wszChannelName.IsEmpty()) {
			int i = 0;
			for (auto &it : root["recipients"]) {
				CMStringW wszName = getName(it);
				if (wszName.IsEmpty())
					continue;

				if (!pUser->wszChannelName.IsEmpty())
					pUser->wszChannelName += L", ";
				pUser->wszChannelName += wszName;

				if (i++ > 3)
					break;
			}
		}

		if (auto *si = pUser->si = Chat_NewSession(GCW_CHATROOM, m_szModuleName, pUser->wszUsername, pUser->wszChannelName)) {
			pUser->hContact = si->hContact;

			Chat_AddGroup(si, LPGENW("Owners"));
			Chat_AddGroup(si, LPGENW("Participants"));
			{
				SnowFlake ownerId = ::getId(root["owner_id"]);
				setId(pUser->hContact, DB_KEY_OWNERID, ownerId);

				CheckAvatarChange(si->hContact, root["icon"].as_mstring());

				bool bHasMe = false;
				GCEVENT gce = { si, GC_EVENT_JOIN };
				for (auto &it : root["recipients"]) {
					CMStringW wszId = it["id"].as_mstring();
					CMStringW wszNick = it["global_name"].as_mstring();
					if (wszNick.IsEmpty())
						wszNick = getNick(it);

					gce.bIsMe = _wtoi64(wszId) == m_ownId;
					gce.pszUID.w = wszId;
					gce.pszNick.w = wszNick;
					gce.pszStatus.w = (_wtoi64(wszId) == ownerId) ? L"Owners" : L"Participants";
					Chat_Event(&gce);

					if (gce.bIsMe)
						bHasMe = true;
				}

				if (!bHasMe) {
					CMStringW wszId(FORMAT, L"%lld", getId(DB_KEY_ID)), wszNick;
					if (auto iDiscr = getDword(DB_KEY_DISCR))
						wszNick.Format(L"%s#%d", getMStringW(DB_KEY_NICK).c_str(), iDiscr);
					else
						wszNick = getMStringW(DB_KEY_NICK);

					gce.bIsMe = true;
					gce.pszUID.w = wszId;
					gce.pszNick.w = wszNick;
					gce.pszStatus.w = (_wtoi64(wszId) == ownerId) ? L"Owners" : L"Participants";
					Chat_Event(&gce);
				}
			}

			Chat_Control(si, m_bHideGroupchats ? WINDOW_HIDDEN : SESSION_INITDONE);
			Chat_Control(si, SESSION_ONLINE);
		}
		break;

	default:
		debugLogA("Invalid channel type: %d, exiting", type);
		return;
	}

	pUser->channelId = channelId;
	pUser->lastMsgId = ::getId(root["last_message_id"]);
	pUser->bIsPrivate = true;

	setId(pUser->hContact, DB_KEY_CHANNELID, pUser->channelId);

	SnowFlake oldMsgId = getId(pUser->hContact, DB_KEY_LASTMSGID);
	if (pUser->lastMsgId > oldMsgId)
		RetrieveHistory(pUser, MSG_AFTER, oldMsgId, 99);
}

CDiscordUser* CDiscordProto::PrepareUser(const JSONNode &user)
{
	SnowFlake id = ::getId(user["id"]);
	if (id == m_ownId)
		return g_myUser;

	int iDiscriminator = _wtoi(user["discriminator"].as_mstring());
	CMStringW username = user["username"].as_mstring();

	CDiscordUser *pUser = FindUser(id);
	if (pUser == nullptr) {
		MCONTACT tmp = INVALID_CONTACT_ID;

		// no user found by userid, try to find him via username+discriminator
		pUser = FindUser(username, iDiscriminator);
		if (pUser != nullptr) {
			// if found, remove the object from list to resort it (its userid==0)
			if (pUser->hContact != 0)
				tmp = pUser->hContact;
			arUsers.remove(pUser);
		}
		pUser = new CDiscordUser(id);
		pUser->wszUsername = username;
		pUser->iDiscriminator = iDiscriminator;
		if (tmp != INVALID_CONTACT_ID) {
			// if we previously had a recently added contact without userid, write it down
			pUser->hContact = tmp;
			setId(pUser->hContact, DB_KEY_ID, id);
		}
		arUsers.insert(pUser);
	}

	if (pUser->hContact == 0) {
		MCONTACT hContact = db_add_contact();
		Proto_AddToContact(hContact, m_szModuleName);

		Clist_SetGroup(hContact, m_wszDefaultGroup);
		setId(hContact, DB_KEY_ID, id);
		setWString(hContact, DB_KEY_NICK, username);
		setDword(hContact, DB_KEY_DISCR, iDiscriminator);

		pUser->hContact = hContact;
	}

	CMStringW wszName(user["global_name"].as_mstring());
	int idx = wszName.ReverseFind(' ');
	if (idx != -1) {
		setWString(pUser->hContact, "FirstName", wszName.Left(idx));
		setWString(pUser->hContact, "LastName", wszName.Mid(idx + 1));
	}
	else setWString(pUser->hContact, "FirstName", wszName);

	CheckAvatarChange(pUser->hContact, user["avatar"].as_mstring());
	return pUser;
}

/////////////////////////////////////////////////////////////////////////////////////////

CMStringW CDiscordProto::PrepareMessageText(const JSONNode &pRoot, CDiscordUser *pUser)
{
	CMStringW wszText = pRoot["content"].as_mstring();
	CMStringA szUserId = pRoot["author"]["id"].as_mstring();

	bool bFilesAdded = false;
	for (auto &it : pRoot["attachments"]) {
		CMStringA szUrl = it["url"].as_mstring();
		if (szUrl.IsEmpty())
			continue;

		bFilesAdded = true;
		CMStringA szId = it["id"].as_mstring();
		
		auto *pFile = new CDiscordAttachment();
		pFile->szUrl = szUrl;
		pFile->szFileName = it["filename"].as_mstring();
		pFile->iFileSize = it["size"].as_int();

		T2Utf szDescr(wszText);
	
		DB::EventInfo dbei(db_event_getById(m_szModuleName, szId));
		dbei.flags = DBEF_TEMPORARY;
		dbei.timestamp = (uint32_t)StringToDate(pRoot["timestamp"].as_mstring());
		dbei.szId = szId;
		dbei.szUserId = szUserId;
		if (_atoi64(szUserId) == m_ownId)
			dbei.flags |= DBEF_READ | DBEF_SENT;

		if (dbei) {
			DB::FILE_BLOB blob(dbei);
			OnReceiveOfflineFile(dbei, blob);
			blob.write(dbei);
			db_event_edit(dbei.getEvent(), &dbei, true);
			delete pFile;
		}
		else ProtoChainRecvFile(pUser->hContact, DB::FILE_BLOB(pFile, pFile->szFileName, szDescr), dbei);
	}

	if (bFilesAdded)
		return L"";

	for (auto &it : pRoot["embeds"]) {
		wszText.Append(L"\n-----------------");

		CMStringW str = it["url"].as_mstring();
		wszText.AppendFormat(L"\n%s: %s", TranslateT("Embed"), str.c_str());

		str = it["provider"]["name"].as_mstring() + L" " + it["type"].as_mstring();
		if (str.GetLength() > 1)
			wszText.AppendFormat(L"\n\t%s", str.c_str());

		str = it["description"].as_mstring();
		if (!str.IsEmpty())
			wszText.AppendFormat(L"\n\t%s", str.c_str());

		str = it["thumbnail"]["url"].as_mstring();
		if (!str.IsEmpty())
			wszText.AppendFormat(L"\n%s: %s", TranslateT("Preview"), str.c_str());
	}

	auto &nAuthor = pRoot["author"];
	SnowFlake mentionId = 0, userId = ::getId(nAuthor["id"]);
	CMStringW wszMentioned, wszAuthor = getName(nAuthor);

	for (auto &it : pRoot["mentions"]) {
		wszMentioned = getName(it);
		mentionId = ::getId(it["id"]);
		break;
	}

	switch (pRoot["type"].as_int()) {
	case 1: // user was added to chat
		if (mentionId != userId)
			wszText.Format(TranslateT("%s added %s to the group"), wszAuthor.c_str(), wszMentioned.c_str());
		else
			wszText.Format(TranslateT("%s joined the group"), wszMentioned.c_str());
		break;

	case 2: // user was removed from chat
		if (mentionId != userId)
			wszText.Format(TranslateT("%s removed %s from the group"), wszAuthor.c_str(), wszMentioned.c_str());
		else
			wszText.Format(TranslateT("%s left the group"), wszMentioned.c_str());
		break;

	case 3: // call
		break;

	case 4: // chat was renamed
		if (pUser->si)
			setWString(pUser->si->hContact, "Nick", wszText);
		break;

	case 5: // chat icon is changed
		wszText.Format(TranslateT("%s changed the group icon"), wszAuthor.c_str());
		break;
	}

	return wszText;
}

/////////////////////////////////////////////////////////////////////////////////////////

void CDiscordProto::ProcessType(CDiscordUser *pUser, const JSONNode &pRoot)
{
	switch (pRoot["type"].as_int()) {
	case 1: // confirmed
		Contact::PutOnList(pUser->hContact);
		delSetting(pUser->hContact, DB_KEY_REQAUTH);
		delSetting(pUser->hContact, "ApparentMode");
		break;

	case 3: // expecting authorization
		Contact::RemoveFromList(pUser->hContact);
		if (!getByte(pUser->hContact, DB_KEY_REQAUTH, 0)) {
			setByte(pUser->hContact, DB_KEY_REQAUTH, 1);

			CMStringA szId(FORMAT, "%lld", pUser->id);
			DB::AUTH_BLOB blob(pUser->hContact, T2Utf(pUser->wszUsername), nullptr, nullptr, szId, nullptr);

			DB::EventInfo dbei;
			dbei.timestamp = (uint32_t)time(0);
			dbei.cbBlob = blob.size();
			dbei.pBlob = blob;
			ProtoChainRecv(pUser->hContact, PSR_AUTH, 0, (LPARAM)&dbei);
		}
		break;
	}
}

/////////////////////////////////////////////////////////////////////////////////////////

void CDiscordProto::ParseSpecialChars(SESSION_INFO *si, CMStringW &str)
{
	for (int i = 0; (i = str.Find('<', i)) != -1; i++) {
		int iEnd = str.Find('>', i + 1);
		if (iEnd == -1)
			return;

		CMStringW wszWord = str.Mid(i + 1, iEnd - i - 1);
		if (wszWord[0] == '@') { // member highlight
			int iStart = 1;
			if (wszWord[1] == '!')
				iStart++;

			USERINFO *ui = g_chatApi.UM_FindUser(si, wszWord.c_str() + iStart);
			if (ui != nullptr)
				str.Replace(L"<" + wszWord + L">", CMStringW(ui->pszNick) + L": ");
		}
		else if (wszWord[0] == '#') {
			CDiscordUser *pUser = FindUserByChannel(_wtoi64(wszWord.c_str() + 1));
			if (pUser != nullptr) {
				ptrW wszNick(getWStringA(pUser->hContact, DB_KEY_NICK));
				if (wszNick != nullptr)
					str.Replace(L"<" + wszWord + L">", wszNick);
			}
		}
	}
}