// Grep.cpp
// 
// Copyright (c) 2009 - 2010 Accenture. All rights reserved.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are
// met:
//
//     * Redistributions of source code must retain the above copyright
// notice, this list of conditions and the following disclaimer.
//     * Redistributions in binary form must reproduce the above
// copyright notice, this list of conditions and the following disclaimer
// in the documentation and/or other materials provided with the
// distribution.
//     * Neither the name of Accenture nor the names of its
// contributors may be used to endorse or promote products derived from
// this software without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
//
#include <fshell/ioutils.h>
#include <fshell/common.mmh>
#include <fshell/descriptorutils.h>
#include <fshell/pcre/cregex.h>

using namespace IoUtils;
using namespace LtkUtils;

class CCmdGrep : public CCommandBase
	{
public:
	static CCommandBase* NewLC();
	~CCmdGrep();
private:
	CCmdGrep();
private: // From CCommandBase.
	virtual const TDesC& Name() const;
	virtual void DoRunL();
	virtual void ArgumentsL(RCommandArgumentList& aArguments);
	virtual void OptionsL(RCommandOptionList& aOptions);
private:
	HBufC* iPattern;
	TBool iIgnoreCase;
	TBool iInvertMatch;
	TBool iCount;
	TBool iUnicode;

	RLtkBuf8 iNarrowBuf;
	CRegEx* iRegex;
	};

EXE_BOILER_PLATE(CCmdGrep)

CCommandBase* CCmdGrep::NewLC()
	{
	CCmdGrep* self = new(ELeave) CCmdGrep();
	CleanupStack::PushL(self);
	self->BaseConstructL();
	return self;
	}

CCmdGrep::~CCmdGrep()
	{
	delete iPattern;
	delete iRegex;
	iNarrowBuf.Close();
	}

CCmdGrep::CCmdGrep()
	{
	}

const TDesC& CCmdGrep::Name() const
	{
	_LIT(KName, "grep");	
	return KName;
	}

void CCmdGrep::ArgumentsL(RCommandArgumentList& aArguments)
	{
	aArguments.AppendStringL(iPattern, _L("pattern"));
	}

void CCmdGrep::OptionsL(RCommandOptionList& aOptions)
	{
	_LIT(KCmdGrepOptIgnoreCase, "ignore-case");
	_LIT(KCmdGrepOptInvertMatch, "invert-match");
	_LIT(KCmdGrepOptCount, "count");
	_LIT(KCmdGrepOptUnicode, "unicode");
	aOptions.AppendBoolL(iIgnoreCase, KCmdGrepOptIgnoreCase);
	aOptions.AppendBoolL(iInvertMatch, KCmdGrepOptInvertMatch);
	aOptions.AppendBoolL(iCount, KCmdGrepOptCount);
	aOptions.AppendBoolL(iUnicode, KCmdGrepOptUnicode);
	}

void CCmdGrep::DoRunL()
	{
	TRegExOptions options(EPcreExtended|EPcreNewlineAny);
	if (iUnicode)
		{
		iNarrowBuf.CopyAsUtf8L(*iPattern);
		options.SetUtf8(ETrue);
		}
	else
		{
		iNarrowBuf.AppendL(*iPattern);
		}
	if (iIgnoreCase) options.SetCaseless(ETrue);

	iRegex = CRegEx::NewL(iNarrowBuf, options);

	Stdin().SetReadMode(RIoReadHandle::ELine);
	TBuf<0x100> line;
	TInt count = 0;
	while (Stdin().Read(line) == KErrNone)
		{
		iNarrowBuf.Zero();
		if (iUnicode)
			{
			iNarrowBuf.CopyAsUtf8L(line);
			}
		else
			{
			iNarrowBuf.AppendL(line);
			}

		TBool matches = iRegex->PartialMatchL(iNarrowBuf);

		if (iInvertMatch)
			{
			matches = !matches;
			}
		if (matches)
			{
			if (iCount)
				{
				count++;
				}
			else
				{
				Write(line);
				}
			}
		}

	if (iCount)
		{
		Printf(_L("Count = %d"), count);
		}
	}