// Copyright 2022 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/test/permission_test_util.h"

#include "base/feature_list.h"
#include "base/notimplemented.h"
#include "components/content_settings/core/common/content_settings_types.h"
#include "components/content_settings/core/common/content_settings_utils.h"
#include "components/content_settings/core/common/features.h"
#include "components/permissions/content_setting_permission_context_base.h"
#include "components/permissions/contexts/geolocation_permission_context.h"
#include "components/permissions/contexts/window_management_permission_context.h"
#include "components/permissions/permission_manager.h"
#include "content/public/browser/browser_context.h"
#include "services/network/public/mojom/permissions_policy/permissions_policy_feature.mojom.h"

namespace permissions {
namespace {

class FakePermissionContext : public ContentSettingPermissionContextBase {
 public:
  FakePermissionContext(
      content::BrowserContext* browser_context,
      ContentSettingsType content_settings_type,
      network::mojom::PermissionsPolicyFeature permissions_policy_feature)
      : ContentSettingPermissionContextBase(browser_context,
                                            content_settings_type,
                                            permissions_policy_feature) {
#if BUILDFLAG(IS_ANDROID)
    if (content_settings_type == ContentSettingsType::NOTIFICATIONS) {
      enabled_app_level_notification_permission_for_testing_ = true;
    }
#endif  // BUILDFLAG(IS_ANDROID)
  }
};

class FakePermissionContextAlwaysAllow : public FakePermissionContext {
 public:
  FakePermissionContextAlwaysAllow(
      content::BrowserContext* browser_context,
      ContentSettingsType content_settings_type,
      network::mojom::PermissionsPolicyFeature permissions_policy_feature)
      : FakePermissionContext(browser_context,
                              content_settings_type,
                              permissions_policy_feature) {}

  // PermissionContextBase:
  ContentSetting GetContentSettingStatusInternal(
      content::RenderFrameHost* render_frame_host,
      const GURL& requesting_origin,
      const GURL& embedding_origin) const override {
    return CONTENT_SETTING_ALLOW;
  }
};

class FakeStorageAccessPermissionContext : public FakePermissionContext {
 public:
  FakeStorageAccessPermissionContext(
      content::BrowserContext* browser_context,
      ContentSettingsType content_settings_type,
      network::mojom::PermissionsPolicyFeature permissions_policy_feature)
      : FakePermissionContext(browser_context,
                              content_settings_type,
                              permissions_policy_feature) {}

  void MaybeOverridePermissionResultToReturn(
      content::PermissionResult& result) const override {
    if (result.status == content::PermissionStatus::DENIED) {
      result.status = content::PermissionStatus::ASK;
    }
  }
};

class TestGeolocationDelegate : public GeolocationPermissionContext::Delegate {
 public:
  bool DecidePermission(const PermissionRequestData& request_data,
                        BrowserPermissionCallback* callback,
                        GeolocationPermissionContext* context) override {
    return false;
  }

#if BUILDFLAG(IS_ANDROID)
  bool IsInteractable(content ::WebContents* web_contents) override {
    return true;
  }

  PrefService* GetPrefs(content ::BrowserContext* browser_context) override {
    NOTIMPLEMENTED();
    return nullptr;
  }

  bool IsRequestingOriginDSE(content ::BrowserContext* browser_context,
                             const GURL& requesting_origin) override {
    return false;
  }
#endif
};

PermissionManager::PermissionContextMap CreatePermissionContexts(
    content::BrowserContext* browser_context) {
  PermissionManager::PermissionContextMap permission_contexts;
  if (base::FeatureList::IsEnabled(
          content_settings::features::kApproximateGeolocationPermission)) {
    permission_contexts[ContentSettingsType::GEOLOCATION_WITH_OPTIONS] =
        std::make_unique<GeolocationPermissionContext>(
            browser_context, std::make_unique<TestGeolocationDelegate>());
  } else {
    permission_contexts[ContentSettingsType::GEOLOCATION] =
        std::make_unique<FakePermissionContext>(
            browser_context, ContentSettingsType::GEOLOCATION,
            network::mojom::PermissionsPolicyFeature::kGeolocation);
  }
  permission_contexts[ContentSettingsType::NOTIFICATIONS] =
      std::make_unique<FakePermissionContext>(
          browser_context, ContentSettingsType::NOTIFICATIONS,
          network::mojom::PermissionsPolicyFeature::kNotFound);
  permission_contexts[ContentSettingsType::MIDI_SYSEX] =
      std::make_unique<FakePermissionContext>(
          browser_context, ContentSettingsType::MIDI_SYSEX,
          network::mojom::PermissionsPolicyFeature::kMidiFeature);
  permission_contexts[ContentSettingsType::MIDI] =
      std::make_unique<FakePermissionContextAlwaysAllow>(
          browser_context, ContentSettingsType::MIDI,
          network::mojom::PermissionsPolicyFeature::kMidiFeature);
  permission_contexts[ContentSettingsType::STORAGE_ACCESS] =
      std::make_unique<FakeStorageAccessPermissionContext>(
          browser_context, ContentSettingsType::STORAGE_ACCESS,
          network::mojom::PermissionsPolicyFeature::kStorageAccessAPI);
#if BUILDFLAG(IS_ANDROID) || BUILDFLAG(IS_CHROMEOS)
  permission_contexts[ContentSettingsType::PROTECTED_MEDIA_IDENTIFIER] =
      std::make_unique<FakePermissionContext>(
          browser_context, ContentSettingsType::PROTECTED_MEDIA_IDENTIFIER,
          network::mojom::PermissionsPolicyFeature::kEncryptedMedia);
#endif
  permission_contexts[ContentSettingsType::WEB_APP_INSTALLATION] =
      std::make_unique<FakePermissionContext>(
          browser_context, ContentSettingsType::WEB_APP_INSTALLATION,
          network::mojom::PermissionsPolicyFeature::kWebAppInstallation);
  permission_contexts[ContentSettingsType::WINDOW_MANAGEMENT] =
      std::make_unique<WindowManagementPermissionContext>(browser_context);
  permission_contexts[ContentSettingsType::MEDIASTREAM_CAMERA] =
      std::make_unique<FakePermissionContext>(
          browser_context, ContentSettingsType::MEDIASTREAM_CAMERA,
          network::mojom::PermissionsPolicyFeature::kCamera);
  permission_contexts[ContentSettingsType::MEDIASTREAM_MIC] =
      std::make_unique<FakePermissionContext>(
          browser_context, ContentSettingsType::MEDIASTREAM_MIC,
          network::mojom::PermissionsPolicyFeature::kMicrophone);
  permission_contexts[ContentSettingsType::AUTOMATIC_FULLSCREEN] =
      std::make_unique<FakePermissionContext>(
          browser_context, ContentSettingsType::AUTOMATIC_FULLSCREEN,
          network::mojom::PermissionsPolicyFeature::kFullscreen);
  permission_contexts[ContentSettingsType::LOCAL_NETWORK_ACCESS] =
      std::make_unique<FakePermissionContext>(
          browser_context, ContentSettingsType::LOCAL_NETWORK_ACCESS,
          network::mojom::PermissionsPolicyFeature::kNotFound);
  return permission_contexts;
}

}  // namespace

std::unique_ptr<content::PermissionControllerDelegate>
GetPermissionControllerDelegate(content::BrowserContext* context) {
  return std::make_unique<permissions::PermissionManager>(
      context, CreatePermissionContexts(context));
}

}  // namespace permissions