chrome/browser/policy/cloud/user_cloud_policy_invalidator.cc

// Copyright 2013 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/policy/cloud/user_cloud_policy_invalidator.h"

#include <memory>

#include "base/functional/bind.h"
#include "base/task/single_thread_task_runner.h"
#include "base/time/default_clock.h"
#include "build/build_config.h"
#include "chrome/browser/invalidation/profile_invalidation_provider_factory.h"
#include "chrome/browser/policy/policy_util.h"
#include "chrome/browser/profiles/profile.h"
#include "components/invalidation/profile_invalidation_provider.h"
#include "components/policy/core/common/cloud/cloud_policy_manager.h"

namespace {

invalidation::ProfileInvalidationProvider* GetInvalidationProvider(
    Profile* profile) {
  return invalidation::ProfileInvalidationProviderFactory::GetForProfile(
      profile);
}

}  // namespace

namespace policy {

UserCloudPolicyInvalidator::UserCloudPolicyInvalidator(
    Profile* profile,
    CloudPolicyManager* policy_manager)
    : CloudPolicyInvalidator(PolicyInvalidationScope::kUser,
                             policy_manager->core(),
                             base::SingleThreadTaskRunner::GetCurrentDefault(),
                             base::DefaultClock::GetInstance(),
                             0 /* highest_handled_invalidation_version */) {
  DCHECK(profile);

  // Register for notification that profile creation is complete. The
  // invalidator must not be initialized before then because the invalidation
  // service cannot be started because it depends on components initialized
  // after this object is instantiated.
  // TODO(stepco): Delayed initialization can be removed once the request
  // context can be accessed during profile-keyed service creation. Tracked by
  // bug 286209.
  // TODO(crbug.com/40113187): Investigate if this is still required.
  profile_observation_.Observe(profile);
}

UserCloudPolicyInvalidator::~UserCloudPolicyInvalidator() = default;

void UserCloudPolicyInvalidator::Shutdown() {
  profile_observation_.Reset();
  CloudPolicyInvalidator::Shutdown();
}

void UserCloudPolicyInvalidator::OnProfileInitializationComplete(
    Profile* profile) {
  DCHECK(profile_observation_.IsObservingSource(profile));
  profile_observation_.Reset();

  // Initialize now that profile creation is complete and the invalidation
  // service can safely be initialized.
  invalidation::ProfileInvalidationProvider* invalidation_provider =
      GetInvalidationProvider(profile);
  if (!invalidation_provider) {
    return;
  }

  Initialize(invalidation_provider->GetInvalidationServiceOrListener(
      GetPolicyInvalidationProjectNumber(PolicyInvalidationScope::kUser)));
}

}  // namespace policy