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.

[David Trainor- moved to gerrit, 2016/08/30 20:46:29]

Can this just call GetIdentitySource()->Connect()?

[xingliu, 2016/08/30 22:27:36]

Done.

   if (!identity_source_) {
-    identity_source_ = base::MakeUnique<IdentitySource>(
-        delegate_,
-        base::Bind(&BlimpClientContextImpl::ConnectToAssignmentSource,
-                   base::Unretained(this)));
+    CreateIdentitySource();
   }
 
   // Start Blimp authentication flow. The OAuth2 token will be used in
   // assignment source.
   identity_source_->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.";
   assignment_source_->GetAssignment(
       client_auth_token,
@@ skipping 39 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