/*
* Copyright (c) 2008-2009 Nokia Corporation and/or its subsidiary(-ies).
* All rights reserved.
* This component and the accompanying materials are made available
* under the terms of the License "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: 
* tswisstep.cpp
* UPS test step implementation
*
*/




/**
 @file
*/


#include <e32std.h>
#include "tupsstep.h"
#include "testutilclient.h"
#include <test/testexecutelog.h>
#include <swi/launcher.h>
#include "swi/sisregistryentry.h"
#include "swi/sisregistrysession.h"
#include "swi/sisregistrypackage.h"
#include "cleanuputils.h"

using namespace Swi;
using namespace Swi::Test;

//
// CUpsInstallStep
//

CUpsInstallStep::~CUpsInstallStep()
	{
	delete iUi;
	}
	

CUpsInstallStep::CUpsInstallStep(TInstallType aInstallType)
	: iInstallType(aInstallType)
	{
	// Call base class method to set up the human readable name for logging
	SetTestStepName(KUpsInstallStep);
	}

/**
 * Override of base class virtual. Prepares for the test run of SWIS
 * @return TVerdict code
 */
TVerdict CUpsInstallStep::doTestStepPreambleL()
	{
	// get step parameters
	TPtrC str;
	if (!GetStringFromConfig(ConfigSection(), _L("sis"), str))
		{
		ERR_PRINTF1(_L("Missing SIS file name"));
		SetTestStepResult(EFail);
		}
	else
		{
		iSisFileName.Copy(str);
		if (!GetStringFromConfig(ConfigSection(), _L("script"), str))
			{
			ERR_PRINTF1(_L("Missing XML file name"));
			SetTestStepResult(EFail);
			}
		else
			{
			iXmlFileName.Copy(str);
			INFO_PRINTF3(_L("Installing '%S' using script '%S'"), 
				&iSisFileName, &iXmlFileName);
			// create UI handler with default entries
			iUi = CUIScriptAdaptor::NewL(iXmlFileName, Logger());
			}
		}

	return TestStepResult();
	}
	
/**
 * Override of base class pure virtual
 * Demonstrates reading configuration parameters fom an ini file section
 * @return TVerdict code
 */
TVerdict CUpsInstallStep::doTestStepL()
	{
	INFO_PRINTF1(KUpsInstallStep);

	// launch the installation
	CInstallPrefs* prefs = CInstallPrefs::NewLC();

	TInt err = DoInstallL(*prefs);
	INFO_PRINTF2(_L("DoInstallL returned %d"), err);
	if (err != KErrNone)
		{
		// Can't set error to KErrServerBusy, or Testexecute will retry.
		SetTestStepResult((err != KErrServerBusy)? static_cast<TVerdict>(err) : EFail);
		}
	CleanupStack::PopAndDestroy(prefs);
	return TestStepResult();
	}

TInt CUpsInstallStep::DoInstallL(CInstallPrefs& aInstallPrefs)
	{
	switch (iInstallType)
		{
		case EUseFileName:
			return Launcher::Install(*iUi, iSisFileName, aInstallPrefs);
		}
			
	// Shouldn't get here
	return KErrGeneral;
	}

/**
 * Override of base class virtual
 * @return TVerdict code
 */
TVerdict CUpsInstallStep::doTestStepPostambleL()
	{
/*	if((TestStepResult() == EPass))
		{
		RSisRegistrySession registrySession;
		User::LeaveIfError(registrySession.Connect());
		CleanupClosePushL(registrySession);

		RSisRegistryEntry registryEntry;
		TInt error=registryEntry.Open(registrySession, iUid);
		if(error != KErrNone)
			{
			ERR_PRINTF2(_L("RSisRegistrySession::Open failed error=%d\n"), error);
			SetTestStepResult(EFail);
			}
		else
			{
			CleanupClosePushL(registryEntry);
				
			TInt userDrive =0;
			TRAP(error, userDrive = registryEntry.SelectedDriveL());
			if((error != KErrNone) || (userDrive != iExpectedUserDrive))
				{
				ERR_PRINTF4(_L("SelectedDriveL err=%d, userDrive expected %d got %d\n"),
							error, iExpectedUserDrive, userDrive);
				SetTestStepResult(EFail);
				}
			else
				{
				INFO_PRINTF2(_L("SelectedDriveL userDrive = %d OK\n"), userDrive);
				}
		
			CleanupStack::PopAndDestroy(&registryEntry);
			}
		CleanupStack::PopAndDestroy(&registrySession);
		}*/

	return TestStepResult();
	}

//
// CSwisUninstallStep
//

CUpsUninstallStep::~CUpsUninstallStep()
	{
	delete iUi;
	}

CUpsUninstallStep::CUpsUninstallStep(TUninstallType aType)
	: iType(aType)
	{
	// Call base class method to set up the human readable name for logging
	SetTestStepName(KUpsUnInstallStep);
	}

	
	
	
