Implement page action popups.

chrome/browser/views/location_bar_view.cc

 // Copyright (c) 2009 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/views/location_bar_view.h"
 
 #if defined(OS_LINUX)
 #include <gtk/gtk.h>
 #endif
 
 #include "app/gfx/canvas.h"
 #include "app/gfx/color_utils.h"
 #include "app/l10n_util.h"
 #include "app/resource_bundle.h"
 #include "app/theme_provider.h"
 #include "base/stl_util-inl.h"
 #include "chrome/app/chrome_dll_resource.h"
 #include "chrome/browser/alternate_nav_url_fetcher.h"
 #include "chrome/browser/browser_list.h"
 #include "chrome/browser/bubble_positioner.h"
 #include "chrome/browser/command_updater.h"
 #include "chrome/browser/extensions/extension_browser_event_router.h"
 #include "chrome/browser/extensions/extensions_service.h"
 #include "chrome/browser/profile.h"
 #include "chrome/browser/search_engines/template_url_model.h"
 #include "chrome/browser/view_ids.h"
+#include "chrome/browser/views/extensions/extension_popup.h"
 #include "grit/generated_resources.h"
 #include "grit/theme_resources.h"
 
 #if defined(OS_WIN)
 #include "chrome/browser/views/first_run_bubble.h"
 #endif
 
 using views::View;
 
 // static
@@ skipping 651 matching lines @@
     page_action_views_.clear();
   }
 }
 
 void LocationBarView::RefreshPageActionViews() {
   std::vector<ExtensionAction*> page_actions;
   ExtensionsService* service = profile_->GetExtensionsService();
   if (!service)
     return;
 
+  std::map<ExtensionAction*, bool> old_visibility;
+  for (size_t i = 0; i < page_action_views_.size(); i++) {
+    old_visibility[page_action_views_[i]->image_view()->page_action()] =
+        page_action_views_[i]->IsVisible();
+  }
+
+  // Remember the previous visibility of the page actions so that we can
+  // notify when this changes.
   for (size_t i = 0; i < service->extensions()->size(); ++i) {
     if (service->extensions()->at(i)->page_action())
       page_actions.push_back(service->extensions()->at(i)->page_action());
   }
 
   // On startup we sometimes haven't loaded any extensions. This makes sure
   // we catch up when the extensions (and any page actions) load.
   if (page_actions.size() != page_action_views_.size()) {
     DeletePageActionViews();  // Delete the old views (if any).
 
     page_action_views_.resize(page_actions.size());
 
     for (size_t i = 0; i < page_actions.size(); ++i) {
       page_action_views_[i] = new PageActionWithBadgeView(
           new PageActionImageView(this, profile_,
                                   page_actions[i], bubble_positioner_));
       page_action_views_[i]->SetVisible(false);
       page_action_views_[i]->SetParentOwned(false);
       AddChildView(page_action_views_[i]);
     }
   }
 
   TabContents* contents = delegate_->GetTabContents();
   if (!page_action_views_.empty() && contents) {
     GURL url = GURL(WideToUTF8(model_->GetText()));
 
-    for (size_t i = 0; i < page_action_views_.size(); i++)
+    for (size_t i = 0; i < page_action_views_.size(); i++) {
       page_action_views_[i]->UpdateVisibility(contents, url);
+
+      // Check if the visibility of the action changed and notify if it did.
+      ExtensionAction* action =
+          page_action_views_[i]->image_view()->page_action();
+      if (old_visibility.find(action) == old_visibility.end() ||
+          old_visibility[action] != page_action_views_[i]->IsVisible()) {
+        NotificationService::current()->Notify(
+            NotificationType::EXTENSION_PAGE_ACTION_VISIBILITY_CHANGED,
+            Source<ExtensionAction>(action),
+            Details<TabContents>(contents));
+      }
+    }
   }
 }
 
 void LocationBarView::SetInfoText(const std::wstring& text,
                                   ToolbarModel::InfoTextType text_type,
                                   const std::wstring& tooltip_text) {
   info_label_.SetVisible(!text.empty());
   info_label_.SetText(text);
   if (text_type == ToolbarModel::INFO_EV_TEXT)
     info_label_.SetColor(GetColor(true, SECURITY_TEXT));
@@ skipping 525 matching lines @@
         gfx::Size(Extension::kPageActionIconMaxSize,
                   Extension::kPageActionIconMaxSize));
   }
 }
 
 LocationBarView::PageActionImageView::~PageActionImageView() {
   if (tracker_)
     tracker_->StopTrackingImageLoad();
 }
 
