// wget.cpp

// 

// Copyright (c) 2008 - 2010 Accenture. All rights reserved.

// This component and the accompanying materials are made available

// under the terms of the "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:

// Accenture - Initial contribution

//

// TODO: byte-range support

#include <commdbconnpref.h>

#include <httpstringconstants.h>

#include <httperr.h>

#include <mhttpdatasupplier.h>

#include <rhttpheaders.h>

#include <thttphdrfielditer.h>

#include <bautils.h>

#include "wget.h"

using namespace IoUtils;

_LIT8(KHttpProtocol,	"HTTP/TCP");

_LIT(KDateFormat,		"%D%M%Y%/0%1%/1%2%/2%3%/3 %:0%H%:1%T%:2%S.%C%:3");

const TInt KMaxHeaderNameLen = 32;

const TInt KMaxHeaderValueLen = 128;

CCommandBase* CCmdWget::NewLC()

	{

	CCmdWget* self = new(ELeave) CCmdWget();

	CleanupStack::PushL(self);

	self->BaseConstructL();

	return self;

	}

CCmdWget::~CCmdWget()

	{

	delete iUrl;

	delete iUsername;

	delete iPassword;

	if (iSessionIsOpen)

		{

		iSession.Close(); // closes iTransaction also

		}

	if (iConnection.SubSessionHandle() > 0)

		{

		iConnection.Close();

		}

	if (iSocketServ.Handle() > 0)

		{

		iSocketServ.Close();

		}

	if (iLocalFile.SubSessionHandle() > 0)

		{

		iLocalFile.Flush();

		iLocalFile.Close();

		}

	}

CCmdWget::CCmdWget() : CCommandBase(CCommandBase::EManualComplete)

	{

	}

const TDesC& CCmdWget::Name() const

	{

	_LIT(KName, "wget");

	return KName;

	}

void CCmdWget::DoRunL()

	{

	// sanity check parameters

	if (!iUrl || iUrl->Des().Length() > KMaxName)

		{

		if (iVerbose)

			{

			PrintError(KErrArgument, _L("Url not specified or is too large."));

			}

		User::Leave(KErrArgument);

		}		

	if ((iUsername && iUsername->Des().Length() > KMaxName) || (iPassword && iPassword->Des().Length() > KMaxName))

		{

		if (iVerbose)

			{

			PrintError(KErrArgument, _L("Username or Password is too large."));

			}

		User::Leave(KErrArgument);

		}

	if (iUploadFilename.Length())

		{

		iPostData = ETrue; // Keep this member variable because lots of code assumes it

		}

	LaunchConnectionL(); // start a connection - KErrNone returned if successfully started or a compatible connection already exists

	if (iPostData)

		{

		// http post

		PrepareUploadFileL();

		}

	else

		{

		// http get

		PrepareDownloadFileL(); // prep. the local file to receive the downloaded data

		}

	User::LeaveIfError(iLocalFile.Size(iLocalFileSize));

	ConfigureHTTPL(); // setup http session & configure with connection parameters

	LaunchHTTPTransactionL(); // launch a http transaction

	}

void CCmdWget::ArgumentsL(RCommandArgumentList& aArguments)

	{

	_LIT(KArg1, "url");

	aArguments.AppendStringL(iUrl, KArg1);

	}

void CCmdWget::OptionsL(RCommandOptionList& aOptions)

	{

	_LIT(KOptContinue, "continue");

	aOptions.AppendBoolL(iContinue, KOptContinue);

	_LIT(KOptDestinationFilename, "downloadfile");

	aOptions.AppendFileNameL(iDestinationFilename, KOptDestinationFilename);

	_LIT(KOptSourceFilename, "uploadfile");

	aOptions.AppendFileNameL(iUploadFilename, KOptSourceFilename);

	_LIT(KOptIapId, "iap");

	aOptions.AppendIntL(iIapId, KOptIapId);

	_LIT(KOptUsername, "username");

	aOptions.AppendStringL(iUsername, KOptUsername);

	_LIT(KOptPassword, "password");

	aOptions.AppendStringL(iPassword, KOptPassword);

	_LIT(KOptVerbose, "verbose");

	aOptions.AppendBoolL(iVerbose, KOptVerbose);

	}

//

// CCmdWget::PrepareDownloadFileL

// determine the destination file & path and open a handle to a local file

//

