// Copyright (c) 2005-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 <e32base.h>
#include <http.h>
#include <escapeutils.h>

#include "testwebbrowser.h"
#include "httpexampleutils.h"


const TInt KMaxURISize = 256;
const TInt KMaxFilePathLength = 512;
const TInt KHttpSchemeLength = 7;

_LIT ( KDirPath, "c:\\TestWebBrowser\\");
_LIT ( KDefaultFilePath, "index.html" );
_LIT ( KQuit, "quit" );
_LIT ( KSelectOption, " Select an option \n\n" );
_LIT ( KPossibleSelectionsText, " 1 Download URL \n 2 Set Proxy\n 3 Quit \n" );
_LIT ( KPossibleSelections,"123" );
_LIT ( KEnterUrl, "Enter URL: " );
_LIT ( KEnterProxy, "Enter Proxy Address: " );	
		
// Invalid characters that can appear in a URL. These values must be replaced
// with a valid character to generate the full file path.
_LIT ( KInvalidChars, "<>:\"|*?");

const TUint KReplaceChar = '_';

void ChangePathSeparator( TDes& aDesPtr, TUint aPathSeperatorFrom, TUint aPathSeperatorTo )
	{
	for( TInt offset = aDesPtr.Length() - 1; offset >= 0; offset-- )
		{
		if ( aDesPtr[offset] == aPathSeperatorFrom )
			{	
				aDesPtr[offset] = TUint16( aPathSeperatorTo );
			}
		}
	}

CTestWebBrowser::CTestWebBrowser ( CHttpExampleUtils& aUtils )
: CActive ( EPriorityStandard ),
iTestUtils ( aUtils ),
iCurrentState ( EMainMenu )
	{
	//Add this active object to the active schedeler
	CActiveScheduler::Add(this);
	__ASSERT_DEBUG( !IsActive(),User::Panic(_L("Reader"), KErrCorrupt) );
	
	// Wait for request to be handled.
	IssueAndCompleteRequest();		
	}


CTestWebBrowser::~CTestWebBrowser ()
	{
	// Close the HTTP session.
	iHttpSession.Close();
	
	iTransArray.ResetAndDestroy ();
	iTransArray.Close ();
	
	delete iBaseUri;
	}
	
CTestWebBrowser* CTestWebBrowser::NewLC ( CHttpExampleUtils& aUtils )	
	{
	CTestWebBrowser* me = new (ELeave) CTestWebBrowser ( aUtils );
	CleanupStack::PushL ( me );
	me->ConstructL ();
	return me;
	}

void CTestWebBrowser::ConstructL ()
	{
	// Open a HTTP session.
	iHttpSession.OpenL ();
	
	// Install the callback into the session.
	InstallAuthenticationL ( iHttpSession );
	
	iBaseUri = HBufC8::NewL ( KMaxURISize );
	}

CTestWebBrowser* CTestWebBrowser::NewL ( CHttpExampleUtils& aUtils )	
	{
	CTestWebBrowser* me = CTestWebBrowser::NewLC ( aUtils );
	CleanupStack::Pop ( me );
	return me;
	}

/**
	Set HTTP proxy information for the session.
	
	@param aProxyAddr [in] Proxy address to set for the session
 */	
void CTestWebBrowser::SetProxyL ( const TDesC8& aProxyAddr )
	{
	
	RHTTPConnectionInfo connInfo = iHttpSession.ConnectionInfo();
	
	// Set the proxy usage property.
	THTTPHdrVal proxyUsage( iHttpSession.StringPool().StringF( HTTP::EUseProxy, RHTTPSession::GetTable() ) );
	connInfo.SetPropertyL( iHttpSession.StringPool().StringF( HTTP::EProxyUsage, RHTTPSession::GetTable() ), proxyUsage );
	
	RStringF proxyAddr = iHttpSession.StringPool ().OpenFStringL ( aProxyAddr );
	CleanupClosePushL ( proxyAddr );
	
	// Set the proxy address property.
	THTTPHdrVal proxyAddrHdr( proxyAddr );
    connInfo.SetPropertyL( iHttpSession.StringPool().StringF( HTTP::EProxyAddress, RHTTPSession::GetTable() ), proxyAddrHdr );
    
    CleanupStack::PopAndDestroy (); // Pop and destroy proxyAddr
	return;		
	}

