PrintPreview: Print Preview is not staying associated with initiator renderer.

chrome/browser/ui/webui/print_preview_ui.h

Index: chrome/browser/ui/webui/print_preview_ui.h
diff --git a/chrome/browser/ui/webui/print_preview_ui.h b/chrome/browser/ui/webui/print_preview_ui.h
index 99a887873c075d237e4117c5391e0dc4594ec562..dbb5f0212144fc790e77993264408816a11531ba 100644
--- a/chrome/browser/ui/webui/print_preview_ui.h
+++ b/chrome/browser/ui/webui/print_preview_ui.h
@@ -9,16 +9,22 @@
 #include <string>
 
 #include "base/memory/ref_counted.h"
+#include "base/memory/ref_counted_memory.h"
+#include "chrome/browser/printing/print_preview_data_service.h"

[Lei Zhang, 2011/05/29 00:24:23]

forward declare instead?

[kmadhusu, 2011/05/29 00:45:40]

Done.

 #include "content/browser/webui/web_ui.h"
 
-class PrintPreviewUIHTMLSource;
-
 class PrintPreviewUI : public WebUI {
  public:
   explicit PrintPreviewUI(TabContents* contents);
   virtual ~PrintPreviewUI();
 
-  PrintPreviewUIHTMLSource* html_source();
+  // Gets the print preview |data|. The data is valid as long as the
+  // PrintPreviewDataService is valid and SetPrintPreviewData() does not get
+  // called.
+  void GetPrintPreviewData(scoped_refptr<RefCountedBytes>* data);
+
+  // Sets the print preview |data|.
+  void SetPrintPreviewData(const RefCountedBytes* data);
 
   // Notify the Web UI renderer that preview data is available.
   // |expected_pages_count| specifies the total number of pages.
@@ -36,7 +42,11 @@ class PrintPreviewUI : public WebUI {
   void OnInitiatorTabClosed(const std::string& initiator_tab_url);
 
  private:
-  scoped_refptr<PrintPreviewUIHTMLSource> html_source_;
+  // Helper function
+  PrintPreviewDataService* print_preview_data_service();
+
+  // Store the PrintPreviewUI address string.
+  std::string preview_ui_addr_str_;
 
   DISALLOW_COPY_AND_ASSIGN(PrintPreviewUI);
 };