void CCmdWget::PrepareDownloadFileL()

	{

	ASSERT(iUrl);

	ASSERT(!iPostData);

	TFileName fname;

	RFs& fs = FsL(); 

	fname.Zero();

	if (iDestinationFilename.Length() <= 0)

		{

		// use current path

		fname.Append(Env().Pwd());

		// use the same name as the file we're intending to download

		TChar dirMarker('/');

		TInt mark = iUrl->Des().LocateReverse(dirMarker);

		if (mark <= 0)

			{

			User::Leave(KErrNotFound);

			}

		fname.Append(iUrl->Des().Mid(++mark)); // increment mark to step over the '/' 

		}

	else 

		{

		fname.Copy(iDestinationFilename); // user has specified a local file name

		}

	if (BaflUtils::FileExists(fs, fname))

		{

		if (iContinue)

			{

			// open the file & append any new data to what's already in there

			User::LeaveIfError(iLocalFile.Open(fs, fname, EFileShareExclusive));

			}

		else

			{

			// open the file & replace old data with new data

			User::LeaveIfError(iLocalFile.Replace(fs, fname, EFileShareExclusive));

			}

		}

	else

		{

		// create the file for the first time

		User::LeaveIfError(iLocalFile.Create(fs, fname, EFileShareExclusive));

		}

	}

//

// CCmdWget::PrepareUploadFileL

// prep. the local file containing data for upload to the web server

//

void CCmdWget::PrepareUploadFileL()

	{

	ASSERT(iUrl);

	ASSERT(iPostData);

	User::LeaveIfError(iLocalFile.Open(FsL(), iUploadFilename, EFileRead | EFileShareReadersOnly));

	}

//

// CCmdWget::LaunchConnectionL

// establish a connection to the network

//

void CCmdWget::LaunchConnectionL()

	{

	User::LeaveIfError(iSocketServ.Connect());

	User::LeaveIfError(iConnection.Open(iSocketServ));

	TInt connError = KErrNone;	

	TCommDbConnPref prefs;

	if (iIapId > 0)

		{

		prefs.SetIapId(iIapId);

		prefs.SetDialogPreference(ECommDbDialogPrefDoNotPrompt);

		connError = iConnection.Start(prefs);

		}

	else

		{

		connError = iConnection.Start();

		}

	if (iVerbose && (connError != KErrNone))

		{

		PrintError(connError, _L("Unable to establish network connection"));

		User::Leave(connError);

		}	

	}

//

// CCmdWget::ConfigureHTTPL

//

void CCmdWget::ConfigureHTTPL()

	{

	iSession.OpenL(KHttpProtocol);

	iSessionIsOpen = ETrue;

	iSession.SetSessionEventCallback(this);

	RStringPool pool = iSession.StringPool();

	RHTTPConnectionInfo conInf = iSession.ConnectionInfo();

	conInf.SetPropertyL(pool.StringF(HTTP::EHttpSocketServ, RHTTPSession::GetTable()), THTTPHdrVal(iSocketServ.Handle()));

	conInf.SetPropertyL(pool.StringF(HTTP::EHttpSocketConnection, RHTTPSession::GetTable()), THTTPHdrVal(reinterpret_cast<TInt> (&iConnection)));

	InstallAuthenticationL(iSession);

	}

//

// CCmdWget::LaunchHTTPTransactionL

// launches a new http transaction

//

void CCmdWget::LaunchHTTPTransactionL()

	{

	// create the transaction

	TBuf8<KMaxName> url8;

	url8.Copy(iUrl->Des());

	TUriParser8 uri;

	User::LeaveIfError(uri.Parse(url8));

	RStringPool pool = iSession.StringPool();

	RStringF method = pool.StringF(HTTP::EGET,RHTTPSession::GetTable());

	if (iPostData)

		{

		method = pool.StringF(HTTP::EPOST,RHTTPSession::GetTable());

		}

	iTransaction = iSession.OpenTransactionL(uri, *this, method);

	// some transaction configuration is required for posting data

	if (iPostData)

		{

		RHTTPRequest req = iTransaction.Request();

		req.SetBody(*this); // ccmdwget to supply the data for posting

		THTTPHdrVal length(OverallDataSize());

		THTTPHdrVal type(pool.StringF(HTTP::ETextAny, RHTTPSession::GetTable()));

		RHTTPHeaders hdr = req.GetHeaderCollection();

		hdr.SetFieldL(pool.StringF(HTTP::EContentLength, RHTTPSession::GetTable()), length);

		hdr.SetFieldL(pool.StringF(HTTP::EContentType, RHTTPSession::GetTable()), type);

		}

	// launch it

	iTransaction.SubmitL();	

	}

//

// CCmdWget::GetCredentialsL

// request from the underlying HTTP session to provide it with some authentication credentials (if present)

//

TBool CCmdWget::GetCredentialsL(const TUriC8& /*aURI*/, RString aRealm, RStringF /*aAuthenticationType*/, RString& aUsername, RString& aPassword)

	{

	TBool present = EFalse;

	RStringPool pl = aRealm.Pool();

	if (iUsername)

		{

		present = ETrue;

		TBuf8<KMaxName> user;

		user.Copy(iUsername->Des());

		aUsername = pl.OpenStringL(user);

		}

	if (iPassword)

		{

		present = ETrue;

		TBuf8<KMaxName> pwd;

		pwd.Copy(iPassword->Des());

		aPassword = pl.OpenStringL(pwd);

		}

	return present;

	}

