[PresentationAPI] Implementing send() from WebPresentationClient to the PresentationService.

content/browser/presentation/presentation_service_impl.cc

 // Copyright 2015 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 "content/browser/presentation/presentation_service_impl.h"
 
 #include <algorithm>
 
 #include "base/logging.h"
 #include "content/browser/presentation/presentation_type_converters.h"
 #include "content/public/browser/content_browser_client.h"
 #include "content/public/browser/navigation_details.h"
 #include "content/public/browser/presentation_session_message.h"
 #include "content/public/browser/render_frame_host.h"
 #include "content/public/browser/render_process_host.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/common/content_client.h"
 #include "content/public/common/frame_navigate_params.h"
+#include "content/public/common/presentation_constants.h"
 
 namespace {
 
 // The return value takes ownership of the contents of |input|.
 presentation::SessionMessagePtr ToMojoSessionMessage(
     content::PresentationSessionMessage* input) {
   presentation::SessionMessagePtr output(presentation::SessionMessage::New());
   output->presentation_url.Swap(&input->presentation_url);
   output->presentation_id.Swap(&input->presentation_id);
   if (input->is_binary()) {
     // binary data
     output->type = presentation::PresentationMessageType::
         PRESENTATION_MESSAGE_TYPE_ARRAY_BUFFER;
     output->data.Swap(input->data.get());
   } else {
     // string message
     output->type =
         presentation::PresentationMessageType::PRESENTATION_MESSAGE_TYPE_TEXT;
     output->message.Swap(input->message.get());
   }
   return output.Pass();
 }
 
-}  // namespace
+scoped_ptr<content::PresentationSessionMessage> GetPresentationSessionMessage(
+    presentation::SessionMessagePtr input) {
+  DCHECK(!input.is_null());
+  scoped_ptr<content::PresentationSessionMessage> output;
+  if (input->type == presentation::PresentationMessageType::
+                     PRESENTATION_MESSAGE_TYPE_TEXT) {
+    DCHECK(!input->message.is_null());
+    DCHECK(input->data.is_null());
+    // Return null PresentationSessionMessage if size exceeds.
+    if (input->message.size() > content::kMaxPresentationSessionMessageSize)
+      return output.Pass();
+
+    output = content::PresentationSessionMessage::CreateStringMessage(
+        input->presentation_url,
+        input->presentation_id,
+        make_scoped_ptr(new std::string));
+    input->message.Swap(output->message.get());
+
+  } else if (input->type == presentation::PresentationMessageType::
+              PRESENTATION_MESSAGE_TYPE_ARRAY_BUFFER) {
+    DCHECK(!input->data.is_null());
+    DCHECK(input->message.is_null());
+    // Return null PresentationSessionMessage if size exceeds.
+    if (input->data.size() > content::kMaxPresentationSessionMessageSize)
+      return output.Pass();
+
+    output = content::PresentationSessionMessage::CreateBinaryMessage(
+        input->presentation_url,
+        input->presentation_id,
+        make_scoped_ptr(new std::vector<uint8_t>));
+    input->data.Swap(output->data.get());
+  }
+
+  return output.Pass();
+}
+
+} // namespace
 
 namespace content {
 
 PresentationServiceImpl::PresentationServiceImpl(
     RenderFrameHost* render_frame_host,
     WebContents* web_contents,
     PresentationServiceDelegate* delegate)
     : WebContentsObserver(web_contents),
       delegate_(delegate),
       is_start_session_pending_(false),
@@ skipping 278 matching lines @@
 
   // Remove listener for old default presentation URL.
   delegate_->RemoveScreenAvailabilityListener(
       render_process_id_,
       render_frame_id_,
       old_it->second.get());
   availability_contexts_.erase(old_it);
   DoSetDefaultPresentationUrl(new_default_url, default_presentation_id);
 }
 
+
+void PresentationServiceImpl::SendSessionMessage(
+    presentation::SessionMessagePtr session_message,
+    const SendMessageMojoCallback& callback) {
+  DVLOG(2) << "SendSessionMessage";
+  DCHECK(!session_message.is_null());
+  // send_message_callback_ should be null by now, otherwise resetting of
+  // send_message_callback_ with new callback will drop the old callback.
+  if (!delegate_ || send_message_callback_) {
+    callback.Run(false);
+    return;
+  }
+
+  send_message_callback_.reset(new SendMessageMojoCallback(callback));
+  delegate_->SendMessage(
+      render_process_id_,
+      render_frame_id_,
+      GetPresentationSessionMessage(session_message.Pass()),
+      base::Bind(&PresentationServiceImpl::OnSendMessageCallback,
+                 weak_factory_.GetWeakPtr()));
+}
+
+void PresentationServiceImpl::OnSendMessageCallback() {
+  // It is possible that Reset() is invoked before receiving this callback.
+  // So, always check send_message_callback_ for non-null.
+  if (send_message_callback_) {
+    send_message_callback_->Run(true);
+    send_message_callback_.reset();
+  }
+}
+
 void PresentationServiceImpl::CloseSession(
     const mojo::String& presentation_url,
     const mojo::String& presentation_id) {
   NOTIMPLEMENTED();
 }
 
 void PresentationServiceImpl::ListenForSessionStateChange(
     const SessionStateCallback& callback) {
   NOTIMPLEMENTED();
 }
@@ skipping 90 matching lines @@
   default_presentation_id_.clear();
   availability_contexts_.clear();
   queued_start_session_requests_.clear();
   FlushNewSessionCallbacks();
   default_session_start_context_.reset();
   if (on_session_messages_callback_.get()) {
     on_session_messages_callback_->Run(
         mojo::Array<presentation::SessionMessagePtr>());
     on_session_messages_callback_.reset();
   }
+  if (send_message_callback_) {
+    // Run the callback with false, indicating the renderer to stop sending
+    // the requests and invalidate all pending requests.
+    send_message_callback_->Run(false);
+    send_message_callback_.reset();
+  }
 }
 
 // static
 void PresentationServiceImpl::InvokeNewSessionMojoCallbackWithError(
     const NewSessionMojoCallback& callback) {
   callback.Run(
         presentation::PresentationSessionInfoPtr(),
         presentation::PresentationError::From(
             PresentationError(PRESENTATION_ERROR_UNKNOWN, "Internal error")));
 }
@@ skipping 129 matching lines @@
 void PresentationServiceImpl::DefaultSessionStartContext::Reset() {
   ScopedVector<DefaultSessionMojoCallback> callbacks;
   callbacks.swap(callbacks_);
   for (const auto& callback : callbacks)
     callback->Run(presentation::PresentationSessionInfoPtr());
   session_.reset();
 }
 
 }  // namespace content