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

#ifndef CONTENT_BROWSER_GENERIC_SENSOR_FRAME_SENSOR_PROVIDER_PROXY_H_
#define CONTENT_BROWSER_GENERIC_SENSOR_FRAME_SENSOR_PROVIDER_PROXY_H_

#include <optional>

#include "base/memory/weak_ptr.h"
#include "base/scoped_observation.h"
#include "content/browser/generic_sensor/web_contents_sensor_provider_proxy.h"
#include "content/common/content_export.h"
#include "content/public/browser/document_user_data.h"
#include "content/public/browser/permission_controller.h"
#include "content/public/browser/permission_result.h"
#include "content/public/browser/web_contents_observer.h"
#include "mojo/public/cpp/bindings/pending_receiver.h"
#include "mojo/public/cpp/bindings/receiver_set.h"
#include "mojo/public/cpp/bindings/remote_set.h"
#include "services/device/public/mojom/sensor.mojom-shared.h"
#include "services/device/public/mojom/sensor_provider.mojom.h"
#include "third_party/blink/public/mojom/permissions/permission_status.mojom-shared.h"
#include "third_party/blink/public/mojom/sensor/web_sensor_provider.mojom.h"

namespace content {

// Per-RenderFrameHost SensorProvider implementation. It does permission checks
// before forwarding GetSensor() calls in WebContentsSensorProviderProxy.
class FrameSensorProviderProxy final
    : public blink::mojom::WebSensorProvider,
      public WebContentsSensorProviderProxy::Observer,
      public DocumentUserData<FrameSensorProviderProxy>,
      public WebContentsObserver {
 public:
  FrameSensorProviderProxy(const FrameSensorProviderProxy&) = delete;
  FrameSensorProviderProxy& operator=(const FrameSensorProviderProxy&) = delete;

  ~FrameSensorProviderProxy() override;

  void Bind(mojo::PendingReceiver<blink::mojom::WebSensorProvider> receiver);

  // WebContentsSensorProviderProxy::Observer overrides.
  void OnMojoConnectionError() override;

  // WebContentsObserver overrides.
  void OnVisibilityChanged(content::Visibility visibility) override;
  void RenderFrameHostStateChanged(
      RenderFrameHost* render_frame_host,
      RenderFrameHost::LifecycleState old_state,
      RenderFrameHost::LifecycleState new_state) override;
  void OnWebContentsFocused(RenderWidgetHost* render_widget_host) override;
  void OnWebContentsLostFocus(RenderWidgetHost* render_widget_host) override;
  void OnFocusChangedInPage(const FocusedNodeDetails& details) override;

 private:
  explicit FrameSensorProviderProxy(RenderFrameHost* render_frame_host);

  // blink::mojom::WebSensorProvider overrides.
  void GetSensor(device::mojom::SensorType type,
                 bool user_gesture,
                 GetSensorCallback callback) override;

  void OnHardwareCheckCompleted(
      device::mojom::SensorType type,
      bool user_gesture,
      mojo::PendingRemote<device::mojom::SensorClientController> controller,
      GetSensorCallback callback,
      device::mojom::SensorCreationResult result,
      device::mojom::SensorInitParamsPtr params);

  void OnHardwareCheckForBlockedSensor(
      device::mojom::SensorType type,
      device::mojom::SensorCreationResult result,
      device::mojom::SensorInitParamsPtr params);

  void OnPermissionRequestCompleted(
      device::mojom::SensorInitParamsPtr params,
      GetSensorCallback callback,
      mojo::PendingRemote<device::mojom::SensorClientController> controller,
      PermissionResult permission_result);

  void OnPermissionChanged(PermissionResult permission_result);

  void OnSensorDisconnect(mojo::RemoteSetElementId id);

  bool ShouldTrackSensorConnection();

  void FinalizeSensorConnection(
      mojo::PendingRemote<device::mojom::SensorClientController> controller);

  mojo::ReceiverSet<blink::mojom::WebSensorProvider> receiver_set_;

  base::ScopedObservation<WebContentsSensorProviderProxy,
                          WebContentsSensorProviderProxy::Observer>
      scoped_observation_{this};

  bool ShouldSuspendSensors() const;
  void UpdateSensorSessionControllers();

  mojo::RemoteSet<device::mojom::SensorClientController> client_controllers_;
  PermissionController::SubscriptionId permission_subscription_id_;
  std::optional<bool> is_suspended_;

  base::WeakPtrFactory<FrameSensorProviderProxy> weak_factory_{this};

  friend DocumentUserData;
  DOCUMENT_USER_DATA_KEY_DECL();
};

}  // namespace content

#endif  // CONTENT_BROWSER_GENERIC_SENSOR_FRAME_SENSOR_PROVIDER_PROXY_H_