/*

* Copyright (c) 2006-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:  Harvests meta data from oma drm file.*

*/

#include <e32base.h>

#include <caf/caf.h>

#include <pathinfo.h>

#include "mdsutils.h"

#include "harvesterexifutil.h"

#include "harvesteromadrmplugin.h"

#include "harvesterlog.h"

#include "harvestercommon.h"

#include "mdeobjectwrapper.h"

#include "mdscommoninternal.h"

#include <harvesterdata.h>

#include <mdenamespacedef.h>

#include <mdeobjectdef.h>

#include <mdeobject.h>

#include <mdetextproperty.h>

#include <mdenamespacedef.h>

#include <mdeconstants.h>

#include <imageconversion.h>

using namespace MdeConstants;

_LIT(KImage, "Image");

_LIT(KVideo, "Video");

_LIT(KAudio, "Audio");

_LIT(KRmMimetype, "realmedia");

_LIT( KSvgMime, "image/svg+xml" );

_LIT( KRingingToneMime, "application/vnd.nokia.ringing-tone" );

_LIT(KInUse, "InUse");

CHarvesterOmaDrmPluginPropertyDefs::CHarvesterOmaDrmPluginPropertyDefs() : CBase(),

    iCreationDatePropertyDef( NULL )

	{

	}

void CHarvesterOmaDrmPluginPropertyDefs::ConstructL( CMdEObjectDef& aObjectDef )

    {

    SetByObjectDefL( aObjectDef );

    }

CHarvesterOmaDrmPluginPropertyDefs* CHarvesterOmaDrmPluginPropertyDefs::NewL()

    {

    CHarvesterOmaDrmPluginPropertyDefs* self = 

        new (ELeave) CHarvesterOmaDrmPluginPropertyDefs();

    return self;

    }

void CHarvesterOmaDrmPluginPropertyDefs::SetByObjectDefL( CMdEObjectDef& aObjectDef )

	{

	CMdENamespaceDef& nsDef = aObjectDef.NamespaceDef();

	// Common property definitions

	CMdEObjectDef& objectDef = nsDef.GetObjectDefL( MdeConstants::Object::KBaseObject );

	iCreationDatePropertyDef = &objectDef.GetPropertyDefL( MdeConstants::Object::KCreationDateProperty );

	iLastModifiedDatePropertyDef = &objectDef.GetPropertyDefL( MdeConstants::Object::KLastModifiedDateProperty );

	iSizePropertyDef = &objectDef.GetPropertyDefL( MdeConstants::Object::KSizeProperty );

	iItemTypePropertyDef = &objectDef.GetPropertyDefL( MdeConstants::Object::KItemTypeProperty );

	iTitlePropertyDef = &objectDef.GetPropertyDefL( MdeConstants::Object::KTitleProperty );

	iDefaultFolderPropertyDef = &objectDef.GetPropertyDefL( MdeConstants::Object::KInDefaultFolder );

	CMdEObjectDef& mediaDef = nsDef.GetObjectDefL( MdeConstants::MediaObject::KMediaObject );

	iDrmPropertyDef = &mediaDef.GetPropertyDefL( MdeConstants::MediaObject::KDRMProperty );

	iDescriptionPropertyDef = &mediaDef.GetPropertyDefL( MdeConstants::MediaObject::KDescriptionProperty );

	iAuthorPropertyDef = &mediaDef.GetPropertyDefL( MdeConstants::MediaObject::KAuthorProperty );

	// Media property definitions

	iWidthPropertyDef = &mediaDef.GetPropertyDefL( MdeConstants::MediaObject::KWidthProperty );

	iHeightPropertyDef = &mediaDef.GetPropertyDefL( MdeConstants::MediaObject::KHeightProperty );

	// Image property definitions

	CMdEObjectDef& imageDef = nsDef.GetObjectDefL( MdeConstants::Image::KImageObject );

	iBitsPerSamplePropertyDef = &imageDef.GetPropertyDefL( MdeConstants::Image::KBitsPerSampleProperty );

	iFrameCountPropertyDef = &imageDef.GetPropertyDefL( MdeConstants::Image::KFrameCountProperty );

	iGenrePropertyDef = &mediaDef.GetPropertyDefL( MdeConstants::MediaObject::KGenreProperty );

	}

