Integrate InstanceID with GCM crypto provider

components/gcm_driver/instance_id/instance_id_impl.cc

 // Copyright 2015 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 "components/gcm_driver/instance_id/instance_id_impl.h"
 
 #include <stdint.h>
 
 #include <algorithm>
 #include <memory>
@@ skipping 31 matching lines @@
     default:
       NOTREACHED() << "Unexpected value of result cannot be converted: "
                    << result;
   }
   return InstanceID::UNKNOWN_ERROR;
 }
 
 }  // namespace
 
 // static
-std::unique_ptr<InstanceID> InstanceID::Create(
-    const std::string& app_id,
-    gcm::InstanceIDHandler* handler) {
-  return base::WrapUnique(new InstanceIDImpl(app_id, handler));
+std::unique_ptr<InstanceID> InstanceID::Create(const std::string& app_id,
+                                               gcm::GCMDriver* gcm_driver) {
+  return base::WrapUnique(new InstanceIDImpl(app_id, gcm_driver));
 }
 
 InstanceIDImpl::InstanceIDImpl(const std::string& app_id,
-                               gcm::InstanceIDHandler* handler)
-    : InstanceID(app_id), handler_(handler), weak_ptr_factory_(this) {
-  DCHECK(handler_);
-  handler_->GetInstanceIDData(
+                               gcm::GCMDriver* gcm_driver)
+    : InstanceID(app_id, gcm_driver), weak_ptr_factory_(this) {
+  Handler()->GetInstanceIDData(
       app_id, base::Bind(&InstanceIDImpl::GetInstanceIDDataCompleted,
                          weak_ptr_factory_.GetWeakPtr()));
 }
 
 InstanceIDImpl::~InstanceIDImpl() {
 }
 
 void InstanceIDImpl::GetID(const GetIDCallback& callback) {
   if (!delayed_task_controller_.CanRunTaskWithoutDelay()) {
     delayed_task_controller_.AddTask(
@@ skipping 50 matching lines @@
   DoGetToken(authorized_entity, scope, options, callback);
 }
 
 void InstanceIDImpl::DoGetToken(
     const std::string& authorized_entity,
     const std::string& scope,
     const std::map<std::string, std::string>& options,
     const GetTokenCallback& callback) {
   EnsureIDGenerated();
 
-  handler_->GetToken(app_id(), authorized_entity, scope, options,
-                     base::Bind(&InstanceIDImpl::OnGetTokenCompleted,
-                                weak_ptr_factory_.GetWeakPtr(), callback));
+  Handler()->GetToken(app_id(), authorized_entity, scope, options,
+                      base::Bind(&InstanceIDImpl::OnGetTokenCompleted,
+                                 weak_ptr_factory_.GetWeakPtr(), callback));
 }
 
-void InstanceIDImpl::DeleteToken(const std::string& authorized_entity,
-                                 const std::string& scope,
-                                 const DeleteTokenCallback& callback) {
+void InstanceIDImpl::DeleteTokenImpl(const std::string& authorized_entity,
+                                     const std::string& scope,
+                                     const DeleteTokenCallback& callback) {
   DCHECK(!authorized_entity.empty());
   DCHECK(!scope.empty());
 
   if (!delayed_task_controller_.CanRunTaskWithoutDelay()) {
     delayed_task_controller_.AddTask(
         base::Bind(&InstanceIDImpl::DoDeleteToken,
                    weak_ptr_factory_.GetWeakPtr(),
                    authorized_entity,
                    scope,
                    callback));
     return;
   }
 
   DoDeleteToken(authorized_entity, scope, callback);
 }
 
 void InstanceIDImpl::DoDeleteToken(
     const std::string& authorized_entity,
     const std::string& scope,
     const DeleteTokenCallback& callback) {
   // Nothing to delete if the ID has not been generated.
   if (id_.empty()) {
     callback.Run(InstanceID::INVALID_PARAMETER);
     return;
   }
 
-  handler_->DeleteToken(app_id(), authorized_entity, scope,
+  Handler()->DeleteToken(app_id(), authorized_entity, scope,
                         base::Bind(&InstanceIDImpl::OnDeleteTokenCompleted,
                                    weak_ptr_factory_.GetWeakPtr(), callback));
 }
 
-void InstanceIDImpl::DeleteID(const DeleteIDCallback& callback) {
+void InstanceIDImpl::DeleteIDImpl(const DeleteIDCallback& callback) {
   if (!delayed_task_controller_.CanRunTaskWithoutDelay()) {
     delayed_task_controller_.AddTask(
         base::Bind(&InstanceIDImpl::DoDeleteID,
                    weak_ptr_factory_.GetWeakPtr(),
                    callback));
     return;
   }
 
   DoDeleteID(callback);
 }
 
 void InstanceIDImpl::DoDeleteID(const DeleteIDCallback& callback) {
   // Nothing to do if ID has not been generated.
   if (id_.empty()) {
     callback.Run(InstanceID::SUCCESS);
     return;
   }
 
-  handler_->DeleteAllTokensForApp(
+  Handler()->DeleteAllTokensForApp(
       app_id(), base::Bind(&InstanceIDImpl::OnDeleteIDCompleted,
                            weak_ptr_factory_.GetWeakPtr(), callback));
 
-  handler_->RemoveInstanceIDData(app_id());
+  Handler()->RemoveInstanceIDData(app_id());
 
   id_.clear();
   creation_time_ = base::Time();
 }
 
 void InstanceIDImpl::OnGetTokenCompleted(const GetTokenCallback& callback,
                                          const std::string& token,
                                          gcm::GCMClient::Result result) {
   callback.Run(token, GCMClientResultToInstanceIDResult(result));
 }
@@ skipping 52 matching lines @@
   base::Base64Encode(
       base::StringPiece(reinterpret_cast<const char*>(bytes), sizeof(bytes)),
       &id_);
   std::replace(id_.begin(), id_.end(), '+', '-');
   std::replace(id_.begin(), id_.end(), '/', '_');
   id_.erase(std::remove(id_.begin(), id_.end(), '='), id_.end());
 
   creation_time_ = base::Time::Now();
 
   // Save to the persistent store.
-  handler_->AddInstanceIDData(
+  Handler()->AddInstanceIDData(
       app_id(), id_, base::Int64ToString(creation_time_.ToInternalValue()));
 }
 
+gcm::InstanceIDHandler* InstanceIDImpl::Handler() {
+  gcm::InstanceIDHandler* handler =
+      gcm_driver()->GetInstanceIDHandlerInternal();
+  DCHECK(handler);
+  return handler;
+}
+
 }  // namespace instance_id