chrome/browser/tpcd/support/origin_trial_service.h

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

#ifndef CHROME_BROWSER_TPCD_SUPPORT_ORIGIN_TRIAL_SERVICE_H_
#define CHROME_BROWSER_TPCD_SUPPORT_ORIGIN_TRIAL_SERVICE_H_

#include "base/memory/raw_ptr.h"
#include "components/content_settings/core/browser/host_content_settings_map.h"
#include "components/keyed_service/core/keyed_service.h"
#include "content/public/browser/origin_trials_controller_delegate.h"
#include "content/public/browser/web_contents_observer.h"
#include "content/public/browser/web_contents_user_data.h"

namespace tpcd::trial {

// A profile-keyed service that maintains TOP_LEVEL_TPCD_ORIGIN_TRIAL content
// settings based on the state of the associated origin trial for a given
// `origin`.
class OriginTrialService
    : public content::OriginTrialsControllerDelegate::Observer,
      public KeyedService {
 public:
  explicit OriginTrialService(content::BrowserContext* browser_context);
  ~OriginTrialService() override;
  OriginTrialService(const OriginTrialService&) = delete;
  OriginTrialService& operator=(const OriginTrialService&) = delete;

  // KeyedService overrides:
  void Shutdown() override;

  void UpdateTopLevelTrialSettingsForTesting(const url::Origin& origin,
                                             bool match_subdomains,
                                             bool enabled);

 private:
  // Updates `ContentSettingsForOneType::TOP_LEVEL_TPCD_ORIGIN_TRIAL` to reflect
  // the status of the trial for `origin`. If `match_subdomains` is true,
  // a custom scope is used for the content setting to match all subdomains of
  // `origin`.
  void UpdateTopLevelTrialSettings(const url::Origin& origin,
                                   bool match_subdomains,
                                   bool enabled);
  void ClearTopLevelTrialSettings();

  // content::OriginTrialsControllerDelegate::Observer overrides:
  void OnStatusChanged(
      const content::OriginTrialStatusChangeDetails& details) override;
  void OnPersistedTokensCleared() override;
  std::string trial_name() override;

  raw_ptr<content::OriginTrialsControllerDelegate> ot_controller_;
  raw_ptr<content::BrowserContext> browser_context_;
};

}  // namespace tpcd::trial

#endif  // CHROME_BROWSER_TPCD_SUPPORT_ORIGIN_TRIAL_SERVICE_H_