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 local_discovery {
+
+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 ServiceDiscoveryHostClient();
+  service_discovery_client_->Start();
+  privet_lister_.reset(new 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);

[Dan Beam, 2013/08/06 21:21:38]

nit: CHECK() or bool rv = ...; DCHECK(rv); etc.

[Noam Samuel, 2013/08/06 23:54:05]

Done.

+  currently_registering_device_ = device_name;
+

[Dan Beam, 2013/08/06 21:21:38]

nit: what if |device_name| isn't in |device_descriptions_|?

[Noam Samuel, 2013/08/06 23:54:05]

Hm. I shouldn't actually be looking this up in the map, since if the domain is
still valid in the cache, resolving it will cause no network hit and will be
more accurate (due to things like expiration).

+  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 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 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,
+    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(
+    PrivetConfirmApiCallFlow::Status status) {
+  if (status == PrivetConfirmApiCallFlow::SUCCESS) {
+    LOG(INFO) << "Confirm success.";

[Dan Beam, 2013/08/06 21:21:38]

nit: DLOG

[Noam Samuel, 2013/08/06 23:54:05]

Done.

+    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 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()) {

[Dan Beam, 2013/08/06 21:21:38]

nit: no curlies

[Noam Samuel, 2013/08/06 23:54:05]

Done.

+    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());

[Dan Beam, 2013/08/06 21:21:38]

nit: don't think you need the .get()

[Noam Samuel, 2013/08/06 23:54:05]

Done.

+}
+
+void LocalDiscoveryUIHandler::LogRegisterErrorToWeb(const std::string& error) {
+  base::StringValue error_value(error);
+  web_ui()->CallJavascriptFunction("registrationFailed", error_value);
+  LOG(ERROR) << error;

[Dan Beam, 2013/08/06 21:21:38]

nit: DLOG

[Noam Samuel, 2013/08/06 23:54:05]

Done.

+}
+
+void LocalDiscoveryUIHandler::LogRegisterDoneToWeb(const std::string& id) {
+  base::StringValue id_value(id);
+  web_ui()->CallJavascriptFunction("registrationSuccess", id_value);
+  LOG(INFO) << "Registered " << id;

[Dan Beam, 2013/08/06 21:21:38]

nit: DLOG

[Noam Samuel, 2013/08/06 23:54:05]

Done.

+}
+
+}  // namespace local_discovery