/*

Jabber Protocol Plugin for Miranda IM
Copyright ( C ) 2002-04  Santithorn Bunchua
Copyright ( C ) 2005-06  George Hazan

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, write to the Free Software
Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.

File name      : $Source: /cvsroot/miranda/miranda/protocols/JabberG/jabber_list.cpp,v $
Revision       : $Revision: 3651 $
Last change on : $Date: 2006-08-30 16:54:52 +0400 (Срд, 30 Авг 2006) $
Last change by : $Author: ghazan $

*/

#include "jabber.h"
#include "jabber_list.h"

static int count;
static JABBER_LIST_ITEM *lists;
static CRITICAL_SECTION csLists;

static void JabberListFreeItemInternal( JABBER_LIST_ITEM *item );

void JabberListInit( void )
{
	lists = NULL;
	count = 0;
	InitializeCriticalSection( &csLists );
}

void JabberListUninit( void )
{
	JabberListWipe();
	DeleteCriticalSection( &csLists );
}

void JabberListWipe( void )
{
	int i;

	EnterCriticalSection( &csLists );
	for( i=0; i<count; i++ )
		JabberListFreeItemInternal( &( lists[i] ));
	if ( lists != NULL ) {
		mir_free( lists );
		lists = NULL;
	}
	count=0;
	LeaveCriticalSection( &csLists );
}

static void JabberListFreeItemInternal( JABBER_LIST_ITEM *item )
{
	int i;

	if ( item == NULL )
		return;

	if ( item->jid ) mir_free( item->jid );
	if ( item->nick ) mir_free( item->nick );

	JABBER_RESOURCE_STATUS* r = item->resource;
	for ( i=0; i<item->resourceCount; i++, r++ ) {
		if ( r->resourceName ) mir_free( r->resourceName );
		if ( r->statusMessage ) mir_free( r->statusMessage );
		if ( r->software ) mir_free( r->software );
		if ( r->version ) mir_free( r->version );
		if ( r->system ) mir_free( r->system );
	}
	if ( item->resource ) mir_free( item->resource );
	if ( item->statusMessage ) mir_free( item->statusMessage );
	if ( item->group ) mir_free( item->group );
	if ( item->photoFileName ) {
		DeleteFileA( item->photoFileName );
		mir_free( item->photoFileName );
	}
	if ( item->messageEventIdStr ) mir_free( item->messageEventIdStr );
	if ( item->name ) mir_free( item->name );
	if ( item->type ) mir_free( item->type );
	if ( item->service ) mir_free( item->service );
	if ( item->list==LIST_ROSTER && item->ft ) delete item->ft;
}

int JabberListExist( JABBER_LIST list, const TCHAR* jid )
{
	TCHAR szSrc[ JABBER_MAX_JID_LEN ];
	JabberStripJid( jid, szSrc, sizeof( szSrc ));

	EnterCriticalSection( &csLists );
	for ( int i=0; i<count; i++ ) {
		if ( lists[i].list == list ) {
			TCHAR szTempJid[ JABBER_MAX_JID_LEN ];
			if ( !_tcsicmp( szSrc, JabberStripJid( lists[i].jid, szTempJid, sizeof( szTempJid )))) {
			  	LeaveCriticalSection( &csLists );
				return i+1;
	}	}	}

	LeaveCriticalSection( &csLists );
	return 0;
}

JABBER_LIST_ITEM *JabberListAdd( JABBER_LIST list, const TCHAR* jid )
{
	TCHAR* s, *p, *q;
	JABBER_LIST_ITEM *item;

	EnterCriticalSection( &csLists );
	if (( item=JabberListGetItemPtr( list, jid )) != NULL ) {
		LeaveCriticalSection( &csLists );
		return item;
	}

	s = mir_tstrdup( jid );
	// strip resource name if any
	if (( p = _tcschr( s, '@' )) != NULL )
		if (( q = _tcschr( p, '/' )) != NULL )
			*q = '\0';

	lists = ( JABBER_LIST_ITEM * ) mir_realloc( lists, sizeof( JABBER_LIST_ITEM )*( count+1 ));
	item = &( lists[count] );
	ZeroMemory( item, sizeof( JABBER_LIST_ITEM ));
	item->list = list;
	item->jid = s;
	item->status = ID_STATUS_OFFLINE;
	item->resource = NULL;
	item->resourceMode = RSMODE_LASTSEEN;
	item->defaultResource = -1;
	if ( list == LIST_ROSTER )
		item->cap = CLIENT_CAP_CHATSTAT;
	count++;
	LeaveCriticalSection( &csLists );

	return item;
}

