Animate the script badges.

chrome/browser/extensions/script_badge_controller.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/extensions/script_badge_controller.h"
 
 #include "chrome/browser/extensions/extension_browser_event_router.h"
 #include "chrome/browser/extensions/extension_service.h"
 #include "chrome/browser/extensions/extension_system.h"
 #include "chrome/browser/extensions/extension_tab_helper.h"
 #include "chrome/browser/ui/tab_contents/tab_contents.h"
 #include "chrome/common/extensions/extension.h"
 #include "chrome/common/extensions/extension_action.h"
 #include "chrome/common/extensions/extension_messages.h"
 #include "chrome/common/extensions/extension_set.h"
-#include "chrome/common/extensions/extension_switch_utils.h"
 #include "chrome/common/chrome_notification_types.h"
 #include "content/public/browser/navigation_controller.h"
 #include "content/public/browser/navigation_entry.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/browser/web_contents.h"
 #include "ipc/ipc_message.h"
 #include "ipc/ipc_message_macros.h"
 
 namespace extensions {
 
 ScriptBadgeController::ScriptBadgeController(TabContents* tab_contents)
     : content::WebContentsObserver(tab_contents->web_contents()),
       script_executor_(tab_contents->web_contents()),
       tab_contents_(tab_contents) {
   registrar_.Add(this,
                  chrome::NOTIFICATION_EXTENSION_UNLOADED,
                  content::Source<Profile>(tab_contents->profile()));
 }
 
 ScriptBadgeController::~ScriptBadgeController() {}
 
-std::vector<ExtensionAction*> ScriptBadgeController::GetCurrentActions() {
+std::vector<ExtensionAction*> ScriptBadgeController::GetCurrentActions() const {
   return current_actions_;
 }
 
 LocationBarController::Action ScriptBadgeController::OnClicked(
     const std::string& extension_id, int mouse_button) {
   ExtensionService* service = GetExtensionService();
   if (!service)
     return ACTION_NONE;
 
   const Extension* extension = service->extensions()->GetByID(extension_id);
@@ skipping 121 matching lines @@
     return false;
 
   ExtensionService* service = GetExtensionService();
   if (!service)
     return  false;
 
   const Extension* extension = service->extensions()->GetByID(extension_id);
   if (!extension)
     return false;
 
-  current_actions_.push_back(extension->script_badge());
+  ExtensionAction* script_badge = extension->script_badge();
+  current_actions_.push_back(script_badge);
+  script_badge->RunIconAnimation(
+      tab_contents_->extension_tab_helper()->tab_id());
+
   return true;
 }
 
-
 bool ScriptBadgeController::EraseExtension(const Extension* extension) {
   if (extensions_executing_scripts_.erase(extension->id()) == 0)
     return false;
 
   size_t size_before = current_actions_.size();
 
   for (std::vector<ExtensionAction*>::iterator it = current_actions_.begin();
        it != current_actions_.end(); ++it) {
     // Safe to -> the extension action because we still have a handle to the
     // owner Extension.
     //
     // Also note that this means that when extensions are uninstalled their
     // script badges will disappear, even though they're still acting on the
     // page (since they would have already acted).
     if ((*it)->extension_id() == extension->id()) {
       current_actions_.erase(it);
       break;
     }
   }
 
   CHECK_EQ(size_before, current_actions_.size() + 1);
   return true;
 }
 
 }  // namespace extensions