Add infrastructure for Chrome Actions

chrome/browser/ui/views/location_bar/page_action_image_view.cc

 // Copyright (c) 2012 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/ui/views/location_bar/page_action_image_view.h"
 
 #include "base/strings/utf_string_conversions.h"
 #include "chrome/browser/extensions/extension_action.h"
 #include "chrome/browser/platform_util.h"
 #include "chrome/browser/profiles/profile.h"
+#include "chrome/browser/sessions/session_tab_helper.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/views/location_bar/location_bar_view.h"
 #include "extensions/browser/extension_registry.h"
 #include "ui/accessibility/ax_view_state.h"
 #include "ui/events/event.h"
 #include "ui/gfx/canvas.h"
 #include "ui/gfx/image/image.h"
 
 // static
 const char PageActionImageView::kViewClassName[] = "PageActionImageView";
@@ skipping 59 matching lines @@
 }
 
 void PageActionImageView::OnGestureEvent(ui::GestureEvent* event) {
   if (event->type() == ui::ET_GESTURE_TAP) {
     view_controller_->ExecuteActionByUser();
     event->SetHandled();
   }
 }
 
 void PageActionImageView::UpdateVisibility(content::WebContents* contents) {
-  int tab_id = view_controller_->GetCurrentTabId();
-
+  int tab_id = GetCurrentTabId();
   if (!contents ||
       tab_id == -1 ||
       (!preview_enabled_ && !extension_action()->GetIsVisible(tab_id))) {
     SetVisible(false);
     return;
   }
 
   // Set the tooltip.
   tooltip_ = extension_action()->GetTitle(tab_id);
   SetTooltipText(base::UTF8ToUTF16(tooltip_));
 
   // Set the image.
   gfx::Image icon = view_controller_->GetIcon(tab_id);
   if (!icon.IsEmpty())
     SetImage(*icon.ToImageSkia());
 
   SetVisible(true);
 }
 
 void PageActionImageView::PaintChildren(gfx::Canvas* canvas,
                                         const views::CullSet& cull_set) {
   View::PaintChildren(canvas, cull_set);
-  int tab_id = view_controller_->GetCurrentTabId();
+  int tab_id = GetCurrentTabId();
   if (tab_id >= 0) {
     view_controller_->extension_action()->PaintBadge(
         canvas, GetLocalBounds(), tab_id);
   }
 }
 
 void PageActionImageView::OnIconUpdated() {
-  UpdateVisibility(GetCurrentWebContents());
+  UpdateVisibility(owner_->GetWebContents());
 }
 
 views::View* PageActionImageView::GetAsView() {
   return this;
 }
 
 bool PageActionImageView::IsShownInMenu() {
   return false;
 }
 
@@ skipping 11 matching lines @@
 }
 
 views::View* PageActionImageView::GetReferenceViewForPopup() {
   return this;
 }
 
 views::MenuButton* PageActionImageView::GetContextMenuButton() {
   return NULL;  // No menu button for page action views.
 }
 
-content::WebContents* PageActionImageView::GetCurrentWebContents() {
-  return owner_->GetWebContents();
+int PageActionImageView::GetCurrentTabId() const {
+  content::WebContents* web_contents = owner_->GetWebContents();
+  return web_contents ? SessionTabHelper::IdForTab(web_contents) : -1;
 }
 
 void PageActionImageView::HideActivePopup() {
   // The only popup that will be active is this popup.
   view_controller_->HidePopup();
 }