/*
* Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies). 
* All rights reserved.
* This component and the accompanying materials are made available
* under the terms of "Eclipse Public License v1.0"
* which accompanies this distribution, and is available
* at the URL "http://www.eclipse.org/legal/epl-v10.html".
*
* Initial Contributors:
* Nokia Corporation - initial contribution.
*
* Contributors:
*
* Description:
*
*/
#include <msvstd.h>
#include <msvfind.h>
#include <msvapi.h>
#include <msvids.h>
#include "mdesession.h"
#include "mdeconstants.h"
#include "mdenamespacedef.h"
#include "messagescannerao.h"
#include "harvesterlog.h"
#include "messagemonitorplugin.h"
#include "clientkeywatcherkeys.h"

CMessageScannerAO* CMessageScannerAO::NewL( CMsvSession* aMsvSession, 
		CMdESession* aMdeSession,
		MMonitorPluginObserver* aMonitorPluginObserver )
	{
	CMessageScannerAO* self = new (ELeave) CMessageScannerAO( 
			aMsvSession, aMdeSession, aMonitorPluginObserver );
	CleanupStack::PushL( self );
	self->ConstructL();
	CleanupStack::Pop( self );
	return self;
	}

CMessageScannerAO::CMessageScannerAO( CMsvSession* aMsvSession, 
		CMdESession* aMdeSession, 
		MMonitorPluginObserver* aMonitorPluginObserver ) :
		CActive( CActive::EPriorityStandard ),
		iFindOperation( NULL ), iObjectQuery( NULL )
	{
	CActiveScheduler::Add( this );
	iMsvSession = aMsvSession;
	iMdeSession = aMdeSession;
	iMonitorPluginObserver = aMonitorPluginObserver;
	iState = EScannerIdle;
	}

CMessageScannerAO::~CMessageScannerAO()
	{
	Cancel();
	Cleanup();
	}

void CMessageScannerAO::StartL()
	{
	WRITELOG("CMessageScannerAO::Start()");
	_LIT( KEmpty, "" );
	iFindOperation = CMsvFindOperation::FindInChildrenL( *iMsvSession, KEmpty, 
			KMsvRootIndexEntryId, KMsvMessagePartNone, iStatus );
	SetActive();
	iState = EScannerRunning;
	WRITELOG("CMessageScannerAO::Start() - ends");
	}
		
void CMessageScannerAO::ConstructL()
	{
	WRITELOG("CMessageScannerAO::ConstructL()");
	User::LeaveIfError( iProperty.Attach( KPSRestoreWatcherCategory,
			KPSRestoreWatcherClientsKey, EOwnerThread ) );
	
	TInt registeredClients = 0;
	iProperty.Get( registeredClients );
	TInt error = iProperty.Set( registeredClients + 1 );
	
	WRITELOG1("CMessageScannerAO::ConstructL() - error: %d", error);
	}

void CMessageScannerAO::RunL()
	{
	WRITELOG("CMessageScannerAO::RunL()");
	if ( iStatus.Int() != KErrNone )
		{
#ifdef _DEBUG
		WRITELOG1("CMessageScannerAO::RunL() . error %d", iStatus.Int());
#endif
		return;
		}
	
	switch ( iState )
		{
		case EScannerRunning:
			{
			QueryAllMessagesL();
			break;
			}
		case EScannerFinished:
			{
			Cleanup();
			break;
			}
		}
	}

void CMessageScannerAO::DoCancel()
	{
	iFindOperation->Cancel();
	iProperty.Cancel();
	}

#ifdef _DEBUG
TInt CMessageScannerAO::RunError( TInt aError )
#else
TInt CMessageScannerAO::RunError( TInt /*aError*/ )
#endif
	{
	WRITELOG1( "CMessageScannerAO::RunError - error: %d", aError );
	return KErrNone;
	}

void CMessageScannerAO::QueryAllMessagesL()
	{
	WRITELOG("CMessageScannerAO::QueryAllMessagesL()");
	CMdENamespaceDef& namespaceDef = iMdeSession->GetDefaultNamespaceDefL();
	CMdEObjectDef& messageDef = namespaceDef.GetObjectDefL( MdeConstants::Message::KMessageObject );
	
	iObjectQuery = iMdeSession->NewObjectQueryL( namespaceDef, messageDef, this );
	iObjectQuery->SetResultMode( EQueryResultModeItem );
	
	iObjectQuery->FindL();
	WRITELOG("CMessageScannerAO::QueryAllMessagesL() - ends");
	}

void CMessageScannerAO::HandleQueryNewResults( CMdEQuery& /*aQuery*/, TInt /*aFirstNewItemIndex*/,
		TInt /*aNewItemCount*/ )
	{
	}

void CMessageScannerAO::HandleQueryCompleted( CMdEQuery& aQuery, TInt /*aError*/ )
	{
	WRITELOG("CMessageScannerAO::HandleQueryCompleted()");
	RArray<TMessage> messages;
	TInt error = KErrNone;
	
	if ( aQuery.Count() > 0 )
		{
#ifdef _DEBUG
		WRITELOG1("CMessageScannerAO::HandleQueryCompleted() - aQuery count %d", aQuery.Count());
#endif
		TRAP( error, ProcessMessagesL( aQuery, messages ) );
		}

	if ( error == KErrNone )
		{
		TRAP_IGNORE( ScanMessagesL( messages ) );
		}
	
	messages.Close();
	Unregister();
	iState = EScannerFinished;
	
	SetActive();
	TRequestStatus* pStatus = &iStatus;
    User::RequestComplete( pStatus, KErrNone );
	}

