// gobble.cpp

// 

// Copyright (c) 2007 - 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

//

#include "gobble.h"

#include <fshell/ltkutils.h>

CCommandBase* CCmdGobble::NewLC()

	{

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

	CleanupStack::PushL(self);

	self->BaseConstructL();

	return self;

	}

CCmdGobble::~CCmdGobble()

	{

	}

CCmdGobble::CCmdGobble()

	: iBlockSize(512)

	{

	}

const TDesC& CCmdGobble::Name() const

	{

	_LIT(KName, "gobble");

	return KName;

	}

void CCmdGobble::DoRunL()

	{

	RFs& fs = FsL();

	if (iAmount == 0)

		{

		TInt drive = EDriveC;

		if (iFileName.Length() == 0) LeaveIfErr(KErrBadName, _L("Bad file name"));

		RFs::CharToDrive(iFileName[0], drive);

		TVolumeInfo volInfo;

		LeaveIfErr(fs.Volume(volInfo, drive), _L("Couldn't get volume information for drive %c"), iFileName[0]);

		iAmount = volInfo.iFree;

		}

	if (!iNoWrite && iAmount < iBlockSize)

		{

		LeaveIfErr(KErrArgument, _L("The amount to consume must be greater than the block size (%d)"), iBlockSize);

		}

	if (iAmount & 0x80000000)

		{

		LeaveIfErr(KErrArgument, _L("The amount to consume is too large (maximum is %d)"), KMaxTInt);

		}

	if (iBlockSize & 0x80000000)

		{

		LeaveIfErr(KErrArgument, _L("The block size is too large (maximum is %d)"), KMaxTInt);

		}

	fs.MkDirAll(iFileName);

	RFile file;

	TInt err = file.Open(fs, iFileName, EFileWrite);

	if (err == KErrNotFound)

		{

		err = file.Create(fs, iFileName, EFileWrite);

		}

	LeaveIfErr(err, _L("Couldn't create file %S"), &iFileName);

	CleanupClosePushL(file);

	if (iNoWrite)

		{

		TInt size = 0;

		LeaveIfErr(file.Size(size), _L("Couldn't get file size"));

		size += iAmount;

		LeaveIfErr(file.SetSize(size), _L("Couldn't set filesize to %d"), size);

		}

	else

		{

		TInt pos = 0;

		User::LeaveIfError(file.Seek(ESeekEnd, pos));

		HBufC8* buf = HBufC8::NewLC(iBlockSize);

		TPtr8 ptr(buf->Des());

		ptr.Fill(TChar('x'), iBlockSize);

		TInt toWrite = static_cast<TInt>(iAmount);

		do

			{

			TInt writeSize;

			if (toWrite > static_cast<TInt>(iBlockSize))

				{

				writeSize = static_cast<TInt>(iBlockSize);

				}

			else

				{

				writeSize = toWrite;

				}

			ptr.SetLength(writeSize);

			err = file.Write(ptr);

			if (err == KErrNone)

				{

				if (iVerbose)

					{

					Printf(_L("\rWrote %d"), iAmount - toWrite);

					}

				toWrite -= writeSize;

				}

			}

		while ((err == KErrNone) && (toWrite > 0));

		if (iVerbose)

			{

			Printf(_L("\rWrote %d"), iAmount - toWrite);

			}

		CleanupStack::PopAndDestroy(buf);

		}

	CleanupStack::PopAndDestroy(&file);

	}

void CCmdGobble::OptionsL(RCommandOptionList& aOptions)

	{

	_LIT(KCmdOptVerbose, "verbose");

	aOptions.AppendBoolL(iVerbose, KCmdOptVerbose);

	_LIT(KCmdOptNoWrite, "no-write");

	aOptions.AppendBoolL(iNoWrite, KCmdOptNoWrite);

	}

void CCmdGobble::ArgumentsL(RCommandArgumentList& aArguments)

	{

	_LIT(KArgFileName, "file_name");

	aArguments.AppendFileNameL(iFileName, KArgFileName);

	_LIT(KArgAmount, "amount");

	aArguments.AppendUintL(iAmount, KArgAmount);

	_LIT(KArgBlockSize, "block_size");

	aArguments.AppendUintL(iBlockSize, KArgBlockSize);

	}

#ifdef EXE_BUILD

EXE_BOILER_PLATE(CCmdGobble)

#endif