// 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 FILES

#include "TestContentHandler.h"

#include "xmlengtester.h"

#include <e32svr.h>

#include <Xml\Parser.h>

#include <Xml\ParserFeature.h>

#include <libc\string.h>

#include <charconv.h>

#include <UTF.H>

#include "xmlengtesterdef.h"

#include "TestUserData.h"

using namespace Xml;

class MXmlEngTestNodeFilter : public MXmlEngNodeFilter

{

    public: 

        TXmlEngNodeFilterResult AcceptNode(TXmlEngNode aNode)

            {

            _LIT8(KNodeName1,"name");

            _LIT8(KNodeName2,"phone");

            _LIT8(KNodeName3,"title");

            TPtrC8 node1=aNode.AsElement().Name();

          if (!aNode.AsElement().Name().Compare(KNodeName2))

                {

                return EReject;

                }

            else if (!aNode.AsElement().Name().Compare(KNodeName1))

                {

                return ESkip;

                }

            else if (!aNode.AsElement().Name().Compare(KNodeName3))

                {

                return ESkipContents;

                }

            else

            return EAccept;

            }

};

/************************************** DOM ****************************************/

// -----------------------------------------------------------------------------

// CLibxml2Tester::ParseFileDOML

// test method function.

// (other items were commented in a header).

// -----------------------------------------------------------------------------

//

TInt CLibxml2Tester::ParseFileDomL( CStifItemParser& aItem )

{

	_LIT(KSourceFileErrorName,"No source file name.");

    _LIT(KEndParsing,"Parsing end.");

	TPtrC SrcFile;

	TPtrC tmp;

	TBool map = TRUE;

    if ( aItem.GetNextString(SrcFile) != KErrNone )

        {

        TestModuleIf().Printf( infoNum++,KTEST_NAME,KSourceFileErrorName);

     	iLog->Log((TDesC)KSourceFileErrorName);

        return KErrGeneral;

        }

    if ( aItem.GetNextString(tmp) == KErrNone )

        {

        map = FALSE;

        }

	TFileName srcFileName;

	srcFileName.Copy(SrcFile);

	TInt err;

	TRAP(err, iDoc = parser.ParseFileL(srcFileName));

	if(map && err == KXmlEngErrParsing)

		{

		iLastError = parser.GetLastParsingError();	

		}

	else

		{

		iLastError = err;		

		}

	TestModuleIf().Printf( infoNum++,KTEST_NAME,KEndParsing);

	iLog->Log((TDesC)KEndParsing);

    return KErrNone;    

    }   

// -----------------------------------------------------------------------------

// CLibxml2Tester::SaveXMLtoFileDomL

// test method function.

// (other items were commented in a header).

// -----------------------------------------------------------------------------

//

TInt CLibxml2Tester::SaveXmlToFileDomL( CStifItemParser& aItem )

{

    _LIT(KOutputFileErrorName,"No output file name.");

    _LIT(KOutputFileError,"Output file error. Error code: %d");

    _LIT(KRootError,"Document has no root element. Error code: %d");

    _LIT(KSaveEnd,"File saved.");

    TPtrC OutputFile;

    if ( aItem.GetNextString(OutputFile) != KErrNone )

    	{

        TestModuleIf().Printf( infoNum++,KTEST_NAME,KOutputFileErrorName);

     	iLog->Log((TDesC)KOutputFileErrorName);

        return KErrGeneral;

    	}

	TXmlEngElement root = iDoc.DocumentElement();

	if(root.NotNull())

		{

		iDoc.SaveL(OutputFile);

		}

	else

		{

		TestModuleIf().Printf( infoNum++, KTEST_NAME,KRootError,iLastError);

		iLog->Log(KRootError, iLastError);	

		RFs fs;

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

		CleanupClosePushL(fs);

		RFile outputFile;

		TFileName outFileName;

		outFileName.Copy(OutputFile);

		// Open output file

		TInt err = outputFile.Replace(fs,outFileName,EFileShareAny|EFileWrite);

    	// If output file is correct

		if (err == KErrNone)

	    	{

			outFileName.Format(KRootError,iLastError);

			HBufC8* buf = NULL;

			buf = HBufC8::NewLC(outFileName.Length());

			buf->Des().Copy(outFileName);

			outputFile.Write(buf->Des());

			outputFile.Close();

			CleanupStack::PopAndDestroy(buf);

			}

		else

			{	

			TestModuleIf().Printf( infoNum++,KTEST_NAME, KOutputFileError, err);

			iLog->Log(KOutputFileError, err);

			}

		CleanupStack::PopAndDestroy(&fs);

		}

	TestModuleIf().Printf( infoNum++,KTEST_NAME,KSaveEnd);

	iLog->Log((TDesC)KSaveEnd);

    return KErrNone;   

    }