void JabberListRemove( JABBER_LIST list, const TCHAR* jid )
{
	EnterCriticalSection( &csLists );
	int i = JabberListExist( list, jid );
	if ( !i ) {
		LeaveCriticalSection( &csLists );
		return;
	}
	i--;
	JabberListFreeItemInternal( &( lists[i] ));
	count--;
	memmove( lists+i, lists+i+1, sizeof( JABBER_LIST_ITEM )*( count-i ));
	lists = ( JABBER_LIST_ITEM * ) mir_realloc( lists, sizeof( JABBER_LIST_ITEM )*count );
	LeaveCriticalSection( &csLists );
}

void JabberListRemoveList( JABBER_LIST list )
{
	int i = 0;
	while (( i=JabberListFindNext( list, i )) >= 0 )
		JabberListRemoveByIndex( i );
}

void JabberListRemoveByIndex( int index )
{
	EnterCriticalSection( &csLists );
	if ( index>=0 && index<count ) {
		JabberListFreeItemInternal( &( lists[index] ));
		count--;
		memmove( lists+index, lists+index+1, sizeof( JABBER_LIST_ITEM )*( count-index ));
		lists = ( JABBER_LIST_ITEM * ) mir_realloc( lists, sizeof( JABBER_LIST_ITEM )*count );
	}
	LeaveCriticalSection( &csLists );
}

int JabberListAddResource( JABBER_LIST list, const TCHAR* jid, int status, const TCHAR* statusMessage )
{
	int j;
	const TCHAR* p, *q;

	EnterCriticalSection( &csLists );
	int i = JabberListExist( list, jid );
	if ( !i ) {
		LeaveCriticalSection( &csLists );
		return 0;
	}
	JABBER_LIST_ITEM* LI = &lists[i-1];

	int bIsNewResource = false;

	if (( p = _tcschr( jid, '@' )) != NULL ) {
		if (( q = _tcschr( p, '/' )) != NULL ) {
			const TCHAR* resource = q+1;
			if ( resource[0] ) {
				JABBER_RESOURCE_STATUS* r = LI->resource;
				for ( j=0; j < LI->resourceCount; j++, r++ ) {
					if ( !_tcscmp( r->resourceName, resource )) {
						// Already exist, update status and statusMessage
						r->status = status;
						replaceStr( r->statusMessage, statusMessage );
						break;
				}	}

				if ( j >= LI->resourceCount ) {
					// Not already exist, add new resource
					LI->resource = ( JABBER_RESOURCE_STATUS * ) mir_realloc( LI->resource, ( LI->resourceCount+1 )*sizeof( JABBER_RESOURCE_STATUS ));
					bIsNewResource = true;
					r = LI->resource + LI->resourceCount++;
					memset( r, 0, sizeof( JABBER_RESOURCE_STATUS ));
					r->status = status;
					r->affiliation = AFFILIATION_NONE;
					r->role = ROLE_NONE;
					r->resourceName = mir_tstrdup( resource );
					if ( statusMessage )
						r->statusMessage = mir_tstrdup( statusMessage );
			}	}
		}
		// No resource, update the main statusMessage
		else replaceStr( LI->statusMessage, statusMessage );
	}

	LeaveCriticalSection( &csLists );
	return bIsNewResource;
}

void JabberListRemoveResource( JABBER_LIST list, const TCHAR* jid )
{
	int j;
	const TCHAR* p, *q;

	EnterCriticalSection( &csLists );
	int i = JabberListExist( list, jid );
	JABBER_LIST_ITEM* LI = &lists[i-1];
	if ( !i || LI == NULL ) {
		LeaveCriticalSection( &csLists );
		return;
	}

	if (( p = _tcschr( jid, '@' )) != NULL ) {
		if (( q = _tcschr( p, '/' )) != NULL ) {
			const TCHAR* resource = q+1;
			if ( resource[0] ) {
				JABBER_RESOURCE_STATUS* r = LI->resource;
				for ( j=0; j < LI->resourceCount; j++, r++ ) {
					if ( !_tcsicmp( r->resourceName, resource ))
						break;
				}
				if ( j < LI->resourceCount ) {
					// Found resource to be removed
					if ( LI->defaultResource == j )
						LI->defaultResource = -1;
					else if ( LI->defaultResource > j )
						LI->defaultResource--;
					if ( r->resourceName ) mir_free( r->resourceName );
					if ( r->statusMessage ) mir_free( r->statusMessage );
					if ( r->software ) mir_free( r->software );
					if ( r->version ) mir_free( r->version );
					if ( r->system ) mir_free( r->system );
					if ( LI->resourceCount-- == 1 ) {
						mir_free( r );
						LI->resource = NULL;
					}
					else {
						memmove( r, r+1, ( LI->resourceCount-j )*sizeof( JABBER_RESOURCE_STATUS ));
						LI->resource = ( JABBER_RESOURCE_STATUS * )mir_realloc( LI->resource, LI->resourceCount*sizeof( JABBER_RESOURCE_STATUS ));
	}	}	}	}	}

	LeaveCriticalSection( &csLists );
}

