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

#ifndef WVCDM_CORE_KEY_SESSION_H_
#define WVCDM_CORE_KEY_SESSION_H_

#include <vector>

#include "OEMCryptoCENC.h"
#include "metrics_collections.h"
#include "wv_cdm_types.h"

namespace wvcdm {

class CryptoKey;

class KeySession {
 protected:
  KeySession(metrics::CryptoMetrics* metrics) : metrics_(metrics) {}

 public:
  typedef enum { kDefault, kEntitlement } KeySessionType;
  virtual ~KeySession() {}
  virtual KeySessionType Type() = 0;
  virtual OEMCryptoResult GenerateDerivedKeys(const std::string& message) = 0;
  virtual OEMCryptoResult GenerateDerivedKeys(
      const std::string& message, const std::string& session_key) = 0;
  virtual OEMCryptoResult LoadKeys(const std::string& message,
                                   const std::string& signature,
                                   const std::string& mac_key_iv,
                                   const std::string& mac_key,
                                   const std::vector<CryptoKey>& keys,
                                   const std::string& provider_session_token,
                                   const std::string& srm_requirement) = 0;
  virtual OEMCryptoResult LoadEntitledContentKeys(
      const std::vector<CryptoKey>& keys) = 0;
  virtual OEMCryptoResult SelectKey(const std::string& key_id,
                                    CdmCipherMode cipher_mode) = 0;
  virtual OEMCryptoResult Decrypt(
      const OEMCrypto_SampleDescription* samples, size_t samples_length,
      const OEMCrypto_CENCEncryptPatternDesc& pattern) = 0;

 protected:
  metrics::CryptoMetrics* metrics_;
};

}  // namespace wvcdm

#endif  // WVCDM_CORE_KEY_SESSION_H_