chrome/browser/installable/installed_webapp_geolocation_context.cc

// 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/installable/installed_webapp_geolocation_context.h"

#include <utility>
#include "base/containers/cxx20_erase_vector.h"
#include "chrome/browser/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 GURL& requesting_url) {
  impls_.push_back(std::make_unique<InstalledWebappGeolocationBridge>(
      std::move(receiver), requesting_url, this));
  if (geoposition_override_)
    impls_.back()->SetOverride(geoposition_override_.Clone());
  else
    impls_.back()->StartListeningForUpdates();
}

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

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();
  }
}