Implement browserAction.setIcon(ImageData) for extensions.

chrome/browser/extensions/extension_browser_actions_api.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/extensions/extension_browser_actions_api.h"
 
 #include "chrome/browser/browser.h"
 #include "chrome/browser/browser_list.h"
 #include "chrome/common/notification_service.h"
+#include "chrome/common/render_messages.h"
 
 namespace extension_browser_actions_api_constants {
 
 const char kSetNameFunction[] = "browserAction.setName";
 const char kSetIconFunction[] = "browserAction.setIcon";
 const char kSetBadgeTextFunction[] = "browserAction.setBadgeText";
 const char kSetBadgeBackgroundColorFunction[] =
     "browserAction.setBadgeBackgroundColor";
 
 }  // namespace extension_browser_actions_api_constants
@@ skipping 17 matching lines @@
   extension->browser_action_state()->set_title(name);
 
   NotificationService::current()->Notify(
       NotificationType::EXTENSION_BROWSER_ACTION_UPDATED,
       Source<ExtensionAction>(extension->browser_action()),
       Details<ExtensionActionState>(extension->browser_action_state()));
   return true;
 }
 
 bool BrowserActionSetIconFunction::RunImpl() {
-  int icon_index = -1;
-  EXTENSION_FUNCTION_VALIDATE(args_->GetAsInteger(&icon_index));
+  // setIcon can take a variant argument: either a canvas ImageData, or an
+  // icon index.
+  EXTENSION_FUNCTION_VALIDATE(args_->IsType(Value::TYPE_BINARY) ||
+                              args_->IsType(Value::TYPE_INTEGER));
 
   Extension* extension = dispatcher()->GetExtension();
   if (!extension->browser_action()) {
     error_ = kNoBrowserActionError;
     return false;
   }
 
-  extension->browser_action_state()->set_icon_index(icon_index);
+  if (args_->IsType(Value::TYPE_BINARY)) {
+    BinaryValue* binary = static_cast<BinaryValue*>(args_);
+    IPC::Message bitmap_pickle(binary->GetBuffer(), binary->GetSize());
+    void* iter = NULL;
+    scoped_ptr<SkBitmap> bitmap(new SkBitmap);
+    EXTENSION_FUNCTION_VALIDATE(
+        IPC::ReadParam(&bitmap_pickle, &iter, bitmap.get()));
+    extension->browser_action_state()->set_icon(bitmap.release());
+  } else {
+    int icon_index = -1;
+    EXTENSION_FUNCTION_VALIDATE(args_->GetAsInteger(&icon_index));
+    extension->browser_action_state()->set_icon_index(icon_index);
+  }
 
   NotificationService::current()->Notify(
       NotificationType::EXTENSION_BROWSER_ACTION_UPDATED,
       Source<ExtensionAction>(extension->browser_action()),
       Details<ExtensionActionState>(extension->browser_action_state()));
   return true;
 }
 
 bool BrowserActionSetBadgeTextFunction::RunImpl() {
   std::string badge_text;
@@ skipping 34 matching lines @@
   }
 
   extension->browser_action_state()->set_badge_background_color(color);
 
   NotificationService::current()->Notify(
       NotificationType::EXTENSION_BROWSER_ACTION_UPDATED,
       Source<ExtensionAction>(extension->browser_action()),
       Details<ExtensionActionState>(extension->browser_action_state()));
   return true;
 }