kurlnavigatorschemecombo.cpp

/*
    SPDX-FileCopyrightText: 2006 Aaron J. Seigo <aseigo@kde.org>
    SPDX-FileCopyrightText: 2009 Peter Penz <peter.penz@kde.org>
 
    SPDX-License-Identifier: LGPL-2.0-or-later
*/
 
#include "kurlnavigatorschemecombo_p.h"
 
#include <QAction>
#include <QMenu>
#include <QPaintEvent>
#include <QPainter>
#include <QStyleOption>
 
#include <KLocalizedString>
#include <kprotocolinfo.h>
#include <kprotocolmanager.h>
#include <kurlnavigator.h>
 
namespace
{
const int ArrowSize = 10;
}
 
namespace KDEPrivate
{
KUrlNavigatorSchemeCombo::KUrlNavigatorSchemeCombo(const QString &scheme, KUrlNavigator *parent)
    : KUrlNavigatorButtonBase(parent)
    , m_menu(nullptr)
    , m_schemes()
    , m_categories()
{
    m_menu = new QMenu(this);
    connect(m_menu, &QMenu::triggered, this, &KUrlNavigatorSchemeCombo::setSchemeFromMenu);
    setText(scheme);
    setMenu(m_menu);
}
 
void KUrlNavigatorSchemeCombo::setSupportedSchemes(const QStringList &schemes)
{
    m_schemes = schemes;
    m_menu->clear();
 
    for (const QString &scheme : schemes) {
        QAction *action = m_menu->addAction(scheme);
        action->setData(scheme);
    }
}
 
QSize KUrlNavigatorSchemeCombo::sizeHint() const
{
    const QSize size = KUrlNavigatorButtonBase::sizeHint();
 
    int width = fontMetrics().boundingRect(KLocalizedString::removeAcceleratorMarker(text())).width();
    width += (3 * BorderWidth) + ArrowSize;
 
    return QSize(width, size.height());
}
 
void KUrlNavigatorSchemeCombo::setScheme(const QString &scheme)
{
    setText(scheme);
}
 
QString KUrlNavigatorSchemeCombo::currentScheme() const
{
    return text();
}
 
void KUrlNavigatorSchemeCombo::showEvent(QShowEvent *event)
{
    KUrlNavigatorButtonBase::showEvent(event);
    if (!event->spontaneous() && m_schemes.isEmpty()) {
        m_schemes = KProtocolInfo::protocols();
 
        auto it = std::remove_if(m_schemes.begin(), m_schemes.end(), [](const QString &s) {
            QUrl url;
            url.setScheme(s);
            return !KProtocolManager::supportsListing(url);
        });
        m_schemes.erase(it, m_schemes.end());
 
        std::sort(m_schemes.begin(), m_schemes.end());
 
        updateMenu();
    }
}
 
void KUrlNavigatorSchemeCombo::paintEvent(QPaintEvent *event)
{
    Q_UNUSED(event);
 
    QPainter painter(this);
    const int buttonWidth = width();
    const int buttonHeight = height();
 
    drawHoverBackground(&painter);
 
    const QColor fgColor = foregroundColor();
    painter.setPen(fgColor);
 
    // draw arrow
    const int arrowX = buttonWidth - ArrowSize - BorderWidth;
    const int arrowY = (buttonHeight - ArrowSize) / 2;
 
    QStyleOption option;
    option.rect = QRect(arrowX, arrowY, ArrowSize, ArrowSize);
    option.palette = palette();
    option.palette.setColor(QPalette::Text, fgColor);
    option.palette.setColor(QPalette::WindowText, fgColor);
    option.palette.setColor(QPalette::ButtonText, fgColor);
    style()->drawPrimitive(QStyle::PE_IndicatorArrowDown, &option, &painter, this);
 
    // draw text
    const int textWidth = arrowX - (2 * BorderWidth);
    int alignment = Qt::AlignCenter | Qt::TextShowMnemonic;
    if (!style()->styleHint(QStyle::SH_UnderlineShortcut, &option, this)) {
        alignment |= Qt::TextHideMnemonic;
    }
    style()->drawItemText(&painter, QRect(BorderWidth, 0, textWidth, buttonHeight), alignment, option.palette, isEnabled(), text());
}
 
void KUrlNavigatorSchemeCombo::setSchemeFromMenu(QAction *action)
{
    const QString scheme = action->data().toString();
    setText(scheme);
    Q_EMIT activated(scheme);
}
 
void KUrlNavigatorSchemeCombo::updateMenu()
{
    initializeCategories();
    std::sort(m_schemes.begin(), m_schemes.end());
 
    // move all schemes into the corresponding category of 'items'
    QList<QString> items[CategoryCount];
    for (const QString &scheme : std::as_const(m_schemes)) {
        if (m_categories.contains(scheme)) {
            const SchemeCategory category = m_categories.value(scheme);
            items[category].append(scheme);
        } else {
            items[OtherCategory].append(scheme);
        }
    }
 
    // Create the menu that includes all entries from 'items'. The categories
    // CoreCategory and PlacesCategory are placed at the top level, the remaining
    // categories are placed in sub menus.
    QMenu *menu = m_menu;
    for (int category = 0; category < CategoryCount; ++category) {
        if (!items[category].isEmpty()) {
            switch (category) {
            case DevicesCategory:
                menu = m_menu->addMenu(i18nc("@item:inmenu", "Devices"));
                break;
 
            case SubversionCategory:
                menu = m_menu->addMenu(i18nc("@item:inmenu", "Subversion"));
                break;
 
            case OtherCategory:
                menu = m_menu->addMenu(i18nc("@item:inmenu", "Other"));
                break;
 
            case CoreCategory:
            case PlacesCategory:
            default:
                break;
            }
 
            for (const QString &scheme : std::as_const(items[category])) {
                QAction *action = menu->addAction(scheme);
                action->setData(scheme);
            }
 
            if (menu == m_menu) {
                menu->addSeparator();
            }
        }
    }
}
 
void KUrlNavigatorSchemeCombo::initializeCategories()
{
    if (m_categories.isEmpty()) {
        m_categories.insert(QStringLiteral("file"), CoreCategory);
        m_categories.insert(QStringLiteral("ftp"), CoreCategory);
        m_categories.insert(QStringLiteral("fish"), CoreCategory);
        m_categories.insert(QStringLiteral("nfs"), CoreCategory);
        m_categories.insert(QStringLiteral("sftp"), CoreCategory);
        m_categories.insert(QStringLiteral("smb"), CoreCategory);
        m_categories.insert(QStringLiteral("webdav"), CoreCategory);
 
        m_categories.insert(QStringLiteral("desktop"), PlacesCategory);
        m_categories.insert(QStringLiteral("fonts"), PlacesCategory);
        m_categories.insert(QStringLiteral("programs"), PlacesCategory);
        m_categories.insert(QStringLiteral("settings"), PlacesCategory);
        m_categories.insert(QStringLiteral("trash"), PlacesCategory);
 
        m_categories.insert(QStringLiteral("floppy"), DevicesCategory);
        m_categories.insert(QStringLiteral("camera"), DevicesCategory);
        m_categories.insert(QStringLiteral("remote"), DevicesCategory);
 
        m_categories.insert(QStringLiteral("svn"), SubversionCategory);
        m_categories.insert(QStringLiteral("svn+file"), SubversionCategory);
        m_categories.insert(QStringLiteral("svn+http"), SubversionCategory);
        m_categories.insert(QStringLiteral("svn+https"), SubversionCategory);
        m_categories.insert(QStringLiteral("svn+ssh"), SubversionCategory);
    }
}
 
} // namespace KDEPrivate
 
#include "moc_kurlnavigatorschemecombo_p.cpp"