TCHAR* JabberListGetBestResourceNamePtr( const TCHAR* jid )
{
	TCHAR* res;

	EnterCriticalSection( &csLists );
	int i = JabberListExist( LIST_ROSTER, jid );
	JABBER_LIST_ITEM* LI = &lists[i-1];
	if ( !i || LI == NULL ) {
		LeaveCriticalSection( &csLists );
		return NULL;
	}

	if ( LI->resourceCount == 1 )
		res = LI->resource[0].resourceName;
	else {
		res = NULL;
		if ( LI->resourceMode == RSMODE_MANUAL || LI->resourceMode == RSMODE_LASTSEEN ) {
			if ( LI->defaultResource>=0 && LI->defaultResource < LI->resourceCount )
				res = LI->resource[ LI->defaultResource ].resourceName;
	}	}

	LeaveCriticalSection( &csLists );
	return res;
}

TCHAR* JabberListGetBestClientResourceNamePtr( const TCHAR* jid )
{
	EnterCriticalSection( &csLists );
	int i = JabberListExist( LIST_ROSTER, jid );
	JABBER_LIST_ITEM* LI = &lists[i-1];

	if ( !i || LI == NULL ) {
		LeaveCriticalSection( &csLists );
		return NULL;
	}

	TCHAR* res = JabberListGetBestResourceNamePtr( jid );
	if ( res == NULL ) {
		JABBER_RESOURCE_STATUS* r = LI->resource;
		int status = ID_STATUS_OFFLINE;
		res = NULL;
		for ( i=0; i < LI->resourceCount; i++ ) {
			int s = r[i].status;
			BOOL foundBetter = FALSE;
			switch ( s ) {
			case ID_STATUS_FREECHAT:
				foundBetter = TRUE;
				break;
			case ID_STATUS_ONLINE:
				if ( status != ID_STATUS_FREECHAT )
					foundBetter = TRUE;
				break;
			case ID_STATUS_DND:
				if ( status != ID_STATUS_FREECHAT && status != ID_STATUS_ONLINE )
					foundBetter = TRUE;
				break;
			case ID_STATUS_AWAY:
				if ( status != ID_STATUS_FREECHAT && status != ID_STATUS_ONLINE && status != ID_STATUS_DND )
					foundBetter = TRUE;
				break;
			case ID_STATUS_NA:
				if ( status != ID_STATUS_FREECHAT && status != ID_STATUS_ONLINE && status != ID_STATUS_DND && status != ID_STATUS_AWAY )
					foundBetter = TRUE;
				break;
			}
			if ( foundBetter ) {
				res = r[i].resourceName;
				status = s;
	}	}	}

	LeaveCriticalSection( &csLists );
	return res;
}

int JabberListFindNext( JABBER_LIST list, int fromOffset )
{
	EnterCriticalSection( &csLists );
	int i = ( fromOffset>=0 ) ? fromOffset : 0;
	for( ; i<count; i++ )
		if ( lists[i].list == list ) {
		  	LeaveCriticalSection( &csLists );
			return i;
		}
	LeaveCriticalSection( &csLists );
	return -1;
}

JABBER_LIST_ITEM *JabberListGetItemPtr( JABBER_LIST list, const TCHAR* jid )
{
	EnterCriticalSection( &csLists );
	int i = JabberListExist( list, jid );
	if ( !i ) {
		LeaveCriticalSection( &csLists );
		return NULL;
	}
	i--;
	LeaveCriticalSection( &csLists );
	return &( lists[i] );
}

JABBER_LIST_ITEM *JabberListGetItemPtrFromIndex( int index )
{
	EnterCriticalSection( &csLists );
	if ( index>=0 && index<count ) {
		LeaveCriticalSection( &csLists );
		return &( lists[index] );
	}
	LeaveCriticalSection( &csLists );
	return NULL;
}