Print Preview: Store preview data in the print preview handler....

chrome/browser/dom_ui/print_preview_ui.cc

 // Copyright (c) 2010 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 "chrome/browser/dom_ui/print_preview_ui.h"
 
 #include <algorithm>
+#include <string>
+#include <vector>
 
 #include "base/message_loop.h"
+#include "base/shared_memory.h"
 #include "base/singleton.h"
 #include "base/string_piece.h"
 #include "base/values.h"
 #include "chrome/browser/browser_thread.h"
 #include "chrome/browser/dom_ui/chrome_url_data_manager.h"
 #include "chrome/browser/dom_ui/dom_ui_theme_source.h"
 #include "chrome/browser/dom_ui/print_preview_handler.h"
 #include "chrome/browser/tab_contents/tab_contents.h"
 #include "chrome/common/jstemplate_builder.h"
 #include "chrome/common/url_constants.h"
@@ skipping 39 matching lines @@
 }  // namespace
 
 ////////////////////////////////////////////////////////////////////////////////
 //
 // PrintPreviewUIHTMLSource
 //
 ////////////////////////////////////////////////////////////////////////////////
 
 class PrintPreviewUIHTMLSource : public ChromeURLDataManager::DataSource {
  public:
-  PrintPreviewUIHTMLSource();
+  explicit PrintPreviewUIHTMLSource(PrintPreviewHandler* handler);
   virtual ~PrintPreviewUIHTMLSource();
 
   // Called when the network layer has requested a resource underneath
   // the path we registered.
   virtual void StartDataRequest(const std::string& path,
                                 bool is_off_the_record,
                                 int request_id);
   virtual std::string GetMimeType(const std::string&) const;
 
  private:
+  PrintPreviewHandler* handler_;
+
   DISALLOW_COPY_AND_ASSIGN(PrintPreviewUIHTMLSource);
 };
 
-PrintPreviewUIHTMLSource::PrintPreviewUIHTMLSource()
-    : DataSource(chrome::kChromeUIPrintHost, MessageLoop::current()) {
+PrintPreviewUIHTMLSource::PrintPreviewUIHTMLSource(PrintPreviewHandler* handler)

[James Hawkins, 2011/01/31 18:39:38]

After reading through the flow, it seems the best place to store
PrintPreviewData is actually PrintPreviewUIHTMLSource. That will simplify things
a bit, and get rid of a level of indirection by not having to store a pointer to
PrintPreviewHandler here.

+    : DataSource(chrome::kChromeUIPrintHost, MessageLoop::current()),
+      handler_(handler) {
 }
 
 PrintPreviewUIHTMLSource::~PrintPreviewUIHTMLSource() {}
 
 void PrintPreviewUIHTMLSource::StartDataRequest(const std::string& path,
                                                 bool is_off_the_record,
                                                 int request_id) {
-  // Print Preview Index page.
   if (path.empty()) {
+    // Print Preview Index page.
     DictionaryValue localized_strings;
     SetLocalizedStrings(&localized_strings);
     SetFontAndTextDirection(&localized_strings);
 
     static const base::StringPiece print_html(
         ResourceBundle::GetSharedInstance().GetRawDataResource(
             IDR_PRINT_PREVIEW_HTML));
     const std::string full_html = jstemplate_builder::GetI18nTemplateHtml(
         print_html, &localized_strings);
 
     scoped_refptr<RefCountedBytes> html_bytes(new RefCountedBytes);
     html_bytes->data.resize(full_html.size());
     std::copy(full_html.begin(), full_html.end(), html_bytes->data.begin());
 
     SendResponse(request_id, html_bytes);
     return;
+  } else if (path == "print.pdf") {
+    // Print Preview data.
+    PrintPreviewHandler::PrintPreviewData data;
+    handler_->GetPrintPreviewData(&data);
+
+    uint32 preview_data_size = data.second;
+    char* preview_data = reinterpret_cast<char*>(data.first->memory());
+
+    scoped_refptr<RefCountedBytes> html_bytes(new RefCountedBytes);
+    html_bytes->data.resize(preview_data_size);
+    std::vector<unsigned char>::iterator it = html_bytes->data.begin();
+    for (uint32 i = 0; i < preview_data_size; ++i, ++it)
+      *it = *(preview_data + i);
+    SendResponse(request_id, html_bytes);
+    return;
+  } else {
+    // Invalid request.
+    scoped_refptr<RefCountedBytes> empty_bytes(new RefCountedBytes);
+    SendResponse(request_id, empty_bytes);
+    return;
   }
-
-  // Print Preview data.
-  NOTIMPLEMENTED() << "No PDF for you";
-  scoped_refptr<RefCountedBytes> data_bytes(new RefCountedBytes);
-  SendResponse(request_id, data_bytes);
 }
 
 std::string PrintPreviewUIHTMLSource::GetMimeType(
     const std::string& path) const {
   // Print Preview Index page.
   if (path.empty())
     return "text/html";
   // Print Preview data.
   return "application/pdf";
 }
 
 ////////////////////////////////////////////////////////////////////////////////
 //
 // PrintPreviewUI
 //
 ////////////////////////////////////////////////////////////////////////////////
 
 PrintPreviewUI::PrintPreviewUI(TabContents* contents) : DOMUI(contents) {
   // PrintPreviewUI owns |handler|.
-  PrintPreviewHandler* handler = new PrintPreviewHandler();
-  AddMessageHandler(handler->Attach(this));
+  handler_ = new PrintPreviewHandler();
+  AddMessageHandler(handler_->Attach(this));
 
   // Set up the chrome://print/ source.
   BrowserThread::PostTask(
       BrowserThread::IO, FROM_HERE,
       NewRunnableMethod(
           ChromeURLDataManager::GetInstance(),
           &ChromeURLDataManager::AddDataSource,
-          make_scoped_refptr(new PrintPreviewUIHTMLSource())));
+          make_scoped_refptr(new PrintPreviewUIHTMLSource(handler_))));
 }
 
 PrintPreviewUI::~PrintPreviewUI() {
 }
+
+PrintPreviewHandler* PrintPreviewUI::handler() {
+  return handler_;
+}