/*
 * Copyright (c) 2025-2026, Tim Flynn <trflynn89@ladybird.org>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#include <AK/Base64.h>
#include <UI/Qt/Icon.h>
#include <UI/Qt/Menu.h>
#include <UI/Qt/StringUtils.h>
#include <UI/Qt/WebContentView.h>

#include <QAction>
#include <QMenu>
#include <QPointer>
#include <QWidget>

namespace Ladybird {

class ActionObserver final : public WebView::Action::Observer {
public:
    static NonnullOwnPtr<ActionObserver> create(WebView::Action& action, QAction& qaction)
    {
        return adopt_own(*new ActionObserver(action, qaction));
    }

    virtual void on_text_changed(WebView::Action& action) override
    {
        if (m_action)
            m_action->setText(qstring_from_ak_string(action.text()));
    }

    virtual void on_tooltip_changed(WebView::Action& action) override
    {
        if (m_action)
            m_action->setToolTip(qstring_from_ak_string(action.tooltip()));
    }

    virtual void on_enabled_state_changed(WebView::Action& action) override
    {
        if (m_action)
            m_action->setEnabled(action.enabled());
    }

    virtual void on_visible_state_changed(WebView::Action& action) override
    {
        if (m_action)
            m_action->setVisible(action.visible());
    }

    virtual void on_engaged_state_changed(WebView::Action& action) override
    {
        if (!m_action)
            return;

        switch (action.id()) {
        case WebView::ActionID::ToggleBookmark:
        case WebView::ActionID::ToggleBookmarkViaToolbar:
            if (auto* parent = as_if<QWidget>(m_action->parent())) {
                auto const* icon = action.engaged() ? "star-filled" : "star-countour";
                m_action->setIcon(create_tvg_icon_with_theme_colors(icon, parent->palette()));
            }
            break;
        default:
            break;
        }
    }

    virtual void on_checked_state_changed(WebView::Action& action) override
    {
        if (m_action)
            m_action->setChecked(action.checked());
    }

private:
    ActionObserver(WebView::Action& action, QAction& qaction)
        : m_action(&qaction)
    {
        QObject::connect(m_action, &QAction::triggered, [weak_action = action.make_weak_ptr()](bool checked) {
            if (auto action = weak_action.strong_ref()) {
                if (action->is_checkable())
                    action->set_checked(checked);
                action->activate();
            }
        });
        QObject::connect(m_action->parent(), &QObject::destroyed, [this, weak_action = action.make_weak_ptr()]() {
            if (auto action = weak_action.strong_ref())
                action->remove_observer(*this);
        });
    }

    QPointer<QAction> m_action;
};

template<typename T>
static void add_properties(QObject& object, T& menu_or_action)
{
    for (auto const& [key, value] : menu_or_action.properties())
        object.setProperty(key.to_byte_string().characters(), qstring_from_ak_string(value));
}

static QIcon icon_from_base64_png(StringView favicon_base64_png)
{
    static constexpr int const MENU_ICON_SIZE = 16;

    auto decoded = decode_base64(favicon_base64_png);
    if (decoded.is_error())
        return {};

    QPixmap pixmap;
    if (!pixmap.loadFromData(decoded.value().data(), static_cast<uint>(decoded.value().size()), "PNG"))
        return {};

    return pixmap.scaled(MENU_ICON_SIZE, MENU_ICON_SIZE, Qt::KeepAspectRatio, Qt::SmoothTransformation);
}

static void initialize_native_control(WebView::Action& action, QAction& qaction, QPalette const& palette)
{
    switch (action.id()) {
    case WebView::ActionID::NavigateBack:
        qaction.setIcon(create_tvg_icon_with_theme_colors("back", palette));
        qaction.setShortcut(QKeySequence::StandardKey::Back);
        break;
    case WebView::ActionID::NavigateForward:
        qaction.setIcon(create_tvg_icon_with_theme_colors("forward", palette));
        qaction.setShortcut(QKeySequence::StandardKey::Forward);
        break;
    case WebView::ActionID::Reload:
        qaction.setIcon(create_tvg_icon_with_theme_colors("reload", palette));
        qaction.setShortcuts({ QKeySequence(Qt::CTRL | Qt::Key_R), QKeySequence(Qt::Key_F5) });
        break;

    case WebView::ActionID::CopySelection:
        qaction.setIcon(load_icon_from_uri("resource://icons/16x16/edit-copy.png"sv));
        qaction.setShortcut(QKeySequence::StandardKey::Copy);
        break;
    case WebView::ActionID::Paste:
        qaction.setIcon(load_icon_from_uri("resource://icons/16x16/paste.png"sv));
        qaction.setShortcut(QKeySequence::StandardKey::Paste);
        break;
    case WebView::ActionID::SelectAll:
        qaction.setIcon(load_icon_from_uri("resource://icons/16x16/select-all.png"sv));
        qaction.setShortcut(QKeySequence::StandardKey::SelectAll);
        break;

    case WebView::ActionID::SearchSelectedText:
        qaction.setIcon(load_icon_from_uri("resource://icons/16x16/find.png"sv));
        break;

    case WebView::ActionID::ToggleBookmark:
        qaction.setShortcut(QKeySequence(Qt::CTRL | Qt::Key_D));
        break;
    case WebView::ActionID::ToggleBookmarksBar:
        qaction.setShortcut(QKeySequence(Qt::CTRL | Qt::SHIFT | Qt::Key_B));
        break;
    case WebView::ActionID::BookmarkItem:
        if (auto icon = action.base64_png_icon(); icon.has_value())
            qaction.setIcon(icon_from_base64_png(*icon));
        else
            qaction.setIcon(create_tvg_icon_with_theme_colors("globe", palette));
        break;

    case WebView::ActionID::OpenProcessesPage:
        qaction.setIcon(load_icon_from_uri("resource://icons/16x16/app-system-monitor.png"sv));
        qaction.setShortcut(QKeySequence(Qt::CTRL | Qt::SHIFT | Qt::Key_M));
        break;
    case WebView::ActionID::OpenSettingsPage:
        qaction.setIcon(load_icon_from_uri("resource://icons/16x16/settings.png"sv));
        qaction.setShortcut(QKeySequence::StandardKey::Preferences);
        break;
    case WebView::ActionID::ToggleDevTools:
        qaction.setIcon(load_icon_from_uri("resource://icons/browser/dom-tree.png"sv));
        qaction.setShortcuts({
            QKeySequence(Qt::CTRL | Qt::SHIFT | Qt::Key_I),
            QKeySequence(Qt::CTRL | Qt::SHIFT | Qt::Key_C),
            QKeySequence(Qt::Key_F12),
        });
        break;
    case WebView::ActionID::ViewSource:
        qaction.setIcon(load_icon_from_uri("resource://icons/16x16/filetype-html.png"sv));
        qaction.setShortcut(QKeySequence(Qt::CTRL | Qt::Key_U));
        break;

    case WebView::ActionID::TakeVisibleScreenshot:
    case WebView::ActionID::TakeFullScreenshot:
        qaction.setIcon(load_icon_from_uri("resource://icons/16x16/filetype-image.png"sv));
        break;

    case WebView::ActionID::OpenInNewTab:
        qaction.setIcon(load_icon_from_uri("resource://icons/16x16/new-tab.png"sv));
        break;
    case WebView::ActionID::CopyURL:
        qaction.setIcon(load_icon_from_uri("resource://icons/16x16/edit-copy.png"sv));
        break;

    case WebView::ActionID::OpenImage:
        qaction.setIcon(load_icon_from_uri("resource://icons/16x16/filetype-image.png"sv));
        break;
    case WebView::ActionID::SaveImage:
        qaction.setIcon(load_icon_from_uri("resource://icons/16x16/download.png"sv));
        break;
    case WebView::ActionID::CopyImage:
        qaction.setIcon(load_icon_from_uri("resource://icons/16x16/edit-copy.png"sv));
        break;

    case WebView::ActionID::OpenAudio:
        qaction.setIcon(load_icon_from_uri("resource://icons/16x16/filetype-sound.png"sv));
        break;
    case WebView::ActionID::OpenVideo:
        qaction.setIcon(load_icon_from_uri("resource://icons/16x16/filetype-video.png"sv));
        break;
    case WebView::ActionID::PlayMedia:
        qaction.setIcon(load_icon_from_uri("resource://icons/16x16/play.png"sv));
        break;
    case WebView::ActionID::PauseMedia:
        qaction.setIcon(load_icon_from_uri("resource://icons/16x16/pause.png"sv));
        break;
    case WebView::ActionID::MuteMedia:
        qaction.setIcon(load_icon_from_uri("resource://icons/16x16/audio-volume-muted.png"sv));
        break;
    case WebView::ActionID::UnmuteMedia:
        qaction.setIcon(load_icon_from_uri("resource://icons/16x16/audio-volume-high.png"sv));
        break;
    case WebView::ActionID::EnterFullscreen:
    case WebView::ActionID::ExitFullscreen: // FIXME: Create a separate icon for exiting fullscreen.
        qaction.setIcon(load_icon_from_uri("resource://icons/16x16/fullscreen.png"sv));
        break;

    case WebView::ActionID::ZoomIn: {
        qaction.setIcon(load_icon_from_uri("resource://icons/16x16/zoom-in.png"sv));

        auto zoom_in_shortcuts = QKeySequence::keyBindings(QKeySequence::StandardKey::ZoomIn);
        auto secondary_zoom_in_shortcut = QKeySequence(Qt::CTRL | Qt::Key_Equal);

        if (!zoom_in_shortcuts.contains(secondary_zoom_in_shortcut))
            zoom_in_shortcuts.append(move(secondary_zoom_in_shortcut));

        qaction.setShortcuts(zoom_in_shortcuts);
        break;
    }
    case WebView::ActionID::ZoomOut:
        qaction.setIcon(load_icon_from_uri("resource://icons/16x16/zoom-out.png"sv));
        qaction.setShortcut(QKeySequence::StandardKey::ZoomOut);
        break;
    case WebView::ActionID::ResetZoom:
        qaction.setIcon(load_icon_from_uri("resource://icons/16x16/zoom-reset.png"sv));
        qaction.setShortcut(QKeySequence(Qt::CTRL | Qt::Key_0));
        break;

    case WebView::ActionID::DumpSessionHistoryTree:
        qaction.setIcon(load_icon_from_uri("resource://icons/16x16/history.png"sv));
        break;
    case WebView::ActionID::DumpDOMTree:
        qaction.setIcon(load_icon_from_uri("resource://icons/browser/dom-tree.png"sv));
        break;
    case WebView::ActionID::DumpLayoutTree:
    case WebView::ActionID::DumpPaintTree:
    case WebView::ActionID::DumpDisplayList:
        qaction.setIcon(load_icon_from_uri("resource://icons/16x16/layout.png"sv));
        break;
    case WebView::ActionID::DumpStackingContextTree:
        qaction.setIcon(load_icon_from_uri("resource://icons/16x16/layers.png"sv));
        break;
    case WebView::ActionID::DumpStyleSheets:
    case WebView::ActionID::DumpStyles:
        qaction.setIcon(load_icon_from_uri("resource://icons/16x16/filetype-css.png"sv));
        break;
    case WebView::ActionID::DumpCSSErrors:
        qaction.setIcon(load_icon_from_uri("resource://icons/16x16/error.png"sv));
        break;
    case WebView::ActionID::DumpCookies:
        qaction.setIcon(load_icon_from_uri("resource://icons/browser/cookie.png"sv));
        break;
    case WebView::ActionID::DumpLocalStorage:
        qaction.setIcon(load_icon_from_uri("resource://icons/browser/local-storage.png"sv));
        break;
    case WebView::ActionID::ShowLineBoxBorders:
        qaction.setIcon(load_icon_from_uri("resource://icons/16x16/box.png"sv));
        break;
    case WebView::ActionID::CollectGarbage:
        qaction.setIcon(load_icon_from_uri("resource://icons/16x16/trash-can.png"sv));
        qaction.setShortcut(QKeySequence(Qt::CTRL | Qt::SHIFT | Qt::Key_G));
        break;

    default:
        break;
    }

    if (action.is_checkable())
        qaction.setCheckable(true);

    action.add_observer(ActionObserver::create(action, qaction));
    add_properties(qaction, action);
}

static void add_items_to_menu(QMenu& qmenu, QWidget& parent, WebView::Menu& menu)
{
    add_properties(qmenu, menu);

    for (auto& menu_item : menu.items()) {
        menu_item.visit(
            [&](NonnullRefPtr<WebView::Action>& action) {
                auto* qaction = create_application_action(parent, action);
                qmenu.addAction(qaction);

                if (action->id() == WebView::ActionID::SpoofUserAgent || action->id() == WebView::ActionID::NavigatorCompatibilityMode) {
                    if (qmenu.icon().isNull())
                        qmenu.setIcon(load_icon_from_uri("resource://icons/16x16/spoof.png"sv));
                }
            },
            [&](NonnullRefPtr<WebView::Menu> const& submenu) {
                auto* qsubmenu = new QMenu(qstring_from_ak_string(submenu->title()), &qmenu);
                add_items_to_menu(*qsubmenu, parent, submenu);

                if (submenu->render_group_icon())
                    qsubmenu->setIcon(create_tvg_icon_with_theme_colors("folder", parent.palette()));

                add_properties(*qsubmenu, *submenu);
                qmenu.addMenu(qsubmenu);
            },
            [&](WebView::Separator) {
                qmenu.addSeparator();
            });
    }
}

QMenu* create_application_menu(QWidget& parent, WebView::Menu& menu)
{
    auto* application_menu = new QMenu(qstring_from_ak_string(menu.title()), &parent);
    add_items_to_menu(*application_menu, parent, menu);
    return application_menu;
}

void repopulate_application_menu(QMenu& menu, QWidget& parent, WebView::Menu& source)
{
    menu.clear();
    add_items_to_menu(menu, parent, source);
}

QMenu* create_context_menu(QWidget& parent, WebContentView& view, WebView::Menu& menu)
{
    auto* application_menu = create_application_menu(parent, menu);

    menu.on_activation = [view = QPointer { &view }, application_menu = QPointer { application_menu }](Gfx::IntPoint position) {
        if (view && application_menu)
            application_menu->exec(view->map_point_to_global_position(position));
    };

    return application_menu;
}

QAction* create_application_action(QWidget& parent, WebView::Action& action)
{
    auto* qaction = new QAction(&parent);
    initialize_native_control(action, *qaction, parent.palette());
    return qaction;
}

}