Integrate UI with authentication flow.

blimp/client/core/blimp_client_context_impl.cc

 // Copyright 2016 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 "blimp/client/core/blimp_client_context_impl.h"
 
 #include "base/bind.h"
 #include "base/memory/ptr_util.h"
 #include "base/message_loop/message_loop.h"
 #include "base/threading/sequenced_task_runner_handle.h"
@@ skipping 75 matching lines @@
 }
 
 std::unique_ptr<BlimpContents> BlimpClientContextImpl::CreateBlimpContents() {
   std::unique_ptr<BlimpContents> blimp_contents =
       blimp_contents_manager_->CreateBlimpContents();
   delegate_->AttachBlimpContentsHelpers(blimp_contents.get());
   return blimp_contents;
 }
 
 void BlimpClientContextImpl::Connect() {
-  // Lazy initialization of IdentitySource.
-  if (!identity_source_) {
-    identity_source_ = base::MakeUnique<IdentitySource>(
-        delegate_,
-        base::Bind(&BlimpClientContextImpl::ConnectToAssignmentSource,
-                   base::Unretained(this)));
-  }
-
   // Start Blimp authentication flow. The OAuth2 token will be used in
   // assignment source.
-  identity_source_->Connect();
+  GetIdentitySource()->Connect();
+}
+
+IdentitySource* BlimpClientContextImpl::GetIdentitySource() {
+  if (!identity_source_) {
+    CreateIdentitySource();
+  }
+  return identity_source_.get();
 }
 
 void BlimpClientContextImpl::ConnectToAssignmentSource(
     const std::string& client_auth_token) {
   if (!assignment_source_) {
     assignment_source_.reset(new AssignmentSource(
         GetAssignerURL(), io_thread_task_runner_, file_thread_task_runner_));
   }
 
   VLOG(1) << "Trying to get assignment.";
@@ skipping 41 matching lines @@
       thread_pipe_manager_->RegisterFeature(BlimpMessage::kIme,
                                             ime_feature_.get()));
   navigation_feature_->set_outgoing_message_processor(
       thread_pipe_manager_->RegisterFeature(BlimpMessage::kNavigation,
                                             navigation_feature_.get()));
   tab_control_feature_->set_outgoing_message_processor(
       thread_pipe_manager_->RegisterFeature(BlimpMessage::kTabControl,
                                             tab_control_feature_.get()));
 }
 
+void BlimpClientContextImpl::CreateIdentitySource() {
+  identity_source_ = base::MakeUnique<IdentitySource>(
+      delegate_, base::Bind(&BlimpClientContextImpl::ConnectToAssignmentSource,
+                            base::Unretained(this)));
+}
+
 }  // namespace client
 }  // namespace blimp