qBittorrent/src/gui/scanfoldersdelegate.cpp

/*
 * Bittorrent Client using Qt and libtorrent.
 * Copyright (C) 2015  sledgehammer999
 *
 * 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., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 *
 * In addition, as a special exception, the copyright holders give permission to
 * link this program with the OpenSSL project's "OpenSSL" library (or with
 * modified versions of it that use the same license as the "OpenSSL" library),
 * and distribute the linked executables. You must obey the GNU General Public
 * License in all respects for all of the code used other than "OpenSSL".  If you
 * modify file(s), you may extend this exception to your version of the file(s),
 * but you are not obligated to do so. If you do not wish to do so, delete this
 * exception statement from your version.
 *
 * Contact : hammered999@gmail.com
 */

#include <QPainter>
#include <QComboBox>
#include <QFileDialog>
#include <QTreeView>
#include <QItemSelectionModel>

#include "base/scanfoldersmodel.h"
#include "base/preferences.h"
#include "scanfoldersdelegate.h"


ScanFoldersDelegate::ScanFoldersDelegate(QObject *parent, QTreeView *foldersView)
    : QItemDelegate(parent)
    , m_folderView(foldersView)
{
}

ScanFoldersDelegate::~ScanFoldersDelegate() {}

void ScanFoldersDelegate::paint(QPainter *painter, const QStyleOptionViewItem &option, const QModelIndex &index) const
{
    painter->save();

    switch(index.column()) {
    case ScanFoldersModel::WATCH:
        QItemDelegate::paint(painter, option, index);
        break;

    case ScanFoldersModel::DOWNLOAD: {
        QStyleOptionViewItemV2 opt = QItemDelegate::setOptions(index, option);
        QItemDelegate::drawBackground(painter, opt, index);
        QString text;

        switch (index.data().toInt()) {
        case ScanFoldersModel::DOWNLOAD_IN_WATCH_FOLDER:
            text = tr("Watch Folder");
            break;
        case ScanFoldersModel::DEFAULT_LOCATION:
            text = tr("Default Folder");
            break;
        case ScanFoldersModel::CUSTOM_LOCATION:
            text = index.data(Qt::UserRole).toString();
            break;
        }
        QItemDelegate::drawDisplay(painter, opt, option.rect, text);
        break;
    }

    default:
        break;
    }

    painter->restore();
}

void ScanFoldersDelegate::setEditorData(QWidget *editor, const QModelIndex &index) const
{
    QComboBox *combobox = static_cast<QComboBox*>(editor);
    // Set combobox index
    if (index.data().toInt() == ScanFoldersModel::CUSTOM_LOCATION)
        combobox->setCurrentIndex(4); // '4' is the index of the item after the separator in the QComboBox menu
    else
        combobox->setCurrentIndex(index.data().toInt());
}

QWidget *ScanFoldersDelegate::createEditor(QWidget *parent, const QStyleOptionViewItem &, const QModelIndex &index) const
{
    if (index.column() != ScanFoldersModel::DOWNLOAD) return 0;

    QComboBox* editor = new QComboBox(parent);
    editor->setFocusPolicy(Qt::StrongFocus);
    editor->addItem(tr("Watch Folder"));
    editor->addItem(tr("Default Folder"));
    editor->addItem(tr("Browse..."));
    if (index.data().toInt() == ScanFoldersModel::CUSTOM_LOCATION) {
        editor->insertSeparator(3);
        editor->addItem(index.data(Qt::UserRole).toString());
    }

    connect(editor, SIGNAL(currentIndexChanged(int)), this, SLOT(comboboxIndexChanged(int)));
    return editor;
}

void ScanFoldersDelegate::setModelData(QWidget *editor, QAbstractItemModel *model, const QModelIndex &index) const
{
    QComboBox *combobox = static_cast<QComboBox*>(editor);
    int value = combobox->currentIndex();

    switch(value)  {
    case ScanFoldersModel::DOWNLOAD_IN_WATCH_FOLDER:
    case ScanFoldersModel::DEFAULT_LOCATION:
        model->setData(index, value, Qt::DisplayRole);
        break;

    case 4: // '4' is the index of the item after the separator in the QComboBox menu
        model->setData(index, combobox->currentText(), Qt::UserRole);
        break;

    default:
        break;
    }
}

void ScanFoldersDelegate::updateEditorGeometry(QWidget *editor, const QStyleOptionViewItem &option, const QModelIndex &) const
{
    qDebug("UpdateEditor Geometry called");
    editor->setGeometry(option.rect);
}

void ScanFoldersDelegate::comboboxIndexChanged(int index)
{
    if (index < 2)
        return;

    if (index == 2) {
        QString path;
        QString newPath;
        QModelIndexList selIndex = m_folderView->selectionModel()->selectedRows(1);

        if (selIndex.isEmpty())
            return;

        ScanFoldersModel::PathType type = static_cast<ScanFoldersModel::PathType>(selIndex.at(0).data().toInt());

        if (type == ScanFoldersModel::CUSTOM_LOCATION)
            path = selIndex.at(0).data(Qt::UserRole).toString();
        else
            path = Preferences::instance()->getSavePath();

        newPath = QFileDialog::getExistingDirectory(0, tr("Choose save path"), path);
        QComboBox *combobox = static_cast<QComboBox*>(sender());
        disconnect(combobox, SIGNAL(currentIndexChanged(int)), this, SLOT(comboboxIndexChanged(int)));

        if (!newPath.isEmpty()) {
            if (combobox->count() != 5) {
                combobox->insertSeparator(3);
                combobox->addItem(newPath);
            }
            else {
                combobox->setItemText(4, newPath);
            }
            combobox->setCurrentIndex(4);
        }
        else {
            if (type == ScanFoldersModel::CUSTOM_LOCATION)
                combobox->setCurrentIndex(4);
            else
                combobox->setCurrentIndex(type);
        }

        connect(combobox, SIGNAL(currentIndexChanged(int)), this, SLOT(comboboxIndexChanged(int)));
    }
}