Modify pdf_scripting_api.js to support postMessage using BrowserPlugin

chrome/browser/resources/pdf/pdf_scripting_api.js

 // 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.
 
 /**
  * Create a new PDFScriptingAPI. This provides a scripting interface to
  * the PDF viewer so that it can be customized by things like print preview.
  * @param {Window} window the window of the page containing the pdf viewer.
- * @param {string} extensionUrl the url of the PDF extension.
+ * @param {Object} plugin the plugin element containing the pdf viewer.
  */
-function PDFScriptingAPI(window, extensionUrl) {
-  this.extensionUrl_ = extensionUrl;
-  this.messageQueue_ = [];
+function PDFScriptingAPI(window, plugin) {
+  this.loaded_ = false;
+  this.pendingScriptingMessages_ = [];
+  this.setPlugin(plugin);
+
   window.addEventListener('message', function(event) {
-    if (event.origin != this.extensionUrl_) {
+    if (event.origin != 'chrome-extension://mhjfbmdgcfjbbpaeojofohoefgiehjai') {
       console.error('Received message that was not from the extension: ' +
                     event);
       return;
     }
     switch (event.data.type) {
-      case 'readyToReceive':
-        this.setDestinationWindow(event.source);
-        break;
       case 'viewport':
         if (this.viewportChangedCallback_)
           this.viewportChangedCallback_(event.data.pageX,
                                         event.data.pageY,
                                         event.data.pageWidth,
                                         event.data.viewportWidth,
                                         event.data.viewportHeight);
         break;
       case 'documentLoaded':
+        this.loaded_ = true;
         if (this.loadCallback_)
           this.loadCallback_();
         break;
       case 'getAccessibilityJSONReply':
         if (this.accessibilityCallback_) {
           this.accessibilityCallback_(event.data.json);
           this.accessibilityCallback_ = null;
         }
         break;
     }
   }.bind(this), false);
 }
 
 PDFScriptingAPI.prototype = {
   /**
    * @private
-   * Send a message to the extension. If we try to send messages prior to the
-   * extension being ready to receive messages (i.e. before it has finished
-   * loading) we queue up the messages and flush them later.
+   * Send a message to the extension. If messages try to get sent before there
+   * is a plugin element set, then we queue them up and send them later (this
+   * can happen in print preview).
    * @param {Object} message The message to send.
    */
   sendMessage_: function(message) {
-    if (!this.pdfWindow_) {
-      this.messageQueue_.push(message);
-      return;
-    }
-
-    this.pdfWindow_.postMessage(message, this.extensionUrl_);
+    if (this.plugin_)
+      this.plugin_.postMessage(message, '*');
+    else
+      this.pendingScriptingMessages_.push(message);
   },
 
-  /**
-   * Sets the destination window containing the PDF viewer. This will be called
-   * when a 'readyToReceive' message is received from the PDF viewer or it can
-   * be called during tests. It then flushes any pending messages to the window.
-   * @param {Window} pdfWindow the window containing the PDF viewer.
-   */
-  setDestinationWindow: function(pdfWindow) {
-    this.pdfWindow_ = pdfWindow;
-    while (this.messageQueue_.length != 0) {
-      this.pdfWindow_.postMessage(this.messageQueue_.shift(),
-                                  this.extensionUrl_);
+ /**
+  * Sets the plugin element containing the PDF viewer. The element will usually
+  * be passed into the PDFScriptingAPI constructor but may also be set later.
+  * @param {Object} plugin the plugin element containing the PDF viewer.
+  */
+  setPlugin: function(plugin) {
+    this.plugin_ = plugin;
+
+    // Send an initialization message to the plugin indicating the window to
+    // respond to.
+    if (this.plugin_) {
+      this.sendMessage_({
+        type: 'setParentWindow'
+      });
+
+      // Now we can flush pending messages
+      while (this.pendingScriptingMessages_.length > 0)
+        this.sendMessage_(this.pendingScriptingMessages_.shift());
     }
   },
 
   /**
    * Sets the callback which will be run when the PDF viewport changes.
    * @param {Function} callback the callback to be called.
    */
   setViewportChangedCallback: function(callback) {
     this.viewportChangedCallback_ = callback;
   },
 
   /**
    * Sets the callback which will be run when the PDF document has finished
-   * loading.
+   * loading. If the document is already loaded, it will be run immediately.
    * @param {Function} callback the callback to be called.
    */
   setLoadCallback: function(callback) {
     this.loadCallback_ = callback;
+    if (this.loaded_ && callback)
+      callback();
   },
 
   /**
    * Resets the PDF viewer into print preview mode.
    * @param {string} url the url of the PDF to load.
    * @param {boolean} grayscale whether or not to display the PDF in grayscale.
    * @param {Array.<number>} pageNumbers an array of the page numbers.
    * @param {boolean} modifiable whether or not the document is modifiable.
    */
   resetPrintPreviewMode: function(url, grayscale, pageNumbers, modifiable) {
@@ skipping 56 matching lines @@
 
 /**
  * Creates a PDF viewer with a scripting interface. This is basically 1) an
  * iframe which is navigated to the PDF viewer extension and 2) a scripting
  * interface which provides access to various features of the viewer for use
  * by print preview and accessibility.
  * @param {string} src the source URL of the PDF to load initially.
  * @return {HTMLIFrameElement} the iframe element containing the PDF viewer.
  */
 function PDFCreateOutOfProcessPlugin(src) {
-  var EXTENSION_URL = 'chrome-extension://mhjfbmdgcfjbbpaeojofohoefgiehjai';
   var iframe = window.document.createElement('iframe');
-  iframe.setAttribute('src', EXTENSION_URL + '/index.html?' + src);
-  var client = new PDFScriptingAPI(window, EXTENSION_URL);
-
+  iframe.setAttribute(
+      'src',
+      'chrome-extension://mhjfbmdgcfjbbpaeojofohoefgiehjai/index.html?' + src);
+  var client = new PDFScriptingAPI(window);
+  iframe.onload = function() {
+    client.setPlugin(iframe.contentWindow);
+  };
   // Add the functions to the iframe so that they can be called directly.
   iframe.setViewportChangedCallback =
       client.setViewportChangedCallback.bind(client);
   iframe.setLoadCallback = client.setLoadCallback.bind(client);
   iframe.resetPrintPreviewMode = client.resetPrintPreviewMode.bind(client);
   iframe.loadPreviewPage = client.loadPreviewPage.bind(client);
   iframe.sendKeyEvent = client.sendKeyEvent.bind(client);
   return iframe;
 }