blob: 313827a98936f30958d1660552a4cf171d94738c [file] [log] [blame]
// Copyright 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/storage/storage_info_fetcher.h"
#include "base/task/post_task.h"
#include "chrome/browser/profiles/profile.h"
#include "content/public/browser/browser_context.h"
#include "content/public/browser/browser_task_traits.h"
#include "content/public/browser/browser_thread.h"
#include "content/public/browser/storage_partition.h"
#include "storage/browser/quota/quota_manager.h"
using content::BrowserContext;
using content::BrowserThread;
StorageInfoFetcher::StorageInfoFetcher(Profile* profile) {
quota_manager_ = content::BrowserContext::GetDefaultStoragePartition(
profile)->GetQuotaManager();
}
StorageInfoFetcher::~StorageInfoFetcher() = default;
void StorageInfoFetcher::FetchStorageInfo(const FetchCallback& fetch_callback) {
// Balanced in OnFetchCompleted.
AddRef();
fetch_callback_ = fetch_callback;
// QuotaManager must be called on IO thread, but the callback must then be
// called on the UI thread.
base::PostTaskWithTraits(
FROM_HERE, {BrowserThread::IO},
base::BindOnce(
&StorageInfoFetcher::GetUsageInfo, this,
base::BindOnce(&StorageInfoFetcher::OnGetUsageInfoInternal, this)));
}
void StorageInfoFetcher::ClearStorage(const std::string& host,
blink::mojom::StorageType type,
const ClearCallback& clear_callback) {
// Balanced in OnUsageCleared.
AddRef();
clear_callback_ = clear_callback;
type_to_delete_ = type;
base::PostTaskWithTraits(
FROM_HERE, {BrowserThread::IO},
base::BindOnce(
&storage::QuotaManager::DeleteHostData, quota_manager_, host, type,
storage::QuotaClient::kAllClientsMask,
base::BindOnce(&StorageInfoFetcher::OnUsageClearedInternal, this)));
}
void StorageInfoFetcher::GetUsageInfo(storage::GetUsageInfoCallback callback) {
DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
quota_manager_->GetUsageInfo(std::move(callback));
}
void StorageInfoFetcher::OnGetUsageInfoInternal(
storage::UsageInfoEntries entries) {
DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
entries_ = std::move(entries);
base::PostTaskWithTraits(
FROM_HERE, {BrowserThread::UI},
base::BindOnce(&StorageInfoFetcher::OnFetchCompleted, this));
}
void StorageInfoFetcher::OnFetchCompleted() {
DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
fetch_callback_.Run(entries_);
Release();
}
void StorageInfoFetcher::OnUsageClearedInternal(
blink::mojom::QuotaStatusCode code) {
DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
quota_manager_->ResetUsageTracker(type_to_delete_);
base::PostTaskWithTraits(
FROM_HERE, {BrowserThread::UI},
base::BindOnce(&StorageInfoFetcher::OnClearCompleted, this, code));
}
void StorageInfoFetcher::OnClearCompleted(blink::mojom::QuotaStatusCode code) {
DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
clear_callback_.Run(code);
Release();
}