/*
Plugin of Miranda IM for communicating with users of the MSN Messenger protocol.

Copyright (c) 2012-2017 Miranda NG Team
Copyright (c) 2006-2012 Boris Krasnovskiy.
Copyright (c) 2003-2005 George Hazan.
Copyright (c) 2002-2003 Richard Hughes (original version).

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"
#include "msn_proto.h"

/////////////////////////////////////////////////////////////////////////////////////////
// MSN_AddGroup - adds new server group to the list

void CMsnProto::MSN_AddGroup(const char* grpName, const char *grpId, bool init)
{
	ServerGroupItem* p = m_arGroups.find((ServerGroupItem*)&grpId);
	if (p != nullptr) return;

	p = (ServerGroupItem*)mir_alloc(sizeof(ServerGroupItem));
	p->id = mir_strdup(grpId);
	p->name = mir_strdup(grpName);

	m_arGroups.insert(p);

	if (init)
		Clist_GroupCreate(0, ptrW(mir_utf8decodeW(grpName)));
}

/////////////////////////////////////////////////////////////////////////////////////////
// MSN_DeleteGroup - deletes a group from the list

void CMsnProto::MSN_DeleteGroup(const char* pId)
{
	int i = m_arGroups.getIndex((ServerGroupItem*)&pId);
	if (i > -1) {
		ServerGroupItem* p = m_arGroups[i];
		mir_free(p->id);
		mir_free(p->name);
		mir_free(p);
		m_arGroups.remove(i);
	}
}

/////////////////////////////////////////////////////////////////////////////////////////
// MSN_DeleteServerGroup - deletes group from the server

void CMsnProto::MSN_DeleteServerGroup(LPCSTR szId)
{
	if (!MyOptions.ManageServer) return;

	MSN_ABAddDelContactGroup(nullptr, szId, "ABGroupDelete");

	int count = -1;
	for (;;) {
		MsnContact *msc = Lists_GetNext(count);
		if (msc == nullptr) break;

		char szGroupID[100];
		if (!db_get_static(msc->hContact, m_szModuleName, "GroupID", szGroupID, sizeof(szGroupID))) {
			if (mir_strcmp(szGroupID, szId) == 0)
				delSetting(msc->hContact, "GroupID");
		}
	}
	MSN_DeleteGroup(szId);
}

/////////////////////////////////////////////////////////////////////////////////////////
// MSN_FreeGroups - clears the server groups list

void CMsnProto::MSN_FreeGroups(void)
{
	for (int i = 0; i < m_arGroups.getCount(); i++) {
		ServerGroupItem* p = m_arGroups[i];
		mir_free(p->id);
		mir_free(p->name);
		mir_free(p);
	}
	m_arGroups.destroy();
}

/////////////////////////////////////////////////////////////////////////////////////////
// MSN_GetGroupById - tries to return a group name associated with given UUID

LPCSTR CMsnProto::MSN_GetGroupById(const char* pId)
{
	ServerGroupItem* p = m_arGroups.find((ServerGroupItem*)&pId);
	return p ? p->name : nullptr;
}

/////////////////////////////////////////////////////////////////////////////////////////
// MSN_GetGroupByName - tries to return a group UUID associated with the given name

LPCSTR CMsnProto::MSN_GetGroupByName(const char* pName)
{
	for (int i = 0; i < m_arGroups.getCount(); i++) {
		const ServerGroupItem* p = m_arGroups[i];
		if (mir_strcmp(p->name, pName) == 0)
			return p->id;
	}

	return nullptr;
}

/////////////////////////////////////////////////////////////////////////////////////////
// MSN_SetGroupName - sets a new name to a server group

void CMsnProto::MSN_SetGroupName(const char* pId, const char* pNewName)
{
	ServerGroupItem* p = m_arGroups.find((ServerGroupItem*)&pId);
	if (p != nullptr)
		replaceStr(p->name, pNewName);
}

/////////////////////////////////////////////////////////////////////////////////////////
// MSN_MoveContactToGroup - sends a contact to the specified group

void CMsnProto::MSN_MoveContactToGroup(MCONTACT hContact, const char* grpName)
{
	if (!MyOptions.ManageServer) return;

	LPCSTR szId = nullptr;
	char szContactID[100], szGroupID[100];
	if (db_get_static(hContact, m_szModuleName, "ID", szContactID, sizeof(szContactID)))
		return;

	if (db_get_static(hContact, m_szModuleName, "GroupID", szGroupID, sizeof(szGroupID)))
		szGroupID[0] = 0;

	bool bInsert = false, bDelete = szGroupID[0] != 0;

	if (grpName != nullptr) {
		szId = MSN_GetGroupByName(grpName);
		if (szId == nullptr) {
			MSN_ABAddGroup(grpName);
			szId = MSN_GetGroupByName(grpName);
		}
		if (szId == nullptr) return;
		if (_stricmp(szGroupID, szId) == 0) bDelete = false;
		else                                bInsert = true;
	}

	if (bDelete) {
		MSN_ABAddDelContactGroup(szContactID, szGroupID, "ABGroupContactDelete");
		delSetting(hContact, "GroupID");
	}

	if (bInsert) {
		MSN_ABAddDelContactGroup(szContactID, szId, "ABGroupContactAdd");
		setString(hContact, "GroupID", szId);
	}

	if (bDelete)
		MSN_RemoveEmptyGroups();
}

/////////////////////////////////////////////////////////////////////////////////////////
// MSN_RemoveEmptyGroups - removes empty groups from the server list

void CMsnProto::MSN_RemoveEmptyGroups(void)
{
	if (!MyOptions.ManageServer) return;

	unsigned *cCount = (unsigned*)mir_calloc(m_arGroups.getCount() * sizeof(unsigned));

	int count = -1;
	for (;;) {
		MsnContact *msc = Lists_GetNext(count);
		if (msc == nullptr) break;

		char szGroupID[100];
		if (!db_get_static(msc->hContact, m_szModuleName, "GroupID", szGroupID, sizeof(szGroupID))) {
			const char *pId = szGroupID;
			int i = m_arGroups.getIndex((ServerGroupItem*)&pId);
			if (i > -1) ++cCount[i];
		}
	}

	for (int i = m_arGroups.getCount(); i--;) {
		if (cCount[i] == 0) MSN_DeleteServerGroup(m_arGroups[i]->id);
	}
	mir_free(cCount);
}

/////////////////////////////////////////////////////////////////////////////////////////
// MSN_RenameServerGroup - renames group on the server

void CMsnProto::MSN_RenameServerGroup(LPCSTR szId, const char* newName)
{
	MSN_SetGroupName(szId, newName);
	MSN_ABRenameGroup(newName, szId);
}


/////////////////////////////////////////////////////////////////////////////////////////
// MSN_UploadServerGroups - adds a group to the server list and contacts into the group

void CMsnProto::MSN_UploadServerGroups(char* group)
{
	if (!MyOptions.ManageServer) return;

	int count = -1;
	for (;;) {
		MsnContact *msc = Lists_GetNext(count);
		if (msc == nullptr) break;

		DBVARIANT dbv;
		if (!db_get_utf(msc->hContact, "CList", "Group", &dbv)) {
			char szGroupID[100];
			if (group == nullptr || (mir_strcmp(group, dbv.pszVal) == 0 &&
				db_get_static(msc->hContact, m_szModuleName, "GroupID", szGroupID, sizeof(szGroupID)) != 0)) {
				MSN_MoveContactToGroup(msc->hContact, dbv.pszVal);
			}
			db_free(&dbv);
		}
	}
}

/////////////////////////////////////////////////////////////////////////////////////////
// MSN_SyncContactToServerGroup - moves contact into appropriate group according to server
// if contact in multiple server groups it get removed from all of them other them it's
// in or the last one

void CMsnProto::MSN_SyncContactToServerGroup(MCONTACT hContact, const char* szContId, ezxml_t cgrp)
{
	if (!MyOptions.ManageServer) return;

	const char* szGrpName = "";

	DBVARIANT dbv;
	if (!db_get_utf(hContact, "CList", "Group", &dbv)) {
		szGrpName = NEWSTR_ALLOCA(dbv.pszVal);
		db_free(&dbv);
	}

	const char* szGrpIdF = nullptr;
	while (cgrp != nullptr) {
		const char* szGrpId = ezxml_txt(cgrp);
		cgrp = ezxml_next(cgrp);

		const char* szGrpNameById = MSN_GetGroupById(szGrpId);

		if (szGrpNameById && (mir_strcmp(szGrpNameById, szGrpName) == 0 ||
			(cgrp == nullptr && szGrpIdF == nullptr)))
			szGrpIdF = szGrpId;
		else
			MSN_ABAddDelContactGroup(szContId, szGrpId, "ABGroupContactDelete");
	}

	if (szGrpIdF != nullptr) {
		setString(hContact, "GroupID", szGrpIdF);
		const char* szGrpNameById = MSN_GetGroupById(szGrpIdF);
		if (mir_strcmp(szGrpNameById, szGrpName))
			db_set_utf(hContact, "CList", "Group", szGrpNameById);
	}
	else {
		if (szGrpName[0])
			db_unset(hContact, "CList", "Group");
		delSetting(hContact, "GroupID");
	}
}

/////////////////////////////////////////////////////////////////////////////////////////
// Msn_SendNickname - update our own nickname on the server

void CMsnProto::MSN_SendNicknameUtf(const char* nickname)
{
	if (nickname[0])
		setStringUtf(NULL, "Nick", nickname);
	else
		delSetting("Nick");

	ForkThread(&CMsnProto::msn_storeProfileThread, (void*)1);
}

/////////////////////////////////////////////////////////////////////////////////////////
// msn_storeAvatarThread - update our own avatar on the server

void CMsnProto::msn_storeAvatarThread(void* arg)
{
	StoreAvatarData* dat = (StoreAvatarData*)arg;
	ptrA szEncBuf;

	if (dat)
		szEncBuf = mir_base64_encode(dat->data, (unsigned)dat->dataSize);

	if (photoid[0] && dat)
		MSN_StoreUpdateDocument(dat->szName, dat->szMimeType, szEncBuf);
	else {
		MSN_StoreUpdateProfile(nullptr, nullptr, 1);

		if (photoid[0]) {
			MSN_StoreDeleteRelationships(true);
			MSN_StoreDeleteRelationships(false);
			photoid[0] = 0;
		}

		if (dat) {
			MSN_StoreCreateDocument(dat->szName, dat->szMimeType, szEncBuf);
			MSN_StoreCreateRelationships();
		}

		MSN_StoreUpdateProfile(nullptr, nullptr, 0);
	}

	MSN_ABUpdateDynamicItem();

	if (dat) {
		mir_free(dat->szName);
		mir_free(dat->data);
		mir_free(dat);
	}
}

/////////////////////////////////////////////////////////////////////////////////////////
// msn_storeProfileThread - update our own avatar on the server

void CMsnProto::msn_storeProfileThread(void* param)
{
	DBVARIANT dbv;
	char *szNick = nullptr;
	bool needFree = false;
	if (!getStringUtf("Nick", &dbv)) {
		szNick = dbv.pszVal[0] ? dbv.pszVal : nullptr;
		needFree = true;
	}

	char** msgptr = GetStatusMsgLoc(m_iStatus);
	char *szStatus = msgptr ? *msgptr : nullptr;

	if (param || (msnLastStatusMsg != szStatus && (msnLastStatusMsg && szStatus && mir_strcmp(msnLastStatusMsg, szStatus))))
		if (MSN_StoreUpdateProfile(szNick, szStatus, false))
			MSN_ABUpdateDynamicItem();

	if (needFree) db_free(&dbv);
}