// 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/autofill_ml_internals/autofill_ml_internals_page_handler.h"

#include "base/memory/raw_ptr.h"
#include "components/autofill/core/browser/ml_model/logging/autofill_ml_internals.mojom.h"
#include "components/autofill/core/browser/ml_model/logging/ml_log_router.h"

AutofillMlInternalsPageHandlerImpl::AutofillMlInternalsPageHandlerImpl(
    mojo::PendingReceiver<autofill_ml_internals::mojom::PageHandler> receiver,
    mojo::PendingRemote<autofill_ml_internals::mojom::Page> page,
    autofill::MlLogRouter* log_router)
    : receiver_(this, std::move(receiver)), page_(std::move(page)) {
  if (log_router) {
    log_router_observation_.Observe(log_router);
  }
}

void AutofillMlInternalsPageHandlerImpl::ProcessLog(
    const autofill_ml_internals::mojom::MlPredictionLog& log) {
  page_->OnLogAdded(log.Clone());
}

AutofillMlInternalsPageHandlerImpl::~AutofillMlInternalsPageHandlerImpl() =
    default;