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

#include "chromeos/ash/experiences/arc/keymint/cert_store_bridge_keymint.h"

#include <cstdint>
#include <utility>

#include "base/functional/bind.h"
#include "base/functional/callback.h"
#include "base/logging.h"
#include "mojo/public/cpp/bindings/pending_remote.h"

namespace arc::keymint {

CertStoreBridgeKeyMint::CertStoreBridgeKeyMint(content::BrowserContext* context)
    : weak_ptr_factory_(this) {}

CertStoreBridgeKeyMint::~CertStoreBridgeKeyMint() = default;

void CertStoreBridgeKeyMint::UpdatePlaceholderKeysInKeyMint(
    std::vector<mojom::ChromeOsKeyPtr> keys,
    mojom::CertStoreInstance::UpdatePlaceholderKeysCallback callback) {
  if (cert_store_proxy_.is_bound()) {
    cert_store_proxy_->UpdatePlaceholderKeys(std::move(keys),
                                             std::move(callback));
  } else {
    LOG(ERROR) << "Tried to update placeholders but cert store is not bound";
    std::move(callback).Run(/*success=*/false);
  }
}

void CertStoreBridgeKeyMint::SetSerialNumber(const std::string& serial_number) {
  if (!cert_store_proxy_.is_bound()) {
    LOG(ERROR) << "Failed to update Serial Number. Cert store not bound";
    return;
  }
  cert_store_proxy_->SetSerialNumber(serial_number);
}

bool CertStoreBridgeKeyMint::IsProxyBound() const {
  return cert_store_proxy_.is_bound();
}

void CertStoreBridgeKeyMint::BindToInvitation(
    mojo::OutgoingInvitation* invitation) {
  VLOG(2) << "CertStoreBridgeKeyMint::BootstrapMojoConnection";

  constexpr uint64_t kCertStorePipeAttachment = 1;
  mojo::ScopedMessagePipeHandle pipe =
      invitation->AttachMessagePipe(kCertStorePipeAttachment);

  if (!pipe.is_valid()) {
    LOG(ERROR) << "CertStoreBridgeKeyMint could not bind to invitation";
    return;
  }

  cert_store_proxy_.Bind(mojo::PendingRemote<keymint::mojom::CertStoreInstance>(
      std::move(pipe), 0u));
  VLOG(2) << "Bound remote CertStoreInstance interface to pipe.";
  cert_store_proxy_.set_disconnect_handler(
      base::BindOnce(&mojo::Remote<keymint::mojom::CertStoreInstance>::reset,
                     base::Unretained(&cert_store_proxy_)));
}

}  // namespace arc::keymint