/*
Copyright (c) 2013-24 Miranda NG team (https://miranda-ng.org)

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 version 2
of the License.

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"

//////////////////////////// History services ///////////////////////////////////////////
INT_PTR CVkProto::SvcEmptyServerHistory(WPARAM hContact, LPARAM)
{
	debugLogA("CVkProto::SvcEmptyServerHistory was called for %s", Clist_GetContactDisplayName(hContact));

	VKUserID_t iUserId = ReadVKUserID(hContact);
	if (iUserId == VK_INVALID_USER || !IsOnline() || iUserId == VK_FEED_USER || isChatRoom(hContact))
		return 1;

	Push(new AsyncHttpRequest(this, REQUEST_GET, "/method/messages.deleteConversation.json", true, &CVkProto::OnReceiveSmth, AsyncHttpRequest::rpLow)
		<< INT_PARAM("peer_id", iUserId));
	
	return 0;
}


INT_PTR __cdecl CVkProto::SvcGetAllServerHistoryForContact(WPARAM hContact, LPARAM)
{
	debugLogA("CVkProto::SvcGetAllServerHistoryForContact");
	if (!IsOnline())
		return 0;

	if (getBool(hContact, "ActiveHistoryTask"))
		return 0;

	LPCWSTR str = TranslateT("Are you sure to reload all messages from vk.com?\nLocal contact history will be deleted and reloaded from the server.\nIt may take a long time.\nDo you want to continue?");
	if (IDNO == MessageBoxW(nullptr, str, TranslateT("Attention!"), MB_ICONWARNING | MB_YESNO))
		return 0;

	VKUserID_t iUserId = ReadVKUserID(hContact);
	if (iUserId == VK_INVALID_USER || iUserId == VK_FEED_USER)
		return 0;

	setByte(hContact, "ActiveHistoryTask", 1);

	DB::ECPTR pCursor(DB::Events(hContact));
	while (pCursor.FetchNext())
		pCursor.DeleteEvent();

	m_bNotifyForEndLoadingHistory = true;

	db_unset(hContact, m_szModuleName, "lastmsgid");
	GetServerHistory(hContact, 0, MAXHISTORYMIDSPERONE, 0, 0);
	return 1;
}

INT_PTR __cdecl CVkProto::SvcGetAllServerHistory(WPARAM, LPARAM)
{
	debugLogA("CVkProto::SvcGetAllServerHistory start");
	if (!IsOnline())
		return 0;
	LPCWSTR str = TranslateT("Are you sure you want to reload all messages for all contacts from vk.com?\nLocal contact history will be deleted and reloaded from the server.\nIt may take a very long time and/or corrupt Miranda database.\nDo you want to continue?");
	if (IDNO == MessageBoxW(nullptr, str, TranslateT("Attention!"), MB_ICONWARNING | MB_YESNO))
		return 0;

	for (auto &hContact : AccContacts()) {
		VKUserID_t iUserId = ReadVKUserID(hContact);
		if (iUserId == VK_INVALID_USER || iUserId == VK_FEED_USER)
			continue;

		if (getBool(hContact, "ActiveHistoryTask"))
			break;
		setByte(hContact, "ActiveHistoryTask", 1);

		DB::ECPTR pCursor(DB::Events(hContact));
		while (pCursor.FetchNext())
			pCursor.DeleteEvent();

		{
			mir_cslock lck(m_csLoadHistoryTask);
			m_iLoadHistoryTask++;
			m_bNotifyForEndLoadingHistoryAllContact = m_bNotifyForEndLoadingHistory = true;
			debugLogA("CVkProto::SvcGetAllServerHistory for ID=%d m_iLoadHistoryTask=%d", iUserId, m_iLoadHistoryTask);
		}

		db_unset(hContact, m_szModuleName, "lastmsgid");
		GetServerHistory(hContact, 0, MAXHISTORYMIDSPERONE, 0, 0);
	}

	return 1;
}

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

void CVkProto::GetServerHistoryLastNDay(MCONTACT hContact, int iNDay)
{
	debugLogA("CVkProto::SvcGetServerHistoryLastNDay %d", iNDay);

	if (getBool(hContact, "ActiveHistoryTask"))
		return;
	setByte(hContact, "ActiveHistoryTask", 1);

	time_t tTime = time(0) - 60 * 60 * 24 * iNDay;

	if (iNDay > 3) {
		DB::ECPTR pCursor(DB::Events(hContact));
		while (MEVENT hDbEvent = pCursor.FetchNext()) {
			DBEVENTINFO dbei = {};
			db_event_get(hDbEvent, &dbei);
			if (dbei.timestamp > tTime && dbei.eventType != VK_USER_DEACTIVATE_ACTION)
				pCursor.DeleteEvent();
		}

		{
			mir_cslock lck(m_csLoadHistoryTask);
			m_iLoadHistoryTask++;
			m_bNotifyForEndLoadingHistory = true;
		}

	}

	WriteQSWord(hContact, "oldlastmsgid", ReadQSWord(hContact, "lastmsgid", -1));
	db_unset(hContact, m_szModuleName, "lastmsgid");
	GetServerHistory(hContact, 0, MAXHISTORYMIDSPERONE, tTime, 0);
}

void CVkProto::GetServerHistory(MCONTACT hContact, int iOffset, int iCount, time_t tTime, VKMessageID_t iLastMsgId, bool bOnce)
{
	debugLogA("CVkProto::GetServerHistory");
	if (!IsOnline() || iCount == 0)
		return;

	VKUserID_t iUserId = isChatRoom(hContact) ? VK_CHAT_MIN + ReadVKUserID(hContact) : ReadVKUserID(hContact);

	debugLogA("CVkProto::GetServerHistory %ld %d %d %d %d %d", iUserId, iOffset, iCount, tTime, iLastMsgId, (int)bOnce);
	if (VK_INVALID_USER == iUserId || iUserId == VK_FEED_USER)
		return;

	Push(new AsyncHttpRequest(this, REQUEST_GET, "/method/execute.GetServerConversationHistory", true, &CVkProto::OnReceiveHistoryMessages, AsyncHttpRequest::rpLow)
		<< INT_PARAM("reqcount", iCount)
		<< INT_PARAM("offset", iOffset)
		<< INT_PARAM("userid", iUserId)
		<< INT_PARAM("time", tTime)
		<< INT_PARAM("lastmid", iLastMsgId)
		<< INT_PARAM("once", (int)bOnce)
	)->pUserInfo = new CVkSendMsgParam(hContact, iLastMsgId, iOffset);
}

void CVkProto::GetHistoryDlg(MCONTACT hContact, VKMessageID_t iLastMsg)
{
	debugLogA("CVkProto::GetHistoryDlg %d", iLastMsg);
	VKMessageID_t iLastMsgId = -1;
	switch (m_vkOptions.iSyncHistoryMetod) {
	case syncAuto:
		iLastMsgId = ReadQSWord(hContact, "lastmsgid", -1);
		if (iLastMsgId == -1 || !IsOnline()) {
			WriteQSWord(hContact, "lastmsgid", iLastMsg);
			return;
		}
		m_bNotifyForEndLoadingHistory = false;
		if (getBool(hContact, "ActiveHistoryTask"))
			return;
		setByte(hContact, "ActiveHistoryTask", 1);
		GetServerHistory(hContact, 0, MAXHISTORYMIDSPERONE, 0, iLastMsgId);
		break;
	case sync1Days:
		GetServerHistoryLastNDay(hContact, 1);
		break;
	case sync3Days:
		GetServerHistoryLastNDay(hContact, 3);
		break;
	}
}

void CVkProto::OnReceiveHistoryMessages(MHttpResponse *reply, AsyncHttpRequest *pReq)
{
	debugLogA("CVkProto::OnReceiveHistoryMessages %d", reply->resultCode);
	if (reply->resultCode != 200 || !pReq->pUserInfo) {
		mir_cslock lck(m_csLoadHistoryTask);
		if (m_iLoadHistoryTask > 0)
			m_iLoadHistoryTask--;
		debugLogW(L"CVkProto::OnReceiveHistoryMessages error m_iLoadHistoryTask=%d", m_iLoadHistoryTask);
		MsgPopup(TranslateT("Error loading message history from server."), TranslateT("Error"), true);

		if (m_iLoadHistoryTask == 0 && m_bNotifyForEndLoadingHistoryAllContact) {
			MsgPopup(TranslateT("Loading messages for all contacts is completed."), TranslateT("Loading history"));
			m_bNotifyForEndLoadingHistoryAllContact = m_bNotifyForEndLoadingHistory = false;
		}
		if (pReq->pUserInfo)
			setByte(((CVkSendMsgParam*)pReq->pUserInfo)->hContact, "ActiveHistoryTask", 0);
		return;
	}

	JSONNode jnRoot;
	CVkSendMsgParam *param = (CVkSendMsgParam*)pReq->pUserInfo;
	const JSONNode &jnResponse = CheckJsonResponse(pReq, reply, jnRoot);
	if (!jnResponse) {
		setByte(param->hContact, "ActiveHistoryTask", 0);
		if (!pReq->bNeedsRestart || m_bTerminated) {
			mir_cslock lck(m_csLoadHistoryTask);
			if (m_iLoadHistoryTask > 0)
				m_iLoadHistoryTask--;

			ptrW pwszNick(db_get_wsa(param->hContact, m_szModuleName, "Nick"));
			CMStringW str(FORMAT, TranslateT("Error loading message history for %s from server."), pwszNick);

			MsgPopup(param->hContact, str, TranslateT("Error"), true);
			debugLogW(L"CVkProto::OnReceiveHistoryMessages error for %s m_iLoadHistoryTask=%d", pwszNick.get(), m_iLoadHistoryTask);

			if (m_iLoadHistoryTask == 0 && m_bNotifyForEndLoadingHistoryAllContact) {
				MsgPopup(TranslateT("Loading messages for all contacts is completed."), TranslateT("Loading history"));
				m_bNotifyForEndLoadingHistoryAllContact = m_bNotifyForEndLoadingHistory = false;
			}

			delete param;
			pReq->pUserInfo = nullptr;
		}
		return;
	}

	int iTime = jnResponse["datetime"].as_int();
	const JSONNode &jnMsgs = jnResponse["items"];
	const JSONNode &jnFUsers = jnResponse["fwd_users"];
	VKMessageID_t iLastMsgId = ReadQSWord(param->hContact, "lastmsgid", -1);
	time_t tLastReadMessageTime = 0;
	MEVENT hDbEvent = 0;
	int count = 0;

	for (auto it = jnMsgs.rbegin(); it != jnMsgs.rend(); ++it) {
		const JSONNode &jnMsg = (*it);
		VKMessageID_t iMessageId = jnMsg["id"].as_int();
		if (iLastMsgId < iMessageId)
			iLastMsgId = iMessageId;

		VKUserID_t iUserId = jnMsg["peer_id"].as_int();
		VKUserID_t iChatId = (GetVKPeerType(iUserId) == VKPeerType::vkPeerMUC) ? iUserId % VK_CHAT_MIN : 0;
		VKMessageID_t iReadMsg = ReadQSWord(param->hContact, "in_read", 0);
		bool bIsRead = (iMessageId <= iReadMsg);
		bool bIsOut = jnMsg["out"].as_int();
		time_t tDateTime = jnMsg["date"].as_int();
		
		if (iChatId != 0){
			AppendChatConversationMessage(iChatId, jnMsg, jnFUsers, true);

			if (bIsRead && bIsOut && tDateTime > tLastReadMessageTime)
				tLastReadMessageTime = tDateTime;

			count++;
			continue;
		}

		char szMid[40], szReplyId[40] = "";
		_ltoa(iMessageId, szMid, 10);

		CMStringW wszBody(jnMsg["text"].as_mstring());
		MCONTACT hContact = FindUser(iUserId, true);

		const JSONNode &jnFwdMessages = jnMsg["fwd_messages"];
		if (jnFwdMessages && !jnFwdMessages.empty()) {
			CMStringW wszFwdMessages = GetFwdMessages(jnFwdMessages, jnFUsers, m_vkOptions.BBCForAttachments());
			if (!wszBody.IsEmpty())
				wszFwdMessages = L"\n" + wszFwdMessages;
			wszBody += wszFwdMessages;
		}

		const JSONNode& jnReplyMessages = jnMsg["reply_message"];
		if (jnReplyMessages && !jnReplyMessages.empty())
			if (m_vkOptions.bShowReplyInMessage) {
				CMStringW wszReplyMessages = GetFwdMessages(jnReplyMessages, jnFUsers, m_vkOptions.BBCForAttachments());
				if (!wszBody.IsEmpty())
					wszReplyMessages = L"\n" + wszReplyMessages;
				wszBody += wszReplyMessages;
			} 
			else if (jnReplyMessages["id"])
				_ltoa(jnReplyMessages["id"].as_int(), szReplyId, 10);

		const JSONNode &jnAttachments = jnMsg["attachments"];
		if (jnAttachments && !jnAttachments.empty()) {
			CMStringW wszAttachmentDescr = GetAttachmentDescr(jnAttachments, m_vkOptions.BBCForAttachments(), hContact, iMessageId);

			if (wszAttachmentDescr == L"== FilterAudioMessages ==") {
				count++;
				continue;
			}

			if (!wszBody.IsEmpty())
				wszAttachmentDescr = L"\n" + wszAttachmentDescr;
			wszBody += wszAttachmentDescr;
		}

		if (m_vkOptions.bAddMessageLinkToMesWAtt && ((jnAttachments && !jnAttachments.empty()) || (jnFwdMessages && !jnFwdMessages.empty()) || (jnReplyMessages && !jnReplyMessages.empty() && m_vkOptions.bShowReplyInMessage)))
			wszBody += SetBBCString(TranslateT("Message link"), m_vkOptions.BBCForAttachments(), vkbbcUrl,
				CMStringW(FORMAT, L"https://vk.com/im?sel=%d&msgid=%d", iUserId, iMessageId));

		DB::EventInfo dbei;
		if (bIsRead)
			dbei.flags |= DBEF_READ;
		if (bIsOut)
			dbei.flags |= DBEF_SENT;

		time_t tUpdateTime = (time_t)jnMsg["update_time"].as_int();
		if (tUpdateTime) {
			CMStringW wszEditTime;

			wchar_t ttime[64];
			_locale_t locale = _create_locale(LC_ALL, "");
			_wcsftime_l(ttime, _countof(ttime), TranslateT("%x at %X"), localtime(&tUpdateTime), locale);
			_free_locale(locale);

			wszEditTime.Format(TranslateT("Edited message (updated %s):\n"), ttime);

			wszBody = SetBBCString(
				wszEditTime,
				m_vkOptions.BBCForAttachments(), vkbbcB) +
				wszBody;
		}

		T2Utf pszBody(wszBody);

		dbei.timestamp = tDateTime;
		dbei.pBlob = pszBody;
		dbei.szId = szMid;

		if (!m_vkOptions.bShowReplyInMessage &&	szReplyId)
			dbei.szReplyId = szReplyId;

		ProtoChainRecvMsg(hContact, dbei);

		hDbEvent = db_event_getById(m_szModuleName, strcat(szMid, "_"));
		if (hDbEvent)
			db_event_delete(hDbEvent, CDF_FROM_SERVER);

		if (bIsRead && bIsOut && tDateTime > tLastReadMessageTime)
			tLastReadMessageTime = tDateTime;

		count++;
	}

	db_event_delivered(param->hContact, hDbEvent);

	WriteQSWord(param->hContact, "lastmsgid", iLastMsgId);

	int once = jnResponse["once"].as_int();
	int iRCount = jnResponse["rcount"].as_int();
	if (count == iRCount && once == 0 && iRCount > 0)
		GetServerHistory(param->hContact, param->iCount + count, iRCount, iTime, param->iMsgID);
	else {
		mir_cslock lck(m_csLoadHistoryTask);
		if (m_iLoadHistoryTask > 0)
			m_iLoadHistoryTask--;

		WriteQSWord(param->hContact, "lastmsgid", iLastMsgId == -1 ? ReadQSWord(param->hContact, "oldlastmsgid", -1) : iLastMsgId);
		db_unset(param->hContact, m_szModuleName, "oldlastmsgid");

		ptrW pwszNick(db_get_wsa(param->hContact, m_szModuleName, "Nick"));
		CMStringW str(FORMAT, TranslateT("Loading messages for %s is completed."), pwszNick);
		debugLogW(L"CVkProto::OnReceiveHistoryMessages for %s m_iLoadHistoryTask=%d", pwszNick.get(), m_iLoadHistoryTask);

		if (m_bNotifyForEndLoadingHistory)
			MsgPopup(param->hContact, str, TranslateT("Loading history"));

		if (m_iLoadHistoryTask == 0 && m_bNotifyForEndLoadingHistoryAllContact) {
			MsgPopup(TranslateT("Loading messages for all contacts is completed."), TranslateT("Loading history"));
			m_bNotifyForEndLoadingHistoryAllContact = m_bNotifyForEndLoadingHistory = false;
		}
		setByte(param->hContact, "ActiveHistoryTask", 0);
	}

	if (!pReq->bNeedsRestart || m_bTerminated) {
		delete param;
		pReq->pUserInfo = nullptr;
	}
}