/*
* Copyright (c) 2010 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 "fbdrivemodel.h"
#include "enginewrapper.h"
#include "fbdriveentry.h"
#include "filebrowsersettings.h"
#include "FB.hrh"

#include <QModelIndex>
#include <QFileIconProvider>

/**
  Constructs a file browser custom system model with the given \a engineWrapper and \a parent.
  */
FbDriveModel::FbDriveModel(EngineWrapper *engineWrapper, QObject *parent) :
    QAbstractListModel(parent),
    mEngineWrapper(engineWrapper),
    mFileIconProvider(0)
{
    mFileIconProvider = new QFileIconProvider();
}

/**
  Destroys this file browser custom system model.
  */
FbDriveModel::~FbDriveModel()
{  
    if (mFileIconProvider) {
        delete mFileIconProvider;
    }
}

/**
  \reimp
  */
int FbDriveModel::rowCount(const QModelIndex &parent) const
{
    Q_UNUSED(parent);
    return mEngineWrapper->itemCount();
}

/**
  \reimp
  */
QVariant FbDriveModel::data(const QModelIndex &index, int role) const
{
    if (!index.isValid() || index.model() != this)
        return QVariant();

    switch (role) {
    case Qt::EditRole:
    case Qt::DisplayRole: {
            QStringList listItem;
            FbDriveEntry driveEntry(mEngineWrapper->getDriveEntry(index));
            if (mEngineWrapper->settings().fileViewMode() == EFileViewModeSimple)
            {
                const QString SimpleDriveEntry("%1: <%2>");
                listItem /*<< driveEntry.IconId() */
                        << SimpleDriveEntry.arg(driveEntry.driveLetter())
                                           .arg(driveEntry.mediaTypeString());
            } else if (mEngineWrapper->settings().fileViewMode() == EFileViewModeExtended) {
                const QString SimpleDriveEntry("%1: <%2>");
                const QString ExtendedDriveEntry("%1/%2 kB");
                listItem /*<< driveEntry.IconId()*/
                        << SimpleDriveEntry.arg(driveEntry.driveLetter())
                                           .arg(driveEntry.mediaTypeString())
                        << ExtendedDriveEntry.arg(QString::number(driveEntry.volumeInfoFree()/1024))
                                             .arg(QString::number(driveEntry.volumeInfoSize()/1024));

            }
            return listItem;
        }
    case Qt::DecorationRole: {
            if (mEngineWrapper) {
                QIcon icon;
                //TODO Drive ico has to be provided, for some reason it is not visible
                icon = mFileIconProvider->icon(QFileIconProvider::Drive);
                return QVariant(icon);
            }
        }
    }
    return QVariant();
}

/**
  \reimp
  */
QVariant FbDriveModel::headerData(int section, Qt::Orientation orientation, int role) const
{
    Q_UNUSED(section);
    Q_UNUSED(orientation);
    Q_UNUSED(role);

    // TODO, implement or remove
    return QVariant();
}


FbDriveEntry FbDriveModel::driveEntry(const QModelIndex &index) const
{
    return mEngineWrapper->getDriveEntry(index);
}

QString FbDriveModel::driveLetter(const QModelIndex &index) const
{
    QString diskLetter;
    if (index.row() >= 0 && index.row() < mEngineWrapper->itemCount()) {
        FbDriveEntry driveEntry(mEngineWrapper->getDriveEntry(index));
        diskLetter = driveEntry.driveLetter();
    }
    return diskLetter;
}

QString FbDriveModel::mediaTypeString(const QModelIndex &index) const
{
    QString mediaTypeString;
    if (index.row() >= 0 && index.row() < mEngineWrapper->itemCount()) {
        FbDriveEntry driveEntry(mEngineWrapper->getDriveEntry(index));
        mediaTypeString = driveEntry.mediaTypeString();
    }
    return mediaTypeString;
}

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