+void LocationBarView::PageActionImageView::ExecuteAction(int button) {
+  if (page_action_->has_popup()) {
+    gfx::Point origin;
+    View::ConvertPointToScreen(this, &origin);
+    gfx::Rect rect = bounds();
+    rect.set_x(origin.x());
+    rect.set_y(origin.y());
+
+    // In tests, GetLastActive could return NULL, so we need to have
+    // a fallback.
+    // TODO(erikkay): Find a better way to get the Browser that this
+    // button is in.
+    Browser* browser = BrowserList::GetLastActiveWithProfile(profile_);
+    if (!browser)
+      browser = BrowserList::FindBrowserWithProfile(profile_);
+    ExtensionPopup::Show(page_action_->popup_url(), browser, rect);
+  } else {
+    ExtensionBrowserEventRouter::GetInstance()->PageActionExecuted(
+        profile_, page_action_->extension_id(), page_action_->id(),
+        current_tab_id_, current_url_.spec(), button);
+  }
+}
+
+void LocationBarView::PageActionImageView::OnMouseMoved(
+    const views::MouseEvent& event) {
+  // PageActionImageView uses normal tooltips rather than the info bubble,
+  // so just do nothing here rather than letting LocationBarImageView start
+  // its hover timer.
+}
+
 bool LocationBarView::PageActionImageView::OnMousePressed(
     const views::MouseEvent& event) {
   int button = -1;
   if (event.IsLeftMouseButton())
     button = 1;
   else if (event.IsMiddleMouseButton())
     button = 2;
   else if (event.IsRightMouseButton())
     button = 3;
-  // Our PageAction icon was clicked on, notify proper authorities.
-  ExtensionBrowserEventRouter::GetInstance()->PageActionExecuted(
-      profile_, page_action_->extension_id(), page_action_->id(),
-      current_tab_id_, current_url_.spec(), button);
+
+  ExecuteAction(button);
   return true;
 }
 
 void LocationBarView::PageActionImageView::ShowInfoBubble() {
   ShowInfoBubbleImpl(ASCIIToWide(tooltip_), GetColor(false, TEXT));
 }
 
 void LocationBarView::PageActionImageView::OnImageLoaded(SkBitmap* image,
                                                          size_t index) {
   // We loaded icons()->size() icons, plus one extra if the page action had
@@ skipping 23 matching lines @@
     TabContents* contents, const GURL& url) {
   // Save this off so we can pass it back to the extension when the action gets
   // executed. See PageActionImageView::OnMousePressed.
   current_tab_id_ = ExtensionTabUtil::GetTabId(contents);
   current_url_ = url;
 
   bool visible = page_action_->GetIsVisible(current_tab_id_);
   if (visible) {
     // Set the tooltip.
     tooltip_ = page_action_->GetTitle(current_tab_id_);
+    SetTooltipText(ASCIIToWide(tooltip_));
 
     // Set the image.
     // It can come from three places. In descending order of priority:
     // - The developer can set it dynamically by path or bitmap. It will be in
     //   page_action_->GetIcon().
     // - The developer can set it dynamically by index. It will be in
     //   page_action_->GetIconIndex().
     // - It can be set in the manifest by path. It will be in page_action_->
     //   default_icon_path().
 
@@ skipping 69 matching lines @@
 }
 
 int LocationBarView::PageActionVisibleCount() {
   int result = 0;
   for (size_t i = 0; i < page_action_views_.size(); i++) {
     if (page_action_views_[i]->IsVisible())
       ++result;
   }
   return result;
 }
+
+ExtensionAction* LocationBarView::GetPageAction(size_t index) {
+  if (index < page_action_views_.size())
+    return page_action_views_[index]->image_view()->page_action();
+
+  NOTREACHED();
+  return NULL;
+}
+
+ExtensionAction* LocationBarView::GetVisiblePageAction(size_t index) {
+  if (index >= 0) {
+    size_t current = 0;
+    for (size_t i = 0; i < page_action_views_.size(); ++i) {
+      if (page_action_views_[i]->IsVisible()) {
+        if (current == index)
+          return page_action_views_[i]->image_view()->page_action();
+
+        ++current;
+      }
+    }
+  }
+
+  NOTREACHED();
+  return NULL;
+}
+
+void LocationBarView::TestPageActionPressed(size_t index) {
+  if (index >= 0) {
+    size_t current = 0;
+    for (size_t i = 0; i < page_action_views_.size(); ++i) {
+      if (page_action_views_[i]->IsVisible()) {
+        if (current == index) {
+          const int button = 1;  // Left mouse button.
+          page_action_views_[i]->image_view()->ExecuteAction(button);
+          return;
+        }
+        ++current;
+      }
+    }
+  }
+  NOTREACHED();
+}