kicad-source/common/widgets/webview_panel.cpp

#include <build_version.h>
#include <widgets/webview_panel.h>
#include <wx/sizer.h>
#include <wx/webviewarchivehandler.h>
#include <wx/webviewfshandler.h>
#include <wx/utils.h>
#include <wx/log.h>

WEBVIEW_PANEL::WEBVIEW_PANEL( wxWindow* aParent, wxWindowID aId, const wxPoint& aPos,
                             const wxSize& aSize, const int aStyle )
    : wxPanel( aParent, aId, aPos, aSize, aStyle ),
      m_initialized( false ),
      m_browser( wxWebView::New() )
{
    wxBoxSizer* sizer = new wxBoxSizer( wxVERTICAL );

    if( !wxGetEnv( wxT( "WEBKIT_DISABLE_COMPOSITING_MODE" ), nullptr ) )
    {
        wxSetEnv( wxT( "WEBKIT_DISABLE_COMPOSITING_MODE" ), wxT( "1" ) );
    }

#ifdef __WXMAC__
    m_browser->RegisterHandler( wxSharedPtr<wxWebViewHandler>( new wxWebViewArchiveHandler( "wxfs" ) ) );
    m_browser->RegisterHandler( wxSharedPtr<wxWebViewHandler>( new wxWebViewFSHandler( "memory" ) ) );
#endif
    m_browser->SetUserAgent( wxString::Format( "KiCad/%s WebView/%s", GetMajorMinorPatchVersion(), wxGetOsDescription() ) );
    m_browser->Create( this, wxID_ANY, wxEmptyString, wxDefaultPosition, wxDefaultSize );
    sizer->Add( m_browser, 1, wxEXPAND );
    SetSizer( sizer );

#ifndef __WXMAC__
    m_browser->RegisterHandler( wxSharedPtr<wxWebViewHandler>( new wxWebViewArchiveHandler( "wxfs" ) ) );
    m_browser->RegisterHandler( wxSharedPtr<wxWebViewHandler>( new wxWebViewFSHandler( "memory" ) ) );
#endif

    Bind( wxEVT_WEBVIEW_NAVIGATING, &WEBVIEW_PANEL::OnNavigationRequest, this, m_browser->GetId() );
    Bind( wxEVT_WEBVIEW_NEWWINDOW, &WEBVIEW_PANEL::OnNewWindow, this, m_browser->GetId() );
    Bind( wxEVT_WEBVIEW_SCRIPT_MESSAGE_RECEIVED, &WEBVIEW_PANEL::OnScriptMessage, this, m_browser->GetId() );
    Bind( wxEVT_WEBVIEW_SCRIPT_RESULT, &WEBVIEW_PANEL::OnScriptResult, this, m_browser->GetId() );
    Bind( wxEVT_WEBVIEW_ERROR, &WEBVIEW_PANEL::OnError, this, m_browser->GetId() );
    Bind( wxEVT_WEBVIEW_LOADED, &WEBVIEW_PANEL::OnWebViewLoaded, this, m_browser->GetId() );
}

WEBVIEW_PANEL::~WEBVIEW_PANEL()
{
}

void WEBVIEW_PANEL::LoadURL( const wxString& aURL )
{
    if( aURL.starts_with( "file:/" ) && !aURL.starts_with( "file:///" ) )
    {
        wxString new_url = wxString( "file:///" ) + aURL.AfterFirst( '/' );
        m_browser->LoadURL( new_url );
        return;
    }

    if( !aURL.StartsWith( "http://" ) && !aURL.StartsWith( "https://" ) && !aURL.StartsWith( "file://" ) )
    {
        wxLogError( "Invalid URL: %s", aURL );
        return;
    }

    m_browser->LoadURL( aURL );
}

void WEBVIEW_PANEL::SetPage( const wxString& aHtmlContent )
{
    m_browser->SetPage( aHtmlContent, "file://" );
}

bool WEBVIEW_PANEL::AddMessageHandler( const wxString& aName, MESSAGE_HANDLER aHandler )
{
    m_msgHandlers.emplace( aName, std::move(aHandler) );
    return true;
}

