Enable page selection for WebView printing

android_webview/browser/aw_print_manager.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 "android_webview/browser/aw_print_manager.h"
 
 #include "components/printing/browser/print_manager_utils.h"
-#include "components/printing/common/print_messages.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/render_frame_host.h"
 
 DEFINE_WEB_CONTENTS_USER_DATA_KEY(android_webview::AwPrintManager);
 
 namespace android_webview {
 
 // static
 AwPrintManager* AwPrintManager::CreateForWebContents(
     content::WebContents* contents,
@@ skipping 27 matching lines @@
   return rfh->Send(new PrintMsg_PrintPages(rfh->GetRoutingID()));
 }
 
 bool AwPrintManager::OnMessageReceived(
     const IPC::Message& message,
     content::RenderFrameHost* render_frame_host) {
   bool handled = true;
   IPC_BEGIN_MESSAGE_MAP_WITH_PARAM(AwPrintManager, message, render_frame_host)
     IPC_MESSAGE_HANDLER_WITH_PARAM_DELAY_REPLY(
         PrintHostMsg_GetDefaultPrintSettings, OnGetDefaultPrintSettings)
+    IPC_MESSAGE_HANDLER_WITH_PARAM_DELAY_REPLY(PrintHostMsg_ScriptedPrint,
+                                               OnScriptedPrint)
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP()
   return handled ? true
                  : PrintManager::OnMessageReceived(message, render_frame_host);
 }
 
 void AwPrintManager::OnGetDefaultPrintSettings(
     content::RenderFrameHost* render_frame_host,
     IPC::Message* reply_msg) {
   // Unlike the printing_message_filter, we do process this in UI thread.
   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
   PrintMsg_Print_Params params;
   printing::RenderParamsFromPrintSettings(settings_, &params);
   params.document_cookie = cookie_;
   PrintHostMsg_GetDefaultPrintSettings::WriteReplyParams(reply_msg, params);
   render_frame_host->Send(reply_msg);
 }
 
+void AwPrintManager::OnScriptedPrint(
+    content::RenderFrameHost* render_frame_host,
+    const PrintHostMsg_ScriptedPrint_Params& scripted_params,
+    IPC::Message* reply_msg) {
+  DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
+  PrintMsg_PrintPages_Params params;
+  printing::RenderParamsFromPrintSettings(settings_, &params.params);
+  params.params.document_cookie = scripted_params.cookie;
+  params.pages = printing::PageRange::GetPages(settings_.ranges());
+  PrintHostMsg_UpdatePrintSettings::WriteReplyParams(reply_msg, params, false);

[Lei Zhang, 2017/03/22 22:26:43]

Shouldn't this be PrintHostMsg_ScriptedPrint::WriteReplyParams() ?

[Shimi Zhang, 2017/03/23 23:21:56]

Done.

+  render_frame_host->Send(reply_msg);
+}
+
 }  // namespace android_webview