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

#include "cmtppkgidstore.h"
#include "dbutility.h"
#include "mtpdebug.h"
#include "OstTraceDefinitions.h"
#ifdef OST_TRACE_COMPILER_IN_USE
#include "cmtppkgidstoreTraces.h"
#endif


/**
Two-phase construction
@param aDatabase    The reference to the database object
@return pointer to the created <pkgid, dpid> map instance
*/
CMTPPkgIDStore* CMTPPkgIDStore::NewL(RDbDatabase& aDatabase)
    {
    CMTPPkgIDStore* self = new (ELeave) CMTPPkgIDStore(aDatabase);
    CleanupStack::PushL(self);
    self->ConstructL();
    CleanupStack::Pop(self);
    return self;
    }


/**
Standard c++ constructor
*/  
CMTPPkgIDStore::CMTPPkgIDStore(RDbDatabase& aDatabase)
    :iDatabase(aDatabase)
    {
    }

/**
Second phase constructor.
*/    
void CMTPPkgIDStore::ConstructL()
    {
    CreatePkgIDStoreTableL();
    }

/**
Create table to store the mapping from object handle to other properties 
(data provider id, storage id, formatcode, etc.)
*/    
void CMTPPkgIDStore::CreatePkgIDStoreTableL()
    {   
    _LIT(KSQLPkgIDTableName, "PkgIDStore");
    if (!DBUtility::IsTableExistsL(iDatabase, KSQLPkgIDTableName))
        {
        _LIT(KSQLCreatePkgIDTableText,
            "CREATE TABLE PkgIDStore (DataProviderId UNSIGNED INTEGER, PkgId UNSIGNED INTEGER)");
        LEAVEIFERROR(iDatabase.Execute(KSQLCreatePkgIDTableText),
                OstTrace0( TRACE_ERROR, CMTPPKGIDSTORE_CREATEPKGIDSTORETABLEL, "TABLE PkgIDStore create failed!"));
        }
    _LIT(KSQLGetPKGID, "SELECT * FROM PkgIDStore");
    iSqlStatement.Format(KSQLGetPKGID);    
    RDbView view;
    CleanupClosePushL(view);
    LEAVEIFERROR(view.Prepare(iDatabase, TDbQuery(iSqlStatement)),
            OstTrace0( TRACE_ERROR, DUP1_CMTPPKGIDSTORE_CREATEPKGIDSTORETABLEL, "view for PkgIDStore prepare failed!" ));   
    LEAVEIFERROR(view.Evaluate(),
            OstTrace0( TRACE_ERROR, DUP2_CMTPPKGIDSTORE_CREATEPKGIDSTORETABLEL, "view evaluate failed!"));   
    while (view.NextL())
        {
        view.GetL();
        iDPIDs.AppendL(view.ColInt64(EPkgIDStoreDataProviderId));
        iPkgIDs.AppendL(view.ColInt64(EPkgIDStorePKGID));
        }
    CleanupStack::PopAndDestroy(&view); 
    }
    
/**
Destructor.
*/    
CMTPPkgIDStore::~CMTPPkgIDStore()
    {
    iDPIDs.Close();
    iPkgIDs.Close();
    }


void CMTPPkgIDStore::InsertPkgIdL(TUint aDPId, TUint aPkgId)
    {       
    TInt index = 0;
    TRAPD(err, index = iDPIDs.FindL(aDPId));
    if(KErrNotFound == err)
        {
        _LIT(KSQLInsertPkgIDObjectText, "INSERT INTO PkgIDStore (DataProviderId, PkgId) VALUES (%u, %u)");
        iSqlStatement.Format(KSQLInsertPkgIDObjectText, aDPId, aPkgId);
        LEAVEIFERROR(iDatabase.Execute(iSqlStatement),
                OstTrace0( TRACE_ERROR, CMTPPKGIDSTORE_INSERTPKGIDL, "INSERT INTO PkgIDStore failed!" ));
        iDPIDs.AppendL(aDPId);
        iPkgIDs.AppendL(aPkgId);
        }
    else
        {
        if(aPkgId != iPkgIDs[index])
            {
            _LIT(KSQLSetPkgIDObjectText, "UPDATE PkgIDStore SET PkgId = %u WHERE DataProviderId = %u");
            iSqlStatement.Format(KSQLSetPkgIDObjectText, aPkgId,aDPId);
            LEAVEIFERROR(iDatabase.Execute(iSqlStatement),
                    OstTrace0( TRACE_ERROR, DUP1_CMTPPKGIDSTORE_INSERTPKGIDL, "UPDATE PkgIDStore SET PkgId failed!" ));    
            iPkgIDs[index] = aPkgId;
            }
        }
    }

const RArray<TUint>& CMTPPkgIDStore::DPIDL() const
    {
    __ASSERT_DEBUG((iPkgIDs.Count() == iDPIDs.Count()), User::Invariant());
    return iDPIDs;
    }

TUint CMTPPkgIDStore::PKGIDL (TUint aIndex) const
    {
    __ASSERT_DEBUG((aIndex < iPkgIDs.Count()), User::Invariant());
    return iPkgIDs[aIndex];
    }

TInt CMTPPkgIDStore::RemoveL(TUint aDpId)
    {
    TInt index = iDPIDs.Find(aDpId);
    if(KErrNotFound != index)
        {
        iDPIDs.Remove(index);
        iPkgIDs.Remove(index);
        _LIT(KSQLDeleteObjectText, "DELETE FROM PkgIDStore WHERE DataProviderId = %u");
        iSqlStatement.Format(KSQLDeleteObjectText, aDpId);
        LEAVEIFERROR(iDatabase.Execute(iSqlStatement),
                OstTrace0( TRACE_ERROR, CMTPPKGIDSTORE_REMOVEL, "DELETE FROM PkgIDStore failed!" ));
        }
    return index;
    }