src/chrome/browser/automation/automation_provider_json.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/automation/automation_provider_json.h"
   6
   7 #include "base/json/json_writer.h"
   8 #include "base/json/string_escape.h"
   9 #include "base/values.h"
  10 #include "chrome/browser/autocomplete/autocomplete_match.h"
  11 #include "chrome/browser/automation/automation_provider.h"
  12 #include "chrome/browser/automation/automation_util.h"
  13 #include "chrome/browser/extensions/extension_service.h"
  14 #include "chrome/browser/extensions/extension_system.h"
  15 #include "chrome/browser/profiles/profile.h"
  16 #include "chrome/common/automation_messages.h"
  17 #include "chrome/common/extensions/extension.h"
  18 #include "content/public/browser/web_contents.h"
  19
  20 using content::WebContents;
  21
  22 AutomationJSONReply::AutomationJSONReply(AutomationProvider* provider,
  23                                          IPC::Message* reply_message)
  24   : provider_(provider),
  25     message_(reply_message) {
  26 }
  27
  28 AutomationJSONReply::~AutomationJSONReply() {
  29   DCHECK(!message_) << "JSON automation request not replied!";
  30 }
  31
  32 void AutomationJSONReply::SendSuccess(const Value* value) {
  33   DCHECK(message_) << "Resending reply for JSON automation request";
  34   std::string json_string = "{}";
  35   if (value)
  36     base::JSONWriter::Write(value, &json_string);
  37   AutomationMsg_SendJSONRequest::WriteReplyParams(
  38       message_, json_string, true);
  39   provider_->Send(message_);
  40   message_ = NULL;
  41 }
  42
  43 void AutomationJSONReply::SendError(const std::string& error_message) {
  44   DCHECK(message_) << "Resending reply for JSON automation request";
  45
  46   base::DictionaryValue dict;
  47   dict.SetString("error", error_message);
  48   std::string json;
  49   base::JSONWriter::Write(&dict, &json);
  50
  51   AutomationMsg_SendJSONRequest::WriteReplyParams(message_, json, false);
  52   provider_->Send(message_);
  53   message_ = NULL;
  54 }
  55
  56 bool GetBrowserFromJSONArgs(
  57     DictionaryValue* args,
  58     Browser** browser,
  59     std::string* error) {
  60   int browser_index;
  61   if (!args->GetInteger("windex", &browser_index)) {
  62     *error = "'windex' missing or invalid";
  63     return false;
  64   }
  65   *browser = automation_util::GetBrowserAt(browser_index);
  66   if (!*browser) {
  67     *error = "Cannot locate browser from given index";
  68     return false;
  69   }
  70   return true;
  71 }
  72
  73 bool GetTabFromJSONArgs(
  74     DictionaryValue* args,
  75     WebContents** tab,
  76     std::string* error) {
  77   int browser_index, tab_index;
  78   if (!args->GetInteger("windex", &browser_index)) {
  79     *error = "'windex' missing or invalid";
  80     return false;
  81   }
  82   if (!args->GetInteger("tab_index", &tab_index)) {
  83     *error = "'tab_index' missing or invalid";
  84     return false;
  85   }
  86   *tab = automation_util::GetWebContentsAt(browser_index, tab_index);
  87   if (!*tab) {
  88     *error = "Cannot locate tab from given indices";
  89     return false;
  90   }
  91   return true;
  92 }
  93
  94 bool GetBrowserAndTabFromJSONArgs(
  95     DictionaryValue* args,
  96     Browser** browser,
  97     WebContents** tab,
  98     std::string* error) {
  99   return GetBrowserFromJSONArgs(args, browser, error) &&
 100          GetTabFromJSONArgs(args, tab, error);
 101 }
 102
 103 bool GetRenderViewFromJSONArgs(
 104     DictionaryValue* args,
 105     Profile* profile,
 106     content::RenderViewHost** rvh,
 107     std::string* error) {
 108   WebContents* tab = NULL;
 109   if (!GetTabFromJSONArgs(args, &tab, error))
 110     return false;
 111   *rvh = tab->GetRenderViewHost();
 112   return true;
 113 }
 114
 115 namespace {
 116
 117 bool GetExtensionFromJSONArgsHelper(
 118     base::DictionaryValue* args,
 119     const std::string& key,
 120     Profile* profile,
 121     bool include_disabled,
 122     const extensions::Extension** extension,
 123     std::string* error) {
 124   std::string id;
 125   if (!args->GetString(key, &id)) {
 126     *error = base::StringPrintf("Missing or invalid key: %s", key.c_str());
 127     return false;
 128   }
 129   ExtensionService* service = extensions::ExtensionSystem::Get(profile)->
 130       extension_service();
 131   if (!service) {
 132     *error = "No extensions service.";
 133     return false;
 134   }
 135   if (!service->GetInstalledExtension(id)) {
 136     // The extension ID does not correspond to any extension, whether crashed
 137     // or not.
 138     *error = base::StringPrintf("Extension %s is not installed.",
 139                                 id.c_str());
 140     return false;
 141   }
 142   const extensions::Extension* installed_extension =
 143       service->GetExtensionById(id, include_disabled);
 144   if (!installed_extension) {
 145     *error = "Extension is disabled or has crashed.";
 146     return false;
 147   }
 148   *extension = installed_extension;
 149   return true;
 150 }
 151
 152 }  // namespace
 153
 154 bool GetExtensionFromJSONArgs(
 155     base::DictionaryValue* args,
 156     const std::string& key,
 157     Profile* profile,
 158     const extensions::Extension** extension,
 159     std::string* error) {
 160   return GetExtensionFromJSONArgsHelper(
 161       args, key, profile, true /* include_disabled */, extension, error);
 162 }
 163
 164 bool GetEnabledExtensionFromJSONArgs(
 165     base::DictionaryValue* args,
 166     const std::string& key,
 167     Profile* profile,
 168     const extensions::Extension** extension,
 169     std::string* error) {
 170   return GetExtensionFromJSONArgsHelper(
 171       args, key, profile, false /* include_disabled */, extension, error);
 172 }