// -----------------------------------------------------------------------------

// CLibxml2Tester::SaveXMLtoBufferUTF16L

// test method function.

// (other items were commented in a header).

// -----------------------------------------------------------------------------

//

TInt CLibxml2Tester::SaveXmlToBufferUTF16L( CStifItemParser& aItem )

{

    _LIT(KOutputFileErrorName,"No output file name.");

    _LIT(KOutputFileError,"Output file error. Error code: %d");

    _LIT(KRootError,"Document has no root element. Error code: %d");

    _LIT(KSaveEnd,"File saved.");

    _LIT8(enc,"UTF-16");

    TPtrC OutputFile;

    if ( aItem.GetNextString(OutputFile) != KErrNone )

    	{

        TestModuleIf().Printf( infoNum++,KTEST_NAME,KOutputFileErrorName);

     	iLog->Log((TDesC)KOutputFileErrorName);

        return KErrGeneral;

    	}

	TPtrC flag;

	TBool useFlag = EFalse;

	if ( aItem.GetNextString(flag) == KErrNone )

    	{

        useFlag = ETrue;

    	}

	RBuf8 buf;

	TXmlEngElement root = iDoc.DocumentElement();

	if(root.NotNull())

		{

		if(useFlag)

		    {

		    TXmlEngSerializationOptions opt(TXmlEngSerializationOptions::KOptionEncoding | TXmlEngSerializationOptions::KOptionStandalone,enc);

		    iDoc.SaveL(buf,iDoc,opt);

		    CleanupClosePushL( buf );

		    }

		else

		    {

		    TXmlEngSerializationOptions opt(0,enc);

		    iDoc.SaveL(buf,iDoc,opt);

		    CleanupClosePushL( buf );

		    }

		}

	else

		{

		TestModuleIf().Printf( infoNum++, KTEST_NAME,KRootError,iLastError);

	  	iLog->Log(KRootError,iLastError);

		TBuf<256> buffer;

		buffer.Format(KRootError,iLastError);

		TBuf8<256> buffer8;

		buffer8.Copy(buffer);

		buf.CreateL(buffer8);

		CleanupClosePushL(buf);

		}

	RFs fs;

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

	CleanupClosePushL(fs);

	RFile outputFile;

	TFileName outFileName;

	outFileName.Copy(OutputFile);

	// Open output file

	TInt err = outputFile.Replace(fs,outFileName,EFileShareAny|EFileWrite);

	// If output file is correct

	if (err == KErrNone)

	   	{

		outputFile.Write(buf);

		outputFile.Close();

		}

	else

		{	

		TestModuleIf().Printf( infoNum++,KTEST_NAME, KOutputFileError, err);

		iLog->Log(KOutputFileError, err);

		}

	CleanupStack::PopAndDestroy(&fs);

	CleanupStack::PopAndDestroy(&buf);

	TestModuleIf().Printf( infoNum++,KTEST_NAME,KSaveEnd);

	iLog->Log((TDesC)KSaveEnd);

    return KErrNone;   

    }

// -----------------------------------------------------------------------------

// CLibxml2Tester::TestNodeFilterL

// test method function.

// (other items were commented in a header).

// -----------------------------------------------------------------------------

//

TInt CLibxml2Tester::TestNodeFilterL( CStifItemParser& aItem )

