1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49
   50
   51
   52
   53
   54
   55
   56
   57
   58
   59
   60
   61
   62
   63
   64
   65
   66
   67
   68
   69
   70
   71
   72
   73
   74
   75
   76
   77
   78
   79
   80
   81
   82
   83
   84
   85
   86
   87
   88
   89
   90
   91
   92
   93
   94
   95
   96
   97
   98
   99


content / web_test / browser / web_test_sensor_provider_manager.cc [blame]

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

#include "content/web_test/browser/web_test_sensor_provider_manager.h"

#include "base/notreached.h"
#include "content/browser/generic_sensor/web_contents_sensor_provider_proxy.h"
#include "content/browser/web_contents/web_contents_impl.h"
#include "content/public/browser/web_contents.h"

namespace content {

WebTestSensorProviderManager::WebTestSensorProviderManager(
    WebContents* web_contents)
    : web_contents_(static_cast<WebContentsImpl*>(web_contents)->GetWeakPtr()) {
}

WebTestSensorProviderManager::~WebTestSensorProviderManager() = default;

void WebTestSensorProviderManager::Bind(
    mojo::PendingReceiver<blink::test::mojom::WebSensorProviderAutomation>
        receiver) {
  receivers_.Add(this, std::move(receiver));
}

void WebTestSensorProviderManager::CreateVirtualSensor(
    device::mojom::SensorType type,
    device::mojom::VirtualSensorMetadataPtr metadata,
    device::mojom::SensorProvider::CreateVirtualSensorCallback callback) {
  if (!web_contents_) {
    return;
  }

  if (sensor_overrides_.contains(type)) {
    std::move(callback).Run(
        device::mojom::CreateVirtualSensorResult::kSensorTypeAlreadyOverridden);
    return;
  }

  auto virtual_sensor =
      WebContentsSensorProviderProxy::GetOrCreate(web_contents_.get())
          ->CreateVirtualSensorForDevTools(type, std::move(metadata));
  CHECK(virtual_sensor);
  sensor_overrides_[type] = std::move(virtual_sensor);

  std::move(callback).Run(device::mojom::CreateVirtualSensorResult::kSuccess);
}

void WebTestSensorProviderManager::UpdateVirtualSensor(
    device::mojom::SensorType type,
    const device::SensorReading& reading,
    device::mojom::SensorProvider::UpdateVirtualSensorCallback callback) {
  if (!web_contents_) {
    return;
  }

  auto it = sensor_overrides_.find(type);
  if (it == sensor_overrides_.end()) {
    std::move(callback).Run(
        device::mojom::UpdateVirtualSensorResult::kSensorTypeNotOverridden);
    return;
  }

  it->second->UpdateVirtualSensor(std::move(reading), std::move(callback));
}

void WebTestSensorProviderManager::RemoveVirtualSensor(
    device::mojom::SensorType type,
    device::mojom::SensorProvider::RemoveVirtualSensorCallback callback) {
  if (!web_contents_) {
    return;
  }

  sensor_overrides_.erase(type);
  std::move(callback).Run();
}

void WebTestSensorProviderManager::GetVirtualSensorInformation(
    device::mojom::SensorType type,
    device::mojom::SensorProvider::GetVirtualSensorInformationCallback
        callback) {
  if (!web_contents_) {
    return;
  }

  auto it = sensor_overrides_.find(type);
  if (it == sensor_overrides_.end()) {
    std::move(callback).Run(
        device::mojom::GetVirtualSensorInformationResult::NewError(
            device::mojom::GetVirtualSensorInformationError::
                kSensorTypeNotOverridden));
    return;
  }

  it->second->GetVirtualSensorInformation(std::move(callback));
}

}  // namespace content