// Copyright 2018 Google LLC. All Rights Reserved. This file and proprietary
// source code may only be used and distributed under the Widevine License
// Agreement.
#ifndef WVCDM_CORE_CRYPTO_KEY_H_
#define WVCDM_CORE_CRYPTO_KEY_H_

#include <string>

#include "wv_cdm_types.h"
#include "wv_class_utils.h"

namespace wvcdm {

class CryptoKey {
 public:
  CryptoKey() = default;
  WVCDM_DEFAULT_COPY_AND_MOVE(CryptoKey);
  ~CryptoKey() {}

  const std::string& key_id() const { return key_id_; }
  const std::string& key_data() const { return key_data_; }
  const std::string& key_data_iv() const { return key_data_iv_; }
  const std::string& key_control() const { return key_control_; }
  const std::string& key_control_iv() const { return key_control_iv_; }
  const std::string& entitlement_key_id() const { return entitlement_key_id_; }
  const std::string& track_label() const { return track_label_; }
  CdmCipherMode cipher_mode() const { return cipher_mode_; }
  void set_key_id(const std::string& key_id) { key_id_ = key_id; }
  void set_key_data(const std::string& key_data) { key_data_ = key_data; }
  void set_key_data_iv(const std::string& iv) { key_data_iv_ = iv; }
  void set_key_control(const std::string& ctl) { key_control_ = ctl; }
  void set_key_control_iv(const std::string& ctl_iv) {
    key_control_iv_ = ctl_iv;
  }
  void set_cipher_mode(CdmCipherMode cipher_mode) {
    cipher_mode_ = cipher_mode;
  }
  void set_track_label(const std::string& track_label) {
    track_label_ = track_label;
  }
  void set_entitlement_key_id(const std::string& entitlement_key_id) {
    entitlement_key_id_ = entitlement_key_id;
  }

  bool HasKeyControl() const { return key_control_.size() >= 16; }

 private:
  std::string key_id_;
  std::string key_data_iv_;
  std::string key_data_;
  std::string key_control_;
  std::string key_control_iv_;
  std::string track_label_;
  std::string entitlement_key_id_;
  CdmCipherMode cipher_mode_;
};  // class CryptoKey
}  // namespace wvcdm
#endif  // WVCDM_CORE_CRYPTO_KEY_H_