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

#include "remoting/protocol/webrtc_video_stream.h"

#include "base/logging.h"
#include "remoting/protocol/webrtc_video_capturer_adapter.h"
#include "third_party/webrtc/api/mediastreaminterface.h"
#include "third_party/webrtc/api/peerconnectioninterface.h"
#include "third_party/webrtc/api/test/fakeconstraints.h"

namespace remoting {
namespace protocol {

const char kStreamLabel[] = "screen_stream";
const char kVideoLabel[] = "screen_video";

WebrtcVideoStream::WebrtcVideoStream() {}

WebrtcVideoStream::~WebrtcVideoStream() {
  if (stream_) {
    for (const auto& track : stream_->GetVideoTracks()) {
      track->GetSource()->Stop();
      stream_->RemoveTrack(track.get());
    }
    connection_->RemoveStream(stream_.get());
  }

  // MediaStream may still outlive WebrtcVideoStream because it's
  // ref-counted. Reset SizeCallback to make sure it won't be called again.
  if (capturer_adapter_)
    capturer_adapter_->SetSizeCallback(SizeCallback());
}

bool WebrtcVideoStream::Start(
    scoped_ptr<webrtc::DesktopCapturer> desktop_capturer,
    scoped_refptr<webrtc::PeerConnectionInterface> connection,
    scoped_refptr<webrtc::PeerConnectionFactoryInterface>
        peer_connection_factory) {
  scoped_ptr<WebrtcVideoCapturerAdapter> capturer_adapter(
      new WebrtcVideoCapturerAdapter(std::move(desktop_capturer)));
  capturer_adapter_ = capturer_adapter->GetWeakPtr();

  connection_ = connection;

  // Set video stream constraints.
  webrtc::FakeConstraints video_constraints;
  video_constraints.AddMandatory(
      webrtc::MediaConstraintsInterface::kMinFrameRate, 5);

  rtc::scoped_refptr<webrtc::VideoTrackInterface> video_track =
      peer_connection_factory->CreateVideoTrack(
          kVideoLabel, peer_connection_factory->CreateVideoSource(
                           capturer_adapter.release(), &video_constraints));

  stream_ = peer_connection_factory->CreateLocalMediaStream(kStreamLabel);

  if (!stream_->AddTrack(video_track.get()) ||
      !connection_->AddStream(stream_.get())) {
    stream_ = nullptr;
    connection_ = nullptr;
    return false;
  }

  return true;
}

void WebrtcVideoStream::Pause(bool pause) {
  if (capturer_adapter_)
    capturer_adapter_->PauseCapturer(pause);
}

void WebrtcVideoStream::OnInputEventReceived(int64_t event_timestamp) {
  NOTIMPLEMENTED();
}

void WebrtcVideoStream::SetLosslessEncode(bool want_lossless) {
  NOTIMPLEMENTED();
}

void WebrtcVideoStream::SetLosslessColor(bool want_lossless) {
  NOTIMPLEMENTED();
}

void WebrtcVideoStream::SetSizeCallback(const SizeCallback& size_callback) {
  if (capturer_adapter_)
    capturer_adapter_->SetSizeCallback(size_callback);
}

}  // namespace protocol
}  // namespace remoting