Copy url from omnibox instead of webcontent (#17254)

brave · Feb 17, 2023 · d7e38a1 · d7e38a1
1 parent 67d73b3
commit d7e38a1
Show file tree

Hide file tree

Showing 9 changed files with 36 additions and 1 deletion.
diff --git a/browser/brave_app_controller_mac.mm b/browser/brave_app_controller_mac.mm
@@ -78,7 +78,7 @@ - (BOOL)validateUserInterfaceItem:(id<NSValidatedUserInterfaceItem>)item {
 - (void)executeCommand:(id)sender withProfile:(Profile*)profile {
   NSInteger tag = [sender tag];
   if (tag == IDC_COPY_CLEAN_LINK) {
-    chrome::ExecuteCommand([self getBrowser], IDC_COPY_CLEAN_LINK);
+    brave::CleanAndCopySelectedURL([self getBrowser]);
     return;
   }
 

diff --git a/browser/ui/brave_browser_window.cc b/browser/ui/brave_browser_window.cc
@@ -31,4 +31,6 @@ sidebar::Sidebar* BraveBrowserWindow::InitSidebar() {
 bool BraveBrowserWindow::HasSelectedURL() const {
   return false;
 }
+void BraveBrowserWindow::CleanAndCopySelectedURL() {}
+
 #endif
diff --git a/browser/ui/brave_browser_window.h b/browser/ui/brave_browser_window.h
@@ -48,6 +48,7 @@ class BraveBrowserWindow : public BrowserWindow {
 #if defined(TOOLKIT_VIEWS)
   virtual sidebar::Sidebar* InitSidebar();
   virtual bool HasSelectedURL() const;
+  virtual void CleanAndCopySelectedURL();
 #endif
 
   virtual void ShowBraveVPNBubble() {}

diff --git a/browser/ui/browser_commands.cc b/browser/ui/browser_commands.cc
@@ -285,4 +285,13 @@ bool HasSelectedURL(Browser* browser) {
   return brave_browser_window && brave_browser_window->HasSelectedURL();
 }
 
+void CleanAndCopySelectedURL(Browser* browser) {
+  if (!browser) {
+    return;
+  }
+  auto* brave_browser_window = BraveBrowserWindow::From(browser->window());
+  if (brave_browser_window) {
+    brave_browser_window->CleanAndCopySelectedURL();
+  }
+}
 }  // namespace brave
diff --git a/browser/ui/browser_commands.h b/browser/ui/browser_commands.h
@@ -11,6 +11,7 @@ class GURL;
 
 namespace brave {
 bool HasSelectedURL(Browser* browser);
+void CleanAndCopySelectedURL(Browser* browser);
 void NewOffTheRecordWindowTor(Browser*);
 void NewTorConnectionForSite(Browser*);
 void AddNewProfile();

diff --git a/browser/ui/views/frame/brave_browser_view.cc b/browser/ui/views/frame/brave_browser_view.cc
@@ -398,6 +398,18 @@ bool BraveBrowserView::HasSelectedURL() const {
   return brave_omnibox_view && brave_omnibox_view->SelectedTextIsURL();
 }
 
+void BraveBrowserView::CleanAndCopySelectedURL() {
+  if (!GetLocationBarView()) {
+    return;
+  }
+  auto* brave_omnibox_view =
+      static_cast<BraveOmniboxViewViews*>(GetLocationBarView()->omnibox_view());
+  if (!brave_omnibox_view) {
+    return;
+  }
+  brave_omnibox_view->CleanAndCopySelectedURL();
+}
+
 WalletButton* BraveBrowserView::GetWalletButton() {
   return static_cast<BraveToolbarView*>(toolbar())->wallet_button();
 }

diff --git a/browser/ui/views/frame/brave_browser_view.h b/browser/ui/views/frame/brave_browser_view.h
@@ -116,6 +116,7 @@ class BraveBrowserView : public BrowserView {
 
   sidebar::Sidebar* InitSidebar() override;
   bool HasSelectedURL() const override;
+  void CleanAndCopySelectedURL() override;
   void UpdateSideBarHorizontalAlignment();
 
   bool closing_confirm_dialog_activated_ = false;

diff --git a/browser/ui/views/omnibox/brave_omnibox_view_views.cc b/browser/ui/views/omnibox/brave_omnibox_view_views.cc
@@ -47,6 +47,14 @@ bool BraveOmniboxViewViews::SelectedTextIsURL() {
   return GetURLToCopy().has_value();
 }
 
+void BraveOmniboxViewViews::CleanAndCopySelectedURL() {
+  auto url_to_copy = GetURLToCopy();
+  if (!url_to_copy.has_value()) {
+    return;
+  }
+  CopySanitizedURL(url_to_copy.value());
+}
+
 void BraveOmniboxViewViews::CopySanitizedURL(const GURL& url) {
   OnBeforePossibleChange();
   brave::CopySanitizedURL(chrome::FindLastActive(), url);

diff --git a/browser/ui/views/omnibox/brave_omnibox_view_views.h b/browser/ui/views/omnibox/brave_omnibox_view_views.h
@@ -20,6 +20,7 @@ class BraveOmniboxViewViews : public OmniboxViewViews {
   ~BraveOmniboxViewViews() override;
 
   bool SelectedTextIsURL();
+  void CleanAndCopySelectedURL();
 
  protected:
   absl::optional<GURL> GetURLToCopy();