Ticket #157: userlistview_test2.cpp

/* museeq - a Qt client to museekd
 *
 * Copyright (C) 2003-2004 Hyriand <hyriand@thegraveyard.org>
 * Copyright 2008 little blue poney <lbponey@users.sourceforge.net>
 *
 * 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 2 of the License, 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.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

#include "userlistview.h"
#include "userlistitem.h"
#include "usermenu.h"
#include "museeq.h"
#include "images.h"
#include "util.h"

#include <QPixmap>
#include <QDropEvent>
#include <QList>
#include <QUrl>
#include <QPainter>
#include <QApplication>
#include <QDrag>
#include <QMouseEvent>

UserListView::UserListView(bool comments, QWidget * parent, const char * name)
        : QTreeWidget(parent) {

        mUsermenu = new Usermenu(this);

    setAcceptDrops(true);
        setDragEnabled(true);
        setSelectionMode(QAbstractItemView::SingleSelection);
        setColumnCount(4);

        QStringList headers;
        headers  << QString::null << tr("User") << tr("Speed") << tr("Files");
        if(comments) {
                headers << (tr("Comments"));
        setColumnCount(5);
        }
        setHeaderLabels(headers);

        setAlternatingRowColors(true);
        setSortingEnabled(true);
        sortItems(1, Qt::AscendingOrder);
        QPixmap& p(IMG("online"));
        setColumnWidth (0, p.width()+5);
        setColumnWidth ( 1, 100 );
        setColumnWidth ( 2, 100 );
        setColumnWidth ( 3, 100 );
        setAllColumnsShowFocus(true);
        setRootIsDecorated(false);
        setContextMenuPolicy(Qt::CustomContextMenu);
        connect(this, SIGNAL(itemDoubleClicked(QTreeWidgetItem*, int)), SLOT(slotActivate(QTreeWidgetItem*, int)));
        connect(this, SIGNAL(itemActivated(QTreeWidgetItem*, int)), SLOT(slotActivate(QTreeWidgetItem*, int)));
        connect(this, SIGNAL(customContextMenuRequested(const QPoint&)), SLOT(slotContextMenu(const QPoint&)));


        connect(museeq, SIGNAL(sortingEnabled(bool)), this, SLOT(sorting(bool)));
        connect(museeq, SIGNAL(userStatus(const QString&, uint)), SLOT(setStatus(const QString&, uint)));
        connect(museeq, SIGNAL(doUpdateStatus(const QString&)), SLOT(updateStatus(const QString&)));
        connect(museeq, SIGNAL(userData(const QString&, uint, uint)), SLOT(setData(const QString&, uint, uint)));
}

void UserListView::sorting(bool sort) {
        setSortingEnabled(sort);
        if (sort)
                resizeColumnToContents(1);
}
UserListItem* UserListView::findItem(const QString& _u) {
        QList<QTreeWidgetItem *> items = findItems(_u, Qt::MatchExactly, 1);
        if (!items.isEmpty())
        return dynamic_cast<UserListItem *>(items.at(0));
    else
        return NULL;
}

uint UserListView::status(const QString& _u) {
        UserListItem* item = findItem(_u);
        if(! item)
                return 0;
        return item->status();
}

uint UserListView::speed(const QString& _u) {
        UserListItem* item = findItem(_u);
        if(! item)
                return 0;
        return item->speed();
}

uint UserListView::files(const QString& _u) {
        UserListItem* item = findItem(_u);
        if(! item)
                return 0;
        return item->files();
}

QString UserListView::comments(const QString& _u) {
        UserListItem* item = findItem(_u);
        if(! item)
                return QString::null;
        return item->comments();
}

void UserListView::setStatus(const QString& _u, uint _s) {
        qDebug() << _u << _s;
        UserListItem *item = findItem(_u);
        qDebug() << _u << " found" << !item;
        if(! item)
                return;
        item->setStatus(_s);
}

void UserListView::updateStatus(const QString& _u) {
        UserListItem* item = findItem(_u);
        if(! item)
                return;
        item->updateUserStatus();
}

void UserListView::setData(const QString& _u, uint _s, uint _f) {
        UserListItem *item = findItem(_u);
        if(! item)
                return;
        item->setSpeed(_s);
        item->setFiles(_f);
}

void UserListView::setComments(const QString& _u, const QString& _c) {
        UserListItem *item = findItem(_u);
        if(! item)
                return;
        item->setComments(_c);
}

void UserListView::add(const QString& _u, uint _st, uint _s, uint _f, const QString& _c) {
        UserListItem *item = findItem(_u);
        if(item) {
                item->setAll(_st, _s, _f, _c);
                return;
        }
        new UserListItem(this, _u, _st, _s, _f, _c);

}

void UserListView::add(const QString& _u, const QString& _c) {
        UserListItem *item = findItem(_u);
        if(item) {
                item->setComments(_c);
                return;
        }
        new UserListItem(this, _u, 0, 0, 0, _c);

}

void UserListView::remove(const QString& _u) {
        UserListItem *item = findItem(_u);
        if(! item) {
                printf("warning, couldn't find user!\n");
                return;
        }
        delete item;
}

void UserListView::slotActivate(QTreeWidgetItem* item, int column) {
        slotActivate( item);
}
void UserListView::slotActivate(QTreeWidgetItem* item) {
        UserListItem* _item = dynamic_cast<UserListItem*>(item);
        if(! _item)
                return;
        emit activated(_item->user());
        emit activated(_item->user(), _item->comments());
}

/**
  * Search TreeWidget users column for users matching keypresses (only first letter)
  */
