chrome/browser/safe_browsing/chrome_ping_manager_factory.h

// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CHROME_BROWSER_SAFE_BROWSING_CHROME_PING_MANAGER_FACTORY_H_
#define CHROME_BROWSER_SAFE_BROWSING_CHROME_PING_MANAGER_FACTORY_H_

#include "base/no_destructor.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/profiles/profile_keyed_service_factory.h"
#include "components/safe_browsing/core/browser/ping_manager.h"

namespace safe_browsing {

// Factory for creating the KeyedService PingManager for chrome. It returns null
// for incognito mode.
class ChromePingManagerFactory : public ProfileKeyedServiceFactory {
 public:
  static ChromePingManagerFactory* GetInstance();
  static PingManager* GetForBrowserContext(content::BrowserContext* context);

 private:
  friend class base::NoDestructor<ChromePingManagerFactory>;
  friend class ChromePingManagerFactoryTest;

  ChromePingManagerFactory();
  ~ChromePingManagerFactory() override;

  // BrowserContextKeyedServiceFactory override:
  std::unique_ptr<KeyedService> BuildServiceInstanceForBrowserContext(
      content::BrowserContext* context) const override;
  bool ServiceIsCreatedWithBrowserContext() const override;
  bool ServiceIsNULLWhileTesting() const override;

  static bool ShouldFetchAccessTokenForReport(Profile* profile);
  static bool ShouldSendPersistedReport(Profile* profile);
};

// Used only for tests. By default, the factory returns null for tests . To
// override it, create an object of this type and keep it in scope for as long
// as the override should exist. The constructor will set the override, and the
// destructor will clear it.
class ChromePingManagerAllowerForTesting {
 public:
  ChromePingManagerAllowerForTesting();
  ChromePingManagerAllowerForTesting(
      const ChromePingManagerAllowerForTesting&) = delete;
  ChromePingManagerAllowerForTesting& operator=(
      const ChromePingManagerAllowerForTesting&) = delete;
  ~ChromePingManagerAllowerForTesting();
};

}  // namespace safe_browsing
#endif  // CHROME_BROWSER_SAFE_BROWSING_CHROME_PING_MANAGER_FACTORY_H_