// Copyright 2016 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 PACKAGER_MEDIA_BASE_PROTECTION_SYSTEM_SPECIFIC_INFO_H_
#define PACKAGER_MEDIA_BASE_PROTECTION_SYSTEM_SPECIFIC_INFO_H_

#include <stdint.h>
#include <memory>
#include <vector>

#include "packager/base/logging.h"

#define NO_PROTECTION_SYSTEM_FLAG 0x00
#define COMMON_PROTECTION_SYSTEM_FLAG 0x01
#define PLAYREADY_PROTECTION_SYSTEM_FLAG 0x02
#define WIDEVINE_PROTECTION_SYSTEM_FLAG 0x04
#define FAIRPLAY_PROTECTION_SYSTEM_FLAG 0x08
#define MARLIN_PROTECTION_SYSTEM_FLAG 0x10

namespace shaka {
namespace media {

struct ProtectionSystemSpecificInfo {
  std::vector<uint8_t> system_id;
  std::vector<uint8_t> psshs;

  /// Parses multiple PSSH boxes from @a data.  These boxes should be
  /// concatenated together.  Any non-PSSH box is an error.
  /// @return true on success; false on failure.
  static bool ParseBoxes(
      const uint8_t* data,
      size_t data_size,
      std::vector<ProtectionSystemSpecificInfo>* pssh_boxes);
};

class PsshBoxBuilder {
 public:
  PsshBoxBuilder() = default;
  ~PsshBoxBuilder() = default;

  /// Parses the given PSSH box into this object.
  /// @return nullptr on failure.
  static std::unique_ptr<PsshBoxBuilder> ParseFromBox(const uint8_t* data,
                                                      size_t data_size);

  /// Creates a PSSH box for the current data.
  std::vector<uint8_t> CreateBox() const;

  uint8_t pssh_box_version() const { return version_; }
  const std::vector<uint8_t>& system_id() const { return system_id_; }
  const std::vector<std::vector<uint8_t>>& key_ids() const { return key_ids_; }
  const std::vector<uint8_t>& pssh_data() const { return pssh_data_; }

  void set_pssh_box_version(uint8_t version) {
    DCHECK_LT(version, 2);
    version_ = version;
  }
  void set_system_id(const uint8_t* system_id, size_t system_id_size) {
    DCHECK_EQ(16u, system_id_size);
    system_id_.assign(system_id, system_id + system_id_size);
  }
  void add_key_id(const std::vector<uint8_t>& key_id) {
    key_ids_.push_back(key_id);
  }
  void clear_key_ids() { key_ids_.clear(); }
  void set_pssh_data(const std::vector<uint8_t>& pssh_data) {
    pssh_data_ = pssh_data;
  }

 private:
  PsshBoxBuilder(const PsshBoxBuilder&) = delete;
  PsshBoxBuilder& operator=(const PsshBoxBuilder&) = delete;

  uint8_t version_ = 0;
  std::vector<uint8_t> system_id_;
  std::vector<std::vector<uint8_t>> key_ids_;
  std::vector<uint8_t> pssh_data_;
};

}  // namespace media
}  // namespace shaka

#endif  // PACKAGER_MEDIA_BASE_PROTECTION_SYSTEM_SPECIFIC_INFO_H_