Actual cancelation of registration when cancel buttons pressed

chrome/browser/ui/webui/local_discovery/local_discovery_ui_handler.cc

 // Copyright 2013 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/ui/webui/local_discovery/local_discovery_ui_handler.h"
 
 #include "base/bind.h"
 #include "base/strings/stringprintf.h"
 #include "base/values.h"
 #include "chrome/browser/local_discovery/privet_device_lister_impl.h"
@@ skipping 23 matching lines @@
 
 LocalDiscoveryUIHandler::LocalDiscoveryUIHandler() : is_visible_(false) {
 }
 
 LocalDiscoveryUIHandler::LocalDiscoveryUIHandler(
     scoped_ptr<PrivetDeviceLister> privet_lister) {
   privet_lister.swap(privet_lister_);
 }
 
 LocalDiscoveryUIHandler::~LocalDiscoveryUIHandler() {
+  ResetCurrentRegistration();
   SetIsVisible(false);
   if (service_discovery_client_.get()) {
     service_discovery_client_ = NULL;
     ServiceDiscoveryHostClientFactory::ReleaseClient();
   }
 }
 
 // static
 LocalDiscoveryUIHandler* LocalDiscoveryUIHandler::Create() {
   if (g_factory) return g_factory->CreateLocalDiscoveryUIHandler();
@@ skipping 16 matching lines @@
       base::Unretained(this)));
   web_ui()->RegisterMessageCallback("isVisible", base::Bind(
       &LocalDiscoveryUIHandler::HandleIsVisible,
       base::Unretained(this)));
   web_ui()->RegisterMessageCallback("registerDevice", base::Bind(
       &LocalDiscoveryUIHandler::HandleRegisterDevice,
       base::Unretained(this)));
   web_ui()->RegisterMessageCallback("chooseUser", base::Bind(
       &LocalDiscoveryUIHandler::HandleChooseUser,
       base::Unretained(this)));
+  web_ui()->RegisterMessageCallback("cancelRegistration", base::Bind(
+      &LocalDiscoveryUIHandler::HandleCancelRegistration,
+      base::Unretained(this)));
 }
 
 void LocalDiscoveryUIHandler::HandleStart(const base::ListValue* args) {
   // If privet_lister_ is already set, it is a mock used for tests or the result
   // of a reload.
   if (!privet_lister_) {
     service_discovery_client_ = ServiceDiscoveryHostClientFactory::GetClient();
     privet_lister_.reset(new PrivetDeviceListerImpl(
         service_discovery_client_.get(), this));
     privet_http_factory_.reset(new PrivetHTTPAsynchronousFactoryImpl(
@@ skipping 40 matching lines @@
   DCHECK(rv);
 
   privet_resolution_ = privet_http_factory_->CreatePrivetHTTP(
       current_register_device_,
       device_descriptions_[current_register_device_].address,
       base::Bind(&LocalDiscoveryUIHandler::StartRegisterHTTP,
                  base::Unretained(this), user));
   privet_resolution_->Start();
 }
 
+void LocalDiscoveryUIHandler::HandleCancelRegistration(
+    const base::ListValue* args) {
+  ResetCurrentRegistration();
+}
+
 void LocalDiscoveryUIHandler::StartRegisterHTTP(
     const std::string& user,
     scoped_ptr<PrivetHTTPClient> http_client) {
   current_http_client_.swap(http_client);
 
   if (!current_http_client_) {
     SendRegisterError();
     return;
   }
 
@@ skipping 196 matching lines @@
       Profile::FromWebUI(web_ui())->GetRequestContext(),
       user_index,
       xsrf_token,
       automated_claim_url,
       base::Bind(&LocalDiscoveryUIHandler::OnConfirmDone,
                  base::Unretained(this))));
 
   confirm_api_call_flow_->Start();
 }
 
+// TODO(noamsml): Create master object for registration flow.
+void LocalDiscoveryUIHandler::ResetCurrentRegistration() {
+  current_register_device_.clear();
+  if (current_register_operation_.get()) {
+    current_register_operation_->Cancel();
+    current_register_operation_.reset();
+  }
+
+  confirm_api_call_flow_.reset();
+  privet_resolution_.reset();
+  cloud_print_account_manager_.reset();
+  xsrf_token_for_primary_user_.clear();
+  current_register_user_index_ = 0;
+  current_http_client_.reset();
+}
+
 }  // namespace local_discovery