// Copyright 2023 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/ui/android/hats/survey_config_android.h"

#include <optional>

#include "base/android/jni_android.h"
#include "base/android/jni_array.h"
#include "base/android/jni_string.h"
#include "chrome/browser/profiles/profile.h"

// Must come after all headers that specialize FromJniType() / ToJniType().
#include "chrome/browser/ui/android/hats/jni_headers/SurveyConfig_jni.h"

using base::android::AttachCurrentThread;
using base::android::ConvertUTF8ToJavaString;
using base::android::JavaRef;
using base::android::ScopedJavaLocalRef;

namespace hats {

SurveyConfigHolder::SurveyConfigHolder(JNIEnv* env,
                                       const JavaRef<jobject>& obj,
                                       Profile* profile) {
  jobj_.Reset(env, obj);
  GetActiveSurveyConfigs(survey_configs_by_triggers_);
  InitJavaHolder(profile);
}

SurveyConfigHolder::~SurveyConfigHolder() = default;

// Initialize the holder on the java side.
void SurveyConfigHolder::InitJavaHolder(Profile* profile) {
  JNIEnv* env = AttachCurrentThread();

  for (auto entry : survey_configs_by_triggers_) {
    auto survey_config = entry.second;

    ScopedJavaLocalRef<jstring> jtrigger =
        ConvertUTF8ToJavaString(env, survey_config.trigger);
    ScopedJavaLocalRef<jstring> jtrigger_id =
        ConvertUTF8ToJavaString(env, survey_config.trigger_id);
    ScopedJavaLocalRef<jobjectArray> jpsd_bits_data_fields =
        base::android::ToJavaArrayOfStrings(
            env, survey_config.product_specific_bits_data_fields);
    ScopedJavaLocalRef<jobjectArray> jpsd_string_data_fields =
        base::android::ToJavaArrayOfStrings(
            env, survey_config.product_specific_string_data_fields);
    bool juser_prompted = survey_config.user_prompted;
    double jprobability = survey_config.probability;
    int32_t requested_browser_type = survey_config.requested_browser_type;
    int32_t profile_age_requirement =
        static_cast<int32_t>(survey_config.profile_age_requirement);

    Java_SurveyConfig_addActiveSurveyConfigToHolder(
        env, jobj_, jtrigger, jtrigger_id, jprobability, juser_prompted,
        jpsd_bits_data_fields, jpsd_string_data_fields, requested_browser_type,
        profile_age_requirement);
  }
}

void SurveyConfigHolder::Destroy(JNIEnv* env) {
  survey_configs_by_triggers_.clear();
  jobj_.Reset();
}

// static
static int64_t JNI_SurveyConfig_InitHolder(
    JNIEnv* env,
    const base::android::JavaRef<jobject>& caller,
    const JavaRef<jobject>& profile) {
  SurveyConfigHolder* holder =
      new SurveyConfigHolder(env, caller, Profile::FromJavaObject(profile));
  return reinterpret_cast<intptr_t>(holder);
}

}  // namespace hats

DEFINE_JNI(SurveyConfig)