#include "config.h"
#include "MediaStream.h"
#if ENABLE(MEDIA_STREAM)
#include "Document.h"
#include "Event.h"
#include "EventNames.h"
#include "Frame.h"
#include "FrameLoader.h"
#include "Logging.h"
#include "MediaStreamRegistry.h"
#include "MediaStreamTrackEvent.h"
#include "NetworkingContext.h"
#include "Page.h"
#include "RealtimeMediaSource.h"
#include <wtf/URL.h>
namespace WebCore {
Ref<MediaStream> MediaStream::create(ScriptExecutionContext& context)
{
return MediaStream::create(context, MediaStreamPrivate::create({ }));
}
Ref<MediaStream> MediaStream::create(ScriptExecutionContext& context, MediaStream& stream)
{
return adoptRef(*new MediaStream(context, stream.getTracks()));
}
Ref<MediaStream> MediaStream::create(ScriptExecutionContext& context, const MediaStreamTrackVector& tracks)
{
return adoptRef(*new MediaStream(context, tracks));
}
Ref<MediaStream> MediaStream::create(ScriptExecutionContext& context, Ref<MediaStreamPrivate>&& streamPrivate)
{
return adoptRef(*new MediaStream(context, WTFMove(streamPrivate)));
}
static inline MediaStreamTrackPrivateVector createTrackPrivateVector(const MediaStreamTrackVector& tracks)
{
MediaStreamTrackPrivateVector trackPrivates;
trackPrivates.reserveCapacity(tracks.size());
for (auto& track : tracks)
trackPrivates.append(&track->privateTrack());
return trackPrivates;
}
MediaStream::MediaStream(ScriptExecutionContext& context, const MediaStreamTrackVector& tracks)
: ActiveDOMObject(&context)
, m_private(MediaStreamPrivate::create(createTrackPrivateVector(tracks)))
, m_mediaSession(PlatformMediaSession::create(*this))
{
for (auto& track : tracks) {
track->addObserver(*this);
m_trackSet.add(track->id(), track);
}
setIsActive(m_private->active());
m_private->addObserver(*this);
MediaStreamRegistry::shared().registerStream(*this);
suspendIfNeeded();
}
MediaStream::MediaStream(ScriptExecutionContext& context, Ref<MediaStreamPrivate>&& streamPrivate)
: ActiveDOMObject(&context)
, m_private(WTFMove(streamPrivate))
, m_mediaSession(PlatformMediaSession::create(*this))
{
setIsActive(m_private->active());
if (document()->page() && document()->page()->isMediaCaptureMuted())
m_private->setCaptureTracksMuted(true);
m_private->addObserver(*this);
MediaStreamRegistry::shared().registerStream(*this);
for (auto& trackPrivate : m_private->tracks()) {
auto track = MediaStreamTrack::create(context, *trackPrivate);
track->addObserver(*this);
m_trackSet.add(track->id(), WTFMove(track));
}
suspendIfNeeded();
}
MediaStream::~MediaStream()
{
m_isActive = false;
MediaStreamRegistry::shared().unregisterStream(*this);
m_private->removeObserver(*this);
for (auto& track : m_trackSet.values())
track->removeObserver(*this);
if (Document* document = this->document()) {
if (m_isWaitingUntilMediaCanStart)
document->removeMediaCanStartListener(this);
}
}
RefPtr<MediaStream> MediaStream::clone()
{
MediaStreamTrackVector clonedTracks;
clonedTracks.reserveInitialCapacity(m_trackSet.size());
for (auto& track : m_trackSet.values())
clonedTracks.uncheckedAppend(track->clone());
return MediaStream::create(*scriptExecutionContext(), clonedTracks);
}
void MediaStream::addTrack(MediaStreamTrack& track)
{
if (!internalAddTrack(track, StreamModifier::DomAPI))
return;
for (auto& observer : m_observers)
observer->didAddOrRemoveTrack();
}
void MediaStream::removeTrack(MediaStreamTrack& track)
{
if (!internalRemoveTrack(track.id(), StreamModifier::DomAPI))
return;
for (auto& observer : m_observers)
observer->didAddOrRemoveTrack();
}
MediaStreamTrack* MediaStream::getTrackById(String id)
{
auto it = m_trackSet.find(id);
if (it != m_trackSet.end())
return it->value.get();
return nullptr;
}
MediaStreamTrackVector MediaStream::getAudioTracks() const
{
return trackVectorForType(RealtimeMediaSource::Type::Audio);
}
MediaStreamTrackVector MediaStream::getVideoTracks() const
{
return trackVectorForType(RealtimeMediaSource::Type::Video);
}
MediaStreamTrackVector MediaStream::getTracks() const
{
return copyToVector(m_trackSet.values());
}
void MediaStream::trackDidEnd()
{
m_private->updateActiveState(MediaStreamPrivate::NotifyClientOption::Notify);
}
void MediaStream::activeStatusChanged()
{
updateActiveState();
}
void MediaStream::didAddTrack(MediaStreamTrackPrivate& trackPrivate)
{
ScriptExecutionContext* context = scriptExecutionContext();
if (!context)
return;
if (!getTrackById(trackPrivate.id()))
internalAddTrack(MediaStreamTrack::create(*context, trackPrivate), StreamModifier::Platform);
}
void MediaStream::didRemoveTrack(MediaStreamTrackPrivate& trackPrivate)
{
internalRemoveTrack(trackPrivate.id(), StreamModifier::Platform);
}
void MediaStream::addTrackFromPlatform(Ref<MediaStreamTrack>&& track)
{
auto* privateTrack = &track->privateTrack();
internalAddTrack(WTFMove(track), StreamModifier::Platform);
m_private->addTrack(privateTrack, MediaStreamPrivate::NotifyClientOption::Notify);
}
bool MediaStream::internalAddTrack(Ref<MediaStreamTrack>&& trackToAdd, StreamModifier streamModifier)
{
auto result = m_trackSet.add(trackToAdd->id(), WTFMove(trackToAdd));
if (!result.isNewEntry)
return false;
ASSERT(result.iterator->value);
auto& track = *result.iterator->value;
track.addObserver(*this);
updateActiveState();
if (streamModifier == StreamModifier::DomAPI)
m_private->addTrack(&track.privateTrack(), MediaStreamPrivate::NotifyClientOption::DontNotify);
else
dispatchEvent(MediaStreamTrackEvent::create(eventNames().addtrackEvent, Event::CanBubble::No, Event::IsCancelable::No, &track));
return true;
}
bool MediaStream::internalRemoveTrack(const String& trackId, StreamModifier streamModifier)
{
auto track = m_trackSet.take(trackId);
if (!track)
return false;
track->removeObserver(*this);
updateActiveState();
if (streamModifier == StreamModifier::DomAPI)
m_private->removeTrack(track->privateTrack(), MediaStreamPrivate::NotifyClientOption::DontNotify);
else
dispatchEvent(MediaStreamTrackEvent::create(eventNames().removetrackEvent, Event::CanBubble::No, Event::IsCancelable::No, WTFMove(track)));
return true;
}
void MediaStream::setIsActive(bool active)
{
if (m_isActive == active)
return;
m_isActive = active;
statusDidChange();
}
void MediaStream::mediaCanStart(Document& document)
{
ASSERT_UNUSED(document, &document == this->document());
ASSERT(m_isWaitingUntilMediaCanStart);
if (m_isWaitingUntilMediaCanStart) {
m_isWaitingUntilMediaCanStart = false;
startProducingData();
}
}
void MediaStream::startProducingData()
{
Document* document = this->document();
if (!document || !document->page())
return;
if (!document->page()->canStartMedia()) {
LOG(Media, "MediaStream::startProducingData(%p) - not allowed to start in background, waiting", this);
if (m_isWaitingUntilMediaCanStart)
return;
m_isWaitingUntilMediaCanStart = true;
document->addMediaCanStartListener(this);
return;
}
if (m_isProducingData)
return;
m_isProducingData = true;
m_mediaSession->canProduceAudioChanged();
m_private->startProducingData();
if (document->page()->isMediaCaptureMuted())
m_private->setCaptureTracksMuted(true);
}
void MediaStream::stopProducingData()
{
if (!m_isProducingData)
return;
m_isProducingData = false;
m_mediaSession->canProduceAudioChanged();
m_private->stopProducingData();
}
void MediaStream::endCaptureTracks()
{
for (auto& track : m_trackSet.values()) {
if (track->isCaptureTrack())
track->stopTrack(MediaStreamTrack::StopMode::PostEvent);
}
}
MediaProducer::MediaStateFlags MediaStream::mediaState() const
{
MediaProducer::MediaStateFlags state = MediaProducer::IsNotPlaying;
if (!m_isActive || !document() || !document()->page())
return state;
for (const auto& track : m_trackSet.values())
state |= track->mediaState();
return state;
}
void MediaStream::statusDidChange()
{
m_mediaSession->canProduceAudioChanged();
if (Document* document = this->document()) {
if (!m_isActive)
return;
document->updateIsPlayingMedia();
}
}
void MediaStream::characteristicsChanged()
{
auto state = mediaState();
if (m_state != state) {
m_state = state;
statusDidChange();
}
}
void MediaStream::updateActiveState()
{
bool active = false;
for (auto& track : m_trackSet.values()) {
if (!track->ended()) {
active = true;
break;
}
}
if (m_isActive == active)
return;
setIsActive(active);
}
URLRegistry& MediaStream::registry() const
{
return MediaStreamRegistry::shared();
}
MediaStreamTrackVector MediaStream::trackVectorForType(RealtimeMediaSource::Type filterType) const
{
MediaStreamTrackVector tracks;
for (auto& track : m_trackSet.values()) {
if (track->source().type() == filterType)
tracks.append(track);
}
return tracks;
}
void MediaStream::addObserver(MediaStream::Observer* observer)
{
if (m_observers.find(observer) == notFound)
m_observers.append(observer);
}
void MediaStream::removeObserver(MediaStream::Observer* observer)
{
size_t pos = m_observers.find(observer);
if (pos != notFound)
m_observers.remove(pos);
}
Document* MediaStream::document() const
{
return downcast<Document>(scriptExecutionContext());
}
PlatformMediaSession::MediaType MediaStream::mediaType() const
{
if (m_private->hasAudio() && m_isProducingData && m_private->hasCaptureAudioSource())
return PlatformMediaSession::MediaStreamCapturingAudio;
return PlatformMediaSession::None;
}
PlatformMediaSession::MediaType MediaStream::presentationType() const
{
return mediaType();
}
PlatformMediaSession::CharacteristicsFlags MediaStream::characteristics() const
{
PlatformMediaSession::CharacteristicsFlags state = PlatformMediaSession::HasNothing;
if (!m_isProducingData)
return state;
if (m_private->hasAudio())
state |= PlatformMediaSession::HasAudio;
if (m_private->hasVideo())
state |= PlatformMediaSession::HasVideo;
return state;
}
void MediaStream::mayResumePlayback(bool)
{
}
void MediaStream::suspendPlayback()
{
}
String MediaStream::sourceApplicationIdentifier() const
{
Document* document = this->document();
if (document && document->frame()) {
if (NetworkingContext* networkingContext = document->frame()->loader().networkingContext())
return networkingContext->sourceApplicationIdentifier();
}
return emptyString();
}
bool MediaStream::canProduceAudio() const
{
return !muted() && active() && m_private->hasAudio() && m_isProducingData;
}
bool MediaStream::processingUserGestureForMedia() const
{
return document() ? document()->processingUserGestureForMedia() : false;
}
void MediaStream::stop()
{
m_isActive = false;
endCaptureTracks();
}
const char* MediaStream::activeDOMObjectName() const
{
return "MediaStream";
}
bool MediaStream::canSuspendForDocumentSuspension() const
{
return !hasPendingActivity();
}
bool MediaStream::hasPendingActivity() const
{
return m_isActive;
}
}
#endif // ENABLE(MEDIA_STREAM)