/*

* Copyright (c) 2007-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:  This is disk space notifier for Metadata server.*

*/

// INCLUDE FILES

#include "mdsdiskspacenotifier.h"

CMdSDiskSpaceNotifierAO* CMdSDiskSpaceNotifierAO::NewL(

	MMdSDiskSpaceNotifierObserver& aObserver, TInt64 aThreshold, const TDesC& aFilename)

	{

	CMdSDiskSpaceNotifierAO* self = 

		CMdSDiskSpaceNotifierAO::NewLC( aObserver, aThreshold, aFilename );

	CleanupStack::Pop( self );

	return self;

	}

CMdSDiskSpaceNotifierAO* CMdSDiskSpaceNotifierAO::NewLC(

	MMdSDiskSpaceNotifierObserver& aObserver, TInt64 aThreshold, const TDesC& aFilename)

	{

	TDriveNumber driveNumber = GetDriveNumberL( aFilename );

    CMdSDiskSpaceNotifierAO* self = 

    	new ( ELeave ) CMdSDiskSpaceNotifierAO( aObserver, aThreshold, driveNumber );

    CleanupStack::PushL( self );

    self->ConstructL();

    return self;

	}

CMdSDiskSpaceNotifierAO::~CMdSDiskSpaceNotifierAO()

	{

	Cancel();

	iFileServerSession.Close();

	}

void CMdSDiskSpaceNotifierAO::RunL()

	{	

	TVolumeInfo volumeInfo;

	if ( iState == CMdSDiskSpaceNotifierAO::ENormal )

		{

		TInt status = iStatus.Int();

		switch( status )

			{

			case KErrNone:

			    {

				const TInt error = iFileServerSession.Volume( volumeInfo, iDrive );

				if( error != KErrNone )

				    {

				    StartNotifier();

				    break;

				    }

				// Check if free space is less than threshold level

				if( volumeInfo.iFree < iThreshold )

					{

					iObserver.HandleDiskSpaceNotificationL( MMdSDiskSpaceNotifierObserver::ELess );

					iDiskFull = ETrue;

					iState = EIterate;

					iIterationCount = 0;

					SetActive();

					TRequestStatus* status = &iStatus;

					User::RequestComplete( status, KErrNone );

					return;

					}

				else

					{

					iObserver.HandleDiskSpaceNotificationL( MMdSDiskSpaceNotifierObserver::EMore );

					iDiskFull = EFalse;

					}

				StartNotifier();

				break;

			    }

			case KErrArgument:

				User::Leave( status );

				break;

			default:

				break;

			}

		}

	else if ( iState == CMdSDiskSpaceNotifierAO::EIterate )

		{

		const TInt KMaxIterations = 10;

        TInt error = iFileServerSession.Volume( volumeInfo, iDrive );

        if( error != KErrNone )

            {

            iState = ENormal;

            iIterationCount = 0;

            StartNotifier();

            return;

            }

		if ( volumeInfo.iFree < iThreshold )

			{

			iObserver.HandleDiskSpaceNotificationL( MMdSDiskSpaceNotifierObserver::ELess );

			++iIterationCount;

			if ( iIterationCount < KMaxIterations )

				{

				SetActive();

				TRequestStatus* status = &iStatus;

				User::RequestComplete( status, KErrNone );

				return;

				}

			else

				{

		        error = iFileServerSession.Volume( volumeInfo, iDrive );

		        if( error != KErrNone )

		            {

		            iState = ENormal;

		            iIterationCount = 0;

		            StartNotifier();

		            return;

		            }

				if ( volumeInfo.iFree >= iThreshold )

					{

					iDiskFull = EFalse;

					}

				}

			}

		else

			{

			iDiskFull = EFalse;

			}

		iState = ENormal;

		iIterationCount = 0;

		StartNotifier();			

		}

	else

		{

		User::Leave( KErrGeneral );

		}

	}

TInt CMdSDiskSpaceNotifierAO::RunError(TInt aError)

	{

	iObserver.HandleDiskSpaceError( aError );

	return KErrNone;

	}

void CMdSDiskSpaceNotifierAO::DoCancel()

	{

	if( IsActive() )

		{	

		iFileServerSession.NotifyDiskSpaceCancel();

		}

	}

CMdSDiskSpaceNotifierAO::CMdSDiskSpaceNotifierAO(

	MMdSDiskSpaceNotifierObserver& aObserver, TInt64 aThreshold, TDriveNumber aDrive)

	: CActive( CActive::EPriorityStandard ), 

	iObserver( aObserver ), iThreshold( aThreshold ), iDrive( aDrive ), iState( CMdSDiskSpaceNotifierAO::ENormal ), iDiskFull( EFalse )

	{

	CActiveScheduler::Add( this );

	}

void CMdSDiskSpaceNotifierAO::ConstructL()

	{	

	TInt KMessageSlotCount = 2; // slots for NotifyDiskSpace and NotifyDiskSpaceCancel

	User::LeaveIfError( iFileServerSession.Connect( KMessageSlotCount ) );

	TVolumeInfo volumeInfo;

	const TInt error = iFileServerSession.Volume( volumeInfo, iDrive );	

	if ( (error != KErrNone) || volumeInfo.iFree < iThreshold )

		{

		iDiskFull = ETrue;

		}

	StartNotifier();

	}

void CMdSDiskSpaceNotifierAO::StartNotifier()

	{	

	iFileServerSession.NotifyDiskSpace( iThreshold, iDrive, iStatus );

	SetActive();

	}

TDriveNumber CMdSDiskSpaceNotifierAO::GetDriveNumberL( const TDesC& aFilename )

	{

	TLex driveParser( aFilename );

	TChar driveChar = driveParser.Get();

	if( 0 == driveChar || TChar( ':' ) != driveParser.Peek() )

		{

		User::Leave( KErrArgument );

		}

	TInt driveNumber;

	RFs::CharToDrive( driveChar, driveNumber );

	return (TDriveNumber)driveNumber;

	}

TBool CMdSDiskSpaceNotifierAO::DiskFull() const

	{

	return iDiskFull;

	}