// 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.

#ifndef COMPONENTS_PRIVATE_AI_PHOSPHOR_TOKEN_MANAGER_IMPL_H_
#define COMPONENTS_PRIVATE_AI_PHOSPHOR_TOKEN_MANAGER_IMPL_H_

#include <map>
#include <memory>
#include <optional>
#include <string>
#include <utility>

#include "base/functional/callback.h"
#include "base/memory/raw_ptr.h"
#include "base/memory/weak_ptr.h"
#include "base/sequence_checker.h"
#include "base/time/time.h"
#include "components/private_ai/common/private_ai_logger.h"
#include "components/private_ai/phosphor/token_manager.h"

namespace private_ai::phosphor {

class TokenFetcher;

namespace internal {
class FeatureTokenManager;
}  // namespace internal

// An implementation of TokenManager that populates itself
// using a passed in TokenFetcher pointer from the cache.
class TokenManagerImpl : public TokenManager {
 public:
  explicit TokenManagerImpl(std::unique_ptr<TokenFetcher> fetcher,
                            PrivateAiLogger* logger);
  ~TokenManagerImpl() override;

  // TokenManager implementation.
  void GetAuthToken(GetAuthTokenCallback callback) override;
  void PrefetchAuthTokens() override;
  void GetAuthTokenForProxy(GetAuthTokenCallback callback) override;
  void PrefetchAuthTokensForProxy() override;
  void OnAccountStatusChanged(bool available) override;

 private:
  const int batch_size_;
  const size_t cache_low_water_mark_;

  raw_ptr<PrivateAiLogger> logger_;
  std::unique_ptr<TokenFetcher> fetcher_;

  std::unique_ptr<internal::FeatureTokenManager> terminal_token_manager_;
  std::unique_ptr<internal::FeatureTokenManager> proxy_token_manager_;

  SEQUENCE_CHECKER(sequence_checker_);

  base::WeakPtrFactory<TokenManagerImpl> weak_ptr_factory_{this};
};

}  // namespace private_ai::phosphor

#endif  // COMPONENTS_PRIVATE_AI_PHOSPHOR_TOKEN_MANAGER_IMPL_H_