manual_testing_import.cc | Explore in Territory

// 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 "components/autofill/core/browser/manual_testing_import.h"

#include <string>
#include <string_view>

#include "base/check.h"
#include "base/command_line.h"
#include "base/containers/fixed_flat_map.h"
#include "base/files/file_path.h"
#include "base/files/file_util.h"
#include "base/functional/bind.h"
#include "base/json/json_reader.h"
#include "base/location.h"
#include "base/strings/utf_string_conversions.h"
#include "base/task/task_traits.h"
#include "base/task/thread_pool.h"
#include "components/autofill/core/browser/address_data_manager.h"
#include "components/autofill/core/browser/autofill_type.h"
#include "components/autofill/core/browser/data_model/autofill_structured_address_component.h"
#include "components/autofill/core/browser/field_type_utils.h"
#include "components/autofill/core/browser/field_types.h"
#include "components/autofill/core/browser/payments_data_manager.h"

namespace autofill {

namespace {

// Util struct for storing the list of profiles and credit cards to be imported.
// If any of `profiles` or `credit_cards` are std::nullopt, then the data used
// for import is malformed, and this will cause a crash.
// When any of `profiles` or `credit_cards` is empty, it means that the JSON
// file used for import did not include the corresponding key. It'll be treated
// as valid but won't be imported so that existing data in the PDM isn't
// cleared without replacement.
struct AutofillProfilesAndCreditCards { … };

constexpr std::string_view kKeyProfiles = …;
constexpr std::string_view kKeyCreditCards = …;
constexpr std::string_view kKeyRecordType = …;
constexpr std::string_view kKeyNickname = …;
constexpr auto kRecordTypeMapping = …;
constexpr std::string_view kKeyInitialCreatorId = …;

// Checks if the `profile` is changed by `FinalizeAfterImport()`. See
// documentation of `AutofillProfilesFromJSON()` for a rationale.
// The return value of `FinalizeAfterImport()` doesn't suffice to check that,
// since structured address and name components are updated separately.
bool IsFullyStructuredProfile(const AutofillProfile& profile) { … }

// Extracts the `kKeyRecordType` value of the `dict` and translates it into an
// AutofillProfile::RecordType. If no value is present,
// RecordType::kLocalOrSyncable is returned. If a record type with invalid value
// is specified, an error message is logged and std::nullopt is returned.
std::optional<AutofillProfile::RecordType> GetRecordTypeFromDict(
    const base::Value::Dict& dict) { … }

// Given a `dict` of "field-type" : "value" mappings, constructs an
// AutofillProfile where each "field-type"  is set to the provided "value".
// All verification statuses are set to `kObserved`. Setting them to
// `kUserVerified` is problematic, since the data model expects that only root
// level (= setting-visible) nodes are user verified.
// If a field type cannot be mapped, or if the resulting profile is not
// `IsFullyStructuredProfile()`, std::nullopt is returned.
std::optional<AutofillProfile> MakeProfile(const base::Value::Dict& dict) { … }

std::optional<CreditCard> MakeCard(const base::Value::Dict& dict) { … }

// Removes all AutofillProfiles from the `pdm`. Since `PDM::RemoveByGUID()`
// invalidates the pointers returned by `PDM::GetProfiles()`, this is done by
// collecting all GUIDs to remove first.
void RemoveAllExistingProfiles(PersonalDataManager& pdm) { … }

// Sets all of the `pdm`'s profiles or credit cards to `profiles` or
// `credit_cards`, if the `pdm` still exists.
void SetData(
    base::WeakPtr<PersonalDataManager> pdm,
    std::optional<AutofillProfilesAndCreditCards> profiles_or_credit_cards) { … }

// Converts all `entries of `json_array` to a vector of Ts using
// `to_data_model`. In case any conversion fails, nullopt is returned.
template <class T>
std::optional<std::vector<T>> DataModelsFromJSON(
    const base::Value::List* const json_array,
    base::RepeatingCallback<std::optional<T>(const base::Value::Dict&)>
        to_data_model) { … }

// Parses AutofillProfiles from the JSON `content` string.
// If parsing fails the error is logged and std::nullopt is returned.
std::optional<AutofillProfilesAndCreditCards> LoadDataFromJSONContent(
    const std::string& file_content) { … }

std::optional<AutofillProfilesAndCreditCards> LoadDataFromFile(
    base::FilePath file) { … }

}  // namespace

std::optional<std::vector<AutofillProfile>> LoadProfilesFromFile(
    base::FilePath file) { … }

std::optional<std::vector<CreditCard>> LoadCreditCardsFromFile(
    base::FilePath file) { … }

std::optional<std::vector<AutofillProfile>> AutofillProfilesFromJSON(
    const base::Value::List* const profiles_json) { … }

std::optional<std::vector<CreditCard>> CreditCardsFromJSON(
    const base::Value::List* const cards_json) { … }

void MaybeImportDataForManualTesting(base::WeakPtr<PersonalDataManager> pdm) { … }

}  // namespace autofill
chromium/components/autofill/core/browser/manual_testing_import.cc
