/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#if !defined(EMEDecoderModule_h_)
#  define EMEDecoderModule_h_

#  include "MediaDataDecoderProxy.h"
#  include "PlatformDecoderModule.h"
#  include "SamplesWaitingForKey.h"

namespace mozilla {

class CDMProxy;
class PDMFactory;

class EMEDecoderModule : public PlatformDecoderModule {
 public:
  const char* Name() const override { return "EME"; }
  EMEDecoderModule(CDMProxy* aProxy, PDMFactory* aPDM);

 protected:
  RefPtr<CreateDecoderPromise> AsyncCreateDecoder(
      const CreateDecoderParams& aParams) override;

  // Decode thread.
  already_AddRefed<MediaDataDecoder> CreateVideoDecoder(
      const CreateDecoderParams& aParams) override {
    MOZ_CRASH("Not used");
  }

  // Decode thread.
  already_AddRefed<MediaDataDecoder> CreateAudioDecoder(
      const CreateDecoderParams& aParams) override {
    MOZ_CRASH("Not used");
  }

  media::DecodeSupportSet SupportsMimeType(
      const nsACString& aMimeType,
      DecoderDoctorDiagnostics* aDiagnostics) const override;

 private:
  virtual ~EMEDecoderModule();
  RefPtr<CDMProxy> mProxy;
  // Will be null if CDM has decoding capability.
  RefPtr<PDMFactory> mPDM;
};

DDLoggedTypeDeclNameAndBase(EMEMediaDataDecoderProxy, MediaDataDecoderProxy);

class EMEMediaDataDecoderProxy
    : public MediaDataDecoderProxy,
      public DecoderDoctorLifeLogger<EMEMediaDataDecoderProxy> {
 public:
  EMEMediaDataDecoderProxy(const CreateDecoderParams& aParams,
                           already_AddRefed<MediaDataDecoder> aProxyDecoder,
                           already_AddRefed<nsISerialEventTarget> aProxyThread,
                           CDMProxy* aProxy);
  EMEMediaDataDecoderProxy(const CreateDecoderParams& aParams,
                           already_AddRefed<MediaDataDecoder> aProxyDecoder,
                           CDMProxy* aProxy);

  RefPtr<DecodePromise> Decode(MediaRawData* aSample) override;
  RefPtr<FlushPromise> Flush() override;
  RefPtr<ShutdownPromise> Shutdown() override;

 private:
  nsCOMPtr<nsISerialEventTarget> mThread;
  RefPtr<SamplesWaitingForKey> mSamplesWaitingForKey;
  MozPromiseRequestHolder<SamplesWaitingForKey::WaitForKeyPromise> mKeyRequest;
  MozPromiseHolder<DecodePromise> mDecodePromise;
  MozPromiseRequestHolder<DecodePromise> mDecodeRequest;
  RefPtr<CDMProxy> mProxy;
};

}  // namespace mozilla

#endif  // EMEDecoderModule_h_