WebView2View.cpp

#include "WebView2View.h"
#include <QQuickWindow>
#include <QDebug>
#include <windowsx.h>
#include <QStandardPaths>

WebView2View::WebView2View() {
    setFlag(ItemHasContents, false);
    setVisible(true);
}

WebView2View::~WebView2View() {
    if (m_controller) {
        m_controller->Close();
    }
}

void WebView2View::componentComplete() {
    QQuickItem::componentComplete();
    initializeWebView();
}

void WebView2View::initializeWebView() {
    if (!window()) return;

    HWND parent = (HWND)window()->winId();
    RECT bounds;
    GetClientRect(parent, &bounds);

    QString userData = QStandardPaths::writableLocation(QStandardPaths::AppLocalDataLocation) + "/wxfilehelperclientData";
    std::wstring userDataW = std::wstring(reinterpret_cast<const wchar_t*>(userData.utf16()));;

    CreateCoreWebView2EnvironmentWithOptions(
        nullptr, userDataW.c_str(), nullptr,
        Microsoft::WRL::Callback<ICoreWebView2CreateCoreWebView2EnvironmentCompletedHandler>(
            [this, parent, bounds](HRESULT result, ICoreWebView2Environment* env) -> HRESULT {
                if (FAILED(result)) {
                    qWarning() << "Failed to create WebView2 environment:" << result;
                    return result;
                }
                env->CreateCoreWebView2Controller(parent,
                    Microsoft::WRL::Callback<ICoreWebView2CreateCoreWebView2ControllerCompletedHandler>(
                        [this, bounds](HRESULT result, ICoreWebView2Controller* controller) -> HRESULT {
                            if (FAILED(result)) {
                                qWarning() << "Failed to create WebView2 controller:" << result;
                                return result;
                            }
                            m_controller = controller;
                            m_controller->get_CoreWebView2(&m_webview);

                            if (m_webview) {
                                m_webview->add_NavigationCompleted(
                                    Microsoft::WRL::Callback<ICoreWebView2NavigationCompletedEventHandler>(
                                        [this](ICoreWebView2*, ICoreWebView2NavigationCompletedEventArgs* args) -> HRESULT {
                                            BOOL success;
                                            args->get_IsSuccess(&success);
                                            emit loadingChanged(success);
                                            return S_OK;
                                    }).Get(),
                                nullptr);
                            }

                            resizeWebView();
                            if (!m_url.isEmpty())
                                m_webview->Navigate(m_url.toStdWString().c_str());

                            return S_OK;
                    }).Get());
                return S_OK;
            }).Get());
}

void WebView2View::resizeWebView() {
    if (!m_controller) return;
    QPointF topLeft = mapToItem(window()->contentItem(), QPointF(0,0));
    qreal dpr = window()->devicePixelRatio();

    LONG left   = qRound(topLeft.x() * dpr);
    LONG top    = qRound(topLeft.y() * dpr);
    LONG right  = qRound((topLeft.x() + width()) * dpr);
    LONG bottom = qRound((topLeft.y() + height()) * dpr);

    RECT bounds{ left, top, right, bottom };
    HRESULT hr = m_controller->put_Bounds(bounds);
    if (FAILED(hr)) qWarning() << "put_Bounds failed:" << hr;

    m_controller->put_IsVisible(isVisible);
}

void WebView2View::geometryChange(const QRectF &newGeometry, const QRectF &oldGeometry) {
    QQuickItem::geometryChange(newGeometry, oldGeometry);
    resizeWebView();
}

void WebView2View::releaseResources() {
    if (m_controller)
        m_controller->Close();
    m_controller = nullptr;
    m_webview = nullptr;
}

void WebView2View::setUrl(const QString &url) {
    if (url == m_url) return;
    m_url = url;
    if (m_webview)
        m_webview->Navigate(url.toStdWString().c_str());
    emit urlChanged();
}

void WebView2View::runJavaScript(const QString &script) {
    if (!m_webview) {
        qWarning() << "WebView2 not initialized. Cannot execute script.";
        return;
    }

    HRESULT hr = m_webview->ExecuteScript(
        script.toStdWString().c_str(),
        Callback<ICoreWebView2ExecuteScriptCompletedHandler>(
            [script](HRESULT errorCode, LPCWSTR resultObjectAsJson) -> HRESULT {
                if (FAILED(errorCode)) {
                    qWarning().noquote()
                        << QStringLiteral("runJavaScript failed (HRESULT=0x%1) executing script: %2")
                               .arg(QString::number(errorCode, 16).rightJustified(8, '0'))
                               .arg(script);
                } else if (resultObjectAsJson) {
                    QString result = QString::fromWCharArray(resultObjectAsJson);

                    if (result.contains("Error") || result.contains("Exception")) {
                        qWarning().noquote()
                            << QStringLiteral("runJavaScript returned error result: %1")
                                   .arg(result);
                    }
                }
                return S_OK;
            }).Get());

    if (FAILED(hr)) {
        qWarning().noquote()
            << QStringLiteral("ExecuteScript() call failed immediately, HRESULT=0x%1")
                   .arg(QString::number(hr, 16).rightJustified(8, '0'));
    }
}

void WebView2View::resetZoom()
{
    if (m_controller)
        m_controller->put_ZoomFactor(1.0);
}

void WebView2View::show()
{
    if (m_controller) {
        m_controller->put_IsVisible(TRUE);
        isVisible = true;
    }
}

void WebView2View::hide()
{
    if (m_controller) {
        m_controller->put_IsVisible(FALSE);
        isVisible = true;
    }
}