// Copyright (c) 1998-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 "US_STD.H"

// Class CDbStoreIndex::CRecover

NONSHARABLE_CLASS(CDbStoreIndex::CRecover) : public CDbTableDatabase::CStepper
	{
private:
	enum {EClearStep=1};
public:
	CRecover(CDbStoreDatabase& aDatabase,const CDbStoreDef& aTable,const CDbStoreIndexDef& anIndex);
	~CRecover();
	static inline TInt Steps(TInt aCardinality);
private:
	TInt StepL(TInt aStep);
private:
	CDbStoreDatabase& iDatabase;
	const CDbStoreDef& iTable;
	const CDbStoreIndexDef& iIndex;
	CDbTableDatabase::CBuildIndex* iBuild;
	};

inline TInt CDbStoreIndex::CRecover::Steps(TInt aCardinality)
	{return CDbTableDatabase::CBuildIndex::Steps(aCardinality)+EClearStep;}

CDbStoreIndex::CRecover::CRecover(CDbStoreDatabase& aDatabase,const CDbStoreDef& aTable,const CDbStoreIndexDef& anIndex)
	: iDatabase(aDatabase),iTable(aTable),iIndex(anIndex)
	{}

CDbStoreIndex::CRecover::~CRecover()
	{
	delete iBuild;
	}

TInt CDbStoreIndex::CRecover::StepL(TInt aStep)
	{
	if (iBuild==NULL)
		{	// first step
		iBuild=CDbTableDatabase::CBuildIndex::NewL(iDatabase,iTable,iIndex);
		((CDbStoreIndex&)iBuild->Index()).DiscardL();
		return aStep-EClearStep;
		}
	aStep=iBuild->StepL(aStep);
	if (aStep==0)
		{	// final step
		iDatabase.IndexRecoveredL();
		delete iBuild;
		iBuild=NULL;
		}
	return aStep;
	}

// Class CDbStoreIndex::CRepair

NONSHARABLE_CLASS(CDbStoreIndex::CRepair) : public CDbTableDatabase::CStepper
	{
private:
	enum {ERepairStep=1};
public:
	CRepair(CDbStoreDatabase& aDatabase,const CDbStoreDef& aTable,const CDbStoreIndexDef& anIndex);
	static inline TInt Steps();
private:
	TInt StepL(TInt aStep);
private:
	CDbStoreDatabase& iDatabase;
	const CDbStoreDef& iTable;
	const CDbStoreIndexDef& iIndex;
	};

inline TInt CDbStoreIndex::CRepair::Steps()
	{return ERepairStep;}

CDbStoreIndex::CRepair::CRepair(CDbStoreDatabase& aDatabase,const CDbStoreDef& aTable,const CDbStoreIndexDef& anIndex)
	: iDatabase(aDatabase),iTable(aTable),iIndex(anIndex)
	{}

TInt CDbStoreIndex::CRepair::StepL(TInt)
	{
	CDbStoreIndex* index=CDbStoreIndex::NewL(iDatabase,iIndex,iTable);
	CleanupStack::PushL(index);
	index->OpenL();
	index->RepairL();
	CleanupStack::PopAndDestroy();
	return 0;
	}

// Class CDbStoreDatabase

CDbTableDatabase::CStepper* CDbStoreDatabase::RecoverL(TInt& aStep)
	{
	TSglQueIterC<CDbStoreDef> iter(SchemaL());	// ensure we have the schema
	TInt count=0;
	const CDbStoreDef* tDef;
	while ((tDef=iter++)!=NULL)
		count+=tDef->Indexes().Count();
	CMultiStepper* mStepper=CMultiStepper::NewL(count);
	CleanupStack::PushL(mStepper);
// damaged ones first (as they are invoked last
	count=0;
	iter.SetToFirst();
	while ((tDef=iter++)!=NULL)
		{
		TInt records=CDbStoreRecords::CardinalityL(Store(),*tDef);
		TSglQueIterC<CDbStoreIndexDef> xIter(tDef->Indexes().AsQue());
		for (const CDbStoreIndexDef* xDef;(xDef=xIter++)!=NULL;)
			{
			if (!CDbStoreIndex::IsDamagedL(*this,*xDef))
				continue;
			// rebuild from scratch
			CDbStoreIndex::CRecover* rebuild=new(ELeave) CDbStoreIndex::CRecover(*this,*tDef,*xDef);
			mStepper->AddStepper(rebuild,rebuild->Steps(records));
			++count;
			}
		}
	if (count==0)
		{	// no indexes are actually damaged
		aStep=0;
		CleanupStack::PopAndDestroy();
		return NULL;
		}
	// repair the other ones after a reclaim
	iter.SetToFirst();
	while ((tDef=iter++)!=NULL)
		{
		TSglQueIterC<CDbStoreIndexDef> xIter(tDef->Indexes().AsQue());
		for (const CDbStoreIndexDef* xDef;(xDef=xIter++)!=NULL;)
			{
			if (CDbStoreIndex::IsDamagedL(*this,*xDef))
				continue;
			// repair from sequence set
			CDbStoreIndex::CRepair* repair=new(ELeave) CDbStoreIndex::CRepair(*this,*tDef,*xDef);
			mStepper->AddStepper(repair,repair->Steps());
			}
		}
	iPagePool->Discard();
	iPagePool->ReclaimAllL();	// recover all page pool space
	aStep=mStepper->TotalSteps();
	CleanupStack::Pop();
	return mStepper;
	}