UI Changes to support clearing EME/CDM data

chrome/browser/browsing_data/media_license_counter.cc

Index: chrome/browser/browsing_data/media_license_counter.cc
diff --git a/chrome/browser/browsing_data/media_license_counter.cc b/chrome/browser/browsing_data/media_license_counter.cc
new file mode 100644
index 0000000000000000000000000000000000000000..53e13060a1b26fa1e37592b51ea9eaf4cd972867
--- /dev/null
+++ b/chrome/browser/browsing_data/media_license_counter.cc
@@ -0,0 +1,79 @@
+// Copyright (c) 2016 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/browsing_data/media_license_counter.h"
+
+#include <stdint.h>
+
+#include "base/memory/ptr_util.h"
+#include "base/memory/ref_counted.h"
+#include "chrome/common/pref_names.h"
+#include "content/public/browser/browser_thread.h"
+#include "content/public/browser/storage_partition.h"
+#include "storage/browser/fileapi/file_system_context.h"
+
+using OriginsDeterminedCB = base::Callback<void(std::set<GURL>)>;

[xhwang, 2016/06/17 06:24:29]

pass std::set by const-ref

[jrummell, 2016/06/21 00:13:44]

Done.

+
+// Determining the origins must be run on the file task thread. This calls
+// |callback| on the main thread with the results.
+void CountOriginsOnFileTaskRunner(

[xhwang, 2016/06/17 06:24:29]

put this in an anonymous namespace?

[jrummell, 2016/06/21 00:13:44]

Done.

+    scoped_refptr<storage::FileSystemContext> filesystem_context,
+    OriginsDeterminedCB callback) {

[xhwang, 2016/06/17 06:24:30]

pass callback by const-ref

[jrummell, 2016/06/21 00:13:44]

Done.

+  DCHECK(filesystem_context->default_file_task_runner()
+             ->RunsTasksOnCurrentThread());
+
+  storage::FileSystemBackend* backend =
+      filesystem_context->GetFileSystemBackend(
+          storage::kFileSystemTypePluginPrivate);
+  storage::FileSystemQuotaUtil* quota_util = backend->GetQuotaUtil();
+
+  std::set<GURL> origins;
+  quota_util->GetOriginsForTypeOnFileTaskRunner(
+      storage::kFileSystemTypePluginPrivate, &origins);
+  content::BrowserThread::PostTask(content::BrowserThread::UI, FROM_HERE,
+                                   base::Bind(callback, origins));
+}
+
+MediaLicenseCounter::MediaLicenseResult::MediaLicenseResult(
+    const MediaLicenseCounter* source,
+    std::set<GURL> origins)
+    : FinishedResult(source, origins.size()) {
+  if (!origins.empty())
+    one_origin_ = origins.begin()->GetOrigin().spec();
+}
+
+MediaLicenseCounter::MediaLicenseResult::~MediaLicenseResult() {}
+
+const std::string& MediaLicenseCounter::MediaLicenseResult::GetOneOrigin()
+    const {
+  return one_origin_;
+}
+
+MediaLicenseCounter::MediaLicenseCounter()
+    : pref_name_(prefs::kDeleteMediaLicenses), weak_ptr_factory_(this) {}
+
+MediaLicenseCounter::~MediaLicenseCounter() {}
+
+const std::string& MediaLicenseCounter::GetPrefName() const {
+  return pref_name_;
+}
+
+void MediaLicenseCounter::Count() {
+  DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
+  scoped_refptr<storage::FileSystemContext> filesystem_context =
+      make_scoped_refptr(
+          content::BrowserContext::GetDefaultStoragePartition(GetProfile())
+              ->GetFileSystemContext());
+  filesystem_context->default_file_task_runner()->PostTask(
+      FROM_HERE,
+      base::Bind(&CountOriginsOnFileTaskRunner,
+                 base::Passed(std::move(filesystem_context)),
+                 base::Bind(&MediaLicenseCounter::OnContentLicensesObtained,
+                            weak_ptr_factory_.GetWeakPtr())));
+}
+
+void MediaLicenseCounter::OnContentLicensesObtained(std::set<GURL> origins) {
+  DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
+  ReportResult(base::WrapUnique(new MediaLicenseResult(this, origins)));
+}