{

    _LIT(KOutputFileErrorName,"No output file name.");

    _LIT(KOutputFileError,"Output file error. Error code: %d");

//    _LIT(KRootError,"Document has no root element. Error code: %d");

    _LIT(KSaveEnd,"File saved.");

    TPtrC InputFile;

    if ( aItem.GetNextString(InputFile) != KErrNone )

    	{

        TestModuleIf().Printf( infoNum++,KTEST_NAME,KOutputFileErrorName);

     	iLog->Log((TDesC)KOutputFileErrorName);

        return KErrGeneral;

    	}

    TPtrC OutputFile;

    if ( aItem.GetNextString(OutputFile) != KErrNone )

    	{

        TestModuleIf().Printf( infoNum++,KTEST_NAME,KOutputFileErrorName);

     	iLog->Log((TDesC)KOutputFileErrorName);

        return KErrGeneral;

    	}

        TFileName srcFileName;

        srcFileName.Copy( InputFile);

        RBuf8 buf;

        RXmlEngDocument doc;

        doc=parser.ParseFileL( srcFileName );

	MXmlEngTestNodeFilter* nodefilter = new (ELeave) MXmlEngTestNodeFilter();

	CleanupStack::PushL(nodefilter);

	TXmlEngSerializationOptions options(0);

	TXmlEngElement root = doc.DocumentElement();

	if(root.NotNull())

		    {

		    options.SetNodeFilter(nodefilter);

		    doc.SaveL(buf,doc,options);

		    CleanupClosePushL( buf );

		    }

	RFs fs;

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

	CleanupClosePushL(fs);

	RFile outputFile;

	TFileName outFileName;

	outFileName.Copy(OutputFile);

	// Open output file

	TInt err = outputFile.Replace(fs,outFileName,EFileShareAny|EFileWrite);

	// If output file is correct

	if (err == KErrNone)

	   	{

		outputFile.Write(buf);

		outputFile.Close();

		}

	else

		{	

		TestModuleIf().Printf( infoNum++,KTEST_NAME, KOutputFileError, err);

		iLog->Log(KOutputFileError, err);

		}

	CleanupStack::PopAndDestroy(&fs);

	CleanupStack::PopAndDestroy(&buf);

	doc.Close();	

	CleanupStack::PopAndDestroy(); //nodefilter

	TestModuleIf().Printf( infoNum++,KTEST_NAME,KSaveEnd);

	iLog->Log((TDesC)KSaveEnd);

    return KErrNone;   

    }

// -----------------------------------------------------------------------------

// CLibxml2Tester::CreateDocumentL

// test method function.

// (other items were commented in a header).

// -----------------------------------------------------------------------------

//

TInt CLibxml2Tester::CreateDocumentL( CStifItemParser& /*aItem*/ )

	{

	_LIT8(name,"doc");

	_LIT8(uri,"http://test.com");

	_LIT8(arg1,"1.1");

	_LIT8(arg2,"http://test2.org");

	iDoc.OpenL(DOM_impl);

	iDoc.CreateDocumentElementL(name);

	iDoc.DocumentElement().SetDefaultNamespaceL(uri);

	iDoc.SetStandalone(TRUE);

	iDoc.SetXmlVersionL(arg1);

	iDoc.SetDocumentUriL(arg2);

    return KErrNone;    

    }  

// -----------------------------------------------------------------------------

// CLibxml2Tester::CheckDocumentParametersL

// test method function.

// (other items were commented in a header).

// -----------------------------------------------------------------------------

//

