chromium/chrome/browser/ui/ash/birch/refresh_token_waiter.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 "chrome/browser/ui/ash/birch/refresh_token_waiter.h"

#include "base/check.h"
#include "base/functional/callback.h"
#include "base/memory/raw_ptr.h"
#include "base/scoped_observation.h"
#include "chrome/browser/signin/identity_manager_factory.h"
#include "components/signin/public/base/consent_level.h"
#include "components/signin/public/identity_manager/identity_manager.h"

class Profile;

namespace ash {

RefreshTokenWaiter::RefreshTokenWaiter(Profile* profile)
    : identity_manager_(IdentityManagerFactory::GetForProfile(profile)) {
  CHECK(identity_manager_);
}

RefreshTokenWaiter::~RefreshTokenWaiter() = default;

void RefreshTokenWaiter::Wait(base::OnceClosure callback) {
  CHECK(callback_.is_null()) << "Multiple waits are not supported.";

  if (identity_manager_->HasPrimaryAccountWithRefreshToken(
          signin::ConsentLevel::kSignin)) {
    // Refresh tokens already loaded, proceed.
    std::move(callback).Run();
    return;
  }
  // Store the callback for later and observe the identity manager for the
  // refresh token load.
  callback_ = std::move(callback);
  identity_manager_observation_.Observe(identity_manager_);
}

void RefreshTokenWaiter::OnRefreshTokenUpdatedForAccount(
    const CoreAccountInfo& account_info) {
  const CoreAccountInfo& primary_account_info =
      identity_manager_->GetPrimaryAccountInfo(signin::ConsentLevel::kSignin);
  if (account_info != primary_account_info) {
    return;
  }
  identity_manager_observation_.Reset();
  // Refresh tokens are loaded, proceed.
  std::move(callback_).Run();
}

}  // namespace ash