// 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 CORE_INCLUDE_INITIALIZATION_DATA_H_
#define CORE_INCLUDE_INITIALIZATION_DATA_H_

#include <string>

#include "license_protocol.pb.h"
#include "wv_cdm_types.h"

namespace wvcdm {

class WvCdmEngineTest;

class InitializationData {
 public:
  InitializationData(const std::string& type = std::string(),
                     const CdmInitData& data = CdmInitData(),
                     const std::string& oec_version = std::string());

  bool is_supported() const { return is_cenc_ || is_webm_ || is_hls_; }
  bool is_cenc() const { return is_cenc_; }
  bool is_hls() const { return is_hls_; }
  bool is_webm() const { return is_webm_; }
  bool is_audio() const { return is_audio_; }

  bool IsEmpty() const { return data_.empty(); }

  const std::string& type() const { return type_; }
  const CdmInitData& data() const { return data_; }
  std::vector<uint8_t> hls_iv() const { return hls_iv_; }
  CdmHlsMethod hls_method() const { return hls_method_; }
  std::vector<video_widevine::WidevinePsshData_EntitledKey> ExtractWrappedKeys()
      const;
  bool contains_entitled_keys() const { return contains_entitled_keys_; }

 private:
  bool SelectWidevinePssh(const CdmInitData& init_data,
                          bool prefer_entitlements, CdmInitData* output);
  // Helpers used by SelectWidevinePssh().
  bool ExtractWidevinePsshs(const CdmInitData& init_data,
                            std::vector<CdmInitData>* psshs);
  bool ExtractWidevinePsshData(const uint8_t* data, size_t length,
                               CdmInitData* output);

  bool ExtractHlsAttributes(const std::string& attribute_list,
                            CdmHlsMethod* method, std::vector<uint8_t>* iv,
                            std::string* uri);
  static bool ConstructWidevineInitData(CdmHlsMethod method,
                                        const std::string& uri,
                                        CdmInitData* output);
  static bool ExtractQuotedAttribute(const std::string& attribute_list,
                                     const std::string& key,
                                     std::string* value);
  static bool ExtractHexAttribute(const std::string& attribute_list,
                                  const std::string& key,
                                  std::vector<uint8_t>* value);
  static bool ExtractAttribute(const std::string& attribute_list,
                               const std::string& key, std::string* value);

  static std::vector<std::string> ExtractKeyFormatVersions(
      const std::string& key_format_versions);

  static bool DetectEntitlementPreference(
      const std::string& oec_version_string);

// For testing only:
#if defined(UNIT_TEST)
  FRIEND_TEST(HlsAttributeExtractionTest, ExtractAttribute);
  FRIEND_TEST(HlsConstructionTest, InitData);
  FRIEND_TEST(HlsInitDataConstructionTest, InvalidUriDataFormat);
  FRIEND_TEST(HlsInitDataConstructionTest, InvalidUriBase64Encode);
  FRIEND_TEST(HlsHexAttributeExtractionTest, ExtractHexAttribute);
  FRIEND_TEST(HlsKeyFormatVersionsExtractionTest, ExtractKeyFormatVersions);
  FRIEND_TEST(HlsParseTest, Parse);
  FRIEND_TEST(HlsQuotedAttributeExtractionTest, ExtractQuotedAttribute);
  FRIEND_TEST(HlsTest, ExtractHlsAttributes);
#endif

  std::string type_;
  CdmInitData data_;
  bool is_cenc_;
  bool is_hls_;
  bool is_webm_;
  bool is_audio_;
  bool contains_entitled_keys_;

  std::vector<uint8_t> hls_iv_;
  CdmHlsMethod hls_method_;
};

}  // namespace wvcdm

#endif  // CORE_INCLUDE_INITIALIZATION_DATA_H_