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 "base/compiler_specific.h"
 #include "chrome/browser/extensions/extension_service.h"
 #include "chrome/browser/extensions/extension_system.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"
+#include "ui/base/animation/animation_delegate.h"
 
 namespace extensions {
 
+// Wraps an IconAnimation and implements its ui::AnimationDelegate to erase the
+// animation from a map when the animation ends or is cancelled, causing itself
+// and its owned IconAnimation to be deleted.
+class ScriptBadgeController::IconAnimationWrapper
+    : public ui::AnimationDelegate {
+ public:
+  IconAnimationWrapper(ScriptBadgeController* owner, const std::string& key)
+      : owner_(owner),
+        key_(key),
+        ALLOW_THIS_IN_INITIALIZER_LIST(animation_(owner, this)) {}
+
+  virtual ~IconAnimationWrapper() {}
+
+  IconAnimation* animation() {
+    return &animation_;
+  }
+
+ private:
+  virtual void AnimationEnded(const ui::Animation* animation) OVERRIDE {
+    Done();
+  }
+
+  virtual void AnimationCanceled(const ui::Animation* animation) OVERRIDE {
+    Done();
+  }
+
+  void Done() {
+    owner_->icon_animations_.erase(key_);
+    // this will now have been deleted.
+  }
+
+  ScriptBadgeController* owner_;
+  std::string key_;
+  IconAnimation animation_;
+};
+
 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_;
 }
 
+base::WeakPtr<LocationBarController::IconAnimation>
+    ScriptBadgeController::GetIconAnimation(
+        const ExtensionAction* action) const {
+  IconAnimationMap::const_iterator it =
+      icon_animations_.find(action->extension_id());
+  return (it != icon_animations_.end()) ? it->second->animation()->AsWeakPtr()
+                                        : base::WeakPtr<IconAnimation>();
+}
+
 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);
   CHECK(extension);
 
   switch (mouse_button) {
@@ skipping 110 matching lines @@
 
   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());
+
+  IconAnimationWrapper* icon_animation =
+      new IconAnimationWrapper(this, extension_id);
+  icon_animations_[extension_id] = make_linked_ptr(icon_animation);
+  icon_animation->animation()->Start();
+
   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);
+
+  icon_animations_.erase(extension->id());
+
   return true;
 }
 
 }  // namespace extensions