// 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.

#ifndef COMPONENTS_OS_CRYPT_ASYNC_BROWSER_KEY_PROVIDER_H_
#define COMPONENTS_OS_CRYPT_ASYNC_BROWSER_KEY_PROVIDER_H_

#include <optional>
#include <string>

#include "base/functional/callback.h"
#include "base/types/expected.h"
#include "components/os_crypt/async/common/encryptor.h"

namespace os_crypt_async {

// KeyProvider is an interface used by OSCryptAsync to provide
// encryption keys for the Encyptor instance. It is not used for data
// encryption, but just for cryptographic operations and storage related to key
// management.
//
// KeyProvider implementations are passed into the constructor of the
// `OSCryptAsync` class.
class KeyProvider {
 public:
  enum class KeyError {
    // Returned when the provider is unable to provide a key and the error is
    // permanent, e.g. the system does not support the key provider, or a
    // permanent failure has occurred.
    // Caution: returning kPermanentlyUnavailable might cause callers to believe
    // that the any data previously encrypted with a key returned from this
    // provider is permanently lost, and therefore drop the data. Consider
    // returning kTemporarilyUnavailable if there is any doubt about the
    // recoverability of the key.
    kPermanentlyUnavailable,
    // Returned when the provider previously provided a key, but the key is
    // currently unavailable e.g. the system keychain is locked, user
    // authentication has failed, system integrity checks have failed, or some
    // other reason.
    kTemporarilyUnavailable,
  };

  using KeyCallback = base::OnceCallback<void(
      /*tag=*/const std::string&,
      /*key=*/base::expected<Encryptor::Key, KeyError>)>;

  virtual ~KeyProvider() = default;

  // Not copyable or movable.
  KeyProvider(const KeyProvider&) = delete;
  KeyProvider& operator=(const KeyProvider&) = delete;
  KeyProvider(KeyProvider&&) = delete;
  KeyProvider& operator=(KeyProvider&&) = delete;

  // The provider should return a non-empty tag value and a valid Key on the
  // `callback`, or an error status (see above). This Key will be used for
  // cryptographic operations. The `tag` will be used to identify that data has
  // been previously encrypted with the `key` and is typically a constant for a
  // specified provider. The provider should return the same key and tag when
  // running within the same execution context (typically: same OS user, or same
  // running application). It is typically backed by persistent storage, or OS
  // provided, or a combination of the two. The provider should generate the
  // key, if necessary.
  virtual void GetKey(KeyCallback callback) = 0;

  // The provider should return `true` if the Key it provides should be
  // considered valid for encryption of new data. Regardless of the return
  // value, the provider's key will still be considered valid for decryption of
  // data encrypted previously with the key.
  virtual bool UseForEncryption() = 0;

 protected:
  KeyProvider() = default;
};

}  // namespace os_crypt_async

#endif  // COMPONENTS_OS_CRYPT_ASYNC_BROWSER_KEY_PROVIDER_H_