Implemented initial version of server-issued notification client.

chrome/browser/sync/notifier/server_notifier_thread.cc

+// Copyright (c) 2010 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/sync/notifier/server_notifier_thread.h"
+
+#include <string>
+#include <vector>
+
+#include "base/logging.h"
+#include "chrome/browser/sync/notifier/cache_invalidation_packet_handler.h"
+#include "chrome/browser/sync/notifier/chrome_invalidation_client.h"
+#include "chrome/browser/sync/notifier/chrome_system_resources.h"
+#include "chrome/browser/sync/notifier/invalidation_util.h"
+#include "chrome/common/net/notifier/listener/notification_defines.h"
+#include "google/cacheinvalidation/invalidation-client-impl.h"
+#include "talk/xmpp/jid.h"
+
+namespace sync_notifier {
+
+ServerNotifierThread::ServerNotifierThread(
+    chrome_common_net::NetworkChangeNotifierThread*
+        network_change_notifier_thread)
+    : notifier::MediatorThreadImpl(network_change_notifier_thread) {}
+
+ServerNotifierThread::~ServerNotifierThread() {}
+
+void ServerNotifierThread::ListenForUpdates() {
+  DCHECK_EQ(MessageLoop::current(), parent_message_loop_);
+  worker_message_loop()->PostTask(
+      FROM_HERE,
+      NewRunnableMethod(this,
+                        &ServerNotifierThread::StartInvalidationListener));
+}
+
+void ServerNotifierThread::SubscribeForUpdates(
+    const std::vector<std::string>& subscribed_services_list) {
+  DCHECK_EQ(MessageLoop::current(), parent_message_loop_);
+  worker_message_loop()->PostTask(
+      FROM_HERE,
+      NewRunnableMethod(
+          this, &ServerNotifierThread::RegisterTypesAndSignalSubscribed));
+}
+
+void ServerNotifierThread::Logout() {
+  DCHECK_EQ(MessageLoop::current(), parent_message_loop_);
+  worker_message_loop()->PostTask(
+      FROM_HERE,
+      NewRunnableMethod(this,
+                        &ServerNotifierThread::StopInvalidationListener));
+  MediatorThreadImpl::Logout();
+}
+
+void ServerNotifierThread::SendNotification(
+    const OutgoingNotificationData& data) {
+  DCHECK_EQ(MessageLoop::current(), parent_message_loop_);
+  NOTREACHED() << "Shouldn't send notifications if "
+               << "ServerNotifierThread is used";
+}
+
+void ServerNotifierThread::Invalidate(
+    const invalidation::Invalidation& invalidation,
+    invalidation::Closure* callback) {
+  DCHECK_EQ(MessageLoop::current(), worker_message_loop());
+  CHECK(invalidation::IsCallbackRepeatable(callback));
+  LOG(INFO) << "Invalidate: " << InvalidationToString(invalidation);
+  // Signal notification only for the invalidated types.
+  parent_message_loop_->PostTask(
+      FROM_HERE,
+      NewRunnableMethod(
+          this,
+          &ServerNotifierThread::SignalIncomingNotification));
+  RunAndDeleteClosure(callback);
+  // A real implementation would respond to the invalidation for the
+  // given object (e.g., refetch the invalidated object).
+}
+
+void ServerNotifierThread::InvalidateAll(
+    invalidation::Closure* callback) {
+  DCHECK_EQ(MessageLoop::current(), worker_message_loop());
+  CHECK(invalidation::IsCallbackRepeatable(callback));
+  LOG(INFO) << "InvalidateAll";
+  parent_message_loop_->PostTask(
+      FROM_HERE,
+      NewRunnableMethod(
+          this,
+          &ServerNotifierThread::SignalIncomingNotification));
+  RunAndDeleteClosure(callback);
+}
+
+void ServerNotifierThread::AllRegistrationsLost(
+    invalidation::Closure* callback) {
+  DCHECK_EQ(MessageLoop::current(), worker_message_loop());
+  CHECK(invalidation::IsCallbackRepeatable(callback));
+  LOG(INFO) << "AllRegistrationsLost; reregistering";
+  RegisterTypes();
+  RunAndDeleteClosure(callback);
+}
+
+void ServerNotifierThread::RegistrationLost(
+    const invalidation::ObjectId& object_id,
+    invalidation::Closure* callback) {
+  DCHECK_EQ(MessageLoop::current(), worker_message_loop());
+  CHECK(invalidation::IsCallbackRepeatable(callback));
+  LOG(INFO) << "RegistrationLost; reregistering: "
+            << ObjectIdToString(object_id);
+  RegisterTypes();
+  RunAndDeleteClosure(callback);
+}
+
+void ServerNotifierThread::StartInvalidationListener() {
+  DCHECK_EQ(MessageLoop::current(), worker_message_loop());
+
+  StopInvalidationListener();
+  chrome_invalidation_client_.reset(new ChromeInvalidationClient());
+
+  // TODO(akalin): If we can figure out a unique per-session key that
+  // is preserved by the server, we can use that instead of kAppName.
+  // What this buys us is that we then won't receive any notifications
+  // that were generated by ourselves.
+  const std::string kAppName = "cc_sync_listen_notifications";
+  chrome_invalidation_client_->Start(kAppName, this, xmpp_client());
+}
+
+void ServerNotifierThread::RegisterTypesAndSignalSubscribed() {
+  DCHECK_EQ(MessageLoop::current(), worker_message_loop());
+  RegisterTypes();
+  parent_message_loop_->PostTask(
+      FROM_HERE,
+      NewRunnableMethod(
+          this,
+          &ServerNotifierThread::SignalSubscribed));
+}
+
+void ServerNotifierThread::RegisterTypes() {
+  DCHECK_EQ(MessageLoop::current(), worker_message_loop());
+
+  // TODO(akalin): This is a giant hack!  Make this configurable.  Add
+  // a mapping to/from ModelType.
+  std::vector<std::string> data_types;
+  data_types.push_back("AUTOFILL");
+  data_types.push_back("BOOKMARK");
+  data_types.push_back("EXTENSION");
+  data_types.push_back("PASSWORD");
+  data_types.push_back("THEME");
+  data_types.push_back("TYPED_URL");
+  data_types.push_back("PREFERENCE");
+
+  std::vector<invalidation::ObjectId> object_ids;
+
+  for (std::vector<std::string>::const_iterator it = data_types.begin();
+       it != data_types.end(); ++it) {
+    invalidation::ObjectId object_id;
+    object_id.mutable_name()->set_string_value(*it);
+    object_id.set_source(invalidation::ObjectId::CHROME_SYNC);
+    object_ids.push_back(object_id);
+  }
+
+  for (std::vector<invalidation::ObjectId>::const_iterator it =
+           object_ids.begin(); it != object_ids.end(); ++it) {
+    chrome_invalidation_client_->Register(
+        *it,
+        invalidation::NewPermanentCallback(
+            this, &ServerNotifierThread::RegisterCallback));
+  }
+}
+
+void ServerNotifierThread::RegisterCallback(
+    const invalidation::RegistrationUpdateResult& result) {
+  DCHECK_EQ(MessageLoop::current(), worker_message_loop());
+  // TODO(akalin): Do something meaningful here.
+  LOG(INFO) << "Registered: " << RegistrationUpdateResultToString(result);
+}
+
+void ServerNotifierThread::SignalSubscribed() {
+  DCHECK_EQ(MessageLoop::current(), parent_message_loop_);
+  if (delegate_) {
+    delegate_->OnSubscriptionStateChange(true);
+  }
+}
+
+void ServerNotifierThread::SignalIncomingNotification() {
+  DCHECK_EQ(MessageLoop::current(), parent_message_loop_);
+  if (delegate_) {
+    // TODO(akalin): Fill this in with something meaningful.
+    IncomingNotificationData notification_data;
+    delegate_->OnIncomingNotification(notification_data);
+  }
+}
+
+void ServerNotifierThread::StopInvalidationListener() {
+  DCHECK_EQ(MessageLoop::current(), worker_message_loop());
+
+  if (chrome_invalidation_client_.get()) {
+    // TODO(akalin): Need to do unregisters here?
+    chrome_invalidation_client_->Stop();
+  }
+  chrome_invalidation_client_.reset();
+}
+
+}  // namespace sync_notifier