/**
	Get the credentials from the user.
	
	@param aURI [in] The URI being requested
	@param aRealm [out] The realm being requested
	@param aAuthenticationType [out] The type of authentication.
	@param aUsername [out] User name
	@param aPassword [out] Pass word
	
	@return TBool ETrue if credentials being returned else EFalse
 */		
TBool CTestWebBrowser::GetCredentialsL( const TUriC8& aURI, 
										RString aRealm, 
									   	RStringF aAuthenticationType,
									   	RString& aUsername, 
									   	RString& aPassword )
	{
	// Convert to 16 bit to display
	HBufC* uriDesBuf = HBufC::NewLC( aURI.UriDes().Length() );
	TPtr uriDesPtr( uriDesBuf->Des() );
	uriDesPtr.Copy( aURI.UriDes() );

	HBufC* uriRealmBuf = HBufC::NewLC( aRealm.DesC().Length() );
	TPtr uriRealmPtr( uriRealmBuf->Des() );
	uriRealmPtr.Copy( aRealm.DesC() );
	
	HBufC* uriAuthenticationType = HBufC::NewLC ( aAuthenticationType.DesC().Length() );
	TPtr uriAuthenticationPtr( uriAuthenticationType->Des() );
	uriAuthenticationPtr.Copy ( aAuthenticationType.DesC() );
	
	
	// Prompt user for input
	iTestUtils.Test().Printf ( _L( "Enter credentials for URL %S, realm %S\n"), &uriDesPtr, &uriRealmPtr );
	iTestUtils.Test().Printf ( _L("Using %S authentication\n"), &uriAuthenticationPtr );
	CleanupStack::PopAndDestroy ( 3 ); // Pop and destroy uriDesBuf, uriRealmBuf and uriAuthenticationType
	
	HBufC* userDetails16 = HBufC::NewLC ( KMaxUserEntrySize );
	HBufC8* userDetails8 = HBufC8::NewLC ( KMaxUserEntrySize );
	TPtr userDetailsPtr16( userDetails16->Des() );
	TPtr8 userDetailsPtr8( userDetails8->Des() );
	
	iTestUtils.GetAnEntry ( _L( "Username (or QUIT to give up): " ), userDetailsPtr16 );
	TBool set = EFalse;
	if ( userDetailsPtr16.CompareF ( KQuit ) )
		{
		userDetailsPtr8.Copy ( userDetailsPtr16 );
		aUsername = aRealm.Pool().OpenStringL ( userDetailsPtr8 );
		iTestUtils.GetAnEntry( _L( "Password: " ), userDetailsPtr16 );
		userDetailsPtr8.Copy ( userDetailsPtr16 );
		aPassword = aRealm.Pool().OpenStringL ( userDetailsPtr8 );
		set = ETrue;		
		}
	CleanupStack::PopAndDestroy ( 2 ); // Pop and destroy userDetails16 & userDetails8
	return set;	
	}

/**
	Create a new transaction. Generate the full uri from a partial uri.
	Also, generates the file path where the contents to be saved. Adds the transaction
	object into array.
	
	@param aUri [in] URI to download.
	@param aParseHtml [in] Parse the downloaded HTML content. Default value: EFalse. ETrue value to parse 
	the HTML content. Typically the value will be set to ETrue only for the main page.
 */
void CTestWebBrowser::CreateTransactionL ( const TDesC8& aUri, TBool aParseHtml /* = EFalse */ )
	{	
	TUriParser8 baseUri;	
	baseUri.Parse ( iBaseUri->Des() );
	
	TUriParser8 refUri;
	refUri.Parse ( aUri );

	// Resolve uri
	CUri8* uriPtr = CUri8::ResolveL( baseUri, refUri );	
	CleanupStack::PushL ( uriPtr );
	
	// Generate the file path from the full http uri.	
	HBufC* filePath = GenerateFilePathL ( uriPtr->Uri() );
	CleanupStack::PushL ( filePath );
	
	// Create a new HTTP transaction object.
	CBrowserTransaction* newTrans = CBrowserTransaction::NewL ( iHttpSession, iTestUtils, this, *filePath, aParseHtml );
	CleanupStack::PopAndDestroy ( filePath );
	CleanupStack::PushL ( newTrans );
	iTransArray.AppendL ( newTrans );
	
	// Open and submit the transaction.
	newTrans->CreateTransactionL ( uriPtr->Uri().UriDes() );
	newTrans->StartTransactionL ();
	
	CleanupStack::Pop ( newTrans ); 
	CleanupStack::PopAndDestroy ( uriPtr );
	return;		
	}

