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

#ifndef THIRD_PARTY_BLINK_RENDERER_MODULES_MEDIASTREAM_MOCK_MOJO_MEDIA_STREAM_DISPATCHER_HOST_H_
#define THIRD_PARTY_BLINK_RENDERER_MODULES_MEDIASTREAM_MOCK_MOJO_MEDIA_STREAM_DISPATCHER_HOST_H_

#include "build/build_config.h"
#include "media/capture/mojom/video_capture_types.mojom-blink.h"
#include "mojo/public/cpp/bindings/pending_remote.h"
#include "mojo/public/cpp/bindings/receiver.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "third_party/blink/public/common/mediastream/media_stream_controls.h"
#include "third_party/blink/public/common/mediastream/media_stream_request.h"
#include "third_party/blink/public/mojom/mediastream/media_stream.mojom-blink.h"

namespace blink {

class MockMojoMediaStreamDispatcherHost
    : public mojom::blink::MediaStreamDispatcherHost {
 public:
  MockMojoMediaStreamDispatcherHost() = default;

  MockMojoMediaStreamDispatcherHost(const MockMojoMediaStreamDispatcherHost&) =
      delete;
  MockMojoMediaStreamDispatcherHost& operator=(
      const MockMojoMediaStreamDispatcherHost&) = delete;

  ~MockMojoMediaStreamDispatcherHost() override;

  mojo::PendingRemote<mojom::blink::MediaStreamDispatcherHost>
  CreatePendingRemoteAndBind();

  void GenerateStreams(
      int32_t request_id,
      const StreamControls& controls,
      bool user_gesture,
      mojom::blink::StreamSelectionInfoPtr audio_stream_selection_info_ptr,
      GenerateStreamsCallback callback) override;
  void CancelRequest(int32_t request_id) override;
  void StopStreamDevice(
      const String& device_id,
      const std::optional<base::UnguessableToken>& session_id) override;
  void OpenDevice(int32_t request_id,
                  const String& device_id,
                  mojom::blink::MediaStreamType type,
                  OpenDeviceCallback callback) override;

  MOCK_METHOD1(CloseDevice, void(const String&));
  MOCK_METHOD3(SetCapturingLinkSecured,
               void(const std::optional<base::UnguessableToken>&,
                    mojom::blink::MediaStreamType,
                    bool));
  MOCK_METHOD1(OnStreamStarted, void(const String&));
  MOCK_METHOD3(KeepDeviceAliveForTransfer,
               void(const base::UnguessableToken&,
                    const base::UnguessableToken&,
                    KeepDeviceAliveForTransferCallback));
#if !BUILDFLAG(IS_ANDROID) && !BUILDFLAG(IS_IOS)
  MOCK_METHOD2(SendWheel,
               void(const base::UnguessableToken&,
                    mojom::blink::CapturedWheelActionPtr));
  MOCK_METHOD3(UpdateZoomLevel,
               void(const base::UnguessableToken&,
                    mojom::blink::ZoomLevelAction,
                    UpdateZoomLevelCallback));
  MOCK_METHOD2(RequestCapturedSurfaceControlPermission,
               void(const base::UnguessableToken&,
                    RequestCapturedSurfaceControlPermissionCallback));
  MOCK_METHOD2(FocusCapturedSurface, void(const String&, bool));
#endif

  MOCK_METHOD5(ApplySubCaptureTarget,
               void(const base::UnguessableToken&,
                    media::mojom::blink::SubCaptureTargetType,
                    const base::Token&,
                    uint32_t,
                    ApplySubCaptureTargetCallback));
  void GetOpenDevice(int32_t request_id,
                     const base::UnguessableToken&,
                     const base::UnguessableToken&,
                     GetOpenDeviceCallback) override;

  void ResetSessionId() { session_id_ = base::UnguessableToken::Create(); }
  void DoNotRunCallback() { do_not_run_cb_ = true; }
  const base::UnguessableToken& session_id() { return session_id_; }

  int request_stream_counter() const { return request_stream_counter_; }
  int stop_audio_device_counter() const { return stop_audio_device_counter_; }
  int stop_video_device_counter() const { return stop_video_device_counter_; }

  const blink::mojom::blink::StreamDevices& devices() const {
    return stream_devices_;
  }

  void SetStreamDevices(const blink::mojom::blink::StreamDevices& devices) {
    stream_devices_ = devices;
  }

  void SetAppendSessionIdToDeviceIds(bool append_session_id_to_device_ids) {
    append_session_id_to_device_ids_ = append_session_id_to_device_ids;
  }

  // Appends the `session_id_ to the passed `device_id` if
  // `append_session_id_to_device_ids_` is true.
  std::string MaybeAppendSessionId(std::string device_id);

  void SetAudioDeviceEffects(std::optional<int> effects) {
    audio_device_effects_ = effects;
  }

 private:
  int request_id_ = -1;
  int request_stream_counter_ = 0;
  int stop_audio_device_counter_ = 0;
  int stop_video_device_counter_ = 0;
  base::UnguessableToken session_id_ = base::UnguessableToken::Create();
  bool do_not_run_cb_ = false;
  blink::mojom::blink::StreamDevices stream_devices_;
  GenerateStreamsCallback generate_stream_cb_;
  GetOpenDeviceCallback get_open_device_cb_;
  mojo::Receiver<mojom::blink::MediaStreamDispatcherHost> receiver_{this};
  bool append_session_id_to_device_ids_ = false;
  std::optional<int> audio_device_effects_;
};

}  // namespace blink

#endif  // THIRD_PARTY_BLINK_RENDERER_MODULES_MEDIASTREAM_MOCK_MOJO_MEDIA_STREAM_DISPATCHER_HOST_H_