content/shell/renderer/ipc_echo.cc

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "content/shell/renderer/ipc_echo.h"
   6
   7 #include "base/logging.h"
   8 #include "content/shell/common/shell_messages.h"
   9 #include "content/shell/renderer/binding_helpers.h"
  10 #include "gin/object_template_builder.h"
  11 #include "gin/wrappable.h"
  12 #include "ipc/ipc_sender.h"
  13 #include "third_party/WebKit/public/web/WebDOMCustomEvent.h"
  14 #include "third_party/WebKit/public/web/WebDOMEvent.h"
  15 #include "third_party/WebKit/public/web/WebSerializedScriptValue.h"
  16
  17 namespace content {
  18
  19 class IPCEchoBindings : public gin::Wrappable<IPCEchoBindings> {
  20 public:
  21   static gin::WrapperInfo kWrapperInfo;
  22   static void Install(base::WeakPtr<IPCEcho> echo, blink::WebFrame* frame);
  23
  24   explicit IPCEchoBindings(base::WeakPtr<IPCEcho>);
  25
  26   void RequestEcho(int id, int size);
  27   int GetLastEchoId() const;
  28   int GetLastEchoSize() const;
  29
  30 private:
  31  gin::ObjectTemplateBuilder GetObjectTemplateBuilder(v8::Isolate*) override;
  32
  33   base::WeakPtr<IPCEcho> native_;
  34 };
  35
  36 IPCEchoBindings::IPCEchoBindings(base::WeakPtr<IPCEcho> native)
  37     : native_(native) {
  38 }
  39
  40 void IPCEchoBindings::RequestEcho(int id, int size) {
  41   if (!native_)
  42     return;
  43   native_->RequestEcho(id, size);
  44 }
  45
  46 int IPCEchoBindings::GetLastEchoId() const {
  47   if (!native_)
  48     return 0;
  49   return native_->last_echo_id();
  50 }
  51
  52 int IPCEchoBindings::GetLastEchoSize() const {
  53   if (!native_)
  54     return 0;
  55   return native_->last_echo_size();
  56 }
  57
  58 gin::WrapperInfo IPCEchoBindings::kWrapperInfo = {
  59     gin::kEmbedderNativeGin };
  60
  61 gin::ObjectTemplateBuilder IPCEchoBindings::GetObjectTemplateBuilder(
  62     v8::Isolate* isolate) {
  63   return gin::Wrappable<IPCEchoBindings>::GetObjectTemplateBuilder(isolate)
  64       .SetMethod("requestEcho",
  65                  &IPCEchoBindings::RequestEcho)
  66       .SetProperty("lastEchoId",
  67                    &IPCEchoBindings::GetLastEchoId)
  68       .SetProperty("lastEchoSize",
  69                    &IPCEchoBindings::GetLastEchoSize);
  70 }
  71
  72 // static
  73 void IPCEchoBindings::Install(base::WeakPtr<IPCEcho> echo,
  74                               blink::WebFrame* frame) {
  75   std::vector<std::string> names;
  76   names.push_back("ipcEcho");
  77   return InstallAsWindowProperties(
  78       new IPCEchoBindings(echo), frame, names);
  79 }
  80
  81 IPCEcho::IPCEcho(blink::WebDocument document,
  82                  IPC::Sender* sender,
  83                  int routing_id)
  84     : document_(document), sender_(sender), routing_id_(routing_id),
  85       last_echo_id_(0),
  86       weak_factory_(this) {
  87 }
  88
  89 IPCEcho::~IPCEcho() {
  90 }
  91
  92 void IPCEcho::RequestEcho(int id, int size) {
  93   sender_->Send(new ShellViewHostMsg_EchoPing(
  94       routing_id_, id, std::string(size, '*')));
  95 }
  96
  97 void IPCEcho::DidRespondEcho(int id, int size) {
  98   last_echo_id_ = id;
  99   last_echo_size_ = size;
 100   blink::WebString eventName = blink::WebString::fromUTF8("CustomEvent");
 101   blink::WebString eventType = blink::WebString::fromUTF8("pong");
 102   blink::WebDOMEvent event = document_.createEvent(eventName);
 103   event.to<blink::WebDOMCustomEvent>().initCustomEvent(
 104       eventType, false, false, blink::WebSerializedScriptValue());
 105   document_.dispatchEvent(event);
 106 }
 107
 108 void IPCEcho::Install(blink::WebFrame* frame) {
 109   IPCEchoBindings::Install(weak_factory_.GetWeakPtr(), frame);
 110 }
 111
 112 }  // namespace content