/* ============================================================
*
* This file is a part of the rekonq project
*
* Copyright (C) 2007-2008 Trolltech ASA. All rights reserved
* Copyright (C) 2008-2009 by Andrea Diamantini <adjam7 at gmail dot com>
*
*
* This program is free software; you can redistribute it
* and/or modify it under the terms of the GNU General
* Public License as published by the Free Software Foundation;
* either version 3, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* ============================================================ */


// Self Includes
#include "modelmenu.h"
#include "modelmenu.moc"


ModelMenu::ModelMenu(QWidget * parent)
        : KMenu(parent)
        , m_maxRows(7)
        , m_firstSeparator(-1)
        , m_maxWidth(-1)
        , m_hoverRole(0)
        , m_separatorRole(0)
        , m_model(0)
{
    connect(this, SIGNAL(aboutToShow()), this, SLOT(aboutToShow()));
}


bool ModelMenu::prePopulated()
{
    return false;
}


void ModelMenu::postPopulated()
{
}


void ModelMenu::setModel(QAbstractItemModel *model)
{
    m_model = model;
}


QAbstractItemModel *ModelMenu::model() const
{
    return m_model;
}


void ModelMenu::setMaxRows(int max)
{
    m_maxRows = max;
}


int ModelMenu::maxRows() const
{
    return m_maxRows;
}


void ModelMenu::setFirstSeparator(int offset)
{
    m_firstSeparator = offset;
}


int ModelMenu::firstSeparator() const
{
    return m_firstSeparator;
}


void ModelMenu::setRootIndex(const QModelIndex &index)
{
    m_root = index;
}


QModelIndex ModelMenu::rootIndex() const
{
    return m_root;
}


void ModelMenu::setHoverRole(int role)
{
    m_hoverRole = role;
}


int ModelMenu::hoverRole() const
{
    return m_hoverRole;
}


void ModelMenu::setSeparatorRole(int role)
{
    m_separatorRole = role;
}


int ModelMenu::separatorRole() const
{
    return m_separatorRole;
}


Q_DECLARE_METATYPE(QModelIndex)
void ModelMenu::aboutToShow()
{
    if (QMenu *menu = qobject_cast<QMenu*>(sender()))
    {
        QVariant v = menu->menuAction()->data();
        if (v.canConvert<QModelIndex>())
        {
            QModelIndex idx = qvariant_cast<QModelIndex>(v);
            createMenu(idx, -1, menu, menu);
            disconnect(menu, SIGNAL(aboutToShow()), this, SLOT(aboutToShow()));
            return;
        }
    }

    clear();
    if (prePopulated())
        addSeparator();
    int max = m_maxRows;
    if (max != -1)
        max += m_firstSeparator;
    createMenu(m_root, max, this, this);
    postPopulated();
}


// WARNING 
// the code commented out here is to create a second separator in the history menu
// with ALL history, subdivided by days. 
void ModelMenu::createMenu(const QModelIndex &parent, int max, QMenu *parentMenu, QMenu *menu)
{
    Q_UNUSED(parentMenu)

    if (!menu)
    {
        return;
    }

    int end = m_model->rowCount(parent);
    if (max != -1)
        end = qMin(max, end);

    connect(menu, SIGNAL(triggered(QAction*)), this, SLOT(triggered(QAction*)));
    connect(menu, SIGNAL(hovered(QAction*)), this, SLOT(hovered(QAction*)));

    for (int i = 0; i < end; ++i)
    {
        QModelIndex idx = m_model->index(i, 0, parent);

        if( !m_model->hasChildren(idx) && ( m_separatorRole == 0 || !idx.data(m_separatorRole).toBool() ) )
        {
            menu->addAction(makeAction(idx));
        }
    }
}

KAction *ModelMenu::makeAction(const QModelIndex &index)
{
    QIcon icon = qvariant_cast<QIcon>(index.data(Qt::DecorationRole));
    KAction *action = (KAction *) makeAction(KIcon(icon), index.data().toString(), this);
    QVariant v;
    v.setValue(index);
    action->setData(v);
    return action;
}

KAction *ModelMenu::makeAction(const KIcon &icon, const QString &text, QObject *parent)
{
    QFontMetrics fm(font());
    if (-1 == m_maxWidth)
        m_maxWidth = fm.width(QLatin1Char('m')) * 30;
    QString smallText = fm.elidedText(text, Qt::ElideMiddle, m_maxWidth);
    return new KAction(icon, smallText, parent);
}

void ModelMenu::triggered(QAction *action)
{
    QVariant v = action->data();
    if (v.canConvert<QModelIndex>())
    {
        QModelIndex idx = qvariant_cast<QModelIndex>(v);
        emit activated(idx);
    }
}

void ModelMenu::hovered(QAction *action)
{
    QVariant v = action->data();
    if (v.canConvert<QModelIndex>())
    {
        QModelIndex idx = qvariant_cast<QModelIndex>(v);
        QString hoveredString = idx.data(m_hoverRole).toString();
        if (!hoveredString.isEmpty())
            emit hovered(hoveredString);
    }
}