TInt CLibxml2Tester::CheckDocumentParametersL( CStifItemParser& /*aItem*/ )

	{

	_LIT(KStandaloneError,"Standalone checking error.");

    _LIT(KEncodingError,"Encoding checking error");

    _LIT(KVersionError,"Version checking error");

    _LIT(KUriError,"Uri checking error");

	_LIT8(arg,"1.0");

	_LIT8(arg2,"UTF-8");

	_LIT8(KDesNull,"");

	_LIT(KTypeError,"Node type checking error.");

	if(!(iDoc.NodeType() == TXmlEngNode::EDocument))

		{

		TestModuleIf().Printf( infoNum++,KTEST_NAME, KTypeError);

		iLog->Log((TDesC)KTypeError);

		return KErrGeneral;

		}	

	if(!(iDoc.IsStandalone() == TRUE))

		{

		TestModuleIf().Printf( infoNum++,KTEST_NAME, KStandaloneError);

		iLog->Log((TDesC)KStandaloneError);

		return KErrGeneral;

		}

	if(iDoc.XmlVersion().Compare(arg))

		{

		TestModuleIf().Printf( infoNum++,KTEST_NAME, KVersionError);

		iLog->Log((TDesC)KVersionError);

		return KErrGeneral;

		}

	if(iDoc.XmlEncoding().Compare(arg2))

		{

		TestModuleIf().Printf( infoNum++,KTEST_NAME, KEncodingError);

		iLog->Log((TDesC)KEncodingError);

		return KErrGeneral;

		}

	//if(iDoc.DocumentUri().Compare(NULL) )

	if(iDoc.DocumentUri().Compare(KDesNull ))

		{

		TestModuleIf().Printf( infoNum++,KTEST_NAME, KUriError);

		iLog->Log((TDesC)KUriError);

		return KErrGeneral;

		}

	return KErrNone;    

    }  

// -----------------------------------------------------------------------------

// CLibxml2Tester::CheckNotNullElementL

// test method function.

// (other items were commented in a header).

// -----------------------------------------------------------------------------

//

TInt CLibxml2Tester::CheckNotNullElementL( CStifItemParser& /*aItem*/ )

	{

	_LIT8(name,"test");

	TXmlEngElement elem;

	_LIT(KNullError,"Is node null checking error.");

	if(!elem.IsNull())

		{

		TestModuleIf().Printf( infoNum++,KTEST_NAME, KNullError);

		iLog->Log((TDesC) KNullError);

		return KErrGeneral;

		}

	iDoc.OpenL(DOM_impl);

	elem = iDoc.CreateElementL(name);

	_LIT(KNotNullError,"Is node not null checking error.");

	if(!elem.NotNull())

		{

		TestModuleIf().Printf( infoNum++,KTEST_NAME, KNotNullError);

		iLog->Log((TDesC) KNotNullError);

		return KErrGeneral;

		}

	iDoc.Close();

    return KErrNone;

    }

// -----------------------------------------------------------------------------

// CLibxml2Tester::CreateDocumentElementL

// test method function.

// (other items were commented in a header).

// -----------------------------------------------------------------------------

//

TInt CLibxml2Tester::CreateDocumentElementL( CStifItemParser& /*aItem*/ )

	{

	_LIT8(name, "doc");

	_LIT8(uri, "http://test.com");

//	_LIT8(arg, "");

	_LIT8(ver, "1.0");

	iDoc.OpenL(DOM_impl);

	TXmlEngElement root = iDoc.CreateElementL(name);

	root.SetDefaultNamespaceL(uri);

	iDoc.SetStandalone(TRUE);

	iDoc.SetXmlVersionL(ver);

	iDoc.SetDocumentElement(root);

	_LIT(KNamespaceError,"Namespace checking error.");

	if(iDoc.DocumentElement().DefaultNamespaceL().Uri().Compare(uri))

		{

		TestModuleIf().Printf( infoNum++,KTEST_NAME, KNamespaceError);

		iLog->Log((TDesC)KNamespaceError);

		return KErrGeneral;

		}

    return KErrNone;

    }

// -----------------------------------------------------------------------------

// CLibxml2Tester::CreateDocumentFromNodeL

// test method function.

// (other items were commented in a header).

// -----------------------------------------------------------------------------

//

TInt CLibxml2Tester::CreateDocumentFromNodeL( CStifItemParser& /*aItem*/ )

	{

	_LIT8(name,"doc");

	_LIT8(tekst,"test content");

	_LIT8(value,"ATTR_value");