Consolidate data storage and notifications in the cloud policy subsystem

chrome/browser/policy/user_policy_token_cache.cc

Index: chrome/browser/policy/user_policy_token_cache.cc
diff --git a/chrome/browser/policy/user_policy_token_cache.cc b/chrome/browser/policy/user_policy_token_cache.cc
index adba4a4e89c75f5cc9fb03081d753a99fba66216..a5d0a7036c6a811a8627dca8351e05f7549efd3c 100644
--- a/chrome/browser/policy/user_policy_token_cache.cc
+++ b/chrome/browser/policy/user_policy_token_cache.cc
@@ -5,19 +5,20 @@
 #include "chrome/browser/policy/user_policy_token_cache.h"
 
 #include "base/file_util.h"
+#include "chrome/browser/policy/proto/device_management_local.pb.h"
 #include "content/browser/browser_thread.h"
 
 namespace policy {
 
 namespace em = enterprise_management;
 
-UserPolicyTokenCache::Delegate::~Delegate() {}
-
 UserPolicyTokenCache::UserPolicyTokenCache(
-    const base::WeakPtr<Delegate>& delegate,
+    const base::WeakPtr<CloudPolicyDataStore>& data_store,
     const FilePath& cache_file)
-    : delegate_(delegate),
-      cache_file_(cache_file) {}
+    : data_store_(data_store),
+      cache_file_(cache_file) {
+  data_store_->AddObserver(this);
+}
 
 void UserPolicyTokenCache::Load() {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
@@ -38,6 +39,8 @@ void UserPolicyTokenCache::Store(const std::string& token,
 }
 
 UserPolicyTokenCache::~UserPolicyTokenCache() {
+  if (data_store_)
+    data_store_->RemoveObserver(this);
 }
 
 void UserPolicyTokenCache::LoadOnFileThread() {
@@ -66,8 +69,11 @@ void UserPolicyTokenCache::LoadOnFileThread() {
 void UserPolicyTokenCache::NotifyOnUIThread(const std::string& token,
                                             const std::string& device_id) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
-  if (delegate_.get())
-    delegate_->OnTokenCacheLoaded(token, device_id);
+  if (data_store_) {
+    data_store_->set_device_id(device_id);
+    data_store_->SetDeviceToken(token, true);
+    data_store_->NotifyDeviceTokenChanged();
+  }
 }
 
 void UserPolicyTokenCache::StoreOnFileThread(const std::string& token,
@@ -93,4 +99,19 @@ void UserPolicyTokenCache::StoreOnFileThread(const std::string& token,
   file_util::WriteFile(cache_file_, data.c_str(), data.length());
 }
 
+void UserPolicyTokenCache::OnDeviceTokenChanged() {
+  if (!data_store_->device_token().empty() &&
+      !data_store_->device_id().empty()) {
+    // This will also happen after loading the cache.
+    Store(data_store_->device_token(), data_store_->device_id());
+  }
+}
+
+void UserPolicyTokenCache::OnCredentialsChanged() {
+}
+
+void UserPolicyTokenCache::OnDataStoreGoingAway() {
+  data_store_->RemoveObserver(this);
+}
+
 }  // namespace policy