//

// CCmdWget::MHFSessionRunL

// handles any up-calls from the http session

//

void CCmdWget::MHFSessionRunL(const THTTPSessionEvent& aEvent)

	{

	switch (aEvent.iStatus)

		{

		case THTTPSessionEvent::EConnect:

			{

			if (iVerbose)

				{

				Printf(_L("HTTP Session Connect\r\n"));

				}

			}

		break;

		case THTTPSessionEvent::EConnectedOK:

			{

			if (iVerbose)

				{

				Printf(_L("HTTP Session Connected OK\r\n"));

				}

			}

		break;

		case THTTPSessionEvent::EDisconnect:

			{

			if (iVerbose)

				{

				Printf(_L("HTTP Session Disconnect\r\n"));

				}

			}

		break;

		case THTTPSessionEvent::EConnectedWithReducedCapabilities:

			{

			if (iVerbose)

				{

				Printf(_L("HTTP Session Connected with reduced capabilities\r\n"));

				}

			}

		break;

		case THTTPSessionEvent::EDisconnected:

			{

			if (iVerbose)

				{

				Printf(_L("HTTP Session Disconnected\r\n"));

				}

			}

		break;

		case THTTPSessionEvent::EAuthenticatedOK:

			{

			if (iVerbose)

				{

				Printf(_L("HTTP Session Authenticated OK\r\n"));

				}

			}

		break;

		case THTTPSessionEvent::EAuthenticationFailure:

			{

			if (iVerbose)

				{

				Printf(_L("HTTP Session Authentication failure\r\n"));

				}

			}

		break;

		case THTTPSessionEvent::EConnectionTimedOut:

			{

			if (iVerbose)

				{

				Printf(_L("HTTP Session Connection timed out\r\n"));

				}

			}

		break;

		case THTTPSessionEvent::ENotConnected:

			{

			if (iVerbose)

				{

				Printf(_L("HTTP Session Not connected\r\n"));

				}

			}

		break;

		case THTTPSessionEvent::EExceptionInfo:

			{

			if (iVerbose)

				{

				Printf(_L("HTTP Session Exception info\r\n"));

				}

			}

		break;

		case THTTPSessionEvent::ERedirected:

			{

			if (iVerbose)

				{

				Printf(_L("HTTP Session Redirected\r\n"));

				}

			}

		break;

		case THTTPSessionEvent::EAlreadyConnecting:

			{

			if (iVerbose)

				{

				Printf(_L("HTTP Session Already connecting\r\n"));

				}

			}

		break;

		case THTTPSessionEvent::EAlreadyConnected:

			{

			if (iVerbose)

				{

				Printf(_L("HTTP Session Already connected\r\n"));

				}

			}

		break;

		case THTTPSessionEvent::EAlreadyDisconnecting:

			{

			if (iVerbose)

				{

				Printf(_L("HTTP Session Already disconnecting\r\n"));

				}

			}

		break;

		case THTTPSessionEvent::EAlreadyDisconnected:

			{

			if (iVerbose)

				{

				Printf(_L("HTTP Session Already disconnected\r\n"));

				}

			}

		break;

		default:

			{

			if (iVerbose)

				{

				Printf(_L("HTTP Session Event unrecgonised\r\n"));

				}

			}

		break;

		};

	}

//

// CCmdWget::MHFSessionRunError

// handles an error related up-calls 

TInt CCmdWget::MHFSessionRunError(TInt aError, const THTTPSessionEvent& /*aEvent*/)

	{

	if (iVerbose)

		{

		PrintError(aError, _L("HTTP Session Error"));

		}

	return KErrNone;

	}

//

// CCmdWget::MHFRunL

// handles any up-calls regarding a current transaction

//

void CCmdWget::MHFRunL(RHTTPTransaction aTransaction, const THTTPEvent& aEvent)

	{

	if (aTransaction.Id() != iTransaction.Id())

		{

		if (iVerbose)

			{

			PrintError(KErrUnknown, _L("HTTP Transaction Id is invalid"));

			}

		User::Leave(KErrUnknown);

		}

	switch (aEvent.iStatus)

		{

		case THTTPEvent::EGotResponseHeaders:

			{

			if (iVerbose)

				{

				DumpRespHeadersL(aTransaction);

				}

			}

		break;

		case THTTPEvent::EGotResponseBodyData:

			{

			RHTTPResponse response = aTransaction.Response();

			if (iVerbose)

				{

				Printf(_L("HTTP <%d> received\r\n"), response.StatusCode());

				}

			switch (response.StatusCode())

				{

				// informational codes (1xx)

				case HTTPStatus::EContinue:

				case HTTPStatus::ESwitchingProtocols:

				break;

				// successful codes (2xx)