/**

* Default constructor

*/

CHarvesterOMADRMPlugin::CHarvesterOMADRMPlugin() : CHarvesterPlugin()

	{

	WRITELOG("CHarvesterOMADRMPlugin::CHarvesterOMADRMPlugin()");

	}

/**

* Construction

* @return Harvester image plugin

*/

CHarvesterOMADRMPlugin* CHarvesterOMADRMPlugin::NewL()

	{

	WRITELOG("CHarvesterOMADRMPlugin::NewL()");

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

	CleanupStack::PushL(self);

	self->ConstructL();

	CleanupStack::Pop(self);

	return self;

	}

/**

* Destruction

*/

CHarvesterOMADRMPlugin::~CHarvesterOMADRMPlugin()

	{

	WRITELOG("CHarvesterOMADRMPlugin::~CHarvesterOMADRMPlugin()");

	iFs.Close();

	delete iPropDefs;

    iPropDefs = NULL;

    delete iPhoneImagesPath;

    iPhoneImagesPath = NULL;

    delete iMmcImagesPath;

    iMmcImagesPath = NULL;

    delete iPhoneVideosPath;

    iPhoneVideosPath = NULL;

    delete iMmcVideosPath;

    iMmcVideosPath = NULL;

    delete iPhoneSoundsPath;

    iPhoneSoundsPath = NULL;

    delete iMmcSoundsPath;

    iMmcSoundsPath = NULL;

	}

/**

* 2nd phase constructor

*/

void CHarvesterOMADRMPlugin::ConstructL()

	{

	WRITELOG( "CHarvesterOMADRMPlugin::ConstructL()" );

    SetPriority( KHarvesterPriorityHarvestingPlugin - 1 );	

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

    iPropDefs = CHarvesterOmaDrmPluginPropertyDefs::NewL();

	TFileName phoneRoot = PathInfo::PhoneMemoryRootPath();

	TFileName mmcRoot = PathInfo::MemoryCardRootPath();

	TFileName images = PathInfo::ImagesPath();

    TFileName phoneImagePath( phoneRoot );

    phoneImagePath.Append( images );

    iPhoneImagesPath = phoneImagePath.AllocL();

    TFileName mmcImagePath( mmcRoot );

    mmcImagePath.Append( images );

    iMmcImagesPath = mmcImagePath.Right( mmcImagePath.Length() - 1 ).AllocL();

    TFileName videos = PathInfo::VideosPath();

    TFileName phoneVideoPath( phoneRoot );

    phoneVideoPath.Append( videos );

    iPhoneVideosPath = phoneVideoPath.AllocL();

    TFileName mmcVideoPath( mmcRoot );

    mmcVideoPath.Append( videos );

    iMmcVideosPath = mmcVideoPath.Right( mmcVideoPath.Length() - 1 ).AllocL();

    TFileName sounds = PathInfo::SoundsPath();

    TFileName phoneSoundPath( phoneRoot );

    phoneSoundPath.Append( sounds );

    iPhoneSoundsPath = phoneSoundPath.AllocL();

    TFileName mmcSoundPath( mmcRoot );

    mmcSoundPath.Append( sounds );

    iMmcSoundsPath = mmcSoundPath.Right( mmcSoundPath.Length() - 1 ).AllocL();

	}

