PyAuto: Automation hooks to get/set theme

chrome/browser/automation/automation_provider.cc

 // Copyright (c) 2010 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/automation/automation_provider.h"
 
 #include <set>
 
 #include "app/l10n_util.h"
 #include "app/message_box_flags.h"
@@ skipping 2327 matching lines @@
     }
   }
 
   // If we get here, error.
   DCHECK(!json_return.empty());
   AutomationMsg_SendJSONRequest::WriteReplyParams(
       reply_message, json_return, false);
   Send(reply_message);
 }
 
+// Sample json input: { "command": "GetThemeInfo" }
+// Refer GetThemeInfo() in chrome/test/pyautolib/pyauto.py for sample output.
+void AutomationProvider::GetThemeInfo(Browser* browser,
+                                      DictionaryValue* args,
+                                      IPC::Message* reply_message) {
+  scoped_ptr<DictionaryValue> return_value(new DictionaryValue);
+  Extension* theme = browser->profile()->GetTheme();
+  if (theme) {
+    return_value->SetString(L"name", theme->name());
+    return_value->Set(L"images", theme->GetThemeImages()->DeepCopy());
+    return_value->Set(L"colors", theme->GetThemeColors()->DeepCopy());
+    return_value->Set(L"tints", theme->GetThemeTints()->DeepCopy());
+  }
+
+  std::string json_return;
+  base::JSONWriter::Write(return_value.get(), false, &json_return);
+  AutomationMsg_SendJSONRequest::WriteReplyParams(
+      reply_message, json_return, true);
+  Send(reply_message);
+}
+
 /* static */
 std::string AutomationProvider::JSONErrorString(const std::string& err) {
   std::string prefix = "{\"error\": \"";
   std::string no_quote_err;
   std::string suffix = "\"}";
 
   base::JsonDoubleQuote(err, false, &no_quote_err);
   return prefix + no_quote_err + suffix;
 }
 
@@ skipping 57 matching lines @@
   handler_map["SetWindowDimensions"] = &AutomationProvider::SetWindowDimensions;
 
   handler_map["GetDownloadsInfo"] = &AutomationProvider::GetDownloadsInfo;
   handler_map["WaitForAllDownloadsToComplete"] =
       &AutomationProvider::WaitForDownloadsToComplete;
 
   handler_map["GetInitialLoadTimes"] = &AutomationProvider::GetInitialLoadTimes;
 
   handler_map["SaveTabContents"] = &AutomationProvider::SaveTabContents;
 
+  // SetTheme() implemented using InstallExtension().
+  handler_map["GetThemeInfo"] = &AutomationProvider::GetThemeInfo;
+
   if (error_string.empty()) {
     if (handler_map.find(std::string(command)) != handler_map.end()) {
       (this->*handler_map[command])(browser, dict_value, reply_message);
       return;
     } else {
       error_string = "Unknown command. Options: ";
       for (std::map<std::string, JsonHandler>::const_iterator it =
            handler_map.begin(); it != handler_map.end(); ++it) {
         error_string += it->first + ", ";
       }
@@ skipping 1175 matching lines @@
 }
 
 void AutomationProvider::WaitForPopupMenuToOpen(IPC::Message* reply_message) {
   NOTIMPLEMENTED();
 }
 #endif  // !defined(TOOLKIT_VIEWS)
 
 void AutomationProvider::ResetToDefaultTheme() {
   profile_->ClearTheme();
 }