From 944d337fcfde2abb2d3d5f6e53b4bab79d09ae21 Mon Sep 17 00:00:00 2001
From: Lawrence Chen <54008264+lawrencecchen@users.noreply.github.com>
Date: Sun, 1 Mar 2026 21:29:14 -0800
Subject: [PATCH] Keep localhost visible in SSH omnibar URLs

Remote loopback requests still route through the localtest.me alias for proxy transport, but the omnibar now canonicalizes the alias host back to localhost for display.
---
 Sources/Panels/BrowserPanel.swift | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

diff --git a/Sources/Panels/BrowserPanel.swift b/Sources/Panels/BrowserPanel.swift
index f52d091e..4a68ecb4 100644
--- a/Sources/Panels/BrowserPanel.swift
+++ b/Sources/Panels/BrowserPanel.swift
@@ -1658,6 +1658,16 @@ final class BrowserPanel: Panel, ObservableObject {
         return components?.url ?? url
     }
 
+    private func canonicalLoopbackURLForDisplayIfNeeded(_ url: URL) -> URL {
+        guard remoteProxyEndpoint != nil else { return url }
+        guard let host = BrowserInsecureHTTPSettings.normalizeHost(url.host ?? "") else { return url }
+        guard host == Self.remoteLoopbackProxyAliasHost else { return url }
+
+        var components = URLComponents(url: url, resolvingAgainstBaseURL: false)
+        components?.host = "localhost"
+        return components?.url ?? url
+    }
+
     private func preparedNavigationRequest(_ request: URLRequest) -> URLRequest {
         var prepared = browserPreparedNavigationRequest(request)
         guard let url = prepared.url else { return prepared }
@@ -2608,14 +2618,18 @@ extension BrowserPanel {
     /// Returns the most reliable URL string for omnibar-related matching and UI decisions.
     /// `currentURL` can lag behind navigation changes, so prefer the live WKWebView URL.
     func preferredURLStringForOmnibar() -> String? {
-        if let webViewURL = webView.url?.absoluteString
+        if let webViewURL = webView.url
+            .map(canonicalLoopbackURLForDisplayIfNeeded)?
+            .absoluteString
             .trimmingCharacters(in: .whitespacesAndNewlines),
            !webViewURL.isEmpty,
            webViewURL != blankURLString {
             return webViewURL
         }
 
-        if let current = currentURL?.absoluteString
+        if let current = currentURL
+            .map(canonicalLoopbackURLForDisplayIfNeeded)?
+            .absoluteString
             .trimmingCharacters(in: .whitespacesAndNewlines),
            !current.isEmpty,
            current != blankURLString {