/**
	To create a new transaction. This function is a calllback called from 
	the CHTMLHandler when it finds a new URI to be downloaded.
	
	@param aUri [in] URI to download.
 */	
void CTestWebBrowser::OnTransactionCreateL ( const TDesC8& aUri, TBool aParseHtml )
	{
	CreateTransactionL ( aUri, aParseHtml );
	return;		
	}

/**
	Delete the transaction. Removes the transaction object from the transaction array
	
	@param aTrans [out] Transaction object to close.
 */	
void CTestWebBrowser::OnTransactionClose ( CBrowserTransaction* aTrans ) 
	{	
	// Find the transaction object to be deleted and remove it from the 
	// transaction	array.
	TInt numTrans = iTransArray.Count ();
	for ( TInt i = 0; i < numTrans; ++i )
		{
		if ( iTransArray[ i ] == aTrans )			
			{
			delete aTrans;
			iTransArray.Remove ( i );
			break;
			}
		}	
	
	if ( iTransArray.Count () == 0 )
		{
		// All transactions are completed. 
		iCurrentState = ETransactionCompleted;
		IssueAndCompleteRequest ();
		}
	return;		
	}

/**
	Issues a new request if the active object has no outstanding request.
	
 */	
void CTestWebBrowser::IssueAndCompleteRequest()
	{
	if ( !IsActive() )		
		{
		TRequestStatus* status = &iStatus;
		User::RequestComplete( status, KErrNone );
		SetActive();
		}
	}
	
void CTestWebBrowser::RunL ()
	{
	
	switch ( iCurrentState )
		{
		case EMainMenu:
			iTestUtils.Test().Printf ( _L ( "Main menu") );
			ExecuteMainMenuL();
			break;
		
		case EEnterUrl:
			AcceptUrlL ();
			break;
		
		case ESelectProxy:
			AcceptProxyL ();
			break;
		
		case EStop:		
			CActiveScheduler::Stop();
			break;
		
		case EStartTransaction:
			{		
			CreateTransactionL ( iBaseUri->Des(), ETrue );	
			}
			break;
		
		case ETransactionStarted:
			{
			iTestUtils.Test().Printf ( _L ( "Transaction started. Please wait...") );			
			}
			break;
		
		case ETransactionCompleted:
			{
			OnTransactionCompleted();
			}
			break;				
			
		default:
			break;
		}
	}

void CTestWebBrowser::DoCancel ()
	{
	// Do nothing...		
	}

/**
	Checks whether a uri is valid or not. Will check whether the uri contains
	scheme, host and path.
	
	@param aUri [in] URI to validate. Only scheme and host presence is validated.
	
	@return TBool ETrue if the URI is successfully validated else EFalse.
 */	
TBool CTestWebBrowser::ValidateUri ( const TDesC8& aUri )
	{
	TUriParser8 uriParser;
	if ( uriParser.Parse ( aUri ) != KErrNone )
		{
		return EFalse;			
		}
	
	// check for scheme
	if ( !uriParser.IsPresent ( EUriScheme ) )
		{
		return EFalse;				
		}
	
	// check for host
	if ( !uriParser.IsPresent ( EUriHost) )
		{
		return EFalse;			
		}
	return ETrue;				
	}

/**
	Generate the full file path based on the aUri.
	
	@param aUri [in] Full URI to generate the full file path.
	
	@return HBufC* Holds the full file path.
 */	
