/* SPDX-FileCopyrightText: 2022 Noah Davis <noahadvs@gmail.com>
 * SPDX-License-Identifier: LGPL-2.0-or-later
 */

#include "CaptureModeModel.h"
#include "ShortcutActions.h"
#include "SpectacleCore.h"

#include <KGlobalAccel>
#include <KLocalizedString>

#include <QApplication>
#include <qnamespace.h>
#include <utility>

using namespace Qt::StringLiterals;

static QString actionShortcutsToString(QAction *action)
{
    QString value;
    if (!KGlobalAccel::self()->hasShortcut(action)) {
        return value;
    }

    const auto &shortcuts = KGlobalAccel::self()->shortcut(action);
    for (int i = 0; i < shortcuts.length(); ++i) {
        if (i > 0) {
            value += u", ";
        }
        value.append(shortcuts[i].toString(QKeySequence::NativeText));
    }
    return value;
}

CaptureModeModel::CaptureModeModel(QObject *parent)
    : QAbstractListModel(parent)
{
    m_roleNames[CaptureModeRole] = "captureMode"_ba;
    m_roleNames[Qt::DisplayRole] = "display"_ba;
    m_roleNames[ShortcutsRole] = "shortcuts"_ba;

    auto platform = SpectacleCore::instance()->imagePlatform();
    connect(platform, &ImagePlatform::supportedGrabModesChanged, this, [this, platform]() {
        setGrabModes(platform->supportedGrabModes());
    });
    setGrabModes(platform->supportedGrabModes());
}

QHash<int, QByteArray> CaptureModeModel::roleNames() const
{
    return m_roleNames;
}

QVariant CaptureModeModel::data(const QModelIndex &index, int role) const
{
    int row = index.row();
    QVariant ret;
    if (!checkIndex(index, CheckIndexOption::IndexIsValid)) {
        return ret;
    }
    if (role == CaptureModeRole) {
        ret = m_data.at(row).captureMode;
    } else if (role == Qt::DisplayRole) {
        ret = m_data.at(row).label;
    } else if (role == ShortcutsRole) {
        ret = m_data.at(row).shortcuts;
    }
    return ret;
}

int CaptureModeModel::rowCount(const QModelIndex &parent) const
{
    Q_UNUSED(parent)
    return m_data.size();
}

int CaptureModeModel::indexOfCaptureMode(CaptureMode captureMode) const
{
    int finalIndex = -1;
    for (int i = 0; i < m_data.length(); ++i) {
        if (m_data[i].captureMode == captureMode) {
            finalIndex = i;
            break;
        }
    }
    return finalIndex;
}

void CaptureModeModel::setGrabModes(ImagePlatform::GrabModes modes)
{
    if (m_grabModes == modes) {
        return;
    }
    m_grabModes = modes;
    const int oldCount = m_data.size();
    m_data.clear();

    const bool hasCurrentScreen = m_grabModes.testFlag(ImagePlatform::GrabMode::CurrentScreen);

    if (m_grabModes.testFlag(ImagePlatform::GrabMode::PerScreenImageNative)) {
        m_data.append({
            CaptureModeModel::RectangularRegion,
            captureModeLabel(CaptureMode::RectangularRegion),
            actionShortcutsToString(ShortcutActions::self()->regionAction()),
        });
    }
    if (m_grabModes.testFlag(ImagePlatform::GrabMode::AllScreens)) {
        m_data.append({
            CaptureModeModel::AllScreens,
            captureModeLabel(hasCurrentScreen ? CaptureMode::AllScreens : CaptureMode::FullScreen),
            actionShortcutsToString(ShortcutActions::self()->fullScreenAction()),
        });
    }
    if (m_grabModes.testFlag(ImagePlatform::GrabMode::AllScreensScaled)) {
        m_data.append({
            CaptureModeModel::AllScreensScaled,
            captureModeLabel(CaptureMode::AllScreensScaled),
        });
    }
    if (hasCurrentScreen) {
        m_data.append({
            CaptureModeModel::CurrentScreen,
            captureModeLabel(CaptureModeModel::CurrentScreen),
            actionShortcutsToString(ShortcutActions::self()->currentScreenAction()),
        });
    }
    if (m_grabModes.testFlag(ImagePlatform::GrabMode::ActiveWindow)) {
        m_data.append({
            CaptureModeModel::ActiveWindow,
            captureModeLabel(CaptureMode::ActiveWindow),
            actionShortcutsToString(ShortcutActions::self()->activeWindowAction()),
        });
    }
    if (m_grabModes.testFlag(ImagePlatform::GrabMode::WindowUnderCursor)) {
        m_data.append({
            CaptureModeModel::WindowUnderCursor,
            captureModeLabel(CaptureMode::WindowUnderCursor),
            actionShortcutsToString(ShortcutActions::self()->windowUnderCursorAction()),
        });
    }

    Q_EMIT captureModesChanged();
    if (oldCount != m_data.size()) {
        Q_EMIT countChanged();
    }
}

QString CaptureModeModel::captureModeLabel(CaptureMode mode)
{
    switch (mode) {
    case CaptureMode::RectangularRegion:
        return i18nc("@action in capture mode section", "Rectangular Region");
    case CaptureMode::AllScreensScaled:
        return i18nc("@action in capture mode section", "All Screens (Scaled to same size)");
    case CaptureMode::CurrentScreen:
        return i18nc("@action in capture mode section", "Current Screen");
    case CaptureMode::ActiveWindow:
        return i18nc("@action in capture mode section", "Active Window");
    case CaptureMode::WindowUnderCursor: {
        auto wayland = qGuiApp->nativeInterface<QNativeInterface::QWaylandApplication>();
        return wayland ? i18nc("@action in capture mode section", "Select Window") : i18nc("@action in capture mode section", "Window Under Cursor");
    }
    case CaptureMode::AllScreens:
        return i18nc("@action in capture mode section", "All Screens");
    case CaptureMode::FullScreen:
        return i18nc("@action in capture mode section", "Full Screen");
    }
    return QString{};
}

#include "moc_CaptureModeModel.cpp"
