// Copyright 2017 Google Inc. All rights reserved.
//
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file or at
// https://developers.google.com/open-source/licenses/bsd

#ifndef MEDIA_BASE_PLAYREADY_SOURCE_H_
#define MEDIA_BASE_PLAYREADY_SOURCE_H_

#include <memory>
#include <string>
#include <vector>

#include "packager/media/base/key_source.h"

namespace shaka {
namespace media {

// SystemID defined for PlayReady Drm.
const uint8_t kPlayReadySystemId[] = {0x9a, 0x04, 0xf0, 0x79, 0x98, 0x40, 0x42,
    0x86, 0xab, 0x92, 0xe6, 0x5b, 0xe0, 0x88, 0x5f, 0x95};

/// A key source that uses playready for encryption.
class PlayReadyKeySource : public KeySource {
 public:
  /// Creates a new PlayReadyKeySource from the given packaging information.
  /// @param server_url PlayReady packaging server url.
  PlayReadyKeySource(const std::string& server_url);
  /// Creates a new PlayReadyKeySource from the given packaging information.
  /// @param server_url PlayReady packaging server url.
  /// @param client_cert_file absolute path to a client certificate.
  /// @param client_cert_private_key_file absolute path to the private file
  ///     for the client certificate.
  /// @param client_cert_private_key_password password for the private key.
  PlayReadyKeySource(const std::string& server_url,
                     const std::string& client_cert_file,
                     const std::string& client_cert_private_key_file,
                     const std::string& client_cert_private_key_password);
  ~PlayReadyKeySource() override;

  /// @name KeySource implementation overrides.
  /// @{
  Status FetchKeys(const std::vector<uint8_t>& pssh_box) override;
  Status FetchKeys(const std::vector<std::vector<uint8_t>>& key_ids) override;
  Status FetchKeys(uint32_t asset_id) override;

  Status GetKey(TrackType track_type, EncryptionKey* key) override;
  Status GetKey(const std::vector<uint8_t>& key_id,
                EncryptionKey* key) override;
  Status GetCryptoPeriodKey(uint32_t crypto_period_index,
                            TrackType track_type,
                            EncryptionKey* key) override;
  /// @}
  virtual Status FetchKeysWithProgramIdentifier(const std::string& program_identifier);

  /// Creates a new PlayReadyKeySource from the hex string information.
  /// Returns null if the strings are invalid.
  /// @param key_id_hex is the key id in hex string.
  /// @param key_hex is the key in hex string.
  /// Note: GetKey on the created key source will always return the same key
  ///       for all track types.
  static std::unique_ptr<PlayReadyKeySource> CreateFromKeyAndKeyId(
      const std::string& key_id_hex, const std::string& key_hex);
  /// Sets the Certificate Authority file for validating self-signed certificates.
  void SetCaFile(const std::string& ca_file) {
    ca_file_ = ca_file;
  }

 private:
  Status GetKeyInternal();
  Status GetCryptoPeriodKeyInternal();
  explicit PlayReadyKeySource(std::unique_ptr<EncryptionKey> key);

  std::unique_ptr<EncryptionKey> encryption_key_;
  std::string server_url_;
  std::string ca_file_;
  std::string client_cert_file_;
  std::string client_cert_private_key_file_;
  std::string client_cert_private_key_password_;

  DISALLOW_COPY_AND_ASSIGN(PlayReadyKeySource);
};

}  // namespace media
}  // namespace shaka

#endif  // MEDIA_BASE_PLAYREADY_SOURCE_H_