// Copyright 2019 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_bridge.h"

#include <memory>
#include <utility>
#include <variant>

#include "base/android/jni_android.h"
#include "base/android/jni_string.h"
#include "base/android/jni_utils.h"
#include "base/containers/id_map.h"
#include "base/memory/safety_checks.h"
#include "base/no_destructor.h"
#include "components/content_settings/core/common/content_settings.h"
#include "components/content_settings/core/common/content_settings_types.h"
#include "components/content_settings/core/common/content_settings_utils.h"
#include "components/permissions/permission_decision.h"
#include "components/permissions/permission_prompt_decision.h"
#include "components/permissions/resolvers/permission_prompt_options.h"
#include "url/gurl.h"

// Must come after all headers that specialize FromJniType() / ToJniType().
#include "chrome/android/chrome_jni_headers/InstalledWebappBridge_jni.h"

using base::android::ConvertJavaStringToUTF8;
using base::android::ScopedJavaLocalRef;

namespace {

// TODO(crbug.com/391248369): We can just write
//
// using PermissionCallbackJava = base::OnceCallback<
//   void(PermissionDecision setting,
//        bool is_final_decision)>;
//
// and remove ADVANCED_MEMORY_SAFETY_CHECKS() once https://crbug.com/391248369
// is fixed.
class PermissionCallbackWithAMSC
    : public base::OnceCallback<void(PermissionDecision setting,
                                     bool is_final_decision)> {
  ADVANCED_MEMORY_SAFETY_CHECKS();
};

base::IDMap<std::unique_ptr<PermissionCallbackWithAMSC>, int64_t>&
GetPermissionCallbacks() {
  static base::NoDestructor<
      base::IDMap<std::unique_ptr<PermissionCallbackWithAMSC>, int64_t>>
      permission_callbacks;
  return *permission_callbacks;
}

}  // namespace

static void JNI_InstalledWebappBridge_NotifyPermissionsChange(
    JNIEnv* env,
    int64_t j_provider,
    int type_int) {
  ContentSettingsType type = static_cast<ContentSettingsType>(type_int);
  DCHECK(IsKnownEnumValue(type));
  InstalledWebappProvider* provider =
      reinterpret_cast<InstalledWebappProvider*>(j_provider);
  provider->Notify(type);
}

static void JNI_InstalledWebappBridge_RunPermissionCallback(JNIEnv* env,
                                                            int64_t callback_id,
                                                            int setting) {
  DCHECK_LE(setting, static_cast<int>(PermissionDecision::kMaxValue));
  auto* callback = GetPermissionCallbacks().Lookup(callback_id);
  if (!callback) {
    return;
  }
  std::move(*callback).Run(static_cast<PermissionDecision>(setting),
                           /*is_final_decision=*/true);
  GetPermissionCallbacks().Remove(callback_id);
}

InstalledWebappProvider::RuleList
InstalledWebappBridge::GetInstalledWebappPermissions(ContentSettingsType type) {
  JNIEnv* env = base::android::AttachCurrentThread();
  ScopedJavaLocalRef<jobjectArray> j_permissions =
      Java_InstalledWebappBridge_getPermissions(env, static_cast<int>(type));

  InstalledWebappProvider::RuleList rules;
  for (auto j_permission : j_permissions.CreateView(env)) {
    GURL origin(
        Java_InstalledWebappBridge_getOriginFromPermission(env, j_permission));
    ContentSetting setting = IntToContentSetting(
        Java_InstalledWebappBridge_getSettingFromPermission(env, j_permission));
    PermissionSetting permission_setting = setting;
    if (type == ContentSettingsType::GEOLOCATION_WITH_OPTIONS) {
      ContentSetting precise_setting = IntToContentSetting(
          Java_InstalledWebappBridge_getPreciseSettingFromPermission(
              env, j_permission));
      permission_setting = GeolocationSetting{
          content_settings::ToPermissionOption(setting),
          content_settings::ToPermissionOption(precise_setting)};
    }
    rules.emplace_back(origin, permission_setting);
  }

  return rules;
}

void InstalledWebappBridge::SetProviderInstance(
    InstalledWebappProvider* provider) {
  Java_InstalledWebappBridge_setInstalledWebappProvider(
      base::android::AttachCurrentThread(), (int64_t)provider);
}

void InstalledWebappBridge::DecidePermission(ContentSettingsType type,
                                             const GURL& origin_url,
                                             const GURL& last_committed_url,
                                             PermissionCallback callback) {
  JNIEnv* env = base::android::AttachCurrentThread();

  // TODO(crbug.com/470038595): IWAs don't fully support
  // GEOLOCATION_WITH_OPTIONS yet, so we hardcode precise accuracy as the
  // selected value here.
  PromptOptions prompt_options =
      (type == ContentSettingsType::GEOLOCATION_WITH_OPTIONS)
          ? PromptOptions(GeolocationPromptOptions{
                .selected_accuracy = GeolocationAccuracy::kPrecise})
          : std::monostate();

  // Transfers the ownership of the callback to the Java callback. The Java
  // callback is guaranteed to be called unless the user never replies to the
  // dialog, but as the dialog is modal, the only other thing the user can do
  // is quit Chrome which will also free the pointer. The callback pointer will
  // be destroyed in RunPermissionCallback.
  auto callback_with_amsc =
      std::make_unique<PermissionCallbackWithAMSC>(base::BindOnce(
          [](PermissionCallback callback, const PromptOptions& prompt_options,
             PermissionDecision decision, bool is_final_decision) {
            std::move(callback).Run(permissions::PermissionPromptDecision{
                .overall_decision = decision,
                .prompt_options = prompt_options,
                .is_final = is_final_decision});
          },
          std::move(callback), prompt_options));

  int64_t callback_id =
      GetPermissionCallbacks().Add(std::move(callback_with_amsc));

  Java_InstalledWebappBridge_decidePermission(
      env, static_cast<int>(type), origin_url.spec(), last_committed_url.spec(),
      callback_id);
}

DEFINE_JNI(InstalledWebappBridge)