/*
Sessions Management plugin for Miranda IM

Copyright (C) 2007-2008 Danil Mozhar

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 3 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 "sessions.h"

void AddSessionMark(MCONTACT hContact, int mode, char bit)
{
	DBVARIANT dbv;
	unsigned int i;
	char temp_1[1]={'\0'},temp_2[1]={'\0'};
	char szDst[256]={'\0'};
	char* pszBuffer=NULL;
	if (mode == 0) {	
		if (!db_get_s(hContact, MODNAME, "LastSessionsMarks", &dbv) && dbv.pszVal) {
			temp_1[0]=dbv.pszVal[0];
			for (i=0; i < ses_limit; i++) {
				temp_2[0]=dbv.pszVal[i+1];
				dbv.pszVal[i+1]=temp_1[0];
				temp_1[0]=temp_2[0];
			}
			for (i=ses_limit; i < 10; i++)
				dbv.pszVal[i]='0';
			dbv.pszVal[0]=bit;
			db_set_s(hContact, MODNAME, "LastSessionsMarks", dbv.pszVal);
			db_free(&dbv);
		}
		else if (bit == '1') db_set_s(hContact, MODNAME, "LastSessionsMarks", "10000000000");
	}
	else if (mode == 1) {
		if (!db_get_s(hContact, MODNAME, "UserSessionsMarks", &dbv) && dbv.pszVal) {   
			if (strlen(dbv.pszVal)<g_ses_count) {
				pszBuffer = (char*)mir_alloc(g_ses_count+1);
				ZeroMemory(pszBuffer,g_ses_count+1);
				strcpy(pszBuffer,dbv.pszVal);
			}
			else pszBuffer = mir_strdup(dbv.pszVal);
			db_free(&dbv);

			temp_1[0]=pszBuffer[0];
			for (i=0;i<g_ses_count;i++) {
				temp_2[0]=pszBuffer[i+1];
				pszBuffer[i+1]=temp_1[0];
				temp_1[0]=temp_2[0];
			}
			pszBuffer[0]=bit;
			db_set_s(hContact, MODNAME, "UserSessionsMarks", pszBuffer);
			
			mir_free(pszBuffer);
		}
		else if (bit == '1')
			db_set_s(hContact, MODNAME, "UserSessionsMarks", "10000000000");
		else
			db_set_s(hContact, MODNAME, "UserSessionsMarks", "00000000000");
	}
}

void RemoveSessionMark(MCONTACT hContact,int mode,int marknum)
{
	unsigned int i=1;
	char temp_1[1]={'\0'},temp_2[1]={'\0'};
	char szDst[256]={'\0'};
	DBVARIANT dbv;

	if (mode == 0) {
		if (!db_get_s(hContact, MODNAME, "LastSessionsMarks", &dbv) && dbv.pszVal) {
			for (i=marknum;i<ses_limit;i++)		
				dbv.pszVal[i] = dbv.pszVal[i+1];

			for (i=ses_limit;i<10;i++)
				dbv.pszVal[i] = '0';

			db_set_s(hContact, MODNAME, "LastSessionsMarks", dbv.pszVal);
			db_free(&dbv);
		}
	}
	else if (mode == 1) {
		if (!db_get_s(hContact, MODNAME, "UserSessionsMarks", &dbv) && dbv.pszVal) {
			for (i=marknum;i < ses_limit; i++)		
				dbv.pszVal[i] = dbv.pszVal[i+1];

			db_set_s(hContact, MODNAME, "UserSessionsMarks", dbv.pszVal);
			db_free(&dbv);
		}
	}
}

void SetSessionMark(MCONTACT hContact,int mode,char bit,unsigned int marknum)
{
	DBVARIANT dbv;
	char* pszBuffer=NULL;

	if (mode == 0) {	
		if (!db_get_s(hContact, MODNAME, "LastSessionsMarks", &dbv)) {
			dbv.pszVal[marknum]=bit;
			db_set_s(hContact, MODNAME, "LastSessionsMarks", dbv.pszVal);

			db_free(&dbv);
		}
	}
	else if (mode == 1) {
		if (!db_get_s(hContact, MODNAME, "UserSessionsMarks", &dbv)) {
			pszBuffer = mir_strdup(dbv.pszVal);
			pszBuffer[marknum]=bit;
			db_set_s(hContact, MODNAME, "UserSessionsMarks", pszBuffer);
			mir_free(pszBuffer);
			db_free(&dbv);
		}
	}
}

BOOL LoadContactsFromMask(MCONTACT hContact,int mode,int count)
{	
	DBVARIANT dbv;
	if (mode == 0) {	
		if (db_get_s(hContact, MODNAME, "LastSessionsMarks", &dbv))
			return 0;
	}
	else if (mode == 1) {
		if (db_get_s(hContact, MODNAME, "UserSessionsMarks", &dbv))
			return 0;
	}
	else return 0;

	BOOL res = dbv.pszVal[count] == '1';
	db_free(&dbv);
	return res;
}

void AddInSessionOrder(MCONTACT hContact,int mode,int ordernum,int writemode)
{
	int i;
	char temp_1[2]={'\0'},temp_2[2]={'\0'};
	char szDst[256]={'\0'};
	char* pszBuffer=NULL;
	char szFormNumBuf[255]={'\0'};

 	if (ordernum<10)
 		mir_snprintf(szFormNumBuf, SIZEOF(szFormNumBuf), "%u%u", 0, ordernum);
 	else
		mir_snprintf(szFormNumBuf, SIZEOF(szFormNumBuf), "%u", ordernum);
	
	if (mode == 0) {
		DBVARIANT dbv;
		if (!db_get_s(hContact, MODNAME, "LastSessionsOrder", &dbv)) {	  
			char* temp=NULL;
			char* temp2=NULL;
			int len= (int)strlen(dbv.pszVal);
			if (!len)
				len = 20;

			temp2 = (char*)mir_alloc(len-1);
			ZeroMemory(temp2,len-1);
			temp = (char*)mir_alloc(len+1);
			ZeroMemory(temp,len+1);
			if (ordernum < 10) {
				strncpy( temp2, dbv.pszVal, len-2 );
				temp2[len-2]='\0';
				mir_snprintf(temp, len+1, "%u%u%s", 0, ordernum, temp2);
			}
			else mir_snprintf(temp, len+1, "%u%s", ordernum, temp2);

			for (i=(ses_limit*2);i<20;i++)
				temp[i]='0';

			db_set_s(hContact, MODNAME, "LastSessionsOrder", temp);
			mir_free(temp);
			mir_free(temp2);
			db_free(&dbv);
		}
		else if (writemode == 1) {
			ZeroMemory(szFormNumBuf,SIZEOF(szFormNumBuf));
			if (ordernum<10)
				mir_snprintf(szFormNumBuf, SIZEOF(szFormNumBuf), "%u%u%s", 0, ordernum, "000000000000000000");
 			else
				mir_snprintf(szFormNumBuf, SIZEOF(szFormNumBuf), "%u%s", ordernum, "000000000000000000");

			db_set_s(hContact, MODNAME, "LastSessionsOrder", szFormNumBuf);
		}
	}
	else if (mode == 1) {
		DBVARIANT dbv;
		int advlen=0;
		if (!db_get_s(hContact, MODNAME, "UserSessionsOrder", &dbv) && dbv.pszVal) {
			char* temp=NULL;
			char* temp2=NULL;
			int len=0;
			if (strlen(dbv.pszVal)<(g_ses_count*2)) {
				pszBuffer = (char*)mir_alloc(g_ses_count*2+1);
				ZeroMemory(pszBuffer,g_ses_count*2+1);
				strcpy(pszBuffer,dbv.pszVal);
			}
			else pszBuffer = mir_strdup(dbv.pszVal);

			if (!(advlen=len= (int)strlen(pszBuffer)))
				advlen=len=20;
			else
				advlen=len+2;
			
			temp = (char*)mir_alloc(advlen+1);
			ZeroMemory(temp,advlen+1);

			if (ordernum<10)
				mir_snprintf(temp, advlen+1, "%u%u%s", 0, ordernum, dbv.pszVal);
			else 
				mir_snprintf(temp, advlen+1, "%u%s", ordernum, dbv.pszVal);
				
			db_free(&dbv);

			db_set_s(hContact, MODNAME, "UserSessionsOrder", temp);
			mir_free(temp);
			mir_free(pszBuffer);
		}
		else if (writemode == 1)
			db_set_s(hContact, MODNAME, "UserSessionsOrder", szFormNumBuf);
		else
			db_set_s(hContact, MODNAME, "UserSessionsOrder", "00");
	}
}

int GetInSessionOrder(MCONTACT hContact,int mode,int count)
{
	int iOrder = 0;
	char szTemp[3]={'\0'};
	if (mode == 0) {	
		DBVARIANT dbv;
		if (!db_get_s(hContact, MODNAME, "LastSessionsOrder", &dbv) && dbv.pszVal) {
			szTemp[0] = dbv.pszVal[count*2];
			szTemp[1] = dbv.pszVal[count*2+1];
			iOrder = atoi(szTemp);
			db_free(&dbv);
		}
	}
	else if (mode == 1) {
		DBVARIANT dbv;
		if (!db_get_s(hContact, MODNAME, "UserSessionsOrder", &dbv) && dbv.pszVal) {
			szTemp[0] = dbv.pszVal[count*2];
			szTemp[1] = dbv.pszVal[count*2+1];
			iOrder = atoi(szTemp);
			db_free(&dbv);
		}
	}
	return iOrder;
}

void SetInSessionOrder(MCONTACT hContact,int mode,int count,unsigned int ordernum)
{
	int iOrder=0;
	char szTemp[3]={'\0'};

	if (ordernum < 10)
		mir_snprintf(szTemp, SIZEOF(szTemp), "%u%u", 0, ordernum);
	else
		mir_snprintf(szTemp, SIZEOF(szTemp), "%u", ordernum);

	if (mode == 0) {	
		DBVARIANT dbv;
		if (!db_get_s(hContact, MODNAME, "LastSessionsOrder", &dbv) && dbv.pszVal) {
			dbv.pszVal[count*2]=szTemp[0];
			dbv.pszVal[count*2+1]=szTemp[1];
			db_set_s(hContact, MODNAME, "LastSessionsOrder", dbv.pszVal);
			db_free(&dbv);
		}
	}
	else if (mode == 1) {
		DBVARIANT dbv;
		if (!db_get_s(hContact, MODNAME, "UserSessionsOrder", &dbv) && dbv.pszVal) {
			dbv.pszVal[count*2]=szTemp[0];
			dbv.pszVal[count*2+1]=szTemp[1];
			db_set_s(hContact, MODNAME, "UserSessionsOrder", dbv.pszVal);
			db_free(&dbv);
		}
	}
}

BOOL ResaveSettings(char* szName,int iFirst,int iLimit,TCHAR* szBuffer)
{
	BYTE marked,marked_t;
	char szNameBuf[256]={0};

	for (int i = iFirst; i < iLimit; i++) {
		if (szBuffer == NULL)
			break;

		mir_snprintf(szNameBuf, SIZEOF(szNameBuf), "%s_%u", szName, i);

		DBVARIANT dbv_temp;
		int res = db_get_ts(NULL, MODNAME, szNameBuf, &dbv_temp);
		db_set_ts(NULL, MODNAME, szNameBuf, szBuffer);
		mir_free(szBuffer);

		marked = IsMarkedUserDefSession(i);
		MarkUserDefSession(i,(BYTE)((i == iFirst)?IsMarkedUserDefSession(iFirst-1):marked_t));
		marked_t = marked;

		if (res) // read failed
			return 0;
			
		szBuffer = mir_tstrdup(dbv_temp.ptszVal);
		db_free(&dbv_temp);
	}

	mir_free(szBuffer);
	return 1;
}

int AddToCurSession(MCONTACT wparam, LPARAM lparam)
{
	if (CheckForDuplicate(session_list, wparam) == -1) {
		for (int i = 0;; i++) {
			if (session_list[i] == 0) {
				session_list[i] = wparam;
				break;
			}
		}
	}
	return 0;
}

int DelFromCurSession(MCONTACT wparam,LPARAM lparam) 
{
	for (int i=0; session_list[i] != 0; i++) {
		if (session_list[i] == wparam) {
			while(session_list[i+1]!=0) {
				session_list[i]=session_list[i+1];
				i++;
			}
			session_list[i] = 0;
		}
	}
	return 0;
}

int CheckForDuplicate(MCONTACT contact_list[], MCONTACT lparam)
{
	int i=0;
	MCONTACT s_list[255] = {0};
	memcpy(s_list, contact_list, SIZEOF(s_list));
	for (i=0;;i++) {
		if (s_list[i] == lparam)
			return i;
		if (s_list[i] == 0)
			return -1;
	}
	return 0;
}

int LoadSessionToCombobox(HWND hdlg,BOOL mode,int iLimit,char* pszSetting,int iFirstNum)
{
	int ses_count = iFirstNum, index;
	char szBuffer[256]={0};
	if (session_list_recovered[0] && !iFirstNum && !mode) {
		index = SendDlgItemMessage(hdlg, IDC_LIST, CB_ADDSTRING, 0, (LPARAM)TranslateT("Session Before Last Crash"));
		SendDlgItemMessage(hdlg, IDC_LIST, CB_SETITEMDATA, index, 256);
	}
	for(int i=0; i < iLimit; i++) {
		mir_snprintf(szBuffer, SIZEOF(szBuffer), "%s_%u", pszSetting, i);
		TCHAR *pszBuffer = db_get_tsa(NULL, MODNAME, szBuffer);
		if (pszBuffer) {
			if (!IsMarkedUserDefSession(i+iFirstNum)||mode == 1) {
				index = SendDlgItemMessage(hdlg, IDC_LIST, CB_ADDSTRING, 0, (LPARAM)pszBuffer);
				SendDlgItemMessage(hdlg, IDC_LIST, CB_SETITEMDATA, (WPARAM)index, i+iFirstNum);
			}
			else {
				SendDlgItemMessage(hdlg, IDC_LIST, CB_INSERTSTRING, 0, (LPARAM)pszBuffer);
				SendDlgItemMessage(hdlg, IDC_LIST, CB_SETITEMDATA, 0, i+iFirstNum);
			}
			mir_free(pszBuffer);
		}
		else {
			if (i == 0) ses_count=iLimit-iFirstNum;
			break;
		}
	}

	return ses_count;
}

int FillFavoritesMenu (HMENU hMenu,int iLimit)
{
	int iItems=0;

	for(int i=0;i < iLimit; i++) {
		if (IsMarkedUserDefSession(i)) {
			char szBuffer[256];
			mir_snprintf(szBuffer, SIZEOF(szBuffer), "%s_%u", "UserSessionDsc", i);
			TCHAR *pszBuffer = db_get_tsa(NULL, MODNAME, szBuffer);
			if (pszBuffer) {
				AppendMenu(hMenu,MF_STRING,i+1,pszBuffer);
				iItems++;
				mir_free(pszBuffer);
			}
		}
	}
	return iItems;
}

void OffsetWindow(HWND parent, HWND hwnd, int dx, int dy)
{
	POINT pt;
	RECT rc;
	GetWindowRect(hwnd, &rc);
	pt.x = rc.left;
	pt.y = rc.top;
	ScreenToClient(parent, &pt);
	rc.left  =pt.x ;
	rc.top   =pt.y ;

	OffsetRect(&rc, dx, dy);
	SetWindowPos(hwnd, NULL, rc.left, rc.top,0, 0, SWP_NOZORDER |SWP_NOSIZE);
}

int CheckContactVisibility(MCONTACT hContact)
{ 
	if (db_get_b(hContact, "CList", "Hidden", 0))
		return 0;
	return 1;
}

void RenameUserDefSession(int ses_count,TCHAR* ptszNewName)
{
	char szSession[256]={0};
	mir_snprintf(szSession, SIZEOF(szSession), "%s_%u", "UserSessionDsc", ses_count);
	db_set_ts(NULL, MODNAME, szSession, ptszNewName);
}

int MarkUserDefSession(int ses_count,BYTE bCheck)
{
	char szSessionName[256]={0};
	mir_snprintf(szSessionName, SIZEOF(szSessionName), "%s_%u", "FavUserSession", ses_count);
	db_set_b(NULL, MODNAME, szSessionName, bCheck);
	return 0;
}

BYTE IsMarkedUserDefSession(int ses_count)
{
	char szSessionName[256]={0};
	mir_snprintf(szSessionName, SIZEOF(szSessionName), "%s_%u","FavUserSession", ses_count);
	return db_get_b(NULL, MODNAME, szSessionName, 0);
}

void SavePosition(HWND hwnd, char *wndName)
{
	RECT rc;
	GetWindowRect(hwnd, &rc);
	char buffer[512];
	mir_snprintf(buffer, SIZEOF(buffer), "%sPosX", wndName);
	db_set_dw(0, MODNAME, buffer, rc.left);
	mir_snprintf(buffer, SIZEOF(buffer), "%sPosY", wndName);
	db_set_dw(0, MODNAME, buffer, rc.top);
}

void LoadPosition(HWND hWnd, char *wndName)
{	
	char buffer[512];
	mir_snprintf(buffer, SIZEOF(buffer), "%sPosX", wndName);
	int x = db_get_dw(0, MODNAME, buffer, ((GetSystemMetrics(SM_CXSCREEN)) / 2)-130);
	mir_snprintf(buffer, SIZEOF(buffer), "%sPosY", wndName);
	int y = db_get_dw(0, MODNAME, buffer, ((GetSystemMetrics(SM_CYSCREEN))/ 2)-80);
	SetWindowPos(hWnd, NULL, x, y, 0, 0, SWP_NOZORDER | SWP_NOOWNERZORDER | SWP_NOSIZE);
}