// Copyright 2022 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/web_package/signed_web_bundles/signed_web_bundle_integrity_block.h"

#include <string>
#include <utility>

#include "base/types/expected_macros.h"
#include "components/web_package/mojom/web_bundle_parser.mojom.h"
#include "components/web_package/signed_web_bundles/signed_web_bundle_id.h"
#include "components/web_package/signed_web_bundles/signed_web_bundle_signature_stack.h"
#include "components/web_package/signed_web_bundles/signed_web_bundle_signature_stack_entry.h"

namespace web_package {

SignedWebBundleIntegrityBlock::~SignedWebBundleIntegrityBlock() = default;

base::expected<SignedWebBundleIntegrityBlock, std::string>
SignedWebBundleIntegrityBlock::Create(
    mojom::BundleIntegrityBlockPtr integrity_block) {
  if (integrity_block->size == 0) {
    return base::unexpected("Cannot create integrity block with a size of 0.");
  }

  ASSIGN_OR_RETURN(auto signature_stack,
                   SignedWebBundleSignatureStack::Create(
                       std::move(integrity_block->signature_stack)),
                   [](std::string error) {
                     return "Cannot create an integrity block: " +
                            std::move(error);
                   });

  ASSIGN_OR_RETURN(
      auto web_bundle_id,
      SignedWebBundleId::Create(integrity_block->attributes.web_bundle_id()));

  return SignedWebBundleIntegrityBlock(
      integrity_block->size, std::move(signature_stack),
      std::move(web_bundle_id), std::move(integrity_block->attributes.cbor()));
}

SignedWebBundleIntegrityBlock::SignedWebBundleIntegrityBlock(
    const uint64_t size_in_bytes,
    SignedWebBundleSignatureStack&& signature_stack,
    SignedWebBundleId web_bundle_id,
    std::vector<uint8_t> attributes_cbor)
    : size_in_bytes_(size_in_bytes),
      signature_stack_(signature_stack),
      web_bundle_id_(std::move(web_bundle_id)),
      attributes_cbor_(std::move(attributes_cbor)) {
  CHECK_GT(size_in_bytes_, 0ul);
}

SignedWebBundleIntegrityBlock::SignedWebBundleIntegrityBlock(
    const SignedWebBundleIntegrityBlock&) = default;
SignedWebBundleIntegrityBlock& SignedWebBundleIntegrityBlock::operator=(
    const SignedWebBundleIntegrityBlock&) = default;

}  // namespace web_package