1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145
|
/* Copyright (c) MediaArea.net SARL. All Rights Reserved.
*
* Use of this source code is governed by a BSD-style license that can
* be found in the License.html file in the root of the source tree.
*/
// webview2widget.cpp
// From code generated with Qwen2.5-Max
#include "webview2widget.h"
#include <QDebug>
#include <QResizeEvent>
#include <QWindow>
#include <QTimer>
#include <Shlobj.h>
#include <WebView2.h>
#include <windows.h>
#include <wrl.h>
#pragma comment(lib, "User32.lib")
namespace {
std::wstring GetUserDataFolder() {
wchar_t appDataPath[MAX_PATH];
SHGetFolderPathW(nullptr, CSIDL_LOCAL_APPDATA, nullptr, 0, appDataPath);
std::wstring userDataFolder = appDataPath;
userDataFolder += L"\\mediainfo-gui\\WebView2";
// Create the folder if it doesn't exist
CreateDirectoryW(userDataFolder.c_str(), nullptr);
return userDataFolder;
}
}
WebView2Widget::WebView2Widget(QWidget *parent)
: QWidget(parent), m_hwndHost(nullptr), m_isInitialized(false) {
// Set up the widget to host WebView2
m_hwndHost = (HWND)this->winId();
// Initialize WebView2
HRESULT hr = InitializeWebView();
if (FAILED(hr)) {
qWarning() << "Failed to initialize WebView2";
}
}
WebView2Widget::~WebView2Widget() {
if (m_webviewController) {
m_webviewController->Close();
}
}
HRESULT WebView2Widget::InitializeWebView() {
HRESULT hr = CreateCoreWebView2EnvironmentWithOptions(
nullptr, GetUserDataFolder().c_str(), nullptr,
Microsoft::WRL::Callback<ICoreWebView2CreateCoreWebView2EnvironmentCompletedHandler>(
[this](HRESULT result, ICoreWebView2Environment *env) -> HRESULT {
if (result == S_OK) {
env->CreateCoreWebView2Controller(
m_hwndHost,
Microsoft::WRL::Callback<ICoreWebView2CreateCoreWebView2ControllerCompletedHandler>(
[this](HRESULT result, ICoreWebView2Controller *controller) -> HRESULT {
Q_UNUSED(result);
if (controller != nullptr) {
m_webviewController = controller;
m_webviewController->get_CoreWebView2(&m_webview);
RECT bounds;
GetClientRect(m_hwndHost, &bounds);
m_webviewController->put_Bounds(bounds);
m_webviewController->put_IsVisible(TRUE);
// Mark WebView2 as initialized
m_isInitialized = true;
// Use a single-shot timer to delay resizing until after the widget is properly laid out
QTimer::singleShot(0, this, [this]() {
resizeEvent(nullptr); // Trigger resizeEvent manually
});
// Process any pending requests
ProcessPendingRequests();
}
return S_OK;
}
).Get()
);
}
return result;
}
).Get()
);
return hr;
}
void WebView2Widget::resizeEvent(QResizeEvent *event) {
QWidget::resizeEvent(event);
if (m_webviewController) {
RECT bounds;
GetClientRect(m_hwndHost, &bounds);
m_webviewController->put_Bounds(bounds);
}
}
void WebView2Widget::ProcessPendingRequests() {
// Process pending HTML content
while (!m_pendingHtml.isEmpty()) {
QString html = m_pendingHtml.dequeue();
std::wstring htmlWide = html.toStdWString();
m_webview->NavigateToString(htmlWide.c_str());
}
// Process pending URLs
while (!m_pendingUrls.isEmpty()) {
QUrl url = m_pendingUrls.dequeue();
std::wstring urlWide = url.toString().toStdWString();
m_webview->Navigate(urlWide.c_str());
}
}
void WebView2Widget::setHtml(const QString &html) {
if (m_isInitialized && m_webview) {
std::wstring htmlWide = html.toStdWString();
m_webview->NavigateToString(htmlWide.c_str());
} else {
// Queue the HTML content if WebView2 is not yet initialized
m_pendingHtml.enqueue(html);
}
}
void WebView2Widget::load(const QUrl &url) {
if (m_isInitialized && m_webview) {
std::wstring urlWide = url.toString().toStdWString();
m_webview->Navigate(urlWide.c_str());
} else {
// Queue the URL if WebView2 is not yet initialized
m_pendingUrls.enqueue(url);
}
}
void WebView2Widget::load(const QString &url) { load(QUrl(url)); }
|