void CHarvesterOMADRMPlugin::HarvestL( CHarvesterData* aHarvesterData )

	{

    CMdEObject& mdeObject = aHarvesterData->MdeObject();

	CDRMHarvestData* drmHarvestData = CDRMHarvestData::NewL();

	CleanupStack::PushL( drmHarvestData );

    CFileData* fileData = CFileData::NewL();

    CleanupStack::PushL( fileData );

    CHarvestData* harvestData = CHarvestData::NewL();

    CleanupStack::PushL( harvestData );

    TInt errorCode( KErrNone );

    TRAPD( error, errorCode = GatherDataL( mdeObject, *drmHarvestData, *fileData, *harvestData ) );

    if ( error == KErrNone && (errorCode == KErrNone || errorCode == KErrCompletion ) ) // ok, something got harvested

        {

        if ( mdeObject.Id() == 0 || mdeObject.Placeholder() ) // is a new object or placeholder

            {

            TRAP_IGNORE( HandleObjectPropertiesL( *harvestData,  *drmHarvestData, *fileData, *aHarvesterData, ETrue ) );

            mdeObject.SetPlaceholder( EFalse );

            }

        else   // not a new object

            {

            TRAP_IGNORE( HandleObjectPropertiesL( *harvestData, *drmHarvestData, *fileData, *aHarvesterData, EFalse ) );

            }

        if ( error != KErrNone )

            {

            WRITELOG1( "CHarvesterOMADRMPlugin::HarvestL() - Handling object failed: ", error );

            }

    	}

    else	

        {

        WRITELOG1( "CHarvesterOMADRMPlugin::HarvestL() - TRAP error: %d, errorCode %d", error );

        TInt convertedError = KErrNone;

        MdsUtils::ConvertTrapError( error, convertedError );

        aHarvesterData->SetErrorCode( convertedError );

        }

    CleanupStack::PopAndDestroy( 3, drmHarvestData );

	}

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

// GatherDataL

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

//

TInt CHarvesterOMADRMPlugin::GatherDataL( CMdEObject& aMetadataObject, CDRMHarvestData& aDRMharvestData, 

        CFileData& aFileData, CHarvestData& /*aHarvestData*/ )

    {

    WRITELOG( "CHarvesterOMADRMPlugin::GatherDataL" );

    TEntry* entry = new (ELeave) TEntry();

    CleanupStack::PushL( entry );

    const TDesC& uri = aMetadataObject.Uri();

    User::LeaveIfError( iFs.Entry( uri, *entry ) );

    aDRMharvestData.iModified = entry->iModified;

    aDRMharvestData.iFileSize = (TUint)entry->iSize;

    CleanupStack::PopAndDestroy( entry );

    ContentAccess::CContent* content = ContentAccess::CContent::NewLC( uri );   

    ContentAccess::CData* data = content->OpenContentLC( ContentAccess::EPeek );

    ContentAccess::RStringAttributeSet attrSet;

    CleanupClosePushL( attrSet );

    attrSet.AddL( ContentAccess::EDescription );

    attrSet.AddL( ContentAccess::EMimeType );

    attrSet.AddL( ContentAccess::ETitle );

    attrSet.AddL( ContentAccess::EAuthor );

    attrSet.AddL( ContentAccess::EGenre );

    User::LeaveIfError( data->GetStringAttributeSet(attrSet) );

    TInt err = attrSet.GetValue( ContentAccess::EDescription, aDRMharvestData.iDescription );

    if ( err != KErrNone)

        {

        WRITELOG1( "CHarvesterOMADRMPlugin::GatherDataL - ERROR: getting description failed %d", err );

        }

    if ( aDRMharvestData.iDescription.Length() <= 0 )

        {

        WRITELOG( "CHarvesterOMADRMPlugin::GatherDataL - no description" );

        }

    err = attrSet.GetValue( ContentAccess::EMimeType, aDRMharvestData.iMimetype );

    if ( err != KErrNone)

        {

        WRITELOG1( "CHarvesterOMADRMPlugin::GatherDataL - ERROR: getting mimetype failed %d", err );

        }

    if ( aDRMharvestData.iMimetype.Length() <= 0 )

        {

        WRITELOG( "CHarvesterOMADRMPlugin::GatherDataL - no mimetype" );

        }

    err = attrSet.GetValue( ContentAccess::ETitle, aDRMharvestData.iTitle );

    if ( err != KErrNone)

        {

        WRITELOG1( "CHarvesterOMADRMPlugin::GatherDataL - ERROR: getting title failed %d", err );

        }

    if ( aDRMharvestData.iTitle.Length() <= 0 )

        {

        WRITELOG( "CHarvesterOMADRMPlugin::GatherDataL - no title" );

        }

    err = attrSet.GetValue( ContentAccess::EAuthor, aDRMharvestData.iAuthor );

    if ( err != KErrNone)

        {

        WRITELOG1( "CHarvesterOMADRMPlugin::GatherDataL - ERROR: getting author failed %d", err );

        }

    if ( aDRMharvestData.iAuthor.Length() <= 0 )

        {

        WRITELOG( "CHarvesterOMADRMPlugin::GatherDataL - no author" );

        }

    err = attrSet.GetValue( ContentAccess::EGenre, aDRMharvestData.iGenre );

    if ( err != KErrNone)

        {

        WRITELOG1( "CHarvesterOMADRMPlugin::GatherDataL - ERROR: getting genre failed %d", err );

        }

    if ( aDRMharvestData.iGenre.Length() <= 0 )

        {

        WRITELOG( "CHarvesterOMADRMPlugin::GatherDataL - no genre" );

        }

    err = content->GetAttribute( ContentAccess::EIsProtected, aDRMharvestData.iDrmProtected );

    if ( err != KErrNone)

        {

        WRITELOG1( "CHarvesterOMADRMPlugin::GatherDataL - ERROR: getting protection info failed %d", err );

        }

        }

    return KErrNone;

    }

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

