/*

* Copyright (c) 2004-2007 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 <imageprintapp.rsg>

#include <aknlists.h>

#include <aknPopup.h>

#include <StringLoader.h>

#include <AknQueryDialog.h>

#include <aknPopupHeadingPane.h>

#include <badesca.h>

#include <eikenv.h>

#include <StringLoader.h>

#include <AknWaitDialog.h>

#include <AknIconUtils.h>

#include <AknGlobalMsgQuery.h>      // global message query

#include <apgcli.h>                 // rapalssession

#include <centralrepository.h>      // repository

#include <aknmessagequerydialog.h>

#include <bluetooth/hci/hcierrors.h>

#include <imgprintbitmaps.mbg>

#include <upnpsettings.h>

#include <upnprunsetupcommand.h>

#include <upnpshowcommand.h>

#include "cdiscoverydlgmanager.h"

#include "imageprint.h"

#include "mdiscovery.h"

#include "mprintsettings.h"

#include "ciffactory.h"

#include "mprintsettings.h"

#include "imgpprintapputil.h"

#include "cimageprintappui.h"

#include "imageprintconsts.h"

#include "cimageprintdoc.h"

#include "cimageprintengine.h"

#include "tprinter.h"

#include "clog.h"

const TInt KArrayGranularity( 5 );

const TInt KSearchAgainUID ( -1 );

_LIT( KEmptyIconIndex, "0" );

_LIT( KCachedIconIndex, "1" );

_LIT( KBTIconIndex, "2" );

_LIT( KKioskIconIndex, "3" );

_LIT( KMMCIconIndex, "4" );

_LIT( KUSBIconIndex, "5" );

_LIT( KIconFile, "z:\\resource\\apps\\imgprintbitmaps.mif" );

const TUint KSpaceForSingleIcon( 2 ); // icon index + tab = 2 chars

const TUint KSpaceForBothIcons( 4 ); // icon index + tab + cache index + tab = 4 chars

const TInt KNumberOfIcons( 7 );

_LIT( KWLANIconIndex, "6" );

void CleanUpResetAndDestroy(TAny* aArray)

	{

	if(aArray)

		{

		CArrayPtrFlat<TDesC>* array=(CArrayPtrFlat<TDesC>*)aArray;

		array->ResetAndDestroy();

		delete array;

		}

	}

// CONSTRUCTION

CDiscoveryDlgManager* CDiscoveryDlgManager::NewL(

		CIFFactory& aIFFactory )

	{

	CDiscoveryDlgManager* self = NewLC( aIFFactory );

	CleanupStack::Pop(self);

	return self;

	}

CDiscoveryDlgManager* CDiscoveryDlgManager::NewLC(

		CIFFactory& aIFFactory )

	{

	CDiscoveryDlgManager* self =

	new (ELeave) CDiscoveryDlgManager ();

	CleanupStack::PushL(self);

	self->ConstructL( aIFFactory );

	return self;

	}

// Constructor

CDiscoveryDlgManager::CDiscoveryDlgManager()

	{

	}

// 2nd phase constructor

void CDiscoveryDlgManager::ConstructL( CIFFactory& aIFFactory )

	{

	iUidArray = new ( ELeave ) CArrayFixFlat<TInt>( KArrayGranularity );

	iBrandArray = new ( ELeave ) CArrayFixFlat<TUint>( KArrayGranularity );

	iProtocolArray = new ( ELeave ) CArrayFixFlat<TPrinterType>( KArrayGranularity );

	iListOfDeviceNames = new ( ELeave ) CDesCArrayFlat( KArrayGranularity );

	iActivePopup = EFalse;  // Popuplist is not activated

	iNameArray = new (ELeave) CDesCArrayFlat( KArrayGranularity );

	iDiscovery = aIFFactory.DiscoveryIF();

	iSettings = aIFFactory.SettingsIF();

	iSelectDeviceText = StringLoader::LoadL( R_QTN_POPUP_LIST_TITLE );

	iSearchingText = StringLoader::LoadL( R_QTN_POPUP_LIST_TITLE2 );

	iActiveTitleText.Set( iSearchingText->Des() );

	}

// Destructor

CDiscoveryDlgManager::~CDiscoveryDlgManager()

	{

	if ( iDiscovery )

		{

		iDiscovery->RemoveDiscoveryObserver();

		}

	if ( iActivePopup )

		{

		if ( iPopupList )

			{

			iPopupList->CancelPopup();

			}

		}

	delete iListBox;

	delete iUidArray;

	delete iBrandArray;

	delete iProtocolArray;

	delete iListOfDeviceNames;

	if ( iNameArray )

		{

		iNameArray->Reset();

		}

	delete iNameArray;

	delete iPrinterName;

	delete iSearchingText;

	delete iSelectDeviceText;

	}

// Called when new device is found

void CDiscoveryDlgManager::NotifyNewPrintDeviceL(

		HBufC* aText,

		TInt aUid,

		TBool aCached,

		TPrinterType aType,

		TUint aVendor )

	{

	LOG("CDiscoveryDlgManager::NotifyNewPrintDevice BEGIN");

	if( iHiddenDiscovery )

		{

		// multi-device hidden discovery (bt/wlan) handled differently

		if ( iMultiDiscovery )

			{

			LOG("CDiscoveryDlgManager::NotifyNewPrintDevice doing hidden multi-discovery");

			TInt uid = iSettings->GetDefaultPrintID();

			LOG2("CDiscoveryDlgManager::NotifyNewPrintDevice found: %d default: %d", aUid, uid);

			if ( uid == aUid )

				{

				if( iHiddenDiscoveryDialog )

					{

					TRAP_IGNORE( iHiddenDiscoveryDialog->ProcessFinishedL() );

					iHiddenDiscoveryDialog = NULL;

					}

				TInt printerErrorCode(0);

				printerErrorCode = iDiscovery->ChoosePrinterL( aUid );

				iSettings->SetCurrentPrinterProtocol( aType );

				iSettings->SetCurrentPrinterVendor( aVendor );

				iSettings->SetCurrentPrinterName( *aText );

				if( printerErrorCode == 0 )

					{

					iPrinterFound = ETrue;

					iDiscoveryCompleted = ETrue;

					delete iPrinterName;

					iPrinterName = NULL;

					// Cancel the discovery when device is found printjob is done.

					iDiscovery->CancelDeviceDiscoveryL();

					iPrinterName = aText->AllocL();

					}

				else

					{

					iPrinterFound = EFalse;

					}

				}

			}

		// USB and MMC hidden discovery only find one printer or not a printer at all

		else

			{

			if( iHiddenDiscoveryDialog )

				{

				iHiddenDiscoveryDialog->ProcessFinishedL();

				iHiddenDiscoveryDialog = NULL;

				}

			TInt printerErrorCode(0);

			printerErrorCode = iDiscovery->ChoosePrinterL( aUid );

			iSettings->SetCurrentPrinterProtocol( aType );

			iSettings->SetCurrentPrinterVendor( aVendor );            

			if( printerErrorCode == 0 )

				{

				iPrinterFound = ETrue;

				iDiscoveryCompleted = ETrue;

				delete iPrinterName;

				iPrinterName = 0;

				if( aType == MDiscoveryObserver::EMMC )

					{

					HBufC* memoryCard = 0;

					memoryCard = StringLoader::LoadLC

					( R_QTN_PRINT_MEMORY_CARD );

					iSettings->SetCurrentPrinterName( *memoryCard );

					iPrinterName = memoryCard->AllocL();                    

					CleanupStack::PopAndDestroy( memoryCard );

					}

				else

					{

					iSettings->SetCurrentPrinterName( *aText );

					iPrinterName = aText->AllocL();

					}

				}

			else

				{

				iPrinterFound = EFalse;

				}

			}

		}

	else

		{

		if ( aType == MDiscoveryObserver::EUSB )

			{

			iUsbPrinterID = aUid;

			}

		TRAPD( err, AddNewDeviceToListL( aText, aUid, aCached, aType, aVendor ) );

		if ( err )

			{

			LOG1("CDiscoveryDlgManager::NotifyNewPrintDevice device adding leaved with: %d", err );

			// Nothing to do with the error code coming from ShowErrorMsgL

			TRAP_IGNORE( IMGPPrintAppUtil::ShowErrorMsgL( err ) );

			}

		}

	LOG("CDiscoveryDlgManager::NotifyNewPrintDevice END");

	}

// Called when error is found

void CDiscoveryDlgManager::DiscoveryError(

		TInt aErrCode )

	{

	if ( aErrCode == ENoConnection )

		{		

		CancelDiscovery();

		}

	}

// Called when process completed

void CDiscoveryDlgManager::DeviceDiscoveryCompleted()

	{

	LOG1("[CDiscoveryDlgManager::DeviceDiscoveryCompleted]\t Before dissmissing the process/wait dialog, iDiscoveryCompleted has value of %d", iDiscoveryCompleted);

	// // Added this to avoid multiple "Search Again" texts, because server sends DeviceDiscoveryCompleted() more than once 

	if( iDiscoveryCompleted )

		{

		return;

		}

	// Searching over, flag is essential

	iSearchingCancelled = ETrue;

	iActiveTitleText.Set( iSelectDeviceText->Des() );

	// Set the title of the pop-up list

	if ( iPopupList )

		{

		TRAP_IGNORE( iPopupList->SetTitleL( iActiveTitleText ) );

		}

	LOG("CDiscoveryDlgManager::DeviceDiscoveryCompleted BEGIN");

	if( iHiddenDiscoveryDialog )

		{

		TRAP_IGNORE( iHiddenDiscoveryDialog->ProcessFinishedL() );

		iHiddenDiscoveryDialog = NULL;

		}

	iDiscoveryCompleted = ETrue;

	if ( iActivePopup )

		{

		iPopupList->Heading()->CancelAnimation();

		TRAP_IGNORE( RemoveCachedIconsL() );

		TRAP_IGNORE( AddSearchAgainTextL() );

		if ( iPopupList )

			{

			TRAP_IGNORE( iPopupList->ButtonGroupContainer()->SetCommandSetL(

					R_AVKON_SOFTKEYS_SELECT_CANCEL );

			iPopupList->ButtonGroupContainer()->DrawNow() );

			}

		}

	if ( iCancelling )

		{

		iCancelling = EFalse;

		TRAP_IGNORE( iCancellingDialog->ProcessFinishedL() );

		iCancellingDialog = NULL;

		}

	LOG("CDiscoveryDlgManager::DeviceDiscoveryCompleted END");

	}

// Displays the printer devices

TBool CDiscoveryDlgManager::StartDlgManagerL( TInt aProtocol, TBool aHidden )

	{

	LOG2("CDiscoveryDlgManager::StartDlgManagerL BEGIN: protocol %d, hidden: %d", aProtocol, aHidden);

	iDiscoveryCompleted = EFalse;

	TBool okToContinue( ETrue );

	iHiddenDiscovery = aHidden;

	// Start the device discovery with all the supported protocols

	if( aProtocol == 0 )

		{

		iSupportedProtocols = iDiscovery->SupportedProtocols();

		// If discovering WLAN, AP must be defined before continuing

		if ( iSupportedProtocols & KImagePrint_PrinterProtocol_UPnP )

			{

			LOG("CDiscoveryDlgManager::StartDlgManagerL: TryWlanDiscoveryL");

			TryDeviceDiscoveryL( iSupportedProtocols, EFalse );

			}

		// Discovery without WLAN    

		else

			{

			LOG("CDiscoveryDlgManager::StartDlgManagerL: StartDeviceDiscoveryL");

			iDiscovery->StartDeviceDiscoveryL( this, iSupportedProtocols );

			}

		}

	else

		{

		iSupportedProtocols = aProtocol;

		// If discovering WLAN

		if ( iSupportedProtocols & KImagePrint_PrinterProtocol_UPnP )

			{

			LOG("CDiscoveryDlgManager::StartDlgManagerL - discovering WLAN");

			if ( !TryDeviceDiscoveryL( aProtocol, ETrue ) )

				{

				LOG("CDiscoveryDlgManager::StartDlgManagerL - discovering WLAN failed");

				// If discovering failed, do not continue

				okToContinue = EFalse;

				return okToContinue;

				}

			}

		else

			{

			LOG("iDiscovery->StartDeviceDiscoveryL START");

			iDiscovery->StartDeviceDiscoveryL( this, aProtocol );

			} 

		LOG("iDiscovery->StartDeviceDiscoveryL END");

		}

	// BT & WLAN cases can find multiple devices which has to be handled

	if ( aProtocol & KImagePrint_PrinterProtocol_BPP ||

			aProtocol & KImagePrint_PrinterProtocol_BIP ||

			aProtocol & KImagePrint_PrinterProtocol_OPP_Printer ||

			aProtocol & KImagePrint_PrinterProtocol_OPP_PC ||

			aProtocol & KImagePrint_PrinterProtocol_UPnP )

		{

		iMultiDiscovery = ETrue;

		}

	if( iHiddenDiscovery )

		{

		okToContinue = StartHiddenDiscoveryL();

		}

	else

		{

		okToContinue = StartNormalDiscoveryL();

		}

	LOG1("CDiscoveryDlgManager::StartDlgManagerL END return: %d", okToContinue);

	return okToContinue;

	}

TBool CDiscoveryDlgManager::StartNormalDiscoveryL()

	{

	iActiveTitleText.Set( iSearchingText->Des() );

	LOG("CDiscoveryDlgManager::StartNormalDiscoveryL BEGIN");

	TInt printerErrorCode = 0;

	TBool continueSearch( ETrue );

	TBool okToContinue( ETrue );

	iSearchingCancelled = EFalse; // ADDED

	TInt uid; // Printer uid

	// Loop until printer is selected or user decides to exit

	while ( continueSearch )

		{

		// Create pop-up list

		if ( !iPopupList )

			{

			iPopupList = CreatePopupListL();

			}

		// Show the pop-up list

		if ( iListOfDeviceNames->Count() > 0 )

			{

			iPopupList->ButtonGroupContainer()->SetCommandSetL(

					R_AVKON_SOFTKEYS_STOP

			);

			}

		// Launch the pop-up list for found devices

		LOG("CDiscoveryDlgManager::StartNormalDiscoveryL before launch popuplist");

		TBool selected = iPopupList->ExecuteLD();

		LOG1("CDiscoveryDlgManager::StartNormalDiscoveryL after popupList, selected: %d", selected);

		iPopupList = NULL;

		iActivePopup = EFalse;

		LOG("CDiscoveryDlgManager::StartNormalDiscoveryL returned from popup");

		TInt index = iListBox->CurrentItemIndex();

		// Search again was selected

		if ( selected && iUidArray->At( index ) == KSearchAgainUID )

			{

			RestartDiscoveryL();

			}

		// Device was selected

		else if ( selected )

			{

			uid = iUidArray->At( index );

			if ( uid == iUsbPrinterID )

				{

				iUSBPrinterSelected = ETrue ;

				}

			else

				{

				iUSBPrinterSelected = EFalse;

				}

			iCurrentPrinterUid = uid;

			printerErrorCode = iDiscovery->ChoosePrinterL( uid );

			if ( printerErrorCode != KErrNone )

				{

				IMGPPrintAppUtil::ShowErrorMsgL( printerErrorCode  );

				User::Leave( printerErrorCode );

				}

			else

				{

				if ( iUidArray->At( 0 ) == KSearchAgainUID )

					{

					iSettings->SetCurrentPrinterProtocol( iProtocolArray->At( index - 1 ) );

					iSettings->SetCurrentPrinterVendor( iBrandArray->At( index - 1 ) );

					iSettings->SetCurrentPrinterName( (*iNameArray)[index-1] );

					}

				else

					{

					iSettings->SetCurrentPrinterProtocol( iProtocolArray->At( index ) );

					iSettings->SetCurrentPrinterVendor( iBrandArray->At( index ));

					iSettings->SetCurrentPrinterName( (*iNameArray)[index] );

					}

				continueSearch = EFalse;

				// If there wasn't default protocol then set selected printer and protocol as default

				if ( iSettings->GetDefaultPrinterProtocol() == KErrNotFound )

					{

					LOG1("CDiscoveryDlgManager::StartNormalDiscoveryL default protocol wasn't set, setting uid: %d", uid );

					iSettings->SetDefaultPrinterID( uid );

					// Discovery has completed and 'Search again' was added as a first item

					// access protocol array with index-1

					if ( iUidArray->At( 0 ) == KSearchAgainUID )

						{

						iSettings->SetDefaultPrinterProtocol( iProtocolArray->At( index-1 ) );

						iSettings->SetDefaultPrinterNameL( (*iNameArray)[index-1] );

						}

					else

						{

						iSettings->SetDefaultPrinterProtocol( iProtocolArray->At( index ) );

						iSettings->SetDefaultPrinterNameL( (*iNameArray)[index] );

						}

					}

				else

					{

					LOG("CDiscoveryDlgManager::StartNormalDiscoveryL default protocol was already set" );

					}

				}

			TPtrC name = iListOfDeviceNames->MdcaPoint( index );

			delete iPrinterName;

			iPrinterName = NULL;

			iPrinterName = name.AllocL();

			}

		// Stop/Cancel was selected

		else

			{

			LOG("CDiscoveryDlgManager::StartNormalDiscoveryL Stop/Cancel was selected" );

			// Searching cancelled by user, cancel device discovery

			if ( !iSearchingCancelled )

				{

				LOG("CDiscoveryDlgManager::StartNormalDiscoveryL iSearchingCancelled EFalse" );

				iSearchingCancelled = ETrue;

				iDiscovery->CancelDeviceDiscoveryL();

				}

			// User cancelled the whole discovery dialog

			else

				{

				LOG("CDiscoveryDlgManager::StartNormalDiscoveryL iSearchingCancelled ETrue" );

				continueSearch = EFalse;

				okToContinue = EFalse;

				iCancelling = ETrue;

				// User cancelled the discovery dialog before all the printers were found

				if ( !iDiscoveryCompleted )

					{

					//ShowCancelWaitNoteL();

					}

				}

			}

		}

	LOG("CDiscoveryDlgManager::StartNormalDiscoveryL END");

	return okToContinue;

	}

TBool CDiscoveryDlgManager::StartHiddenDiscoveryL()

	{

	LOG("CDiscoveryDlgManager::StartHiddenDiscoveryL BEGIN");

	iHiddenDiscoveryDialog =