qBittorrent/src/gui/watchedfolderoptionsdialog.cpp

/*
 * Bittorrent Client using Qt and libtorrent.
 * Copyright (C) 2021  Vladimir Golovnev <glassez@yandex.ru>
 *
 * 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.
 */

#include "watchedfolderoptionsdialog.h"

#include <QDir>
#include <QPushButton>

#include "base/bittorrent/session.h"
#include "base/global.h"
#include "base/utils/fs.h"
#include "ui_watchedfolderoptionsdialog.h"
#include "utils.h"

#define SETTINGS_KEY(name) "WatchedFolderOptionsDialog/" name

WatchedFolderOptionsDialog::WatchedFolderOptionsDialog(
        const TorrentFilesWatcher::WatchedFolderOptions &watchedFolderOptions, QWidget *parent)
    : QDialog {parent}
    , m_ui {new Ui::WatchedFolderOptionsDialog}
    , m_savePath {watchedFolderOptions.addTorrentParams.savePath}
    , m_downloadPath {watchedFolderOptions.addTorrentParams.downloadPath}
    , m_storeDialogSize {SETTINGS_KEY("DialogSize")}
{
    m_ui->setupUi(this);

    m_ui->savePath->setMode(FileSystemPathEdit::Mode::DirectorySave);
    m_ui->savePath->setDialogCaption(tr("Choose save path"));

    const auto *session = BitTorrent::Session::instance();

    m_ui->downloadPath->setMode(FileSystemPathEdit::Mode::DirectorySave);
    m_ui->downloadPath->setDialogCaption(tr("Choose save path"));
    m_ui->groupBoxDownloadPath->setChecked(watchedFolderOptions.addTorrentParams.useDownloadPath.value_or(session->isDownloadPathEnabled()));

    connect(m_ui->comboTTM, qOverload<int>(&QComboBox::currentIndexChanged), this, &WatchedFolderOptionsDialog::onTMMChanged);
    connect(m_ui->categoryComboBox, qOverload<int>(&QComboBox::currentIndexChanged), this, &WatchedFolderOptionsDialog::onCategoryChanged);

    m_ui->checkBoxRecursive->setChecked(watchedFolderOptions.recursive);
    populateSavePaths();

    const BitTorrent::AddTorrentParams &torrentParams = watchedFolderOptions.addTorrentParams;
    m_ui->startTorrentCheckBox->setChecked(!torrentParams.addPaused.value_or(session->isAddTorrentPaused()));
    m_ui->skipCheckingCheckBox->setChecked(torrentParams.skipChecking);
    m_ui->comboTTM->setCurrentIndex(torrentParams.useAutoTMM.value_or(!session->isAutoTMMDisabledByDefault()));
    m_ui->contentLayoutComboBox->setCurrentIndex(
        static_cast<int>(torrentParams.contentLayout.value_or(session->torrentContentLayout())));

    // Load categories
    QStringList categories = session->categories();
    std::sort(categories.begin(), categories.end(), Utils::Compare::NaturalLessThan<Qt::CaseInsensitive>());

    if (!torrentParams.category.isEmpty())
        m_ui->categoryComboBox->addItem(torrentParams.category);
    m_ui->categoryComboBox->addItem(u""_qs);

    for (const QString &category : asConst(categories))
    {
        if (category != torrentParams.category)
            m_ui->categoryComboBox->addItem(category);
    }

    loadState();

    // Default focus
    if (m_ui->comboTTM->currentIndex() == 0) // 0 is Manual mode
        m_ui->savePath->setFocus();
    else
        m_ui->categoryComboBox->setFocus();
}

WatchedFolderOptionsDialog::~WatchedFolderOptionsDialog()
{
    saveState();
    delete m_ui;
}

TorrentFilesWatcher::WatchedFolderOptions WatchedFolderOptionsDialog::watchedFolderOptions() const
{
    TorrentFilesWatcher::WatchedFolderOptions watchedFolderOptions;
    watchedFolderOptions.recursive = m_ui->checkBoxRecursive->isChecked();

    BitTorrent::AddTorrentParams &params = watchedFolderOptions.addTorrentParams;
    const bool useAutoTMM = (m_ui->comboTTM->currentIndex() == 1);
    if (!useAutoTMM)
    {
        params.savePath = m_ui->savePath->selectedPath();
        params.useDownloadPath = m_ui->groupBoxDownloadPath->isChecked();
        if (params.useDownloadPath)
            params.downloadPath = m_ui->downloadPath->selectedPath();
    }
    params.useAutoTMM = useAutoTMM;
    params.category = m_ui->categoryComboBox->currentText();
    params.addPaused = !m_ui->startTorrentCheckBox->isChecked();
    params.skipChecking = m_ui->skipCheckingCheckBox->isChecked();
    params.contentLayout = static_cast<BitTorrent::TorrentContentLayout>(m_ui->contentLayoutComboBox->currentIndex());

    return watchedFolderOptions;
}

void WatchedFolderOptionsDialog::loadState()
{
    resize(m_storeDialogSize);
}

void WatchedFolderOptionsDialog::saveState()
{
    m_storeDialogSize = size();
}

void WatchedFolderOptionsDialog::onCategoryChanged(const int index)
{
    Q_UNUSED(index);

    if (m_ui->comboTTM->currentIndex() == 1)
    {
        const auto *btSession = BitTorrent::Session::instance();
        const QString categoryName = m_ui->categoryComboBox->currentText();

        const Path savePath = btSession->categorySavePath(categoryName);
        m_ui->savePath->setSelectedPath(savePath);

        const Path downloadPath = btSession->categoryDownloadPath(categoryName);
        m_ui->downloadPath->setSelectedPath(downloadPath);

        m_ui->groupBoxDownloadPath->setChecked(!downloadPath.isEmpty());
    }
}

void WatchedFolderOptionsDialog::populateSavePaths()
{
    const auto *btSession = BitTorrent::Session::instance();

    const Path defaultSavePath {btSession->savePath()};
    m_ui->savePath->setSelectedPath(!m_savePath.isEmpty() ? m_savePath : defaultSavePath);

    const Path defaultDownloadPath {btSession->downloadPath()};
    m_ui->downloadPath->setSelectedPath(!m_downloadPath.isEmpty() ? m_downloadPath : defaultDownloadPath);

    m_ui->groupBoxDownloadPath->setChecked(m_useDownloadPath);
}

void WatchedFolderOptionsDialog::onTMMChanged(const int index)
{
    if (index != 1)
    { // 0 is Manual mode and 1 is Automatic mode. Handle all non 1 values as manual mode.
        populateSavePaths();
        m_ui->groupBoxSavePath->setEnabled(true);
        m_ui->savePath->blockSignals(false);
        m_ui->downloadPath->blockSignals(false);
    }
    else
    {
        m_ui->groupBoxSavePath->setEnabled(false);

        const auto *btSession = BitTorrent::Session::instance();

        m_ui->savePath->blockSignals(true);
        m_savePath = m_ui->savePath->selectedPath();
        const Path savePath = btSession->categorySavePath(m_ui->categoryComboBox->currentText());
        m_ui->savePath->setSelectedPath(savePath);

        m_ui->downloadPath->blockSignals(true);
        m_downloadPath = m_ui->downloadPath->selectedPath();
        const Path downloadPath = btSession->categoryDownloadPath(m_ui->categoryComboBox->currentText());
        m_ui->downloadPath->setSelectedPath(downloadPath);

        m_useDownloadPath = m_ui->groupBoxDownloadPath->isChecked();
        m_ui->groupBoxDownloadPath->setChecked(!downloadPath.isEmpty());
    }
}