// Copyright 2024 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/test/base/ash/interactive/cellular/cellular_util.h"

#include "base/check.h"
#include "base/strings/stringprintf.h"
#include "base/values.h"
#include "chrome/test/base/ash/interactive/network/shill_service_util.h"
#include "chromeos/ash/components/dbus/hermes/hermes_euicc_client.h"
#include "chromeos/ash/components/dbus/hermes/hermes_profile_client.h"
#include "chromeos/components/onc/onc_utils.h"
#include "components/onc/onc_constants.h"
#include "dbus/object_path.h"
#include "third_party/cros_system_api/dbus/hermes/dbus-constants.h"

namespace ash {
namespace {

// Template for a cellular network policy. The activation code is arbitrary
// since, if an ICCID is provided, the device will assume it has already
// installed an eSIM profile and should not try again.
const char kCellularPolicyPattern[] =
    R"({
      "GUID": "%s",
      "Type": "Cellular",
      "Name": "cellular_policy",
      "Cellular": {
        "SMDPAddress": "LPA:1$SMDP.GSMA.COM$123",
        "ICCID": "%s"
      }
    })";

}  // namespace

EuiccInfo::EuiccInfo(unsigned int id)
    : path_(base::StringPrintf("/hermes/euicc%u", id)),
      eid_(base::StringPrintf("%032u", id)) {}

EuiccInfo::~EuiccInfo() = default;

SimInfo::SimInfo(unsigned int id)
    : guid_(base::StringPrintf("esim_guid%018u", id)),
      profile_path_(base::StringPrintf("/profile/path%u", id)),
      iccid_(base::StringPrintf("%018u", id)),
      // The name of the profile is the prefix of the nickname to allow tests to
      // search for the name and be able to find nodes that have the name or the
      // nickname.
      // TODO(crbug.com/376485899): Update these to be unique and not have a
      // shared prefix.
      name_(base::StringPrintf("Test Profile %u", id)),
      nickname_(base::StringPrintf("Test Profile %u Nickname", id)),
      service_provider_(base::StringPrintf("Service Provider %u", id)),
      // TODO(crbug.com/339260115): Align Shill and Hermes fakes so that the
      // service path and GUID logic is consistent for cellular networks.
      service_path_(base::StringPrintf("service_path_for_%s", iccid_.c_str())) {
  auto* hermes_euicc_client = HermesEuiccClient::Get()->GetTestInterface();
  CHECK(hermes_euicc_client);
  activation_code_ = hermes_euicc_client->GenerateFakeActivationCode();
}

SimInfo::~SimInfo() = default;

void SimInfo::Connect() const {
  ConnectShillService(service_path_);
}

void SimInfo::Disconnect() const {
  DisconnectShillService(service_path_);
}

void ConfigureEsimProfile(const EuiccInfo& euicc_info,
                          const SimInfo& esim_info,
                          bool connected) {
  auto* hermes_euicc_client = HermesEuiccClient::Get()->GetTestInterface();
  CHECK(hermes_euicc_client);

  // Add an inactive profile.
  hermes_euicc_client->AddCarrierProfile(
      dbus::ObjectPath(esim_info.profile_path()),
      dbus::ObjectPath(euicc_info.path()), esim_info.iccid(), esim_info.name(),
      esim_info.nickname(), esim_info.service_provider(),
      esim_info.activation_code(),
      /*network_service_path=*/esim_info.service_path(),
      /*state=*/hermes::profile::State::kInactive,
      /*profile_class=*/hermes::profile::ProfileClass::kOperational,
      /*add_carrier_profile_behavior=*/
      HermesEuiccClient::TestInterface::AddCarrierProfileBehavior::
          kAddProfileWithService);

  if (connected) {
    // Explicitly enable the newly added profile.
    auto* hermes_profile_client = HermesProfileClient::Get();
    CHECK(hermes_profile_client);
    hermes_profile_client->EnableCarrierProfile(
        dbus::ObjectPath(esim_info.profile_path()), base::DoNothing());

    esim_info.Connect();
  }
}

base::DictValue GenerateCellularPolicy(const SimInfo& info,
                                       bool allow_apn_modification,
                                       bool allow_roaming) {
  auto network_config =
      chromeos::onc::ReadDictionaryFromJson(base::StringPrintf(
          kCellularPolicyPattern, info.guid().c_str(), info.iccid().c_str()));
  CHECK(network_config.has_value());
  auto* cellular = network_config->FindDict(onc::network_type::kCellular);
  CHECK(cellular);
  if (allow_apn_modification) {
    base::ListValue recommended;
    recommended.Append(base::Value(onc::cellular::kCustomAPNList));
    cellular->Set(onc::kRecommended, std::move(recommended));
  } else {
    cellular->Set(::onc::cellular::kCustomAPNList, base::ListValue());
  }
  cellular->Set(::onc::cellular::kAllowRoaming, allow_roaming);
  return std::move(*network_config);
}

}  // namespace ash