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

#include "third_party/blink/renderer/modules/mediasource/same_thread_media_source_attachment.h"

#include "third_party/blink/renderer/core/html/media/html_media_element.h"
#include "third_party/blink/renderer/core/html/track/audio_track_list.h"
#include "third_party/blink/renderer/core/html/track/video_track_list.h"
#include "third_party/blink/renderer/modules/mediasource/attachment_creation_pass_key_provider.h"
#include "third_party/blink/renderer/modules/mediasource/media_source.h"
#include "third_party/blink/renderer/modules/mediasource/same_thread_media_source_tracer.h"

namespace {
// Downcasts |tracer| to the expected same-thread attachment's tracer type.
// Includes a debug-mode check that the tracer matches the expected attachment
// semantic.
blink::SameThreadMediaSourceTracer* GetTracerImpl(
    blink::MediaSourceTracer* tracer) {
  DCHECK(!tracer || !tracer->IsCrossThreadForDebugging());
  return static_cast<blink::SameThreadMediaSourceTracer*>(tracer);
}

blink::MediaSource* GetMediaSource(blink::MediaSourceTracer* tracer) {
  return GetTracerImpl(tracer)->GetMediaSource();
}

blink::HTMLMediaElement* GetMediaElement(blink::MediaSourceTracer* tracer) {
  return GetTracerImpl(tracer)->GetMediaElement();
}

}  // namespace

