#pragma once
#if ENABLE(ENCRYPTED_MEDIA)
#include "ActiveDOMObject.h"
#include "CDMInstance.h"
#include "EventTarget.h"
#include "GenericEventQueue.h"
#include "GenericTaskQueue.h"
#include "JSDOMPromiseDeferred.h"
#include "MediaKeyMessageType.h"
#include "MediaKeySessionType.h"
#include "MediaKeyStatus.h"
#include <wtf/RefCounted.h>
#include <wtf/Vector.h>
#include <wtf/WeakPtr.h>
#include <wtf/text/WTFString.h>
namespace WebCore {
class BufferSource;
class CDM;
class MediaKeyStatusMap;
class MediaKeys;
class SharedBuffer;
class MediaKeySession final : public RefCounted<MediaKeySession>, public EventTargetWithInlineData, public ActiveDOMObject {
public:
static Ref<MediaKeySession> create(ScriptExecutionContext&, MediaKeySessionType, bool useDistinctiveIdentifier, Ref<CDM>&&, Ref<CDMInstance>&&);
virtual ~MediaKeySession();
using RefCounted<MediaKeySession>::ref;
using RefCounted<MediaKeySession>::deref;
const String& sessionId() const;
double expiration() const;
Ref<MediaKeyStatusMap> keyStatuses() const;
void generateRequest(const AtomicString&, const BufferSource&, Ref<DeferredPromise>&&);
void load(const String&, Ref<DeferredPromise>&&);
void update(const BufferSource&, Ref<DeferredPromise>&&);
void close(Ref<DeferredPromise>&&);
void remove(Ref<DeferredPromise>&&);
using ClosedPromise = DOMPromiseDeferred<void>;
void registerClosedPromise(ClosedPromise&&);
const Vector<std::pair<Ref<SharedBuffer>, MediaKeyStatus>>& statuses() const { return m_statuses; }
private:
MediaKeySession(ScriptExecutionContext&, MediaKeySessionType, bool useDistinctiveIdentifier, Ref<CDM>&&, Ref<CDMInstance>&&);
void enqueueMessage(MediaKeyMessageType, const SharedBuffer&);
void updateKeyStatuses(CDMInstance::KeyStatusVector&&);
void updateExpiration(double);
void sessionClosed();
EventTargetInterface eventTargetInterface() const override { return MediaKeySessionEventTargetInterfaceType; }
ScriptExecutionContext* scriptExecutionContext() const override { return ActiveDOMObject::scriptExecutionContext(); }
void refEventTarget() override { ref(); }
void derefEventTarget() override { deref(); }
bool hasPendingActivity() const override;
const char* activeDOMObjectName() const override;
bool canSuspendForDocumentSuspension() const override;
void stop() override;
String m_sessionId;
double m_expiration;
std::optional<ClosedPromise> m_closedPromise;
Ref<MediaKeyStatusMap> m_keyStatuses;
bool m_closed { false };
bool m_uninitialized { true };
bool m_callable { false };
bool m_useDistinctiveIdentifier;
MediaKeySessionType m_sessionType;
Ref<CDM> m_implementation;
Ref<CDMInstance> m_instance;
GenericEventQueue m_eventQueue;
GenericTaskQueue<Timer> m_taskQueue;
Vector<Ref<SharedBuffer>> m_recordOfKeyUsage;
double m_firstDecryptTime { 0 };
double m_latestDecryptTime { 0 };
Vector<std::pair<Ref<SharedBuffer>, MediaKeyStatus>> m_statuses;
WeakPtrFactory<MediaKeySession> m_weakPtrFactory;
};
}
#endif // ENABLE(ENCRYPTED_MEDIA)