ContentSettings now uses the JSON Schema Compiler.

chrome/browser/extensions/api/content_settings/content_settings_api.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/api/content_settings/content_settings_api.h"
 
 #include <vector>
 
 #include "base/bind.h"
 #include "base/command_line.h"
 #include "base/values.h"
 #include "chrome/browser/extensions/api/content_settings/content_settings_api_constants.h"
 #include "chrome/browser/extensions/api/content_settings/content_settings_helpers.h"
 #include "chrome/browser/extensions/api/content_settings/content_settings_store.h"
 #include "chrome/browser/content_settings/cookie_settings.h"
 #include "chrome/browser/content_settings/host_content_settings_map.h"
 #include "chrome/browser/extensions/extension_preference_api_constants.h"
 #include "chrome/browser/extensions/extension_preference_helpers.h"
 #include "chrome/browser/extensions/extension_service.h"
 #include "chrome/browser/prefs/pref_service.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/common/chrome_switches.h"
+#include "chrome/common/extensions/api/content_settings.h"
 #include "chrome/common/extensions/extension_error_utils.h"
 #include "content/public/browser/plugin_service.h"
 #include "webkit/plugins/npapi/plugin_group.h"
 
 using content::BrowserThread;
 using content::PluginService;
 
+namespace Clear = extensions::api::content_settings::ContentSetting::Clear;
+namespace Get = extensions::api::content_settings::ContentSetting::Get;
+namespace Set = extensions::api::content_settings::ContentSetting::Set;
 namespace pref_helpers = extension_preference_helpers;
 namespace pref_keys = extension_preference_api_constants;
 
 namespace {
 
 const std::vector<webkit::npapi::PluginGroup>* g_testing_plugin_groups_;
 
+bool ExtractContentType(ListValue* args, ContentSettingsType* content_type) {
+  std::string content_type_str;
+  args->GetString(0, &content_type_str);

[chebert, 2012/06/15 23:28:02]

EXTENSION_FUNCTION_VALIDATE relies on a bad_message_ member variable, so I
removed it for this line. 

I'm not sure it is necessary in this case, since the ContentSettingsType param
gets added by the renderer.

[battre, 2012/06/18 15:19:07]

you could still return false here, right?

+  args->Remove(0, NULL);

[chebert, 2012/06/15 23:28:02]

We need to remove the ContentSettingsType param since the compiled schema code
is expecting exactly 1 argument.

[battre, 2012/06/18 15:19:07]

can you please make this a comment?

+  *content_type =
+      extensions::content_settings_helpers::StringToContentSettingsType(
+          content_type_str);
+  return *content_type != CONTENT_SETTINGS_TYPE_DEFAULT;
+}
+
 }  // namespace
 
 namespace extensions {
 
 namespace helpers = content_settings_helpers;
 namespace keys = content_settings_api_constants;
 
 bool ClearContentSettingsFunction::RunImpl() {
-  std::string content_type_str;
-  EXTENSION_FUNCTION_VALIDATE(args_->GetString(0, &content_type_str));
-  ContentSettingsType content_type =
-      helpers::StringToContentSettingsType(content_type_str);
-  EXTENSION_FUNCTION_VALIDATE(content_type != CONTENT_SETTINGS_TYPE_DEFAULT);
+  ContentSettingsType content_type;
+  EXTENSION_FUNCTION_VALIDATE(ExtractContentType(args_.get(), &content_type));
+
+  scoped_ptr<Clear::Params> params(Clear::Params::Create(*args_));
+  EXTENSION_FUNCTION_VALIDATE(params.get());
 
   DictionaryValue* details = NULL;
   EXTENSION_FUNCTION_VALIDATE(args_->GetDictionary(1, &details));

[battre, 2012/06/18 15:19:07]

This is not used any more.

 
   ExtensionPrefsScope scope = kExtensionPrefsScopeRegular;
-  if (details->HasKey(pref_keys::kScopeKey)) {
-    std::string scope_str;
-    EXTENSION_FUNCTION_VALIDATE(details->GetString(pref_keys::kScopeKey,
-                                                   &scope_str));
-
-    EXTENSION_FUNCTION_VALIDATE(pref_helpers::StringToScope(scope_str, &scope));
-    EXTENSION_FUNCTION_VALIDATE(
-        scope != kExtensionPrefsScopeIncognitoPersistent);
+  if (params->details.scope ==
+          Clear::Params::Details::SCOPE_INCOGNITO_SESSION_ONLY) {
+    scope = kExtensionPrefsScopeIncognitoSessionOnly;
   }
 
   bool incognito = (scope == kExtensionPrefsScopeIncognitoPersistent ||
                     scope == kExtensionPrefsScopeIncognitoSessionOnly);
   if (incognito) {
     // We don't check incognito permissions here, as an extension should be
     // always allowed to clear its own settings.
   } else {
     // Incognito profiles can't access regular mode ever, they only exist in
     // split mode.
     if (profile()->IsOffTheRecord()) {
       error_ = keys::kIncognitoContextError;
       return false;
     }
   }
 
   ContentSettingsStore* store =
       profile_->GetExtensionService()->GetContentSettingsStore();
   store->ClearContentSettingsForExtension(extension_id(), scope);
 
   return true;
 }
 
 bool GetContentSettingFunction::RunImpl() {
-  std::string content_type_str;
-  EXTENSION_FUNCTION_VALIDATE(args_->GetString(0, &content_type_str));
-  ContentSettingsType content_type =
-      helpers::StringToContentSettingsType(content_type_str);
-  EXTENSION_FUNCTION_VALIDATE(content_type != CONTENT_SETTINGS_TYPE_DEFAULT);
+  ContentSettingsType content_type;
+  EXTENSION_FUNCTION_VALIDATE(ExtractContentType(args_.get(), &content_type));
 
-  DictionaryValue* details = NULL;
-  EXTENSION_FUNCTION_VALIDATE(args_->GetDictionary(1, &details));
+  scoped_ptr<Get::Params> params(Get::Params::Create(*args_));
+  EXTENSION_FUNCTION_VALIDATE(params.get());
 
-  std::string primary_url_spec;
-  EXTENSION_FUNCTION_VALIDATE(
-      details->GetString(keys::kPrimaryUrlKey, &primary_url_spec));
-  GURL primary_url(primary_url_spec);
+  GURL primary_url(params->details.primary_url);
   if (!primary_url.is_valid()) {
     error_ = ExtensionErrorUtils::FormatErrorMessage(keys::kInvalidUrlError,
-                                                     primary_url_spec);
+        params->details.primary_url);
     return false;
   }
 
   GURL secondary_url(primary_url);
-  std::string secondary_url_spec;
-  if (details->GetString(keys::kSecondaryUrlKey, &secondary_url_spec)) {
-    secondary_url = GURL(secondary_url_spec);
+  if (params->details.secondary_url.get()) {
+    secondary_url = GURL(*params->details.secondary_url);
     if (!secondary_url.is_valid()) {
       error_ = ExtensionErrorUtils::FormatErrorMessage(keys::kInvalidUrlError,
-                                                       secondary_url_spec);
+        *params->details.secondary_url);
       return false;
     }
   }
 
   std::string resource_identifier;
-  if (details->HasKey(keys::kResourceIdentifierKey)) {
-    DictionaryValue* resource_identifier_dict = NULL;
+  if (params->details.resource_identifier.get()) {
     EXTENSION_FUNCTION_VALIDATE(
-        details->GetDictionary(keys::kResourceIdentifierKey,
-                               &resource_identifier_dict));
-    EXTENSION_FUNCTION_VALIDATE(
-        resource_identifier_dict->GetString(keys::kIdKey,
-                                            &resource_identifier));
+        params->details.resource_identifier->ToValue()->GetString(keys::kIdKey,
+            &resource_identifier));
   }
 
   bool incognito = false;
-  if (details->HasKey(pref_keys::kIncognitoKey)) {
-    EXTENSION_FUNCTION_VALIDATE(
-        details->GetBoolean(pref_keys::kIncognitoKey, &incognito));
-  }
+  if (params->details.incognito.get())
+    incognito = *params->details.incognito;
   if (incognito && !include_incognito()) {
     error_ = pref_keys::kIncognitoErrorMessage;
     return false;
   }
 
   HostContentSettingsMap* map;
   CookieSettings* cookie_settings;
   if (incognito) {
     if (!profile()->HasOffTheRecordProfile()) {
       // TODO(bauerb): Allow reading incognito content settings
@@ skipping 23 matching lines @@
   DictionaryValue* result = new DictionaryValue();
   result->SetString(keys::kContentSettingKey,
                     helpers::ContentSettingToString(setting));
 
   result_.reset(result);
 
   return true;
 }
 
 bool SetContentSettingFunction::RunImpl() {
-  std::string content_type_str;
-  EXTENSION_FUNCTION_VALIDATE(args_->GetString(0, &content_type_str));
-  ContentSettingsType content_type =
-      helpers::StringToContentSettingsType(content_type_str);
-  EXTENSION_FUNCTION_VALIDATE(content_type != CONTENT_SETTINGS_TYPE_DEFAULT);
+  ContentSettingsType content_type;
+  EXTENSION_FUNCTION_VALIDATE(ExtractContentType(args_.get(), &content_type));
 
-  DictionaryValue* details = NULL;
-  EXTENSION_FUNCTION_VALIDATE(args_->GetDictionary(1, &details));
+  scoped_ptr<Set::Params> params(Set::Params::Create(*args_));
+  EXTENSION_FUNCTION_VALIDATE(params.get());
 
-  std::string primary_pattern_str;
-  EXTENSION_FUNCTION_VALIDATE(
-      details->GetString(keys::kPrimaryPatternKey, &primary_pattern_str));
   std::string primary_error;
   ContentSettingsPattern primary_pattern =
-      helpers::ParseExtensionPattern(primary_pattern_str, &primary_error);
+      helpers::ParseExtensionPattern(params->details.primary_pattern,
+                                     &primary_error);
   if (!primary_pattern.IsValid()) {
     error_ = primary_error;
     return false;
   }
 
   ContentSettingsPattern secondary_pattern = ContentSettingsPattern::Wildcard();
   std::string secondary_pattern_str;
-  if (details->GetString(keys::kSecondaryPatternKey, &secondary_pattern_str)) {
+  if (params->details.secondary_pattern.get()) {
     std::string secondary_error;
     secondary_pattern =
-        helpers::ParseExtensionPattern(secondary_pattern_str, &secondary_error);
+        helpers::ParseExtensionPattern(*params->details.secondary_pattern,
+                                       &secondary_error);
     if (!secondary_pattern.IsValid()) {
       error_ = secondary_error;
       return false;
     }
   }
 
   std::string resource_identifier;
-  if (details->HasKey(keys::kResourceIdentifierKey)) {
-    DictionaryValue* resource_identifier_dict = NULL;
-    EXTENSION_FUNCTION_VALIDATE(
-        details->GetDictionary(keys::kResourceIdentifierKey,
-                               &resource_identifier_dict));
+  if (params->details.resource_identifier.get()) {
+    DictionaryValue* resource_identifier_dict =
+        params->details.resource_identifier->ToValue().get();
     EXTENSION_FUNCTION_VALIDATE(
         resource_identifier_dict->GetString(keys::kIdKey,
                                             &resource_identifier));
   }
 
   std::string setting_str;
   EXTENSION_FUNCTION_VALIDATE(
-      details->GetString(keys::kContentSettingKey, &setting_str));
-  ContentSetting setting = CONTENT_SETTING_DEFAULT;
+      params->details.setting.value().GetAsString(&setting_str));
+  ContentSetting setting;
   EXTENSION_FUNCTION_VALIDATE(
       helpers::StringToContentSetting(setting_str, &setting));
   EXTENSION_FUNCTION_VALIDATE(
       HostContentSettingsMap::IsSettingAllowedForType(profile()->GetPrefs(),
                                                       setting,
                                                       content_type));
 
   ExtensionPrefsScope scope = kExtensionPrefsScopeRegular;
-  if (details->HasKey(pref_keys::kScopeKey)) {
-    std::string scope_str;
-    EXTENSION_FUNCTION_VALIDATE(details->GetString(pref_keys::kScopeKey,
-                                                   &scope_str));
-
-    EXTENSION_FUNCTION_VALIDATE(pref_helpers::StringToScope(scope_str, &scope));
-    EXTENSION_FUNCTION_VALIDATE(
-        scope != kExtensionPrefsScopeIncognitoPersistent);
+  if (params->details.scope ==
+          Set::Params::Details::SCOPE_INCOGNITO_SESSION_ONLY) {
+    scope = kExtensionPrefsScopeIncognitoSessionOnly;
   }
 
   bool incognito = (scope == kExtensionPrefsScopeIncognitoPersistent ||
                     scope == kExtensionPrefsScopeIncognitoSessionOnly);
   if (incognito) {
     // Regular profiles can't access incognito unless include_incognito is true.
     if (!profile()->IsOffTheRecord() && !include_incognito()) {
       error_ = pref_keys::kIncognitoErrorMessage;
       return false;
     }
@@ skipping 14 matching lines @@
 
   ContentSettingsStore* store =
       profile_->GetExtensionService()->GetContentSettingsStore();
   store->SetExtensionContentSetting(extension_id(), primary_pattern,
                                     secondary_pattern, content_type,
                                     resource_identifier, setting, scope);
   return true;
 }
 
 bool GetResourceIdentifiersFunction::RunImpl() {
-  std::string content_type_str;
-  EXTENSION_FUNCTION_VALIDATE(args_->GetString(0, &content_type_str));
-  ContentSettingsType content_type =
-      helpers::StringToContentSettingsType(content_type_str);
-  EXTENSION_FUNCTION_VALIDATE(content_type != CONTENT_SETTINGS_TYPE_DEFAULT);
+  ContentSettingsType content_type;
+  EXTENSION_FUNCTION_VALIDATE(ExtractContentType(args_.get(), &content_type));
 
   if (content_type == CONTENT_SETTINGS_TYPE_PLUGINS) {
     if (g_testing_plugin_groups_) {
       OnGotPluginGroups(*g_testing_plugin_groups_);
     } else {
       PluginService::GetInstance()->GetPluginGroups(
           base::Bind(&GetResourceIdentifiersFunction::OnGotPluginGroups, this));
     }
   } else {
     SendResponse(true);
@@ skipping 19 matching lines @@
           &GetResourceIdentifiersFunction::SendResponse, this, true));
 }
 
 // static
 void GetResourceIdentifiersFunction::SetPluginGroupsForTesting(
     const std::vector<webkit::npapi::PluginGroup>* plugin_groups) {
   g_testing_plugin_groups_ = plugin_groups;
 }
 
 }  // namespace extensions