/*
* Copyright (c) 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 "fmdrivemodel.h"
#include "fmfileiconprovider.h"
#include "fmutils.h"

#include <QDir>
#include <QFileInfo>

#include <hbglobal.h>

FmDriveModel::FmDriveModel( QObject *parent, Options options,
        FmDriveListProvider *driveListProvider ) :
    QAbstractListModel( parent ), mOptions( options ), mDriveListProvider( driveListProvider )
{
    mIconProvider = new FmFileIconProvider();
    refresh();
}

FmDriveModel::~FmDriveModel()
{
    delete mIconProvider;
}

void FmDriveModel::refresh()
{
    emit layoutAboutToBeChanged();
    mDriveList.clear();
    
    // if mDriveListProvider existed, use it to fetch drive list
    // otherwise use FmUtils::getDriveList to fetch drive list.
    if( mDriveListProvider ) {
        mDriveListProvider->getDriveList( mDriveList );
    } else {
        if( mOptions & HideUnAvailableDrive ) {
            FM_LOG( QString( "FmDriveModel::refresh HideUnAvailableDrive_true" ) );
            FmUtils::getDriveList( mDriveList, true );
        } else {
            FM_LOG( QString( "FmDriveModel::refresh HideUnAvailableDrive_false" ) );
            FmUtils::getDriveList( mDriveList, false );
        }
    }

	emit layoutChanged();
	for( int i=0; i<mDriveList.count(); i++ ) {
        emit dataChanged(index( i, 0 ), index( i, 0 ));
	}
}


int FmDriveModel::rowCount( const QModelIndex &parent ) const
{
    if (!parent.isValid())
        return mDriveList.size();

    return 0;
}

int FmDriveModel::columnCount( const QModelIndex &parent ) const
{
    Q_UNUSED( parent );
    return 1;
}

QVariant FmDriveModel::data( const QModelIndex &index, int role ) const
{
    if (!indexValid( index ))
        return QVariant();

    if (role == Qt::DisplayRole ) {
       return displayString( index );
    } 
    if (index.column() == 0) {
        if (role == Qt::DecorationRole ) {
            QString path = driveName( index );
            return mIconProvider->icon( QFileInfo( path ) );
        }
    }

    if (index.column() == 1 && role == Qt::TextAlignmentRole) {
        return Qt::AlignRight;
    }

    return QVariant();
}


QVariant FmDriveModel::headerData( int section, Qt::Orientation orientation, int role ) const
{
    if (orientation == Qt::Horizontal) {
        if (role != Qt::DisplayRole)
            return QVariant();

        switch (section) {
            case 0: return hbTrId("Name");
            case 1: return hbTrId("Size");
            case 2: return hbTrId("Type");
            case 3: return hbTrId("Date Modified");
            default: return QVariant();
        }
    }

    return QAbstractItemModel::headerData( section, orientation, role );
}

bool FmDriveModel::indexValid( const QModelIndex &index ) const
{
    if( !(&index) )
        return false;
    return true;
}

QString FmDriveModel::driveName( const QModelIndex &index ) const
{
    QString data;
    if (index.row() >= 0 && index.row() < mDriveList.size()) {
        int row = index.row();
        QString diskName = mDriveList[ row ];
        data = diskName;
    }
    return data;
}

QVariant FmDriveModel::displayString( const QModelIndex &index ) const
{
    QStringList data;
    if (index.row() >= 0 && index.row() < mDriveList.size()) {
        int row = index.row();
        QString diskName = mDriveList[ row ];

        if( mOptions & FillWithVolume ) {
            data << FmUtils::fillDriveVolume( diskName, mOptions & FillWithDefaultVolume );
        } else {
            data << FmUtils::removePathSplash( diskName );
        }
        diskName = FmUtils::fillPathWithSplash( diskName );
        FmDriverInfo driverInfo = FmUtils::queryDriverInfo( diskName );
        if ( mOptions & FillWithTotalSize ) {                    
            if( driverInfo.driveState() & FmDriverInfo::EDriveAvailable ) {
                data << QString( hbTrId ( "Size: " ) + FmUtils::formatStorageSize( driverInfo.size() ) );                
            } else if( driverInfo.driveState() & FmDriverInfo::EDriveLocked ) {
                data << QString( hbTrId ( "Locked" ) );                
            } else if( driverInfo.driveState() & FmDriverInfo::EDriveCorrupted ) {
                data << QString( hbTrId ( "Corrupted" ) );                
            } else if( driverInfo.driveState() & FmDriverInfo::EDriveNotPresent ) {
                data << QString( hbTrId ( "Not Ready" ) );                
            }            
        }
        if ( mOptions & FillWithFreeSize ) {        
            if( driverInfo.driveState() & FmDriverInfo::EDriveAvailable ) {                
                data << QString( hbTrId ( "Free: " ) + FmUtils::formatStorageSize( driverInfo.freeSize() ) );
            } else if( ( driverInfo.driveState() & FmDriverInfo::EDriveLocked ) ||
                    ( driverInfo.driveState() & FmDriverInfo::EDriveCorrupted ) ||
                    ( driverInfo.driveState() & FmDriverInfo::EDriveNotPresent )) {                
                data << QString( QString(" ") );
            }            
        }
    }
    return data;
}