void CMessageScannerAO::ProcessMessagesL( CMdEQuery& aQuery, RArray<TMessage>& aMessages )
	{
	_LIT( KPeriod, "." );

	CMdENamespaceDef& namespaceDef = iMdeSession->GetDefaultNamespaceDefL();
	CMdEObjectDef& messageDef = namespaceDef.GetObjectDefL( MdeConstants::Message::KMessageObject );
	CMdEPropertyDef& creationDateDef = messageDef.GetPropertyDefL( 
			MdeConstants::Object::KCreationDateProperty );

	TInt count = aQuery.Count();
	
	for (TInt i = 0; i < count; i++ )
		{
		CMdEObject& object = static_cast<CMdEObject&>( aQuery.ResultItem( i ) );
		TInt index = object.Uri().Find( KPeriod );
		if ( index != KErrNotFound )
			{
			TLex16 lex( object.Uri().Left( index ) );
			TMessage message;
			message.iMdeId = object.Id();

			CMdEProperty* creationDateProp = NULL;
			object.Property( creationDateDef, creationDateProp, 0 );
			message.iCreationDate = creationDateProp->TimeValueL();

			TInt error = lex.Val( message.iMsvId );
			if ( error == KErrNone )
				{
				aMessages.Append( message );
				}
			}
		}
	}

void CMessageScannerAO::ScanMessagesL( RArray<TMessage>& aMessages )
	{
	WRITELOG("CMessageScannerAO::ScanMessagesL()");
	const CMsvFindResultSelection& findResults = iFindOperation->GetFindResult();
	CMdENamespaceDef& defaultNamespace = iMdeSession->GetDefaultNamespaceDefL();
	TInt msvCount = findResults.Count();
	
	for ( TInt i = 0; i < msvCount; i++ )
		{
		WRITELOG1("CMessageScannerAO::ScanMessagesL() - MSV item %d", i);
		TBool idMatched = EFalse;
		for ( TInt j = 0; j < aMessages.Count(); j++ )
			{
			WRITELOG1("CMessageScannerAO::ScanMessagesL() - MDE item %d", j);
			if ( findResults[i].iId == aMessages[j].iMsvId )
				{
				WRITELOG("CMessageScannerAO::ScanMessagesL() - id matches");
				idMatched = ETrue;
				CMsvEntry* e = iMsvSession->GetEntryL( findResults[i].iId );
				if ( !e )
					{
					WRITELOG("CMessageScannerAO::ScanMessagesL() - message entry is NULL!?");
					continue;
					}
				
				CleanupStack::PushL( e );
				const TMsvEntry& entry = e->Entry();
				WRITELOG("CMessageScannerAO::ScanMessagesL() - comparing dates");
				if ( entry.iDate != aMessages[j].iCreationDate )
					{
					WRITELOG("CMessageScannerAO::ScanMessagesL() - creation dates don't match");
					// Harvest the message with EHarvesterEdit event type.
					HBufC* uri = CMessageMonitorPlugin::CreateUriL( entry.iMtm, entry.Id() );
					CleanupStack::PushL( uri );
					CHarvesterData* hd = CHarvesterData::NewL( uri );
					CleanupStack::Pop( uri );
					hd->SetEventType( EHarvesterEdit );
					hd->SetTakeSnapshot( EFalse );
					hd->SetBinary( EFalse );
					iMonitorPluginObserver->MonitorEvent( hd );
					}
				
				CleanupStack::PopAndDestroy( e );
				aMessages.Remove( j );
				j--;
				break;
				}
			}
		
		if ( !idMatched )
			{
			WRITELOG("CMessageScannerAO::ScanMessagesL() - harvesting new message");
			// Harvest the message as new message.
			CMsvEntry* e = iMsvSession->GetEntryL( findResults[i].iId );
			if ( !e )
				{
				WRITELOG("CMessageScannerAO::ScanMessagesL() - message entry is NULL!?");
				continue;
				}
			
			CleanupStack::PushL( e );
			const TMsvEntry& entry = e->Entry();
			HBufC* uri = CMessageMonitorPlugin::CreateUriL( entry.iMtm, entry.Id() );
			CleanupStack::PushL( uri );
			CHarvesterData* hd = CHarvesterData::NewL( uri );
			CleanupStack::Pop( uri );
			hd->SetEventType( EHarvesterAdd );
			hd->SetTakeSnapshot( EFalse );
			hd->SetBinary( EFalse );
			iMonitorPluginObserver->MonitorEvent( hd );
			
			CleanupStack::PopAndDestroy( e );
			}
		}
	
	// Remove extra messages in MdS db.
	if ( aMessages.Count() > 0 )
		{
		WRITELOG1("CMessageScannerAO::ScanMessagesL() - removing extra messages from mde (%d)", 
				aMessages.Count() );
		RArray<TItemId> extraMessages;
		CleanupClosePushL( extraMessages );
		
		for ( TInt i = 0; i < aMessages.Count(); i++ )
			{
			extraMessages.Append( aMessages[i].iMdeId );
			}
		
		RArray<TItemId> results;
		CleanupClosePushL( results );
		iMdeSession->RemoveObjectsL( extraMessages, results, &defaultNamespace );
		CleanupStack::PopAndDestroy( 2 ); // extramessages, results
		}
	}

void CMessageScannerAO::Unregister()
	{
	TInt registeredClients = 0;
	iProperty.Get( registeredClients );
	WRITELOG1("CMessageScannerAO::Unregister() - registered clients: %d", registeredClients);
	if ( registeredClients > 0 )
		{
		iProperty.Set( registeredClients - 1 );
		}
	}

void CMessageScannerAO::Cleanup()
	{
	iProperty.Close();
	delete iFindOperation;
	iFindOperation = NULL;
	delete iObjectQuery;
	iObjectQuery = NULL;
	}