void UserListView::keyboardSearch(const QString& string) {
    // Search for a user starting by this letter
        QList<QTreeWidgetItem *> searchUsers = QTreeWidget::findItems(string, Qt::MatchStartsWith, 1);

        // If we've already searched for this letter earlier, we should display next one
        if (mLastSearch != string) {
                mSearchPosition = 0;
                mLastSearch = QString(string);
        } else {
                mSearchPosition += 1;
        }

    // Do nothing if we didn't find anything
        if (searchUsers.isEmpty())
        return;

    // If we're on the last user having this letter, next one will be the first one
    if (static_cast<int>(mSearchPosition) >= searchUsers.size())
                mSearchPosition = 0;

    // Get the found item
        QTreeWidgetItem * item = 0;
        item = searchUsers.at(mSearchPosition);

        if (! item)
                return;

    // Select found item
        setCurrentItem(item);
}

/**
  * User have press mouse button in this widget
  */
void UserListView::mousePressEvent(QMouseEvent *event)
{
    event->accept();

    if (event->button() == Qt::LeftButton)
        mDragStartPosition = event->pos();

    QTreeWidget::mousePressEvent(event);
}

/**
  * User have moved his mouse in this widget
  */
void UserListView::mouseMoveEvent(QMouseEvent *event)
{
    event->accept();

    // Should we start dragging?
    if (!(event->buttons() & Qt::LeftButton))
        return;
    if ((event->pos() - mDragStartPosition).manhattanLength() < QApplication::startDragDistance())
        return;

    // Create drag object
    QDrag *drag = new QDrag(this);
    QMimeData *mimeData = new QMimeData;

    QList<QUrl> urls;

        QList<QTreeWidgetItem*> items = selectedItems();
    QList<QTreeWidgetItem*>::const_iterator it = items.begin();
        for(; it != items.end(); ++it) {
            // slsk protocol: in QUrl, hostname is always lower case.
            // So we put username as hostname for compatibility, and as username to have the correct case.
            // Ex: slsk://MuSeEk@museek/path/to/a/file
            // Code should first look at QUrl::userName() and if not present, try QUrl::host()
            UserListItem * item = dynamic_cast<UserListItem*>(*it);
            if (!item)
            continue;
            QUrl url("slsk://" + item->user());
            url.setUserName(item->user());

            // There may be spaces in username so url may not be valid. It will work, but QUrl::isValid() should not be used
            urls.push_back(url);
        }

        if(urls.count() == 0)
                return;

        // Add the urls to the mimedata
    mimeData->setUrls(urls);
    // Add them too in text format if we want to paste it in a text area
    QString textUrls;
    QList<QUrl>::const_iterator uit;
    for(uit = urls.begin(); uit != urls.end(); uit++)
        textUrls += uit->toString() + "\n";
    mimeData->setText(textUrls);

    // And now set this mimedata into drag object
    drag->setMimeData(mimeData);

    // Make visible what we're dragging
    if(items.count() < 6) {
                QSize dest(0, 0);
                for(it = items.begin(); it != items.end(); ++it) {
            UserListItem * item = dynamic_cast<UserListItem*>(*it);
            if (!item)
                continue;

                        QSize textSize = viewport()->fontMetrics().size(Qt::TextSingleLine, item->user());

                        QPixmap icon = item->icon(0).pixmap(50, 50); // 50x50 is the maximum size, if image is smaller, the pixmap will be too
                        QSize iconSize = icon.size();

                        QSize lineSize = textSize.expandedTo(QSize(textSize.width() + iconSize.width() + 2, iconSize.height()));
                        dest = dest.expandedTo(QSize(lineSize.width(), lineSize.height() + 2));
                }

                QSize s(dest.width() + 6, dest.height() * items.count() + 4);

                QPixmap pix(s);
                QPainter p(&pix);
                p.setFont(viewport()->font());
        p.setPen(viewport()->palette().color(QPalette::WindowText));

                p.fillRect(QRect(QPoint(0, 0), s), viewport()->palette().color(QPalette::Background));
                p.drawRect(QRect(QPoint(0, 0), s));

                int y = 2;
                for(it = items.begin(); it != items.end(); ++it) {
            UserListItem * item = dynamic_cast<UserListItem*>(*it);
            if (!item)
                continue;

                        const QPixmap icon = item->icon(0).pixmap(50, 50);
                        p.drawPixmap(3, y + ((dest.height() - icon.height()) / 2), icon);
                        QRect r(QPoint(5 + icon.width(), y),
                                dest - QSize(icon.width() + 2, 0));
                        p.drawText(r, Qt::AlignLeft |  Qt::AlignVCenter, item->user());
                        y += dest.height();
                }

                p.end();

        drag->setHotSpot(QPoint(20, 20));
                drag->setPixmap(pix);
    }
        else {
                QString x(QString(tr("%n user(s)", "", items.count())));
                QSize s = viewport()->fontMetrics().size(Qt::TextSingleLine, x) + QSize(6, 4);

                QPixmap pix(s);
                QPainter p(&pix);
                p.setFont(viewport()->font());
        p.setPen(viewport()->palette().color(QPalette::WindowText));

                p.fillRect(QRect(QPoint(0, 0), s), viewport()->palette().color(QPalette::Background));
                p.drawRect(QRect(QPoint(0, 0), s));
                p.drawText(QRect(QPoint(3, 3), s - QSize(3, 3)), Qt::AlignLeft | Qt::AlignVCenter, x);

                p.end();

        drag->setHotSpot(QPoint(20, 20));
                drag->setPixmap(pix);
        }

    // Begin dragging
    drag->exec();
}

void UserListView::dragMoveEvent(QDragMoveEvent* event)
{
    event->acceptProposedAction();
}

void UserListView::dragEnterEvent(QDragEnterEvent* event) {
    event->acceptProposedAction();
}

void UserListView::dropEvent(QDropEvent* event) {
    event->acceptProposedAction();

    if (Util::hasSlskUrls(event) && acceptDrops())
        emit dropSlsk(event->mimeData()->urls());
}

void UserListView::slotContextMenu(const QPoint& pos) {
        QTreeWidgetItem * item = 0 ;
        item = itemAt(pos) ;
        if (! item ) {
                return;
        }
        mUsermenu->exec(item->text(1), mapToGlobal(pos));
}

void UserListView::slotContextMenu(QTreeWidgetItem* item, const QPoint& pos, int) {
    UserListItem * _item = dynamic_cast<UserListItem *>(item);
        if(_item)
        mUsermenu->exec(_item->user(), pos);
}