path: root/src/dialog/dlgreplacecuecolor.cpp

#include "dialog/dlgreplacecuecolor.h"

#include <QAbstractButton>
#include <QColor>
#include <QDialogButtonBox>
#include <QMessageBox>
#include <QResizeEvent>
#include <QtConcurrent>

#include "engine/controls/cuecontrol.h"
#include "library/dao/cuedao.h"
#include "library/queryutil.h"
#include "preferences/colorpalettesettings.h"
#include "track/track.h"
#include "util/color/predefinedcolorpalettes.h"

namespace {

const QString kColorButtonStyleSheetLight = QStringLiteral(
        "QPushButton { background-color: %1; color: black; }");
const QString kColorButtonStyleSheetDark = QStringLiteral(
        "QPushButton { background-color: %1; color: white; }");

void setButtonColor(QPushButton* button, const QColor& color) {
    button->setText(color.name());
    button->setStyleSheet(
            (Color::isDimColor(color)
                            ? kColorButtonStyleSheetDark
                            : kColorButtonStyleSheetLight)
                    .arg(color.name()));
}

typedef struct {
    int id;
    TrackId trackId;
    mixxx::RgbColor color;
} CueDatabaseRow;

} // namespace

DlgReplaceCueColor::DlgReplaceCueColor(
        UserSettingsPointer pConfig,
        mixxx::DbConnectionPoolPtr dbConnectionPool,
        TrackCollectionManager* pTrackCollectionManager,
        QWidget* pParent)
        : QDialog(pParent),
          m_pConfig(pConfig),
          m_pDbConnectionPool(dbConnectionPool),
          m_pTrackCollectionManager(pTrackCollectionManager),
          m_bDatabaseChangeInProgress(false),
          m_pNewColorMenu(new QMenu(this)),
          m_pCurrentColorMenu(new QMenu(this)),
          m_lastAutoSetNewColor(std::nullopt),
          m_lastAutoSetCurrentColor(std::nullopt),
          m_pStyle(QStyleFactory::create(QStringLiteral("fusion"))) {
    setupUi(this);
    setWindowModality(Qt::ApplicationModal);

    spinBoxHotcueIndex->setMaximum(NUM_HOT_CUES);

    QIcon icon = QIcon::fromTheme("dialog-warning");
    if (!icon.isNull()) {
        labelReplaceAllColorsIcon->setPixmap(icon.pixmap(QSize(32, 32)));
    } else {
        labelReplaceAllColorsIcon->hide();
    }

    // Unfortunately, not all styles supported by Qt support setting a
    // background color for QPushButtons (see
    // https://bugreports.qt.io/browse/QTBUG-11089). For example, when using
    // the gtk2 style all color buttons would be just grey. It's possible to
    // work around this by modifying the button border with a QSS stylesheet,
    // so that the QStyle will be overwritten, but as a sane default for skins
    // without styles for WColorPicker, we're setting the platform-independent
    // "Fusion" style here. This will make the buttons look slightly different
    // from the rest of the application (when not styled via QSS), but that's
    // better than having buttons without any colors (which would make the
    // color picker unusable).
    pushButtonNewColor->setStyle(m_pStyle);
    pushButtonCurrentColor->setStyle(m_pStyle);

    // Set up new color button
    ColorPaletteSettings colorPaletteSettings(pConfig);
    ColorPalette hotcuePalette = colorPaletteSettings.getHotcueColorPalette();
    mixxx::RgbColor firstColor = mixxx::PredefinedColorPalettes::kDefaultCueColor;
    DEBUG_ASSERT(hotcuePalette.size() > 0);
    if (hotcuePalette.size() > 0) { // Should always be true
        firstColor = hotcuePalette.at(0);
    }
    setButtonColor(pushButtonNewColor, mixxx::RgbColor::toQColor(firstColor));

    // Add menu for new color button
    m_pNewColorPickerAction = make_parented<WColorPickerAction>(
            WColorPicker::Option::AllowCustomColor,
            colorPaletteSettings.getHotcueColorPalette(),
            this);
    m_pNewColorPickerAction->setObjectName("HotcueColorPickerAction");
    m_pNewColorPickerAction->setSelectedColor(firstColor);
    connect(m_pNewColorPickerAction,
            &WColorPickerAction::colorPicked,
            this,
            [this](mixxx::RgbColor::optional_t color) {
                if (color) {
                    setButtonColor(pushButtonNewColor, mixxx::RgbColor::toQColor(*color));
                    slotUpdateWidgets();
                }
                m_pNewColorMenu->hide();
            });
    m_pNewColorMenu->addAction(m_pNewColorPickerAction);
    pushButtonNewColor->setMenu(m_pNewColorMenu);

    // Set up current color button
    setButtonColor(pushButtonCurrentColor,
            mixxx::RgbColor::toQColor(
                    mixxx::PredefinedColorPalettes::kDefaultCueColor));

    // Update apply button when the current color comparison combobox is
    // modified
    connect(comboBoxCurrentColorCompare,
            &QComboBox::currentTextChanged,
            this,
            &DlgReplaceCueColor::slotUpdateWidgets);

    // Add menu for current color button
    m_pCurrentColorPickerAction = make_parented<WColorPickerAction>(
            WColorPicker::Option::AllowCustomColor,
            colorPaletteSettings.getHotcueColorPalette(),
            this);
    m_pCurrentColorPickerAction->setObjectName("HotcueColorPickerAction");
    m_pNewColorPickerAction->setSelectedColor(
            mixxx::PredefinedColorPalettes::kDefaultCueColor);
    connect(m_pCurrentColorPickerAction,
            &WColorPickerAction::colorPicked,
            this,
            [this](mixxx::RgbColor::optional_t color) {
                if (color) {
                    setButtonColor(pushButtonCurrentColor,
                            mixxx::RgbColor::toQColor(*color));
                    slotUpdateWidgets();
                }
                m_pCurrentColorMenu->hide();
            });
    m_pCurrentColorMenu->addAction(m_pCurrentColorPickerAction);
    pushButtonCurrentColor->setMenu(m_pCurrentColorMenu);

    // Update dialog widgets when conditions checkboxes are (un)checked
    connect(checkBoxCurrentColorCondition,
            &QCheckBox::stateChanged,
            this,
            &DlgReplaceCueColor::slotUpdateWidgets);
    connect(checkBoxHotcueIndexCondition,
            &QCheckBox::stateChanged,
            this,
            &DlgReplaceCueColor::slotUpdateWidgets);

    connect(buttonBox,
            &QDialogButtonBox::clicked,
            this,
            [this](QAbstractButton* button) {
                switch (buttonBox->buttonRole(button)) {
                case QDialogButtonBox::RejectRole:
                    reject();
                    break;
                case QDialogButtonBox::ApplyRole:
                    slotApply();
                    break;
                default:
                    break;
                };
            });

    slotUpdateWidgets();
}

