Move page actions over to ExtensionAction2

chrome/browser/tab_contents/tab_contents.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/tab_contents/tab_contents.h"
 
 #include "app/l10n_util.h"
 #include "app/resource_bundle.h"
 #include "base/file_version_info.h"
 #include "base/process_util.h"
@@ skipping 37 matching lines @@
 #include "chrome/browser/tab_contents/interstitial_page.h"
 #include "chrome/browser/tab_contents/navigation_entry.h"
 #include "chrome/browser/tab_contents/provisional_load_details.h"
 #include "chrome/browser/tab_contents/tab_contents_delegate.h"
 #include "chrome/browser/tab_contents/tab_contents_view.h"
 #include "chrome/browser/tab_contents/thumbnail_generator.h"
 #include "chrome/browser/thumbnail_store.h"
 #include "chrome/browser/search_engines/template_url_fetcher.h"
 #include "chrome/browser/search_engines/template_url_model.h"
 #include "chrome/common/chrome_switches.h"
-#include "chrome/common/extensions/extension_action.h"
+#include "chrome/common/extensions/extension_action2.h"
 #include "chrome/common/notification_service.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/common/pref_service.h"
 #include "chrome/common/render_messages.h"
 #include "chrome/common/renderer_preferences.h"
 #include "chrome/common/url_constants.h"
 #include "grit/generated_resources.h"
 #include "grit/locale_settings.h"
 #include "net/base/mime_util.h"
 #include "net/base/net_errors.h"
@@ skipping 529 matching lines @@
 }
 
 void TabContents::SetIsCrashed(bool state) {
   if (state == is_crashed_)
     return;
 
   is_crashed_ = state;
   NotifyNavigationStateChanged(INVALIDATE_TAB);
 }
 
-void TabContents::SetPageActionEnabled(const ExtensionAction* page_action,
-                                       bool enable,
-                                       const std::string& title,
-                                       int icon_id) {
-  DCHECK(page_action);
-  ExtensionActionState* state = GetOrCreatePageActionState(page_action);
-  state->set_hidden(!enable);
-  state->set_title(title);
-  state->set_icon_index(icon_id);
-  state->set_icon(NULL);
-}
-
-const ExtensionActionState* TabContents::GetPageActionState(
-    const ExtensionAction* page_action) {
-  if (page_actions_.end() == page_actions_.find(page_action))
-    return NULL;
-
-  return page_actions_[page_action].get();
-}
-
-ExtensionActionState* TabContents::GetOrCreatePageActionState(
-    const ExtensionAction* page_action) {
-  if (page_actions_.end() == page_actions_.find(page_action)) {
-    page_actions_[page_action].reset(
-        new ExtensionActionState(page_action->title(), 0));
-  }
-
-  return page_actions_[page_action].get();
-}
-
 void TabContents::PageActionStateChanged() {
   NotifyNavigationStateChanged(TabContents::INVALIDATE_PAGE_ACTIONS);
 }
 
 void TabContents::NotifyNavigationStateChanged(unsigned changed_flags) {
   if (delegate_)
     delegate_->NavigationStateChanged(this, changed_flags);
 }
 
 void TabContents::DidBecomeSelected() {
@@ skipping 759 matching lines @@
   // The keyword generator uses the navigation entries, so must be called after
   // the commit.
   GenerateKeywordIfNecessary(params);
 
   // Allow the new page to set the title again.
   received_page_title_ = false;
 
   // Get the favicon, either from history or request it from the net.
   fav_icon_helper_.FetchFavIcon(details.entry->url());
 
-  // Disable all page actions, unless this is an in-page navigation.
+  // Clear all page and browser action state for this tab, unless this is an
+  // in-page navigation.
   url_canon::Replacements<char> replacements;
   replacements.ClearRef();
   if (params.url.ReplaceComponents(replacements) !=
       params.referrer.ReplaceComponents(replacements)) {
-    if (!page_actions_.empty())
-      page_actions_.clear();
-
     ExtensionsService* service = profile()->GetExtensionsService();
     if (service) {
       for (size_t i = 0; i < service->extensions()->size(); ++i) {
-        ExtensionAction2* action =
+        ExtensionAction2* browser_action =
             service->extensions()->at(i)->browser_action();
-        if (!action)
-          continue;
+        if (browser_action) {
+          browser_action->ClearAllValuesForTab(controller().session_id().id());
+          NotificationService::current()->Notify(
+              NotificationType::EXTENSION_BROWSER_ACTION_UPDATED,
+              Source<ExtensionAction2>(browser_action),
+              NotificationService::NoDetails());
+        }
 
-        action->ClearAllValuesForTab(controller().session_id().id());
-        NotificationService::current()->Notify(
-            NotificationType::EXTENSION_BROWSER_ACTION_UPDATED,
-            Source<ExtensionAction2>(action),
-            NotificationService::NoDetails());
+        ExtensionAction2* page_action =
+            service->extensions()->at(i)->page_action();
+        if (page_action) {
+          page_action->ClearAllValuesForTab(controller().session_id().id());
+          PageActionStateChanged();
+        }
       }
     }
   }
 
   // Close constrained popups if necessary.
   MaybeCloseChildWindows(details.previous_url, details.entry->url());
 
   // Update the starred state.
   UpdateStarredStateForCurrentURL();
 }
@@ skipping 1202 matching lines @@
 #endif
 
     default:
       NOTREACHED();
   }
 }
 
 void TabContents::set_encoding(const std::string& encoding) {
   encoding_ = CharacterEncoding::GetCanonicalEncodingNameByAliasName(encoding);
 }