namespace blink {

SameThreadMediaSourceAttachment::SameThreadMediaSourceAttachment(
    MediaSource* media_source,
    AttachmentCreationPassKeyProvider::PassKey /* passkey */)
    : registered_media_source_(media_source),
      recent_element_time_(0.0),
      element_has_error_(false),
      element_context_destroyed_(false),
      media_source_context_destroyed_(false) {
  // This kind of attachment only operates on the main thread.
  DCHECK(IsMainThread());

  DVLOG(1) << __func__ << " this=" << this << " media_source=" << media_source;

  // Verify that at construction time, refcounting of this object begins at
  // precisely 1.
  DCHECK(HasOneRef());
}

SameThreadMediaSourceAttachment::~SameThreadMediaSourceAttachment() {
  DVLOG(1) << __func__ << " this=" << this;
}

void SameThreadMediaSourceAttachment::NotifyDurationChanged(
    MediaSourceTracer* tracer,
    double duration) {
  DVLOG(1) << __func__ << " this=" << this;

  VerifyCalledWhileContextsAliveForDebugging();

  HTMLMediaElement* element = GetMediaElement(tracer);

  bool request_seek = element->currentTime() > duration;
  element->DurationChanged(duration, request_seek);
}

base::TimeDelta SameThreadMediaSourceAttachment::GetRecentMediaTime(
    MediaSourceTracer* tracer) {
  DVLOG(1) << __func__ << " this=" << this;

  VerifyCalledWhileContextsAliveForDebugging();

  HTMLMediaElement* element = GetMediaElement(tracer);
  base::TimeDelta result = base::Seconds(element->currentTime());

  DVLOG(2) << __func__ << " this=" << this
           << " -> recent time=" << recent_element_time_
           << ", actual currentTime=" << result;
  return result;
}

bool SameThreadMediaSourceAttachment::GetElementError(
    MediaSourceTracer* tracer) {
  DVLOG(1) << __func__ << " this=" << this;

  VerifyCalledWhileContextsAliveForDebugging();

  HTMLMediaElement* element = GetMediaElement(tracer);
  bool current_element_error_state = !!element->error();

  DCHECK_EQ(current_element_error_state, element_has_error_);

  return current_element_error_state;
}

AudioTrackList* SameThreadMediaSourceAttachment::CreateAudioTrackList(
    MediaSourceTracer* tracer) {
  DVLOG(1) << __func__ << " this=" << this;

  VerifyCalledWhileContextsAliveForDebugging();

  HTMLMediaElement* element = GetMediaElement(tracer);
  return MakeGarbageCollected<AudioTrackList>(*element);
}

VideoTrackList* SameThreadMediaSourceAttachment::CreateVideoTrackList(
    MediaSourceTracer* tracer) {
  DVLOG(1) << __func__ << " this=" << this;

  VerifyCalledWhileContextsAliveForDebugging();

  HTMLMediaElement* element = GetMediaElement(tracer);
  return MakeGarbageCollected<VideoTrackList>(*element);
}

void SameThreadMediaSourceAttachment::AddAudioTrackToMediaElement(
    MediaSourceTracer* tracer,
    AudioTrack* track) {
  DVLOG(3) << __func__ << " this=" << this;

  VerifyCalledWhileContextsAliveForDebugging();

  HTMLMediaElement* element = GetMediaElement(tracer);
  element->audioTracks().Add(track);
}

void SameThreadMediaSourceAttachment::AddVideoTrackToMediaElement(
    MediaSourceTracer* tracer,
    VideoTrack* track) {
  DVLOG(3) << __func__ << " this=" << this;

  VerifyCalledWhileContextsAliveForDebugging();

  HTMLMediaElement* element = GetMediaElement(tracer);
  element->videoTracks().Add(track);
}

void SameThreadMediaSourceAttachment::RemoveAudioTracksFromMediaElement(
    MediaSourceTracer* tracer,
    Vector<String> audio_ids,
    bool enqueue_change_event) {
  DVLOG(3) << __func__ << " this=" << this << ", ids size=" << audio_ids.size()
           << ", enqueue_change_event=" << enqueue_change_event;

  if (element_context_destroyed_ || media_source_context_destroyed_) {
    DVLOG(3) << __func__ << " this=" << this
             << " -> skipping due to context(s) destroyed";
    return;
  }

  HTMLMediaElement* element = GetMediaElement(tracer);
  for (auto& audio_id : audio_ids) {
    if (element->audioTracks().getTrackById(audio_id)) {
      element->audioTracks().Remove(audio_id);
    } else {
      // This case can happen on element noneSupported() after MSE had added
      // some track(s). See https://crbug.com/1204656.
      DVLOG(3) << __func__ << " this=" << this
               << ", skipping removal of missing audio track id " << audio_id;
    }
  }

  if (enqueue_change_event) {
    Event* event = Event::Create(event_type_names::kChange);
    event->SetTarget(&element->audioTracks());
    element->ScheduleEvent(event);
  }
}

void SameThreadMediaSourceAttachment::RemoveVideoTracksFromMediaElement(
    MediaSourceTracer* tracer,
    Vector<String> video_ids,
    bool enqueue_change_event) {
  DVLOG(3) << __func__ << " this=" << this << ", ids size=" << video_ids.size()
           << ", enqueue_change_event=" << enqueue_change_event;

  if (element_context_destroyed_ || media_source_context_destroyed_) {
    DVLOG(3) << __func__ << " this=" << this
             << " -> skipping due to context(s) destroyed";
    return;
  }

  HTMLMediaElement* element = GetMediaElement(tracer);
  for (auto& video_id : video_ids) {
    if (element->videoTracks().getTrackById(video_id)) {
      element->videoTracks().Remove(video_id);
    } else {
      // This case can happen on element noneSupported() after MSE had added
      // some track(s). See https://crbug.com/1204656.
      DVLOG(3) << __func__ << " this=" << this
               << ", skipping removal of missing video track id " << video_id;
    }
  }

  if (enqueue_change_event) {
    Event* event = Event::Create(event_type_names::kChange);
    event->SetTarget(&element->videoTracks());
    element->ScheduleEvent(event);
  }
}

void SameThreadMediaSourceAttachment::OnMediaSourceContextDestroyed() {
  DVLOG(3) << __func__ << " this=" << this;

  // We should only be notified once.
  DCHECK(!media_source_context_destroyed_);

  media_source_context_destroyed_ = true;
}

void SameThreadMediaSourceAttachment::Unregister() {
  DVLOG(1) << __func__ << " this=" << this;

  // The only expected caller is a MediaSourceRegistryImpl on the main thread.
  DCHECK(IsMainThread());

  // Release our strong reference to the MediaSource. Note that revokeObjectURL
  // of the url associated with this attachment could commonly follow this path
  // while the MediaSource (and any attachment to an HTMLMediaElement) may still
  // be alive/active.
  DCHECK(registered_media_source_);
  registered_media_source_ = nullptr;
}

MediaSourceTracer*
SameThreadMediaSourceAttachment::StartAttachingToMediaElement(
    HTMLMediaElement* element,
    bool* success) {
  VerifyCalledWhileContextsAliveForDebugging();
  DCHECK(success);

  if (!registered_media_source_) {
    *success = false;
    return nullptr;
  }

  MediaSourceTracer* tracer =
      registered_media_source_->StartAttachingToMediaElement(
          WrapRefCounted(this), element);

  // For this same-thread attachment start, a non-nullptr tracer indicates
  // success here.
  *success = !!tracer;
  return tracer;
}

void SameThreadMediaSourceAttachment::CompleteAttachingToMediaElement(
    MediaSourceTracer* tracer,
    std::unique_ptr<WebMediaSource> web_media_source) {
  VerifyCalledWhileContextsAliveForDebugging();

  GetMediaSource(tracer)->CompleteAttachingToMediaElement(
      std::move(web_media_source));
}

void SameThreadMediaSourceAttachment::Close(MediaSourceTracer* tracer) {
  // The media element may have already notified us that its context is
  // destroyed, so VerifyCalledWhileContextIsAliveForDebugging() is unusable in
  // this scope. Note that we might be called during main thread context
  // destruction, and the ordering of MediaSource versus HTMLMediaElement
  // context destruction notification is nondeterminate. MediaSource closes
  // itself on its context destruction notification, so elide calling Close()
  // again on it in that case. This affords stronger guarantees on when
  // MediaSource::Close is callable by an attachment.
  if (media_source_context_destroyed_)
    return;

  GetMediaSource(tracer)->Close();
}

WebTimeRanges SameThreadMediaSourceAttachment::BufferedInternal(
    MediaSourceTracer* tracer) const {
  // The ordering of MediaSource versus HTMLMediaElement context destruction
  // notification is nondeterminate.
  if (media_source_context_destroyed_) {
    return {};
  }

  // Since the attached MediaSource, HTMLMediaElement and the element's player's
  // underlying demuxer are all owned by the main thread in this SameThread
  // attachment, it is safe to get an ExclusiveKey here to pass to the attached
  // MediaSource to let it know it is safe to access the underlying demuxer to
  // tell us what is currently buffered.
  return GetMediaSource(tracer)->BufferedInternal(GetExclusiveKey());
}

WebTimeRanges SameThreadMediaSourceAttachment::SeekableInternal(
    MediaSourceTracer* tracer) const {
  VerifyCalledWhileContextsAliveForDebugging();

  // Since the attached MediaSource, HTMLMediaElement and the element's player's
  // underlying demuxer are all owned by the main thread in this SameThread
  // attachment, it is safe to get an ExclusiveKey here to pass to the attached
  // MediaSource to let it know it is safe to access the underlying demuxer to
  // tell us what is currently seekable.
  return GetMediaSource(tracer)->SeekableInternal(GetExclusiveKey());
}

void SameThreadMediaSourceAttachment::OnTrackChanged(MediaSourceTracer* tracer,
                                                     TrackBase* track) {
  // In this same thread implementation, the MSE side of the attachment can loop
  // back into this from SourceBuffer's initialization segment received
  // algorithm notifying the element, which then calls this. Regardless, we are
  // not called as part of execution context teardown, so verification should be
  // stable here.
  VerifyCalledWhileContextsAliveForDebugging();

  GetMediaSource(tracer)->OnTrackChanged(track);
}

void SameThreadMediaSourceAttachment::OnElementTimeUpdate(double time) {
  DVLOG(3) << __func__ << " this=" << this << ", time=" << time;

  // The ordering of MediaSource versus HTMLMediaElement context destruction
  // notification is nondeterminate.
  if (media_source_context_destroyed_) {
    return;
  }

  recent_element_time_ = time;
}

void SameThreadMediaSourceAttachment::OnElementError() {
  DVLOG(3) << __func__ << " this=" << this;

  VerifyCalledWhileContextsAliveForDebugging();

  DCHECK(!element_has_error_)
      << "At most one transition to element error per attachment is expected";

  element_has_error_ = true;
}

void SameThreadMediaSourceAttachment::OnElementContextDestroyed() {
  DVLOG(3) << __func__ << " this=" << this;

  // We should only be notified once.
  DCHECK(!element_context_destroyed_);

  element_context_destroyed_ = true;
}

void SameThreadMediaSourceAttachment::
    VerifyCalledWhileContextsAliveForDebugging() const {
  DCHECK(!element_context_destroyed_);
  DCHECK(!media_source_context_destroyed_);
}

}  // namespace blink