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

#include "components/permissions/contexts/geolocation_permission_context_system.h"

#include "components/content_settings/core/browser/content_settings_observer.h"
#include "components/content_settings/core/browser/content_settings_registry.h"
#include "components/content_settings/core/common/content_settings.h"
#include "components/content_settings/core/common/features.h"
#include "content/public/browser/browser_thread.h"
#include "services/device/public/cpp/geolocation/location_system_permission_status.h"

namespace permissions {
GeolocationPermissionContextSystem::GeolocationPermissionContextSystem(
    content::BrowserContext* browser_context,
    std::unique_ptr<Delegate> delegate)
    : GeolocationPermissionContext(browser_context, std::move(delegate)) {
  DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
  auto* geolocation_system_permission_manager =
      device::GeolocationSystemPermissionManager::GetInstance();
  DCHECK(geolocation_system_permission_manager);
  geolocation_system_permission_manager->AddObserver(this);
  system_permission_ =
      geolocation_system_permission_manager->GetSystemPermission();
}

GeolocationPermissionContextSystem::~GeolocationPermissionContextSystem() {
  DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
  auto* geolocation_system_permission_manager =
      device::GeolocationSystemPermissionManager::GetInstance();
  if (geolocation_system_permission_manager) {
    geolocation_system_permission_manager->RemoveObserver(this);
  }
}

PermissionSetting
GeolocationPermissionContextSystem::GetPermissionStatusInternal(
    content::RenderFrameHost* render_frame_host,
    const GURL& requesting_origin,
    const GURL& embedding_origin) const {
  auto site_permission =
      GeolocationPermissionContext::GetPermissionStatusInternal(
          render_frame_host, requesting_origin, embedding_origin);
  // If the client is in the kApproximateGeolocationPermission experiment, their
  // setting is a PermissionSetting, otherwise it's a content setting.
  bool is_permission_content_setting = !base::FeatureList::IsEnabled(
      content_settings::features::kApproximateGeolocationPermission);

  if (is_permission_content_setting) {
    if (std::get<ContentSetting>(site_permission) != CONTENT_SETTING_ALLOW) {
      return site_permission;
    }
  } else {
    if (std::get<GeolocationSetting>(site_permission).approximate !=
        PermissionOption::kAllowed) {
      return site_permission;
    }
  }

  switch (system_permission_) {
    case LocationSystemPermissionStatus::kNotDetermined:
      return is_permission_content_setting
                 ? PermissionSetting(CONTENT_SETTING_ASK)
                 : GeolocationSetting(PermissionOption::kAsk,
                                      PermissionOption::kAsk);
    case LocationSystemPermissionStatus::kDenied:
      return is_permission_content_setting
                 ? PermissionSetting(CONTENT_SETTING_BLOCK)
                 : GeolocationSetting(PermissionOption::kDenied,
                                      PermissionOption::kDenied);
    case LocationSystemPermissionStatus::kAllowed:
      return site_permission;
  }
}

void GeolocationPermissionContextSystem::OnSystemPermissionUpdated(
    LocationSystemPermissionStatus new_status) {
  system_permission_ = new_status;
  for (permissions::Observer& obs : permission_observers_) {
    obs.OnPermissionChanged(ContentSettingsPattern::Wildcard(),
                            ContentSettingsPattern::Wildcard(),
                            ContentSettingsTypeSet(content_settings_type()));
  }
}

}  // namespace permissions