/* 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/. */

#ifndef DOM_MEDIA_PLATFORM_WMF_MFCDMPROXY_H
#define DOM_MEDIA_PLATFORM_WMF_MFCDMPROXY_H

#include <map>
#include <mfobjects.h>
#include <unknwn.h>
#include <windef.h>
#include <wrl.h>

#include "MFCDMExtra.h"
#include "nsISupportsImpl.h"

namespace mozilla {

/**
 * MFCDMProxy wraps a IMFContentDecryptionModule and provides some high level
 * helper methods in order to allow caller to interact with the wrapped CDM.
 */
class MFCDMProxy {
  NS_INLINE_DECL_REFCOUNTING(MFCDMProxy);

  explicit MFCDMProxy(IMFContentDecryptionModule* aCDM) : mCDM(aCDM) {}

 public:
  // Return a IMediaProtectionPMPServer from the existing CDM.
  HRESULT GetPMPServer(REFIID aRiid, LPVOID* aPMPServerOut);

  // Return a IMFInputTrustAuthority for given stream id, the same stream ID
  // always maps to the same IMFInputTrustAuthority. In addition,
  // `aContentInitData` is optional initialization data as in
  // https://www.w3.org/TR/encrypted-media/#initialization-data
  HRESULT GetInputTrustAuthority(uint32_t aStreamId,
                                 const uint8_t* aContentInitData,
                                 uint32_t aContentInitDataSize, REFIID aRiid,
                                 IUnknown** aInputTrustAuthorityOut);

  // Set IMFContentEnabler to the existing CDM, `aRequest` should be a inherited
  // class of `IMFContentEnabler`.
  HRESULT SetContentEnabler(IUnknown* aRequest, IMFAsyncResult* aResult);

  // Notify the CDM on DRM_E_TEE_INVALID_HWDRM_STATE (0x8004cd12), which happens
  // in cases like OS Sleep. In this case, the CDM should close all sessions
  // because they are in bad state.
  void OnHardwareContextReset();

  // TODO : set last key id in order to let CDM use the key IDs information to
  // perform some optimization.

 private:
  ~MFCDMProxy() = default;

  Microsoft::WRL::ComPtr<IMFContentDecryptionModule> mCDM;

  // The same ITA is always mapping to the same stream Id.
  std::map<uint32_t /* stream Id */,
           Microsoft::WRL::ComPtr<IMFInputTrustAuthority>>
      mInputTrustAuthorities;

  Microsoft::WRL::ComPtr<IMFTrustedInput> mTrustedInput;

  // TODO : need some events? (Eg. significant playback, error, hardware context
  // reset)
};

}  // namespace mozilla

#endif  // DOM_MEDIA_PLATFORM_WMF_MFCDMPROXY_H