// Copyright 2016 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.

#ifndef MEDIA_REMOTING_REMOTE_DEMUXER_STREAM_ADAPTER_H_
#define MEDIA_REMOTING_REMOTE_DEMUXER_STREAM_ADAPTER_H_

#include <stdint.h>

#include <memory>
#include <vector>

#include "base/macros.h"
#include "base/memory/weak_ptr.h"
#include "base/optional.h"
#include "media/base/audio_decoder_config.h"
#include "media/base/demuxer_stream.h"
#include "media/base/video_decoder_config.h"
#include "media/mojo/interfaces/remoting.mojom.h"
#include "media/remoting/rpc/rpc_broker.h"
#include "mojo/public/cpp/system/data_pipe.h"

namespace base {
class SingleThreadTaskRunner;
}

namespace media {

class DemuxerStream;

namespace remoting {

// Class to fetch audio/video buffer from demuxer and send it to browser process
// via mojo::Remoting interface. Note the class is created and run on media
// thread using |media_task_runner|, Mojo data pipe should run on media thread,
// while RPC message should be sent on main thread using |main_task_runner|.
class RemoteDemuxerStreamAdapter {
 public:
  // |main_task_runner|: Task runner to post RPC message on main thread
  // |media_task_runner|: Task runner to run whole class on media thread.
  // |name|: Demuxer stream name. For troubleshooting purposes.
  // |demuxer_stream|: Demuxer component.
  // |rpc_broker|: Broker class to handle incoming and outgoing RPC message. It
  //               is used only on the main thread.
  // |stream_sender_info|: Transfer of pipe binding on the media thread. It is
  //                       to access mojo interface for sending data stream.
  // |producer_handle|: handle to send data using mojo data pipe.
  RemoteDemuxerStreamAdapter(
      scoped_refptr<base::SingleThreadTaskRunner> main_task_runner,
      scoped_refptr<base::SingleThreadTaskRunner> media_task_runner,
      const std::string& name,
      ::media::DemuxerStream* demuxer_stream,
      const base::WeakPtr<RpcBroker>& rpc_broker,
      mojom::RemotingDataStreamSenderPtrInfo stream_sender_info,
      mojo::ScopedDataPipeProducerHandle producer_handle);
  ~RemoteDemuxerStreamAdapter();

  // Rpc handle for this class. This is used for sending/receiving RPC message
  // with specific hanle using Rpcbroker.
  int rpc_handle() const { return rpc_handle_; }

  // Signals if system is in flushing state. The caller uses |flushing| to
  // signal when flush starts and when is done. During flush operation, all
  // fetching data actions will be discarded. The return value indicates frame
  // count in order to signal receiver what frames are in flight before flush,
  // or base::nullopt if the flushing state was unchanged.
  base::Optional<uint32_t> SignalFlush(bool flushing);

 private:
  friend class MockRemoteDemuxerStreamAdapter;

  // Receives RPC message from RpcBroker.
  void OnReceivedRpc(std::unique_ptr<remoting::pb::RpcMessage> message);

  // RPC message tasks.
  void Initialize(int remote_callback_handle);
  void ReadUntil(std::unique_ptr<remoting::pb::RpcMessage> message);
  void RequestBuffer(int callback_handle);
  void SendReadAck(int callback_handle);

  // Callback function when retrieving data from demuxer.
  void OnNewBuffer(int callback_handle,
                   ::media::DemuxerStream::Status status,
                   const scoped_refptr<::media::DecoderBuffer>& input);
  void TryWriteData(int callback_handle, MojoResult result);
  void ResetPendingFrame();

  // Callback function when data pipe error occurs.
  void OnFatalError();

  const scoped_refptr<base::SingleThreadTaskRunner> main_task_runner_;
  const scoped_refptr<base::SingleThreadTaskRunner> media_task_runner_;

  // Name of demuxer stream. Debug only.
  const std::string name_;

  // Weak pointer of RpcBroker. It should use |main_task_runner_| to access the
  // interfaces.
  const base::WeakPtr<RpcBroker> rpc_broker_;

  // RPC handle for this demuxer stream service.
  const int rpc_handle_;

  // Demuxer stream and stream type.
  ::media::DemuxerStream* const demuxer_stream_;
  const ::media::DemuxerStream::Type type_;

  // Remote RPC handles to process demuxer stream data.
  int remote_callback_handle_;

  // Current frame count issued by RPC_DS_READUNTIL RPC message. It should send
  // all frame data with count id smaller than |read_until_count_| before
  // sending RPC_DS_READUNTIL_CALLBACK back to receiver.
  uint32_t read_until_count_;

  // Count id of last frame sent.
  uint32_t last_count_;

  // Indicates if the class is processing RPC_DS_READUNTIL RPC message. The flag
  // is to prevent from receiving multiple RPC_DS_READUNTIL RPC messages. This
  // is set to true after receiving RPC_DS_READUNTIL and before sending
  // RPC_DS_READUNTIL_CALLBACK back to receiver.
  bool processing_read_rpc_;

  // Flag to indicate if it's on flushing operation. All actions should be
  // aborted and data should be discarded when the value is true.
  bool pending_flush_;

  // Frame buffer and its information that is currently in process of writing to
  // Mojo data pipe.
  std::vector<uint8_t> pending_frame_;
  uint32_t current_pending_frame_offset_;
  bool pending_frame_is_eos_;

  // Monitor if data pipe is available to write data.
  mojo::Watcher write_watcher_;

  // Keeps latest demuxer stream status and audio/video decoder config.
  ::media::DemuxerStream::Status media_status_;
  ::media::AudioDecoderConfig audio_config_;
  ::media::VideoDecoderConfig video_config_;

  ::media::mojom::RemotingDataStreamSenderPtr stream_sender_;
  mojo::ScopedDataPipeProducerHandle producer_handle_;

  // WeakPtrFactory only for reading buffer from demuxer stream. This is used
  // for canceling all read callbacks provided to the |demuxer_stream_| before a
  // flush.
  base::WeakPtrFactory<RemoteDemuxerStreamAdapter> request_buffer_weak_factory_;
  // WeakPtrFactory for normal usage.
  base::WeakPtrFactory<RemoteDemuxerStreamAdapter> weak_factory_;

  DISALLOW_COPY_AND_ASSIGN(RemoteDemuxerStreamAdapter);
};

mojo::DataPipe* CreateDataPipe();

}  // namespace remoting
}  // namespace media

#endif  // MEDIA_REMOTING_REMOTE_DEMUXER_STREAM_ADAPTER_H_