ios/chrome/browser/ui/webui/physical_web_ui.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 "ios/chrome/browser/ui/webui/physical_web_ui.h"

#include "base/macros.h"
#include "base/memory/ptr_util.h"
#include "base/metrics/user_metrics.h"
#include "components/grit/components_resources.h"
#include "components/physical_web/data_source/physical_web_data_source.h"
#include "components/physical_web/webui/physical_web_base_message_handler.h"
#include "components/physical_web/webui/physical_web_ui_constants.h"
#include "components/strings/grit/components_strings.h"
#include "ios/chrome/browser/application_context.h"
#include "ios/chrome/browser/browser_state/chrome_browser_state.h"
#include "ios/chrome/browser/chrome_url_constants.h"
#include "ios/web/public/web_ui_ios_data_source.h"
#include "ios/web/public/webui/web_ui_ios.h"
#include "ios/web/public/webui/web_ui_ios_message_handler.h"

namespace {

web::WebUIIOSDataSource* CreatePhysicalWebUIDataSource() {
  web::WebUIIOSDataSource* html_source =
      web::WebUIIOSDataSource::Create(kChromeUIPhysicalWebHost);

  // Localized and data strings.
  html_source->AddLocalizedString(physical_web_ui::kTitle,
                                  IDS_PHYSICAL_WEB_UI_TITLE);
  html_source->AddLocalizedString(physical_web_ui::kEmptyMessage,
                                  IDS_PHYSICAL_WEB_UI_EMPTY_MESSAGE);
  html_source->SetJsonPath("strings.js");
  html_source->AddResourcePath(physical_web_ui::kPhysicalWebJS,
                               IDR_PHYSICAL_WEB_UI_JS);
  html_source->AddResourcePath(physical_web_ui::kPhysicalWebCSS,
                               IDR_PHYSICAL_WEB_UI_CSS);
  html_source->AddResourcePath(physical_web_ui::kPhysicalWebLinkIcon,
                               IDR_PHYSICAL_WEB_UI_LINK_ICON);
  html_source->SetDefaultResource(IDR_PHYSICAL_WEB_UI_HTML);
  return html_source;
}

// Implements all MessageHandler core functionality.  This is extends the
// PhysicalWebBaseMessageHandler and implements functions to manipulate an
// iOS-specific WebUI object.
class MessageHandlerImpl
    : public physical_web_ui::PhysicalWebBaseMessageHandler {
 public:
  explicit MessageHandlerImpl(web::WebUIIOS* web_ui) : web_ui_(web_ui) {}
  ~MessageHandlerImpl() override {}

 private:
  void RegisterMessageCallback(
      const std::string& message,
      const physical_web_ui::MessageCallback& callback) override {
    web_ui_->RegisterMessageCallback(message, callback);
  }
  void CallJavaScriptFunction(const std::string& function,
                              const base::Value& arg) override {
    web_ui_->CallJavascriptFunction(function, arg);
  }
  physical_web::PhysicalWebDataSource* GetPhysicalWebDataSource() override {
    return GetApplicationContext()->GetPhysicalWebDataSource();
  }

  web::WebUIIOS* web_ui_;
  DISALLOW_COPY_AND_ASSIGN(MessageHandlerImpl);
};

////////////////////////////////////////////////////////////////////////////////
//
// PhysicalWebDOMHandler
//
////////////////////////////////////////////////////////////////////////////////

// The handler for Javascript messages for the chrome:physical-web page.
class PhysicalWebDOMHandler : public web::WebUIIOSMessageHandler {
 public:
  PhysicalWebDOMHandler() {}
  ~PhysicalWebDOMHandler() override {}

  void RegisterMessages() override {
    impl_.reset(new MessageHandlerImpl(web_ui()));
    impl_->RegisterMessages();
  }

 private:
  std::unique_ptr<MessageHandlerImpl> impl_;
  DISALLOW_COPY_AND_ASSIGN(PhysicalWebDOMHandler);
};

}  // namespace

////////////////////////////////////////////////////////////////////////////////
//
// PhysicalWebUI
//
////////////////////////////////////////////////////////////////////////////////

PhysicalWebUI::PhysicalWebUI(web::WebUIIOS* web_ui)
    : web::WebUIIOSController(web_ui) {
  web_ui->AddMessageHandler(base::MakeUnique<PhysicalWebDOMHandler>());

  web::WebUIIOSDataSource::Add(ios::ChromeBrowserState::FromWebUIIOS(web_ui),
                               CreatePhysicalWebUIDataSource());

  base::RecordAction(base::UserMetricsAction("PhysicalWeb.WebUI.Open"));
}

PhysicalWebUI::~PhysicalWebUI() {}