chrome/browser/ui/webui/ntp_microsoft_auth/ntp_microsoft_auth_page_handler.cc

// 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/ntp_microsoft_auth/ntp_microsoft_auth_page_handler.h"

#include <utility>

#include "base/metrics/histogram_functions.h"
#include "chrome/browser/new_tab_page/microsoft_auth/microsoft_auth_service.h"
#include "chrome/browser/new_tab_page/microsoft_auth/microsoft_auth_service_factory.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/ui/webui/ntp_microsoft_auth/ntp_microsoft_auth_untrusted_ui.mojom.h"
#include "mojo/public/cpp/bindings/pending_receiver.h"

MicrosoftAuthUntrustedPageHandler::MicrosoftAuthUntrustedPageHandler(
    mojo::PendingReceiver<
        new_tab_page::mojom::MicrosoftAuthUntrustedPageHandler> handler,
    mojo::PendingRemote<new_tab_page::mojom::MicrosoftAuthUntrustedDocument>
        document,
    Profile* profile)
    : handler_(this, std::move(handler)),
      document_(std::move(document)),
      auth_service_(MicrosoftAuthServiceFactory::GetForProfile(profile)) {
  CHECK(auth_service_);
  microsoft_auth_service_observation_.Observe(auth_service_);
}

MicrosoftAuthUntrustedPageHandler::~MicrosoftAuthUntrustedPageHandler() =
    default;

void MicrosoftAuthUntrustedPageHandler::ClearAuthData() {
  auth_service_->ClearAuthData();
}

// TODO(crbug.com/396144770): Update logic to something that simply calls
// `OnAuthStateUpdated` after merges for M134 are complete, instead of the
// renderer calling this and deciding for itself based on the response.
void MicrosoftAuthUntrustedPageHandler::GetAuthState(
    GetAuthStateCallback callback) {
  auto state = auth_service_->GetAuthState();
  if (state == new_tab_page::mojom::AuthState::kNone) {
    base::UmaHistogramEnumeration("NewTabPage.MicrosoftAuth.AuthStarted",
                                  new_tab_page::mojom::AuthType::kSilent);
  }
  std::move(callback).Run(std::move(state));
}

void MicrosoftAuthUntrustedPageHandler::SetAccessToken(
    new_tab_page::mojom::AccessTokenPtr token) {
  auth_service_->SetAccessToken(std::move(token));
}

void MicrosoftAuthUntrustedPageHandler::SetAuthStateError() {
  auth_service_->SetAuthStateError();
}

void MicrosoftAuthUntrustedPageHandler::OnAuthStateUpdated() {
  new_tab_page::mojom::AuthState auth_state = auth_service_->GetAuthState();
  if (auth_state == new_tab_page::mojom::AuthState::kNone) {
    document_->AcquireTokenSilent();
    base::UmaHistogramEnumeration("NewTabPage.MicrosoftAuth.AuthStarted",
                                  new_tab_page::mojom::AuthType::kSilent);
  }
}