/**
 * Override of base class virtual. Prepares for the test run of SWIS
 * @return TVerdict code
 */
TVerdict CUpsUninstallStep::doTestStepPreambleL()
	{
	INFO_PRINTF1(_L("CUpsUninstallStep::doTestStepPreambleL"));
	// get step parameters
	
	TPtrC str;
	if (!GetStringFromConfig(ConfigSection(), _L("script"), str))
		{
		ERR_PRINTF1(_L("Missing XML file name"));
		SetTestStepResult(EFail);
		return EFail;
		}
	iXmlFileName.Copy(str);	
	TInt uid=0;

	if (!GetHexFromConfig(ConfigSection(), _L("uid"), uid))
		{
		ERR_PRINTF1(_L("Missing uid"));
		SetTestStepResult(EFail);
		return EFail;
		}

	iUid.iUid=uid;
		
	if (iType == EByPackage)
		{
		TPtrC vendorName;
		if (!GetStringFromConfig(ConfigSection(), _L("vendorName"), vendorName))
			{
			ERR_PRINTF1(_L("Missing Vendor Name"));
			SetTestStepResult(EFail);
			return EFail;
			}
		iVendorName.Set(vendorName);
		
		TPtrC packageName;
		if (!GetStringFromConfig(ConfigSection(), _L("packageName"), packageName))
			{
			ERR_PRINTF1(_L("Missing Package Name"));
			SetTestStepResult(EFail);
			return EFail;
			}
		iPackageName.Set(packageName);
			
		
		INFO_PRINTF5(_L("Uninstalling %D, %S, %S  using script '%S'"), 
			iUid.iUid, &iPackageName, &iVendorName, &iXmlFileName);
		// create UI handler with default entries
		iUi = CUIScriptAdaptor::NewL(iXmlFileName, Logger());	
		}
	else if (iType== EByUid)
		{
		INFO_PRINTF3(_L("Uninstalling '%D' using script '%S'"), 
			iUid.iUid, &iXmlFileName);
		// create UI handler with default entries
		iUi = CUIScriptAdaptor::NewL(iXmlFileName, Logger());	
		}
	
	return TestStepResult();
	}

/**
 * Override of base class pure virtual
 * Demonstrates reading configuration parameters fom an ini file section
 * @return TVerdict code
 */

TInt CUpsUninstallStep::DoUninstallL()
	{
	TInt error = KErrNotFound;

	if (iType == EByUid)
		{
		// launch the installation
		return Launcher::Uninstall(*iUi, iUid); 
		}
	else if (iType == EByPackage)
		{
		
		TInt err = KErrNotFound;
		// Go through list of packages from base package to get augmentations.
		CSisRegistryPackage* uninstallPackage=CSisRegistryPackage::NewLC(iUid, iPackageName, iVendorName);
		
		RSisRegistrySession registrySession;
		User::LeaveIfError(registrySession.Connect());
		CleanupClosePushL(registrySession);

		RSisRegistryEntry registryEntry;
	
		User::LeaveIfError(registryEntry.Open(registrySession, iUid));
		CleanupClosePushL(registryEntry);

		CSisRegistryPackage* package=registryEntry.PackageL();
		CleanupStack::PushL(package);
		
		if (*package == *uninstallPackage)
			{
				err = -1;
				err=Launcher::Uninstall(*iUi, *package); 
			}
		else
			{
			// check augmenations
			RPointerArray<CSisRegistryPackage> augmentationPackages;
			CleanupResetAndDestroy<RPointerArray<CSisRegistryPackage> >::PushL(augmentationPackages);

			registryEntry.AugmentationsL(augmentationPackages);
			for (TInt i=0; i < augmentationPackages.Count(); ++i)
				{
				if (*augmentationPackages[i] == *uninstallPackage)
					{
					err=User::LeaveIfError(Launcher::Uninstall(*iUi, *augmentationPackages[i])); 
					break;
					}
				}
			CleanupStack::PopAndDestroy(&augmentationPackages);
			}	
		
		CleanupStack::PopAndDestroy(3, &registrySession);
		CleanupStack::PopAndDestroy(uninstallPackage);
		error = err;
		}
	return error;
	}

TVerdict CUpsUninstallStep::doTestStepL()
	{
	INFO_PRINTF1(KUpsUnInstallStep);

	TInt err = DoUninstallL();
	INFO_PRINTF2(_L("DoUninstallL returned %d"), err);
	if (err != KErrNone)
		{
		// Can't set error to KErrServerBusy, or Testexecute will retry.
		SetTestStepResult((err != KErrServerBusy)? static_cast<TVerdict>(err) : EFail);
		}
	return TestStepResult();
	}

/**
 * Override of base class virtual
 * @return TVerdict code
 */
TVerdict CUpsUninstallStep::doTestStepPostambleL()
	{
	return TestStepResult();
	}

// End of file