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

#include "chrome/browser/webapps/installable/installed_webapp_geolocation_context.h"

#include <utility>
#include <vector>

#include "chrome/browser/webapps/installable/installed_webapp_geolocation_bridge.h"
#include "url/origin.h"

InstalledWebappGeolocationContext::InstalledWebappGeolocationContext() =
    default;

InstalledWebappGeolocationContext::~InstalledWebappGeolocationContext() =
    default;

void InstalledWebappGeolocationContext::BindGeolocation(
    mojo::PendingReceiver<device::mojom::Geolocation> receiver,
    const url::Origin& requesting_origin,
    device::mojom::GeolocationClientId client_id,
    bool has_precise_permission) {
  impls_.push_back(std::make_unique<InstalledWebappGeolocationBridge>(
      std::move(receiver), requesting_origin, this));
  if (geoposition_override_)
    impls_.back()->SetOverride(geoposition_override_.Clone());
  else
    impls_.back()->StartListeningForUpdates();
}

void InstalledWebappGeolocationContext::OnPermissionRevoked(
    const url::Origin& origin) {
  std::erase_if(impls_, [&origin](const auto& impl) {
    if (origin != impl->origin()) {
      return false;
    }
    // Invoke the position callback with kPermissionDenied before removing.
    impl->OnPermissionRevoked();
    return true;
  });
}

void InstalledWebappGeolocationContext::OnPermissionUpdated(
    const url::Origin& origin,
    device::mojom::GeolocationPermissionLevel permission_level) {
  // For subclasses that support approximate geolocation, `kPrecise` and
  // `kApproximate` are meant for changing accuracy settings. For subclasses
  // that do not support this feature, these cases are no-ops.
  if (permission_level == device::mojom::GeolocationPermissionLevel::kDenied) {
    OnPermissionRevoked(origin);
  }
}

void InstalledWebappGeolocationContext::OnConnectionError(
    InstalledWebappGeolocationBridge* impl) {
  for (auto it = impls_.begin(); it != impls_.end(); ++it) {
    if (impl == it->get()) {
      impls_.erase(it);
      return;
    }
  }
}

void InstalledWebappGeolocationContext::SetOverride(
    device::mojom::GeopositionResultPtr result) {
  CHECK(result);
  geoposition_override_ = std::move(result);
  for (auto& impl : impls_) {
    impl->SetOverride(geoposition_override_.Clone());
  }
}

void InstalledWebappGeolocationContext::ClearOverride() {
  geoposition_override_.reset();
  for (auto& impl : impls_) {
    impl->ClearOverride();
  }
}