// Copyright 2025 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/webui/regional_capabilities_internals/regional_capabilities_internals_ui.h"

#include "base/check_deref.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/regional_capabilities/regional_capabilities_service_factory.h"
#include "chrome/browser/search_engine_choice/search_engine_choice_service_factory.h"
#include "chrome/browser/search_engines/template_url_service_factory.h"
#include "components/grit/regional_capabilities_internals_resources.h"
#include "components/grit/regional_capabilities_internals_resources_map.h"
#include "components/regional_capabilities/access/country_access_reason.h"
#include "components/regional_capabilities/regional_capabilities_internals_data_holder.h"
#include "components/regional_capabilities/regional_capabilities_service.h"
#include "components/search_engines/search_engine_choice/search_engine_choice_service.h"
#include "components/search_engines/template_url_service.h"
#include "components/webui/regional_capabilities_internals/constants.h"
#include "content/public/browser/browser_context.h"
#include "content/public/browser/web_ui_data_source.h"
#include "ui/webui/webui_util.h"

using regional_capabilities::CountryAccessKey;
using regional_capabilities::CountryAccessReason;
using regional_capabilities::RegionalCapabilitiesServiceFactory;

namespace {

#if BUILDFLAG(IS_ANDROID)
std::u16string GetExternalChoiceTemplateURL(Profile* profile) {
  TemplateURLService* template_url_service =
      TemplateURLServiceFactory::GetForProfile(profile);
  for (const auto& template_url : template_url_service->GetTemplateURLs()) {
    if (template_url->CreatedByRegulatoryProgram()) {
      return template_url->keyword();
    }
  }

  return u"NONE FOUND";
}
#endif

}  // namespace

RegionalCapabilitiesInternalsUI::RegionalCapabilitiesInternalsUI(
    content::WebUI* web_ui,
    const GURL& url)
    : WebUIController(web_ui) {
  Profile* profile = Profile::FromWebUI(web_ui);
  content::WebUIDataSource* source = content::WebUIDataSource::CreateAndAdd(
      profile,
      regional_capabilities::kChromeUIRegionalCapabilitiesInternalsHost);

  regional_capabilities::InternalsDataHolder internals_data_holder =
      RegionalCapabilitiesServiceFactory::GetForProfile(profile)
          ->GetInternalsData();

  search_engines::SearchEngineChoiceService* search_engine_choice_service =
      search_engines::SearchEngineChoiceServiceFactory::GetForProfile(profile);
  internals_data_holder.SetRecordedEligibility(
      search_engine_choice_service
          ->recorded_profile_load_choice_screen_eligibility());

  for (const auto& [key, value] :
       internals_data_holder.GetRestricted(CountryAccessKey(
           CountryAccessReason::
               kRegionalCapabilitiesInternalsDisplayInDebugUi))) {
    source->AddString(key, value);
  }

#if BUILDFLAG(IS_ANDROID)
  source->AddString(regional_capabilities::kExternalChoiceKeywordKey,
                    GetExternalChoiceTemplateURL(profile));
#endif

  webui::SetupWebUIDataSource(source, kRegionalCapabilitiesInternalsResources,
                              IDR_REGIONAL_CAPABILITIES_INTERNALS_INDEX_HTML);
}