Make printing work better with OOPIF. (try 2)

chrome/browser/printing/print_view_manager_base.h

 // Copyright 2013 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.
 
 #ifndef CHROME_BROWSER_PRINTING_PRINT_VIEW_MANAGER_BASE_H_
 #define CHROME_BROWSER_PRINTING_PRINT_VIEW_MANAGER_BASE_H_
 
+#include <memory>
+
 #include "base/macros.h"
 #include "base/memory/ref_counted.h"
 #include "base/strings/string16.h"
 #include "build/build_config.h"
 #include "components/prefs/pref_member.h"
 #include "components/printing/browser/print_manager.h"
 #include "content/public/browser/notification_observer.h"
 #include "content/public/browser/notification_registrar.h"
 #include "printing/features/features.h"
 #include "printing/printed_pages_source.h"
 
 struct PrintHostMsg_DidPrintPage_Params;
 
+namespace content {
+class RenderFrameHost;
+}
+
 namespace printing {
 
 class JobEventDetails;
 class MetafilePlayer;
 class PrintJob;
 class PrintJobWorkerOwner;
 class PrintQueriesQueue;
 
 // Base class for managing the print commands for a WebContents.
 class PrintViewManagerBase : public content::NotificationObserver,
                              public PrintedPagesSource,
                              public PrintManager {
  public:
   ~PrintViewManagerBase() override;
 
 #if BUILDFLAG(ENABLE_BASIC_PRINTING)
   // Prints the current document immediately. Since the rendering is
   // asynchronous, the actual printing will not be completed on the return of
   // this function. Returns false if printing is impossible at the moment.
-  virtual bool PrintNow();
+  virtual bool PrintNow(content::RenderFrameHost* rfh);
 #endif  // ENABLE_BASIC_PRINTING
 
-  // Whether to block scripted printing for our tab or not.
-  void UpdateScriptedPrintingBlocked();
+  // Whether printing is enabled or not.
+  void UpdatePrintingEnabled();
 
   // PrintedPagesSource implementation.
   base::string16 RenderSourceName() override;
 
  protected:
   explicit PrintViewManagerBase(content::WebContents* web_contents);
 
   // Helper method for Print*Now().
-  bool PrintNowInternal(IPC::Message* message);
+  bool PrintNowInternal(content::RenderFrameHost* rfh,
+                        std::unique_ptr<IPC::Message> message);
 
-  // Terminates or cancels the print job if one was pending.
-  void RenderProcessGone(base::TerminationStatus status) override;
+  void SetPrintingRFH(content::RenderFrameHost* rfh);
 
   // content::WebContentsObserver implementation.
-  bool OnMessageReceived(const IPC::Message& message) override;
+  void RenderFrameDeleted(content::RenderFrameHost* render_frame_host) override;
+  bool OnMessageReceived(const IPC::Message& message,
+                         content::RenderFrameHost* render_frame_host) override;
 
  private:
   // content::NotificationObserver implementation.
   void Observe(int type,
                const content::NotificationSource& source,
                const content::NotificationDetails& details) override;
 
   // content::WebContentsObserver implementation.
   void DidStartLoading() override;
 
@@ skipping 24 matching lines @@
   // Creates a new empty print job. It has no settings loaded. If there is
   // currently a print job, safely disconnect from it. Returns false if it is
   // impossible to safely disconnect from the current print job or it is
   // impossible to create a new print job.
   bool CreateNewPrintJob(PrintJobWorkerOwner* job);
 
   // Makes sure the current print_job_ has all its data before continuing, and
   // disconnect from it.
   void DisconnectFromCurrentPrintJob();
 
-  // Notify that the printing is done.
-  void PrintingDone(bool success);
-
   // Terminates the print job. No-op if no print job has been created. If
   // |cancel| is true, cancel it instead of waiting for the job to finish. Will
   // call ReleasePrintJob().
   void TerminatePrintJob(bool cancel);
 
   // Releases print_job_. Correctly deregisters from notifications. No-op if
   // no print job has been created.
   void ReleasePrintJob();
 
   // Runs an inner message loop. It will set inside_inner_message_loop_ to true
   // while the blocking inner message loop is running. This is useful in cases
   // where the RenderView is about to be destroyed while a printing job isn't
   // finished.
   bool RunInnerMessageLoop();
 
   // In the case of Scripted Printing, where the renderer is controlling the
   // control flow, print_job_ is initialized whenever possible. No-op is
   // print_job_ is initialized.
   bool OpportunisticallyCreatePrintJob(int cookie);
 
   // Release the PrinterQuery associated with our |cookie_|.
   void ReleasePrinterQuery();
 
+  // Helper method for UpdatePrintingEnabled().
+  void SendPrintingEnabled(bool enabled, content::RenderFrameHost* rfh);
+
   content::NotificationRegistrar registrar_;
 
+  // The current RFH that is printing with a system printing dialog.
+  content::RenderFrameHost* printing_rfh_;
+
   // Manages the low-level talk to the printer.
   scoped_refptr<PrintJob> print_job_;
 
   // Indication of success of the print job.
   bool printing_succeeded_;
 
   // Running an inner message loop inside RenderAllMissingPagesNow(). This means
   // we are _blocking_ until all the necessary pages have been rendered or the
   // print settings are being loaded.
   bool inside_inner_message_loop_;
 
 #if !defined(OS_MACOSX)
   // Set to true when OnDidPrintPage() should be expecting the first page.
   bool expecting_first_page_;
 #endif
 
   // Whether printing is enabled.
   BooleanPrefMember printing_enabled_;
 
   scoped_refptr<printing::PrintQueriesQueue> queue_;
 
   DISALLOW_COPY_AND_ASSIGN(PrintViewManagerBase);
 };
 
 }  // namespace printing
 
 #endif  // CHROME_BROWSER_PRINTING_PRINT_VIEW_MANAGER_BASE_H_