Add the Copresence API.

chrome/browser/extensions/api/copresence/copresence_translations.cc

+// Copyright 2014 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/extensions/api/copresence/copresence_translations.h"
+
+#include "chrome/common/extensions/api/copresence.h"
+#include "components/copresence/proto/data.pb.h"
+#include "components/copresence/proto/enums.pb.h"
+#include "components/copresence/proto/rpcs.pb.h"
+
+namespace {
+
+const int kDefaultTimeToLiveMs = 5 * 60 * 1000;    // 5 minutes.
+const int kMaxTimeToLiveMs = 24 * 60 * 60 * 1000;  // 24 hours.
+
+// Checks and returns the ttl provided by the user. If invalid, returns -1.
+int SanitizeTtl(int* user_ttl) {
+  return !user_ttl
+             ? kDefaultTimeToLiveMs
+             : *user_ttl <= 0 || *user_ttl > kMaxTimeToLiveMs ? -1 : *user_ttl;
+}
+
+copresence::BroadcastScanConfiguration TranslateStrategy(
+    extensions::api::copresence::Strategy* strategy) {
+  bool only_broadcast = false;
+  bool only_scan = false;
+
+  if (strategy->only_broadcast && strategy->only_broadcast)

[xiyuan, 2014/08/05 21:02:35]

missing '*' for the second "strategy->only_broadcast"

[rkc, 2014/08/05 22:25:22]

Done.

+    only_broadcast = true;
+  if (strategy->only_scan && *strategy->only_scan)
+    only_scan = true;
+
+  if (only_broadcast && only_scan)
+    return copresence::BROADCAST_AND_SCAN;
+  if (only_broadcast)
+    return copresence::BROADCAST_ONLY;
+  if (only_scan)
+    return copresence::SCAN_ONLY;
+
+  return copresence::BROADCAST_SCAN_CONFIGURATION_UNKNOWN;
+}
+
+}  // namespace
+
+namespace extensions {
+
+bool AddPublishToRequest(const std::string& app_id,
+                         const api::copresence::PublishOperation& publish,
+                         copresence::ReportRequest* request) {
+  copresence::PublishedMessage* publish_proto =
+      request->mutable_manage_messages_request()->add_message_to_publish();
+  publish_proto->mutable_access_policy()->mutable_acl()->set_acl_type(
+      copresence::NO_ACL_CHECK);
+  publish_proto->set_id(publish.id);
+  publish_proto->mutable_message()->mutable_type()->set_type(
+      publish.message.type);
+  publish_proto->mutable_message()->set_payload(publish.message.payload);
+
+  int ttl = SanitizeTtl(publish.time_to_live_millis.get());
+  if (ttl < 0)
+    return false;
+  publish_proto->mutable_access_policy()->set_ttl_millis(ttl);
+
+  // TODO(rkc): This is a temporary hack; eventually namespaces will be
+  // completely gone. When that happens, remove this.
+  publish_proto->mutable_message()->mutable_type()->set_namespace_deprecated(
+      app_id);
+  publish_proto->set_strategy(copresence::AGGRESSIVE);
+
+  if (publish.strategies.get()) {
+    copresence::BroadcastScanConfiguration config =
+        TranslateStrategy(publish.strategies.get());
+    if (config != copresence::BROADCAST_SCAN_CONFIGURATION_UNKNOWN) {
+      copresence::TokenExchangeStrategy* strategy_proto =
+          publish_proto->mutable_token_exchange_strategy();
+      strategy_proto->set_broadcast_scan_configuration(config);
+    }
+  }
+
+  DVLOG(2) << "Publishing message of type " << publish.message.type << ":\n"
+           << publish.message.payload;
+  // TODO(ckehoe): Validate that required fields are non-empty, etc.
+  return true;
+}
+
+bool AddUnpublishToRequest(const std::string& publish_id,
+                           copresence::ReportRequest* request) {
+  if (publish_id.empty())
+    return false;
+
+  request->mutable_manage_messages_request()->add_id_to_unpublish(publish_id);
+  DVLOG(2) << "Unpublishing message \"" << publish_id << "\"";
+  return true;
+}
+
+bool AddSubscribeToRequest(
+    const std::string& app_id,
+    const api::copresence::SubscribeOperation& subscription,
+    SubscriptionToAppMap* apps_by_subscription_id,
+    copresence::ReportRequest* request) {
+  // Associate the subscription id with the app id.
+  std::map<std::string, std::string>::iterator previous_subscription =

[xiyuan, 2014/08/05 21:02:35]

nit: SubscriptionToAppMap::iterator since we have the typedef

[rkc, 2014/08/05 22:25:22]

Done.

+      apps_by_subscription_id->find(subscription.id);
+  if (previous_subscription == apps_by_subscription_id->end()) {
+    (*apps_by_subscription_id)[subscription.id] = app_id;
+  } else if (previous_subscription->second == app_id) {
+    VLOG(2) << "Overwriting subscription id \"" << subscription.id
+            << "\" for app \"" << app_id << "\"";
+  } else {
+    // A conflicting association exists already.
+    VLOG(1) << "Subscription id \"" << subscription.id
+            << "\" used by two apps: \"" << previous_subscription->second
+            << "\" and \"" << app_id << "\"";
+    return false;
+  }
+
+  // Convert from client to server subscription format.
+  copresence::Subscription* subscription_proto =
+      request->mutable_manage_subscriptions_request()->add_subscription();
+  subscription_proto->set_id(subscription.id);
+  int ttl = SanitizeTtl(subscription.time_to_live_millis.get());
+  if (ttl < 0)
+    return false;
+  subscription_proto->set_ttl_millis(ttl);
+
+  // TODO(rkc): This is a temporary hack; eventually namespaces will be
+  // completely gone. When that happens, remove this.
+  subscription_proto->mutable_message_type()->set_namespace_deprecated(app_id);
+  subscription_proto->set_strategy(copresence::AGGRESSIVE);
+
+  subscription_proto->mutable_message_type()->set_type(
+      subscription.filter.type);
+
+  if (subscription.strategies.get()) {
+    copresence::BroadcastScanConfiguration config =
+        TranslateStrategy(subscription.strategies.get());
+    if (config != copresence::BROADCAST_SCAN_CONFIGURATION_UNKNOWN) {
+      copresence::TokenExchangeStrategy* strategy_proto =
+          subscription_proto->mutable_token_exchange_strategy();
+      strategy_proto->set_broadcast_scan_configuration(config);
+    }
+  }
+
+  DVLOG(2) << "Subscribing for messages of type " << subscription.filter.type;
+  // TODO(ckehoe): Validate that required fields are non-empty, etc.
+  return true;
+}
+
+bool AddUnsubscribeToRequest(const std::string& app_id,
+                             const std::string& subscription_id,
+                             SubscriptionToAppMap* apps_by_subscription_id,
+                             copresence::ReportRequest* request) {
+  if (subscription_id.empty())
+    return false;
+
+  // Check that this subscription id belongs to this app.
+  std::map<std::string, std::string>::iterator subscription =

[xiyuan, 2014/08/05 21:02:35]

nit: SubscriptionToAppMap::iterator

[rkc, 2014/08/05 22:25:22]

Done.

+      apps_by_subscription_id->find(subscription_id);
+  if (subscription == apps_by_subscription_id->end()) {
+    LOG(ERROR) << "No such subscription \"" << subscription_id
+               << "\". Cannot unsubscribe.";
+    return false;
+  } else if (subscription->second != app_id) {
+    LOG(ERROR) << "Subscription \"" << subscription_id
+               << "\" does not belong to app \"" << app_id
+               << "\". Cannot unsubscribe.";
+    return false;
+  } else {
+    apps_by_subscription_id->erase(subscription);
+  }
+
+  request->mutable_manage_subscriptions_request()->add_id_to_unsubscribe(
+      subscription_id);
+  DVLOG(2) << "Cancelling subscription \"" << subscription_id << "\" for app \""
+           << app_id << "\"";
+  return true;
+}
+
+bool PrepareReportRequestProto(
+    const std::vector<linked_ptr<api::copresence::Operation> >& operations,
+    const std::string& app_id,
+    SubscriptionToAppMap* apps_by_subscription_id,
+    copresence::ReportRequest* request) {
+  for (size_t i = 0; i < operations.size(); ++i) {
+    linked_ptr<api::copresence::Operation> op = operations[i];
+    DCHECK(op.get());
+    switch (op->type) {
+      case api::copresence::OPERATION_TYPE_PUBLISH:
+        if (!op->publish ||
+            !AddPublishToRequest(app_id, *(op->publish), request))
+          return false;
+        break;
+
+      case api::copresence::OPERATION_TYPE_SUBSCRIBE:
+        if (!op->subscribe ||
+            !AddSubscribeToRequest(
+                app_id, *(op->subscribe), apps_by_subscription_id, request))
+          return false;
+        break;
+
+      case api::copresence::OPERATION_TYPE_UNPUBLISH:
+        if (!op->unpublish ||
+            !AddUnpublishToRequest(op->unpublish->unpublish_id, request))
+          return false;
+        break;
+
+      case api::copresence::OPERATION_TYPE_UNSUBSCRIBE:
+        if (!op->unsubscribe ||
+            !AddUnsubscribeToRequest(app_id,
+                                     op->unsubscribe->unsubscribe_id,
+                                     apps_by_subscription_id,
+                                     request))
+          return false;
+        break;
+
+      case api::copresence::OPERATION_TYPE_NONE:
+      default:
+        LOG(ERROR) << "Unrecognized operation type " << op->type;
+    }
+  }
+  return true;
+}
+
+}  // namespace extensions