// Copyright 2021 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "components/services/storage/public/cpp/quota_client_callback_wrapper.h"

#include <utility>

#include "base/check_op.h"
#include "base/sequence_checker.h"
#include "components/services/storage/public/cpp/buckets/bucket_locator.h"
#include "components/services/storage/public/mojom/quota_client.mojom.h"
#include "mojo/public/cpp/bindings/callback_helpers.h"
#include "third_party/blink/public/common/storage_key/storage_key.h"

namespace storage {

QuotaClientCallbackWrapper::QuotaClientCallbackWrapper(
    mojom::QuotaClient* wrapped_client)
    : wrapped_client_(wrapped_client) {
  DCHECK(wrapped_client);
  DETACH_FROM_SEQUENCE(sequence_checker_);
}

QuotaClientCallbackWrapper::~QuotaClientCallbackWrapper() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
}

void QuotaClientCallbackWrapper::GetBucketUsage(
    const BucketLocator& bucket,
    GetBucketUsageCallback callback) {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);

  wrapped_client_->GetBucketUsage(
      bucket,
      mojo::WrapCallbackWithDefaultInvokeIfNotRun(std::move(callback), 0));
}

void QuotaClientCallbackWrapper::GetDefaultStorageKeys(
    GetDefaultStorageKeysCallback callback) {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);

  wrapped_client_->GetDefaultStorageKeys(
      mojo::WrapCallbackWithDefaultInvokeIfNotRun(
          std::move(callback), std::vector<blink::StorageKey>()));
}

void QuotaClientCallbackWrapper::DeleteBucketData(
    const BucketLocator& bucket,
    DeleteBucketDataCallback callback) {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);

  wrapped_client_->DeleteBucketData(
      bucket,
      mojo::WrapCallbackWithDefaultInvokeIfNotRun(
          std::move(callback), blink::mojom::QuotaStatusCode::kErrorAbort));
}

void QuotaClientCallbackWrapper::PerformStorageCleanup(
    PerformStorageCleanupCallback callback) {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);

  wrapped_client_->PerformStorageCleanup(
      mojo::WrapCallbackWithDefaultInvokeIfNotRun(std::move(callback)));
}

}  // namespace storage