Implement cloud policy invalidations using the invalidation service framework.

chrome/browser/policy/cloud/cloud_policy_manager.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // 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/cloud_policy_manager.h"
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
+#include "base/command_line.h"
 #include "base/logging.h"
 #include "base/prefs/pref_service.h"
+#include "chrome/browser/policy/cloud/cloud_policy_invalidator.h"
+#include "chrome/browser/policy/cloud/cloud_policy_refresh_scheduler.h"
 #include "chrome/browser/policy/cloud/cloud_policy_service.h"
 #include "chrome/browser/policy/policy_bundle.h"
 #include "chrome/browser/policy/policy_map.h"
+#include "chrome/common/chrome_switches.h"
 
 namespace policy {
 
 CloudPolicyManager::CloudPolicyManager(const PolicyNamespaceKey& policy_ns_key,
                                        CloudPolicyStore* cloud_policy_store)
     : core_(policy_ns_key, cloud_policy_store),
       waiting_for_policy_refresh_(false) {
   store()->AddObserver(this);
 
   // If the underlying store is already initialized, publish the loaded
   // policy. Otherwise, request a load now.
   if (store()->is_initialized())
     CheckAndPublishPolicy();
   else
     store()->Load();
 }
 
 CloudPolicyManager::~CloudPolicyManager() {}
 
 void CloudPolicyManager::Shutdown() {
+  invalidator_.reset();
   core_.Disconnect();
   store()->RemoveObserver(this);
   ConfigurationPolicyProvider::Shutdown();
 }
 
 bool CloudPolicyManager::IsInitializationComplete(PolicyDomain domain) const {
   if (domain == POLICY_DOMAIN_CHROME)
     return store()->is_initialized();
   return true;
 }
@@ skipping 15 matching lines @@
 }
 
 void CloudPolicyManager::OnStoreError(CloudPolicyStore* cloud_policy_store) {
   DCHECK_EQ(store(), cloud_policy_store);
   // Publish policy (even though it hasn't changed) in order to signal load
   // complete on the ConfigurationPolicyProvider interface. Technically, this
   // is only required on the first load, but doesn't hurt in any case.
   CheckAndPublishPolicy();
 }
 
+void CloudPolicyManager::InvalidatePolicy() {
+  DCHECK(core()->refresh_scheduler());
+  core()->refresh_scheduler()->RefreshSoon();
+}
+
+void CloudPolicyManager::OnInvalidatorStateChanged(bool invalidations_enabled) {
+  DCHECK(core()->refresh_scheduler());
+  // Enable this code once method is added to CloudPolicyRefreshScheduler.
+  // core()->refresh_scheduler()->SetInvalidationServiceAvailability(
+  //     invalidations_enabled);
+}
+
 void CloudPolicyManager::CheckAndPublishPolicy() {
   if (IsInitializationComplete(POLICY_DOMAIN_CHROME) &&
       !waiting_for_policy_refresh_) {
     UpdatePolicy(CreatePolicyBundle());
   }
 }
 
+void CloudPolicyManager::CreateInvalidator(
+    invalidation::InvalidationService* service,
+    bool ignore_switch) {
+  DCHECK(core()->refresh_scheduler());
+#if defined(OS_ANDROID) // Allow invalidations on Android regardless of switch.
+  ignore_switch = true;
+#endif
+  if (ignore_switch || CommandLine::ForCurrentProcess()->HasSwitch(
+      switches::kEnableCloudPolicyPush)) {
+    invalidator_.reset(new CloudPolicyInvalidator(
+        service,
+        client(),
+        store(),
+        base::MessageLoop::current()->message_loop_proxy(),
+        this /* invalidation_handler */));
+  }
+}
+
+void CloudPolicyManager::UnregisterInvalidator() {
+  if (invalidator_.get()) {

[Joao da Silva, 2013/07/24 15:34:07]

.get() not needed to test scoped_ptrs

[Steve Condie, 2013/07/25 01:18:08]

This function is now removed anyway.

+    invalidator_->Unregister();
+    invalidator_.reset();
+  }
+}
+
 scoped_ptr<PolicyBundle> CloudPolicyManager::CreatePolicyBundle() {
   scoped_ptr<PolicyBundle> bundle(new PolicyBundle);
   bundle->Get(PolicyNamespace(POLICY_DOMAIN_CHROME, std::string()))
       .CopyFrom(store()->policy_map());
   return bundle.Pass();
 }
 
 void CloudPolicyManager::OnRefreshComplete(bool success) {
   waiting_for_policy_refresh_ = false;
   CheckAndPublishPolicy();
 }
 
 }  // namespace policy