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

chrome/browser/dom_ui/print_preview_ui.cc

-// Copyright (c) 2010 The Chromium Authors. All rights reserved.
+// Copyright (c) 2011 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 <vector>
 
 #include "app/l10n_util.h"
 #include "app/resource_bundle.h"
 #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/printing/print_preview_manager.h"
 #include "chrome/browser/tab_contents/tab_contents.h"
 #include "chrome/common/jstemplate_builder.h"
 #include "chrome/common/url_constants.h"
 
 #include "grit/browser_resources.h"
 #include "grit/generated_resources.h"
 #include "grit/theme_resources.h"
 
 namespace {
 
@@ skipping 31 matching lines @@
 }  // namespace
 
 ////////////////////////////////////////////////////////////////////////////////
 //
 // PrintPreviewUIHTMLSource
 //
 ////////////////////////////////////////////////////////////////////////////////
 
 class PrintPreviewUIHTMLSource : public ChromeURLDataManager::DataSource {
  public:
-  PrintPreviewUIHTMLSource();
+  explicit PrintPreviewUIHTMLSource(TabContents* contents);
   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:
+  TabContents* contents_;
   DISALLOW_COPY_AND_ASSIGN(PrintPreviewUIHTMLSource);
 };
 
-PrintPreviewUIHTMLSource::PrintPreviewUIHTMLSource()
-    : DataSource(chrome::kChromeUIPrintHost, MessageLoop::current()) {
+PrintPreviewUIHTMLSource::PrintPreviewUIHTMLSource(TabContents* contents)
+    : DataSource(chrome::kChromeUIPrintHost, MessageLoop::current()),
+      contents_(contents) {
 }
 
 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.
+    printing::PrintPreviewManager::PrintPreviewData data;
+    printing::PrintPreviewManager* print_preview_manager =
+        printing::PrintPreviewManager::GetInstance();
+    if (!print_preview_manager->GetPrintPreviewData(contents_, &data)) {
+      LOG(ERROR) << "failed to get pdf";
+      scoped_refptr<RefCountedBytes> empty_bytes(new RefCountedBytes);
+      SendResponse(request_id, empty_bytes);
+      return;
+    }
+
+    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);
+  } else {
+    // Invalid request.
+    scoped_refptr<RefCountedBytes> empty_bytes(new RefCountedBytes);
+    SendResponse(request_id, empty_bytes);
   }
-
-  // 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));
 
   // 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(contents))));
 }
 
 PrintPreviewUI::~PrintPreviewUI() {
 }