Demo UI for device discovery and registration

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/message_loop/message_loop.h"
+#include "base/strings/stringprintf.h"
 #include "base/values.h"
+#include "chrome/browser/local_discovery/privet_device_lister_impl.h"
+#include "chrome/browser/local_discovery/privet_http_impl.h"
+#include "chrome/browser/profiles/profile.h"
+#include "chrome/browser/signin/profile_oauth2_token_service.h"
+#include "chrome/browser/signin/profile_oauth2_token_service_factory.h"
+#include "chrome/browser/signin/signin_manager.h"
+#include "chrome/browser/signin/signin_manager_base.h"
+#include "chrome/browser/signin/signin_manager_factory.h"
 #include "content/public/browser/web_ui.h"
+#include "net/base/host_port_pair.h"
+#include "net/base/net_util.h"
 
-LocalDiscoveryUIHandler::LocalDiscoveryUIHandler()
-    : action_callback_(base::Bind(&LocalDiscoveryUIHandler::OnNewDevice,
-                                  base::Unretained(this))) {
-  content::AddActionCallback(action_callback_);
+namespace {
+// TODO(noamsml): This is a temporary shim until automated_url is in the
+// response.
+const char kPrivetAutomatedClaimURLFormat[] = "%s/confirm?token=%s";
+}  // namepsace
+
+LocalDiscoveryUIHandler::LocalDiscoveryUIHandler() {
 }
 
 LocalDiscoveryUIHandler::~LocalDiscoveryUIHandler() {
-  content::RemoveActionCallback(action_callback_);
 }
 
-void LocalDiscoveryUIHandler::RegisterMessages() {}
+void LocalDiscoveryUIHandler::RegisterMessages() {
+  web_ui()->RegisterMessageCallback("start", base::Bind(
+      &LocalDiscoveryUIHandler::OnStart,
+      base::Unretained(this)));
+  web_ui()->RegisterMessageCallback("register", base::Bind(
+      &LocalDiscoveryUIHandler::OnRegisterDevice,
+      base::Unretained(this)));
+}
 
-void LocalDiscoveryUIHandler::OnNewDevice(const std::string& name) {
-  // TODO(gene): Once we receive information about new locally discovered
-  // device, we should add it to the page.
-  // Here is an example how to do it:
-  //
-  // base::StringValue service_name(name);
-  //
-  // base::DictionaryValue info;
-  // info.SetString("domain", "domain.local");
-  // info.SetString("port", "80");
-  // info.SetString("ip", "XXX.XXX.XXX.XXX");
-  // info.SetString("metadata", "metadata\nmetadata");
-  // info.SetString("lastSeen", "unknown");
-  // info.SetString("registered", "not registered");
-  //
-  // base::StringValue domain("domain.local");
-  // base::StringValue port("80");
-  // base::StringValue ip("XXX.XXX.XXX.XXX");
-  // base::StringValue metadata("metadata<br>metadata");
-  // base::StringValue lastSeen("unknown");
-  // base::StringValue registered("not registered");
-  //
-  // web_ui()->CallJavascriptFunction("onServiceUpdate", service_name, info);
+void LocalDiscoveryUIHandler::OnStart(const base::ListValue* value) {
+  service_discovery_client_ = new local_discovery::ServiceDiscoveryHostClient();
+  service_discovery_client_->Start();
+  privet_lister_.reset(new local_discovery::PrivetDeviceListerImpl(
+      service_discovery_client_.get(), this));
+  privet_lister_->Start();
+  privet_lister_->DiscoverNewDevices(false);
 }
+
+void LocalDiscoveryUIHandler::OnRegisterDevice(const base::ListValue* data) {
+  std::string device_name;
+  data->GetString(0, &device_name);
+  currently_registering_device_ = device_name;
+
+  if (!device_descriptions_[device_name].ip_address.empty()) {
+    OnDomainResolved(true, device_descriptions_[device_name].ip_address);
+  } else {
+    domain_resolver_ = service_discovery_client_->CreateLocalDomainResolver(
+        device_descriptions_[device_name].address.host(),
+        net::ADDRESS_FAMILY_UNSPECIFIED,
+        base::Bind(&LocalDiscoveryUIHandler::OnDomainResolved,
+                   base::Unretained(this)));
+
+    domain_resolver_->Start();
+  }
+}
+
+void LocalDiscoveryUIHandler::OnDomainResolved(bool success,
+    const net::IPAddressNumber& address) {
+  if (!success) {
+    LogRegisterErrorToWeb("Resolution failed");
+    return;
+  }
+
+  Profile* profile = Profile::FromWebUI(web_ui());
+  SigninManagerBase* signin_manager =
+      SigninManagerFactory::GetForProfileIfExists(profile);
+
+  if (!signin_manager) {
+    LogRegisterErrorToWeb("You must be signed in");
+    return;
+  }
+
+  std::string username = signin_manager->GetAuthenticatedUsername();
+
+  std::string address_str = net::IPAddressToString(address);
+  int port = device_descriptions_[currently_registering_device_].address.port();
+
+  current_http_client_.reset(new local_discovery::PrivetHTTPClientImpl(
+      net::HostPortPair(address_str, port),
+      Profile::FromWebUI(web_ui())->GetRequestContext()));
+  current_register_operation_ =
+      current_http_client_->CreateRegisterOperation(username, this);
+  current_register_operation_->Start();
+}
+
+void LocalDiscoveryUIHandler::OnPrivetRegisterClaimToken(
+    const std::string& token,
+    const GURL& url) {
+  GURL automated_claim_url(base::StringPrintf(
+      kPrivetAutomatedClaimURLFormat,
+      device_descriptions_[currently_registering_device_].url.c_str(),
+      token.c_str()));
+
+  Profile* profile = Profile::FromWebUI(web_ui());
+
+  OAuth2TokenService* token_service =
+      ProfileOAuth2TokenServiceFactory::GetForProfile(profile);
+
+  if (!token_service) {
+    LogRegisterErrorToWeb("Could not get token service");
+    return;
+  }
+
+  confirm_api_call_flow_.reset(new local_discovery::PrivetConfirmApiCallFlow(
+      profile->GetRequestContext(),
+      token_service,
+      automated_claim_url,
+      base::Bind(&LocalDiscoveryUIHandler::OnConfirmDone,
+                 base::Unretained(this))));
+
+  confirm_api_call_flow_->Start();
+}
+
+void LocalDiscoveryUIHandler::OnPrivetRegisterError(
+    const std::string& action,
+    local_discovery::PrivetRegisterOperation::FailureReason reason,
+    int printer_http_code,
+    const DictionaryValue* json) {
+  // TODO(noamsml): Add detailed error message.
+  LogRegisterErrorToWeb("Registration error.");
+}
+
+void LocalDiscoveryUIHandler::OnPrivetRegisterDone(
+    const std::string& device_id) {
+  current_register_operation_.reset();
+  current_http_client_.reset();
+
+  LogRegisterDoneToWeb(device_id);
+}
+
+void LocalDiscoveryUIHandler::OnConfirmDone(
+    local_discovery::PrivetConfirmApiCallFlow::Status status) {
+  if (status == local_discovery::PrivetConfirmApiCallFlow::SUCCESS) {
+    LOG(INFO) << "Confirm success.";
+    confirm_api_call_flow_.reset();
+    current_register_operation_->CompleteRegistration();
+  } else {
+    // TODO(noamsml): Add detailed error message.
+    LogRegisterErrorToWeb("Confirm error.");
+  }
+}
+
+void LocalDiscoveryUIHandler::DeviceChanged(
+    bool added,
+    const std::string& name,
+    const local_discovery::DeviceDescription& description) {
+  device_descriptions_[name] = description;
+
+  base::StringValue service_name(name);
+  base::DictionaryValue info;
+  info.SetString("domain", description.address.host());
+  info.SetInteger("port", description.address.port());
+  std::string ip_addr_string;
+  if (!description.ip_address.empty()) {
+    ip_addr_string = net::IPAddressToString(description.ip_address);
+  }
+  info.SetString("ip", ip_addr_string);
+  info.SetString("metadata", "");
+  info.SetString("lastSeen", "unknown");
+  info.SetBoolean("registered", !description.id.empty());
+
+  web_ui()->CallJavascriptFunction("onServiceUpdate", service_name, info);
+}
+
+void LocalDiscoveryUIHandler::DeviceRemoved(const std::string& name) {
+  device_descriptions_.erase(name);
+  scoped_ptr<base::Value> null_value(base::Value::CreateNullValue());
+  base::StringValue name_value(name);
+
+  web_ui()->CallJavascriptFunction("onServiceUpdate",
+                                   name_value, *null_value.get());
+}
+
+void LocalDiscoveryUIHandler::LogRegisterErrorToWeb(const std::string& error) {
+  base::StringValue error_value(error);
+  web_ui()->CallJavascriptFunction("registrationFailed", error_value);
+  LOG(ERROR) << error;
+}
+
+void LocalDiscoveryUIHandler::LogRegisterDoneToWeb(const std::string& id) {
+  base::StringValue id_value(id);
+  web_ui()->CallJavascriptFunction("registrationSuccess", id_value);
+  LOG(INFO) << "Registered " << id;
+}