void WEBVIEW_PANEL::ClearMessageHandlers()
{
    m_msgHandlers.clear();
}

void WEBVIEW_PANEL::OnNavigationRequest( wxWebViewEvent& aEvt )
{
    // Default behavior: open external links in the system browser
    bool isExternal = aEvt.GetURL().StartsWith( "http://" ) || aEvt.GetURL().StartsWith( "https://" );
    if( isExternal )
    {
        wxLaunchDefaultBrowser( aEvt.GetURL() );
        aEvt.Veto();
    }
}

void WEBVIEW_PANEL::OnWebViewLoaded( wxWebViewEvent& aEvt )
{
    if( !m_initialized )
    {
        // Defer handler registration to avoid running during modal dialog/yield
        CallAfter([this]() {
            static bool handler_added_inner = false;
            if (!handler_added_inner) {

                for( const auto& handler : m_msgHandlers )
                {
                    if( !m_browser->AddScriptMessageHandler( handler.first ) )
                    {
                        wxLogDebug( "Could not add script message handler %s", handler.first );
                    }
                }

                handler_added_inner = true;
            }

            // Inject navigation hook for SPA/JS navigation to prevent webkit crashing without new window
            m_browser->AddUserScript(R"(
                (function() {
                    // Change window.open to navigate in the same window
                    window.open = function(url) { if (url) window.location.href = url; return null; };
                    window.showModalDialog = function() { return null; };

                    if (window.external && window.external.invoke) {
                        function notifyHost() {
                            window.external.invoke('navigation:' + window.location.href);
                        }
                        window.addEventListener('popstate', notifyHost);
                        window.addEventListener('pushstate', notifyHost);
                        window.addEventListener('replacestate', notifyHost);
                        ['pushState', 'replaceState'].forEach(function(type) {
                            var orig = history[type];
                            history[type] = function() {
                                var rv = orig.apply(this, arguments);
                                window.dispatchEvent(new Event(type.toLowerCase()));
                                return rv;
                            };
                        });
                    }
                })();
            )");
        });

        m_initialized = true;
    }
}

void WEBVIEW_PANEL::OnNewWindow( wxWebViewEvent& aEvt )
{
    m_browser->LoadURL( aEvt.GetURL() );
    aEvt.Veto(); // Prevent default behavior of opening a new window
    wxLogTrace( "webview", "New window requested for URL: %s", aEvt.GetURL() );
    wxLogTrace( "webview", "Target: %s", aEvt.GetTarget() );
    wxLogTrace( "webview", "Action flags: %d", static_cast<int>(aEvt.GetNavigationAction()) );
    wxLogTrace( "webview", "Message handler: %s", aEvt.GetMessageHandler() );
}

void WEBVIEW_PANEL::OnScriptMessage( wxWebViewEvent& aEvt )
{
    wxLogTrace( "webview", "Script message received: %s for handler %s", aEvt.GetString(), aEvt.GetMessageHandler() );

    if( aEvt.GetMessageHandler().IsEmpty() )
    {
        wxLogDebug( "No message handler specified for script message: %s", aEvt.GetString() );
        return;
    }

    auto it = m_msgHandlers.find( aEvt.GetMessageHandler() );
    if( it == m_msgHandlers.end() )
    {
        wxLogDebug( "No handler registered for message: %s", aEvt.GetMessageHandler() );
        return;
    }

    // Call the registered handler with the message
    wxLogTrace( "webview", "Calling handler for message: %s", aEvt.GetMessageHandler() );
    it->second( aEvt.GetString() );
}

void WEBVIEW_PANEL::OnScriptResult( wxWebViewEvent& aEvt )
{
    if( aEvt.IsError() )
        wxLogDebug( "Async script execution failed: %s", aEvt.GetString() );
}

void WEBVIEW_PANEL::OnError( wxWebViewEvent& aEvt )
{
    wxLogDebug( "WebView error: %s", aEvt.GetString() );
}