chromium/components/autofill_prediction_improvements/core/browser/autofill_prediction_improvements_filling_engine_impl.cc

// 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 "components/autofill_prediction_improvements/core/browser/autofill_prediction_improvements_filling_engine_impl.h"

#include <optional>
#include <vector>

#include "base/functional/callback.h"
#include "base/memory/raw_ptr.h"
#include "base/strings/utf_string_conversions.h"
#include "base/types/expected.h"
#include "components/autofill/core/common/form_data.h"
#include "components/optimization_guide/core/optimization_guide_proto_util.h"
#include "components/optimization_guide/core/optimization_guide_util.h"
#include "components/optimization_guide/proto/features/common_quality_data.pb.h"
#include "components/optimization_guide/proto/features/forms_predictions.pb.h"
#include "components/user_annotations/user_annotations_service.h"
#include "components/user_annotations/user_annotations_types.h"

namespace autofill_prediction_improvements {

AutofillPredictionImprovementsFillingEngineImpl::
    AutofillPredictionImprovementsFillingEngineImpl(
        optimization_guide::OptimizationGuideModelExecutor* model_executor,
        user_annotations::UserAnnotationsService* user_annotations_service)
    : … { … }
AutofillPredictionImprovementsFillingEngineImpl::
    ~AutofillPredictionImprovementsFillingEngineImpl() = default;

void AutofillPredictionImprovementsFillingEngineImpl::GetPredictions(
    autofill::FormData form_data,
    optimization_guide::proto::AXTreeUpdate ax_tree_update,
    PredictionsReceivedCallback callback) { … }

void AutofillPredictionImprovementsFillingEngineImpl::
    OnUserAnnotationsRetrieved(
        autofill::FormData form_data,
        optimization_guide::proto::AXTreeUpdate ax_tree_update,
        PredictionsReceivedCallback callback,
        std::vector<optimization_guide::proto::UserAnnotationsEntry>
            user_annotations) { … }

void AutofillPredictionImprovementsFillingEngineImpl::OnModelExecuted(
    autofill::FormData form_data,
    PredictionsReceivedCallback callback,
    optimization_guide::OptimizationGuideModelExecutionResult execution_result,
    std::unique_ptr<optimization_guide::ModelQualityLogEntry> log_entry) { … }

// static
void AutofillPredictionImprovementsFillingEngineImpl::FillFormDataWithResponse(
    autofill::FormData& form_data,
    const optimization_guide::proto::FilledFormData& form_data_proto) { … }

}  // namespace autofill_prediction_improvements