HBufC* CTestWebBrowser::GenerateFilePathL ( const TUriC8& aUri )
	{
	
	HBufC* filePath = HBufC::NewLC ( KMaxFilePathLength );	
	TPtr filePtr ( filePath->Des() );	
	
	// Append the base path
	filePtr.Append ( KDirPath ); 
	
	// Get the full uri.
	HBufC* completeUri = aUri.DisplayFormL();
	CleanupStack::PushL ( completeUri );
	
	TPtr ptr = completeUri->Des();
	// Chop off the scheme. ( http:// )
	ptr.Delete ( 0, KHttpSchemeLength );
	
	// Reverse the slashes.
	ChangePathSeparator ( ptr, '/', '\\' );
	
	// Replace invalid characters with '_'. Invalid characters: <>:"|* 	
	TPtrC ptrInvalidChars ( KInvalidChars );
	for ( TInt i = 0; i < KInvalidChars().Length(); ++i )
		{
		ChangePathSeparator ( ptr, ptrInvalidChars[i], KReplaceChar );					
		}
	
	// Append the URI file path.	
	filePtr.Append ( ptr );		
	CleanupStack::PopAndDestroy ( completeUri );
	
	HBufC* uriTail = aUri.GetFileNameL ( EUriFileNameTail );
	CleanupStack::PushL ( uriTail );	
	
	TBool fileExtPresent = uriTail->Des().Length() != 0;
	
	CleanupStack::PopAndDestroy ( uriTail );
			
	// Append a '\\' at the end of the completeUri if not present.
	TInt len = filePtr.Length();
	
	if ( ( !fileExtPresent ) && ( len != 0 ) && ( filePtr[ len-1 ] != '\\') )
		{
		filePtr.Append ( '\\' );		
		}

	if ( !fileExtPresent )
		{
		// No file name present in the uri. Set the file name as index.html.		
		filePtr.Append ( KDefaultFilePath() );
		}	

	CleanupStack::Pop ( filePath );
	iTestUtils.LogIt ( _L("File Path: %S"), &filePtr );
	return filePath;
	}
		
void CTestWebBrowser::ExecuteMainMenuL ()
	{
	iTestUtils.Test().Console()->ClearScreen ();

	iTestUtils.Test().Printf(KSelectOption);		
	
	
	TInt selection = iTestUtils.GetSelection ( KPossibleSelectionsText, KPossibleSelections );	
	
	switch ( selection )
		{
		case EDownloadURL:
		iCurrentState = EEnterUrl;
		break;
		
		case ESetProxy:
		iCurrentState = ESelectProxy;
		break;
		
		case EQuit:
		iCurrentState = EStop;
		break;
		
		default:
		break;
		}
		
	IssueAndCompleteRequest ();
	}

/**
	Accept the url from the user and validate.
 */	
void CTestWebBrowser::AcceptUrlL ()
	{

	HBufC* uri = HBufC::NewLC ( KMaxURISize );
	TPtr ptr( uri->Des() );
	iTestUtils.GetAnEntry ( KEnterUrl, ptr );
	if ( iBaseUri->Des().MaxLength() < ptr.Length() )
		{
		iBaseUri = 	iBaseUri->ReAllocL ( ptr.Length() );
		}
		
	iBaseUri->Des().Copy ( ptr );
	
	if ( ValidateUri( iBaseUri->Des() ) )
		{
		iCurrentState = EStartTransaction;
		}
	else
		{
		iTestUtils.Test().Printf ( _L("Invalid entry.") );
		iTestUtils.PressAnyKey ();			
		iCurrentState = EMainMenu;
		}	
	CleanupStack::PopAndDestroy ( uri );
	
	IssueAndCompleteRequest ();		
	}

/**
	Accept proxy information from the user.
 */	
void CTestWebBrowser::AcceptProxyL ()
	{	
	HBufC* proxy16 = HBufC::NewLC ( 128 )		;
	TPtr ptr( proxy16->Des() );
	iTestUtils.GetAnEntry ( KEnterProxy, ptr );
	
	HBufC8* proxy8 = HBufC8::NewLC ( ptr.Length() );
	proxy8->Des().Copy ( ptr );
	
	TRAPD ( err, SetProxyL ( proxy8->Des() ) );	
	
	if ( err != KErrNone )
		{
		iTestUtils.Test().Printf ( _L ( "Setting of proxy address failed." ) );
		}
	
	iCurrentState = EMainMenu;
	IssueAndCompleteRequest ();
	}

/**
	Set the state to EMainMenu and issues a request to be processed by the RunL.	
 */	
void CTestWebBrowser::OnTransactionCompleted ()
	{
	iTestUtils.Test().Printf ( _L ("Transaction completed.") );
	iTestUtils.PressAnyKey();
	iCurrentState = EMainMenu;			
	IssueAndCompleteRequest ();
	}