// HandleObjectPropertiesL

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

//

void CHarvesterOMADRMPlugin::HandleObjectPropertiesL( CHarvestData& /*aHarvestData*/, CDRMHarvestData& aDRMharvestData, CFileData& aFileData, 

        CHarvesterData& aHarvesterData, TBool aIsAdd )

    {

    WRITELOG("CHarvesterOMADRMPlugin - HandleNewObject ");

    CMdEObject& mdeObject = aHarvesterData.MdeObject();

    InitPropDefsL( mdeObject.Def() );

    TTimeIntervalSeconds timeOffset = User::UTCOffset();

    TPtrC objectDefName( mdeObject.Def().Name());

    if( ! mdeObject.Placeholder() )

    	{

    	// Creation date

    	TTime localTime = aDRMharvestData.iModified + timeOffset;

    	CMdeObjectWrapper::HandleObjectPropertyL(mdeObject, 

    			*iPropDefs->iCreationDatePropertyDef, &localTime, aIsAdd );

    	// Last modified date

    	CMdeObjectWrapper::HandleObjectPropertyL(mdeObject, 

    			*iPropDefs->iLastModifiedDatePropertyDef, &aDRMharvestData.iModified, aIsAdd );

    	// File size

    	CMdeObjectWrapper::HandleObjectPropertyL(mdeObject, 

    			*iPropDefs->iSizePropertyDef, &aDRMharvestData.iFileSize, aIsAdd );

    	TPtrC objectDefName( mdeObject.Def().Name());

        if( objectDefName == MdeConstants::Image::KImageObject )

            {

            const TDesC& uri = mdeObject.Uri();

            if( uri.FindF( iMmcImagesPath->Des()) != KErrNotFound ||

                uri.FindF( iPhoneImagesPath->Des()) != KErrNotFound ||

                uri.FindF( KDCIMFolder ) != KErrNotFound )

                {

                TBool inDefaultFolder( ETrue );

                CMdeObjectWrapper::HandleObjectPropertyL(mdeObject, *iPropDefs->iDefaultFolderPropertyDef, &inDefaultFolder, aIsAdd );

                }

            else

                {

                TBool inDefaultFolder( EFalse );

                CMdeObjectWrapper::HandleObjectPropertyL(mdeObject, *iPropDefs->iDefaultFolderPropertyDef, &inDefaultFolder, aIsAdd );    

                }    

            }

        else if( objectDefName == MdeConstants::Video::KVideoObject )

            {

            const TDesC& uri = mdeObject.Uri();

            if( uri.FindF( iMmcVideosPath->Des()) != KErrNotFound ||

                uri.FindF( iPhoneVideosPath->Des()) != KErrNotFound ||

                uri.FindF( KDCIMFolder ) != KErrNotFound )

                {

                TBool inDefaultFolder( ETrue );

                CMdeObjectWrapper::HandleObjectPropertyL(mdeObject, *iPropDefs->iDefaultFolderPropertyDef, &inDefaultFolder, aIsAdd );

                }

            else

                {

                TBool inDefaultFolder( EFalse );

                CMdeObjectWrapper::HandleObjectPropertyL(mdeObject, *iPropDefs->iDefaultFolderPropertyDef, &inDefaultFolder, aIsAdd );    

                }    

            }

        else if( objectDefName == MdeConstants::Audio::KAudioObject )

            {

            const TDesC& uri = mdeObject.Uri();

            if( uri.FindF( iMmcSoundsPath->Des()) != KErrNotFound ||

                uri.FindF( iPhoneSoundsPath->Des()) != KErrNotFound )

                {

                TBool inDefaultFolder( ETrue );

                CMdeObjectWrapper::HandleObjectPropertyL(mdeObject, *iPropDefs->iDefaultFolderPropertyDef, &inDefaultFolder, aIsAdd );

                }

            else

                {

                TBool inDefaultFolder( EFalse );

                CMdeObjectWrapper::HandleObjectPropertyL(mdeObject, *iPropDefs->iDefaultFolderPropertyDef, &inDefaultFolder, aIsAdd );    

                }     

            }

    	}

    // Item Type

    if(aDRMharvestData.iMimetype.Length() > 0)

        {

        TBool isAdd( EFalse );

        CMdEProperty* prop = NULL;

        TInt index = mdeObject.Property( *iPropDefs->iItemTypePropertyDef, prop );

        if( index < 0 )

            {

            isAdd = ETrue;

            }

        CMdeObjectWrapper::HandleObjectPropertyL(mdeObject, 

                *iPropDefs->iItemTypePropertyDef, &aDRMharvestData.iMimetype, isAdd );

        }

    // DRM protection

    CMdeObjectWrapper::HandleObjectPropertyL(mdeObject, 

    		*iPropDefs->iDrmPropertyDef, &aDRMharvestData.iDrmProtected, aIsAdd );

    // Title (is set from URI by default)

    if( aDRMharvestData.iTitle.Length() > 0 && aDRMharvestData.iTitle.Length() < KMaxTitleFieldLength )

    	{

    	CMdeObjectWrapper::HandleObjectPropertyL(mdeObject, 

    			*iPropDefs->iTitlePropertyDef, &aDRMharvestData.iTitle, EFalse );

    	}

    // Description

    if( aDRMharvestData.iDescription.Length() > 0 && aDRMharvestData.iDescription.Length() < iMaxTextLength )

    	{

    	CMdeObjectWrapper::HandleObjectPropertyL(mdeObject, 

    			*iPropDefs->iDescriptionPropertyDef, &aDRMharvestData.iDescription, aIsAdd );

    	}   

    // Author

    if( aDRMharvestData.iAuthor.Length() > 0 && aDRMharvestData.iAuthor.Length() < iMaxTextLength )

    	{

    	CMdeObjectWrapper::HandleObjectPropertyL(mdeObject, 

    			*iPropDefs->iAuthorPropertyDef, &aDRMharvestData.iAuthor, aIsAdd );

    	}

    // Genre

    if( aDRMharvestData.iGenre.Length() > 0 && aDRMharvestData.iGenre.Length() < iMaxTextLength )

        {

        CMdeObjectWrapper::HandleObjectPropertyL(mdeObject, 

                *iPropDefs->iGenrePropertyDef, &aDRMharvestData.iGenre, aIsAdd );

        }

    if( objectDefName == MdeConstants::Image::KImageObject )

        {

      // Image - Bits per Sample

        if (aFileData.iBitsPerPixel != 0)

            {

            CMdeObjectWrapper::HandleObjectPropertyL(mdeObject, *iPropDefs->iBitsPerSamplePropertyDef, &aFileData.iBitsPerPixel, aIsAdd );

            }

        // Image - Framecount

        CMdeObjectWrapper::HandleObjectPropertyL(mdeObject, *iPropDefs->iFrameCountPropertyDef, &aFileData.iFrameCount, aIsAdd );

        // MediaObject - Width

        if (aFileData.iImageWidth != 0)

            {

            CMdeObjectWrapper::HandleObjectPropertyL(mdeObject, *iPropDefs->iWidthPropertyDef, &aFileData.iImageWidth, aIsAdd );

            }