[Presentation API] (4th)(1-UA blink side) Add WebPresentationConnection and WebPresentationConnecti…

third_party/WebKit/Source/modules/presentation/PresentationConnection.cpp

 // Copyright 2014 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 "modules/presentation/PresentationConnection.h"
 
 #include "bindings/core/v8/ScriptPromiseResolver.h"
 #include "core/dom/DOMArrayBuffer.h"
 #include "core/dom/DOMArrayBufferView.h"
 #include "core/dom/Document.h"
@@ skipping 135 matching lines @@
   std::unique_ptr<FileReaderLoader> m_loader;
 };
 
 PresentationConnection::PresentationConnection(LocalFrame* frame,
                                                const String& id,
                                                const KURL& url)
     : ContextClient(frame),
       m_id(id),
       m_url(url),
       m_state(WebPresentationConnectionState::Connecting),
-      m_binaryType(BinaryTypeBlob) {}
+      m_binaryType(BinaryTypeBlob),
+      m_proxy(nullptr) {}
 
 PresentationConnection::~PresentationConnection() {
   ASSERT(!m_blobLoader);
 }
 
+void PresentationConnection::bindProxy(
+    std::unique_ptr<WebPresentationConnectionProxy> proxy) {
+  DCHECK(proxy);
+  // TODO(zhaobin): Restore to DCHECK(!m_proxy) when reconnect() is properly
+  // implemented.
+  if (!m_proxy)
+    m_proxy = std::move(proxy);
+}
+
 // static
 PresentationConnection* PresentationConnection::take(
     ScriptPromiseResolver* resolver,
     const WebPresentationSessionInfo& sessionInfo,
     PresentationRequest* request) {
   ASSERT(resolver);
   ASSERT(request);
   ASSERT(resolver->getExecutionContext()->isDocument());
 
   Document* document = toDocument(resolver->getExecutionContext());
@@ skipping 126 matching lines @@
     throwPresentationDisconnectedError(exceptionState);
     return false;
   }
 
   // The connection can send a message if there is a client available.
   return !!presentationClient(getExecutionContext());
 }
 
 void PresentationConnection::handleMessageQueue() {
   WebPresentationClient* client = presentationClient(getExecutionContext());
-  if (!client)
+  if (!client || !m_proxy)
     return;
 
   while (!m_messages.isEmpty() && !m_blobLoader) {
     Message* message = m_messages.first().get();
     switch (message->type) {
       case MessageTypeText:
-        client->sendString(m_url, m_id, message->text);
+        client->sendString(m_url, m_id, message->text, m_proxy.get());
         m_messages.removeFirst();
         break;
       case MessageTypeArrayBuffer:
-        client->sendArrayBuffer(m_url, m_id, static_cast<const uint8_t*>(
-                                                 message->arrayBuffer->data()),
-                                message->arrayBuffer->byteLength());
+        client->sendArrayBuffer(
+            m_url, m_id,
+            static_cast<const uint8_t*>(message->arrayBuffer->data()),
+            message->arrayBuffer->byteLength(), m_proxy.get());
         m_messages.removeFirst();
         break;
       case MessageTypeBlob:
         ASSERT(!m_blobLoader);
         m_blobLoader = new BlobLoader(message->blobDataHandle, this);
         break;
     }
   }
 }
 
@@ skipping 13 matching lines @@
     m_binaryType = BinaryTypeBlob;
     return;
   }
   if (binaryType == "arraybuffer") {
     m_binaryType = BinaryTypeArrayBuffer;
     return;
   }
   ASSERT_NOT_REACHED();
 }
 
-void PresentationConnection::didReceiveTextMessage(const String& message) {
+void PresentationConnection::didReceiveTextMessage(const WebString& message) {
   if (m_state != WebPresentationConnectionState::Connected)
     return;
 
   dispatchEvent(MessageEvent::create(message));
 }
 
 void PresentationConnection::didReceiveBinaryMessage(const uint8_t* data,
                                                      size_t length) {
   if (m_state != WebPresentationConnectionState::Connected)
     return;
@@ skipping 83 matching lines @@
   m_state = WebPresentationConnectionState::Closed;
   dispatchStateChangeEvent(PresentationConnectionCloseEvent::create(
       EventTypeNames::close, connectionCloseReasonToString(reason), message));
 }
 
 void PresentationConnection::didFinishLoadingBlob(DOMArrayBuffer* buffer) {
   ASSERT(!m_messages.isEmpty() && m_messages.first()->type == MessageTypeBlob);
   ASSERT(buffer && buffer->buffer());
   // Send the loaded blob immediately here and continue processing the queue.
   WebPresentationClient* client = presentationClient(getExecutionContext());
-  if (client)
+  if (client) {
     client->sendBlobData(m_url, m_id,
                          static_cast<const uint8_t*>(buffer->data()),
-                         buffer->byteLength());
+                         buffer->byteLength(), m_proxy.get());
+  }
 
   m_messages.removeFirst();
   m_blobLoader.clear();
   handleMessageQueue();
 }
 
 void PresentationConnection::didFailLoadingBlob(
     FileError::ErrorCode errorCode) {
   ASSERT(!m_messages.isEmpty() && m_messages.first()->type == MessageTypeBlob);
   // FIXME: generate error message?
@@ skipping 21 matching lines @@
 void PresentationConnection::tearDown() {
   // Cancel current Blob loading if any.
   if (m_blobLoader) {
     m_blobLoader->cancel();
     m_blobLoader.clear();
   }
   m_messages.clear();
 }
 
 }  // namespace blink