DlgReplaceCueColor::~DlgReplaceCueColor() {
}

void DlgReplaceCueColor::setColorPalette(const ColorPalette& palette) {
    m_pNewColorPickerAction->setColorPalette(palette);
    QResizeEvent resizeNewColorMenuEvent(QSize(), m_pNewColorMenu->size());
    qApp->sendEvent(m_pNewColorMenu, &resizeNewColorMenuEvent);
    m_pCurrentColorPickerAction->setColorPalette(palette);
    QResizeEvent resizeCurrentColorMenuEvent(QSize(), m_pCurrentColorMenu->size());
    qApp->sendEvent(m_pCurrentColorMenu, &resizeCurrentColorMenuEvent);
}

void DlgReplaceCueColor::setNewColor(mixxx::RgbColor color) {
    mixxx::RgbColor::optional_t buttonColor = mixxx::RgbColor::fromQString(
            pushButtonNewColor->text());

    // Make sure we don't overwrite colors selected by the user
    if (!m_lastAutoSetNewColor || m_lastAutoSetNewColor == buttonColor) {
        setButtonColor(pushButtonNewColor, mixxx::RgbColor::toQColor(color));
        m_pNewColorPickerAction->setSelectedColor(color);
        slotUpdateWidgets();
    }

    m_lastAutoSetNewColor = color;
}

void DlgReplaceCueColor::setCurrentColor(mixxx::RgbColor color) {
    mixxx::RgbColor::optional_t buttonColor = mixxx::RgbColor::fromQString(
            pushButtonCurrentColor->text());

    // Make sure we don't overwrite colors selected by the user
    if (!m_lastAutoSetCurrentColor || m_lastAutoSetCurrentColor == buttonColor) {
        setButtonColor(pushButtonCurrentColor, mixxx::RgbColor::toQColor(color));
        m_pCurrentColorPickerAction->setSelectedColor(color);
        slotUpdateWidgets();
    }

    m_lastAutoSetCurrentColor = color;
}

void DlgReplaceCueColor::slotApply() {
    m_bDatabaseChangeInProgress = false;
    slotUpdateWidgets();

    // Get values for SELECT query
    QProgressDialog progress(this);
    progress.setWindowModality(Qt::ApplicationModal);
    progress.setAutoReset(false);
    progress.setLabelText(tr("Selecting database rows..."));
    progress.setMaximum(0);
    progress.setValue(0);

    Conditions conditions = ConditionFlag::NoConditions;
    if (checkBoxCurrentColorCondition->isChecked()) {
        conditions |= ConditionFlag::CurrentColorCheck;
    }
    if (comboBoxCurrentColorCompare->currentText() == "is not") {
        conditions |= ConditionFlag::CurrentColorNotEqual;
    }

    if (checkBoxHotcueIndexCondition->isChecked()) {
        conditions |= ConditionFlag::HotcueIndexCheck;
    }
    if (comboBoxHotcueIndexCompare->currentText() == "is not") {
        conditions |= ConditionFlag::HotcueIndexNotEqual;
    }

    mixxx::RgbColor::optional_t currentColor =
            mixxx::RgbColor::fromQString(pushButtonCurrentColor->text());
    VERIFY_OR_DEBUG_ASSERT(currentColor) {
        m_bDatabaseChangeInProgr