// 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 "content/browser/buckets/bucket_manager.h"

#include "content/browser/buckets/bucket_context.h"
#include "content/browser/buckets/bucket_manager_host.h"
#include "content/browser/renderer_host/render_frame_host_impl.h"
#include "services/network/public/cpp/is_potentially_trustworthy.h"

namespace content {

BucketManager::BucketManager(StoragePartitionImpl* storage_partition)
    : storage_partition_(storage_partition) {}

BucketManager::~BucketManager() = default;

void BucketManager::BindReceiver(
    base::WeakPtr<BucketContext> context,
    mojo::PendingReceiver<blink::mojom::BucketManagerHost> receiver,
    mojo::ReportBadMessageCallback bad_message_callback) {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  DCHECK(context);

  blink::StorageKey storage_key = context->GetBucketStorageKey();
  auto it = hosts_.find(storage_key);
  if (it != hosts_.end()) {
    it->second->BindReceiver(std::move(receiver), context);
    return;
  }

  if (!network::IsOriginPotentiallyTrustworthy(storage_key.origin())) {
    std::move(bad_message_callback)
        .Run("Called Buckets from an insecure context");
    return;
  }

  auto [insert_it, insert_succeeded] = hosts_.insert(
      {storage_key, std::make_unique<BucketManagerHost>(this, storage_key)});
  DCHECK(insert_succeeded);
  insert_it->second->BindReceiver(std::move(receiver), context);
}

void BucketManager::OnHostReceiverDisconnect(BucketManagerHost* host,
                                             base::PassKey<BucketManagerHost>) {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  DCHECK(host != nullptr);
  DCHECK_GT(hosts_.count(host->storage_key()), 0u);
  DCHECK_EQ(hosts_[host->storage_key()].get(), host);

  if (host->has_connected_receivers())
    return;

  hosts_.erase(host->storage_key());
}

}  // namespace content