// Copyright 2014 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef MEDIA_MOJO_SERVICES_MOJO_CDM_SERVICE_H_
#define MEDIA_MOJO_SERVICES_MOJO_CDM_SERVICE_H_

#include "base/callback.h"
#include "base/macros.h"
#include "base/memory/scoped_ptr.h"
#include "base/memory/weak_ptr.h"
#include "media/base/media_keys.h"
#include "media/mojo/interfaces/content_decryption_module.mojom.h"
#include "third_party/mojo/src/mojo/public/cpp/bindings/interface_impl.h"

namespace media {

// A mojo::ContentDecryptionModule implementation backed by a media::MediaKeys.
class MojoCdmService
    : public mojo::InterfaceImpl<mojo::ContentDecryptionModule> {
 public:
  MojoCdmService(const mojo::String& key_system);
  ~MojoCdmService() final;

  // mojo::ContentDecryptionModule implementation.
  void SetServerCertificate(
      mojo::Array<uint8_t> certificate_data,
      const mojo::Callback<void(mojo::CdmPromiseResultPtr)>& callback) final;
  void CreateSessionAndGenerateRequest(
      mojo::ContentDecryptionModule::SessionType session_type,
      const mojo::String& init_data_type,
      mojo::Array<uint8_t> init_data,
      const mojo::Callback<void(mojo::CdmPromiseResultPtr, mojo::String)>&
          callback) final;
  void LoadSession(mojo::ContentDecryptionModule::SessionType session_type,
                   const mojo::String& session_id,
                   const mojo::Callback<void(mojo::CdmPromiseResultPtr,
                                             mojo::String)>& callback) final;
  void UpdateSession(
      const mojo::String& session_id,
      mojo::Array<uint8_t> response,
      const mojo::Callback<void(mojo::CdmPromiseResultPtr)>& callback) final;
  void CloseSession(
      const mojo::String& session_id,
      const mojo::Callback<void(mojo::CdmPromiseResultPtr)>& callback) final;
  void RemoveSession(
      const mojo::String& session_id,
      const mojo::Callback<void(mojo::CdmPromiseResultPtr)>& callback) final;
  void GetCdmContext(int32_t cdm_id,
                     mojo::InterfaceRequest<mojo::Decryptor> decryptor) final;

 private:
  // Callbacks for firing session events.
  void OnSessionMessage(const std::string& session_id,
                        MediaKeys::MessageType message_type,
                        const std::vector<uint8_t>& message,
                        const GURL& legacy_destination_url);
  void OnSessionKeysChange(const std::string& session_id,
                           bool has_additional_usable_key,
                           CdmKeysInfo keys_info);
  void OnSessionExpirationUpdate(const std::string& session_id,
                                 const base::Time& new_expiry_time);
  void OnSessionClosed(const std::string& session_id);
  void OnSessionError(const std::string& session_id,
                      MediaKeys::Exception exception,
                      uint32_t system_code,
                      const std::string& error_message);

  scoped_ptr<MediaKeys> cdm_;

  base::WeakPtr<MojoCdmService> weak_this_;
  base::WeakPtrFactory<MojoCdmService> weak_factory_;

  DISALLOW_COPY_AND_ASSIGN(MojoCdmService);
};

}  // namespace media

#endif  // MEDIA_MOJO_SERVICES_MOJO_CDM_SERVICE_H_