Store USB device permissions in website settings.

chrome/browser/permissions/chooser_permission_context_base.cc

+// Copyright 2015 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/permissions/chooser_permission_context_base.h"
+
+#include <string>
+
+#include "base/values.h"
+#include "chrome/browser/content_settings/host_content_settings_map_factory.h"
+#include "chrome/browser/permissions/permission_manager.h"
+#include "chrome/browser/permissions/permission_manager_factory.h"
+#include "components/content_settings/core/browser/host_content_settings_map.h"
+#include "content/public/common/permission_status.mojom.h"
+
+namespace {
+const char* const kObjectListKey = "chosen-objects";

[Bernhard Bauer, 2015/10/12 11:03:48]

Use char[] instead of char* (which I think also lets you get rid of one const).
Also, constants automatically get internal linkage, so the anonymous namespace
isn't necessary.

[Reilly Grant (use Gerrit), 2015/10/13 01:32:24]

Done.

+}
+
+ChooserPermissionContextBase::ChooserPermissionContextBase(
+    Profile* profile,
+    const content::PermissionType permission_type,
+    const ContentSettingsType chooser_settings_type)
+    : profile_(profile),
+      permission_type_(permission_type),
+      chooser_settings_type_(chooser_settings_type) {}
+
+ChooserPermissionContextBase::~ChooserPermissionContextBase() {}
+
+scoped_ptr<base::ListValue>
+ChooserPermissionContextBase::GetPreviouslyChosenObjects(
+    const GURL& requesting_origin,
+    const GURL& embedding_origin) {
+  PermissionManager* permission_manager =
+      PermissionManagerFactory::GetForProfile(profile_);
+  content::PermissionStatus permission_status =
+      permission_manager->GetPermissionStatus(
+          permission_type_, requesting_origin, embedding_origin);
+  if (permission_status != content::PERMISSION_STATUS_GRANTED)
+    return make_scoped_ptr(new base::ListValue());
+
+  scoped_ptr<base::DictionaryValue> setting =
+      GetWebsiteSetting(requesting_origin, embedding_origin);
+  scoped_ptr<base::Value> object_list;
+  if (!setting->Remove(kObjectListKey, &object_list))
+    return make_scoped_ptr(new base::ListValue());
+
+  if (object_list->GetType() != base::Value::TYPE_LIST)
+    return make_scoped_ptr(new base::ListValue());
+  return base::ListValue::From(object_list.Pass());
+}
+
+void ChooserPermissionContextBase::GrantObjectPermission(
+    const GURL& requesting_origin,
+    const GURL& embedding_origin,
+    scoped_ptr<base::Value> object) {
+  scoped_ptr<base::DictionaryValue> setting =
+      GetWebsiteSetting(requesting_origin, embedding_origin);
+  base::ListValue* object_list;
+  if (!setting->GetList(kObjectListKey, &object_list)) {
+    object_list = new base::ListValue();
+    setting->Set(kObjectListKey, object_list);
+  }
+  object_list->Append(object.Pass());
+  SetWebsiteSetting(requesting_origin, embedding_origin, setting.Pass());
+}
+
+void ChooserPermissionContextBase::RevokeObjectPermission(
+    const GURL& requesting_origin,
+    const GURL& embedding_origin,
+    const base::Value& object) {
+  scoped_ptr<base::DictionaryValue> setting =
+      GetWebsiteSetting(requesting_origin, embedding_origin);
+  base::ListValue* object_list;
+  if (!setting->GetList(kObjectListKey, &object_list))
+    return;
+  object_list->Remove(object, nullptr);
+  SetWebsiteSetting(requesting_origin, embedding_origin, setting.Pass());
+}
+
+scoped_ptr<base::DictionaryValue>
+ChooserPermissionContextBase::GetWebsiteSetting(const GURL& requesting_origin,
+                                                const GURL& embedding_origin) {
+  HostContentSettingsMap* settings =
+      HostContentSettingsMapFactory::GetForProfile(profile_);
+  DCHECK(settings);
+
+  scoped_ptr<base::DictionaryValue> value =
+      base::DictionaryValue::From(settings->GetWebsiteSetting(
+          requesting_origin, embedding_origin, chooser_settings_type_,
+          std::string(), nullptr));
+  if (!value)
+    return make_scoped_ptr(new base::DictionaryValue());
+
+  return value.Pass();
+}
+
+void ChooserPermissionContextBase::SetWebsiteSetting(
+    const GURL& requesting_origin,
+    const GURL& embedding_origin,
+    scoped_ptr<base::Value> value) {
+  HostContentSettingsMap* settings =
+      HostContentSettingsMapFactory::GetForProfile(profile_);
+  DCHECK(settings);
+
+  ContentSettingsPattern primary_pattern(
+      ContentSettingsPattern::FromURLNoWildcard(requesting_origin));
+  ContentSettingsPattern secondary_pattern(
+      ContentSettingsPattern::FromURLNoWildcard(embedding_origin));
+  if (!primary_pattern.IsValid() || !secondary_pattern.IsValid())
+    return;
+
+  settings->SetWebsiteSetting(primary_pattern, secondary_pattern,
+                              chooser_settings_type_, std::string(),
+                              value.release());
+}