// Copyright 2013 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/webui/flags/pref_service_flags_storage.h"

#include "base/logging.h"
#include "base/values.h"
#include "build/build_config.h"
#include "components/pref_registry/pref_registry_syncable.h"
#include "components/prefs/pref_registry_simple.h"
#include "components/prefs/pref_service.h"
#include "components/prefs/scoped_user_pref_update.h"
#include "components/webui/flags/flags_ui_pref_names.h"

namespace flags_ui {

PrefServiceFlagsStorage::PrefServiceFlagsStorage(PrefService* prefs)
    : prefs_(prefs) {}

PrefServiceFlagsStorage::~PrefServiceFlagsStorage() = default;

std::set<std::string> PrefServiceFlagsStorage::GetFlags() const {
  const base::ListValue& enabled_experiments =
      prefs_->GetList(prefs::kAboutFlagsEntries);
  std::set<std::string> flags;
  for (const auto& entry : enabled_experiments) {
    if (!entry.is_string()) {
      LOG(WARNING) << "Invalid entry in " << prefs::kAboutFlagsEntries;
      continue;
    }
    flags.insert(entry.GetString());
  }
  return flags;
}

bool PrefServiceFlagsStorage::SetFlags(const std::set<std::string>& flags) {
  base::ListValue experiments_list;
  for (const auto& flag : flags) {
    experiments_list.Append(flag);
  }
  prefs_->SetList(prefs::kAboutFlagsEntries, std::move(experiments_list));

  return true;
}

std::string PrefServiceFlagsStorage::GetOriginListFlag(
    const std::string& internal_entry_name) const {
  const base::DictValue& origin_lists =
      prefs_->GetDict(prefs::kAboutFlagsOriginLists);
  if (const std::string* s = origin_lists.FindString(internal_entry_name)) {
    return *s;
  }
  return std::string();
}

void PrefServiceFlagsStorage::SetOriginListFlag(
    const std::string& internal_entry_name,
    const std::string& origin_list_value) {
  ScopedDictPrefUpdate update(prefs_, prefs::kAboutFlagsOriginLists);
  update->SetByDottedPath(internal_entry_name, origin_list_value);
}

std::string PrefServiceFlagsStorage::GetStringFlag(
    const std::string& internal_entry_name) const {
  return GetOriginListFlag(internal_entry_name);
}

void PrefServiceFlagsStorage::SetStringFlag(
    const std::string& internal_entry_name,
    const std::string& string_value) {
  SetOriginListFlag(internal_entry_name, string_value);
}

base::DictValue PrefServiceFlagsStorage::GetCustomizedFlags() const {
  return prefs_->GetDict(prefs::kAboutFlagsOriginLists).Clone();
}

void PrefServiceFlagsStorage::SetCustomizedFlags(
    const base::DictValue& customized_flags) {
  prefs_->SetDict(prefs::kAboutFlagsOriginLists, customized_flags.Clone());
}

void PrefServiceFlagsStorage::CommitPendingWrites() {
  prefs_->CommitPendingWrite();
}

// static
void PrefServiceFlagsStorage::RegisterPrefs(PrefRegistrySimple* registry) {
  registry->RegisterListPref(prefs::kAboutFlagsEntries);
  registry->RegisterDictionaryPref(prefs::kAboutFlagsOriginLists);
}

#if BUILDFLAG(IS_CHROMEOS)
// static
void PrefServiceFlagsStorage::RegisterProfilePrefs(
    user_prefs::PrefRegistrySyncable* registry) {
  registry->RegisterListPref(prefs::kAboutFlagsEntries);
  registry->RegisterDictionaryPref(prefs::kAboutFlagsOriginLists);
}
#endif  // BUILDFLAG(IS_CHROMEOS)

}  // namespace flags_ui