Print Preview: Set a print job's title and url correctly.

chrome/browser/printing/print_view_manager.cc

 // 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/printing/print_view_manager.h"
 
 #include "base/memory/scoped_ptr.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/printing/print_job.h"
 #include "chrome/browser/printing/print_job_manager.h"
+#include "chrome/browser/printing/print_preview_tab_controller.h"
 #include "chrome/browser/printing/printer_query.h"
+#include "chrome/browser/ui/webui/print_preview_ui.h"
 #include "chrome/common/print_messages.h"
 #include "content/browser/renderer_host/render_view_host.h"
 #include "content/browser/tab_contents/navigation_entry.h"
 #include "content/browser/tab_contents/tab_contents.h"
 #include "content/common/notification_details.h"
 #include "content/common/notification_source.h"
 #include "grit/generated_resources.h"
 #include "printing/native_metafile.h"
 #include "printing/native_metafile_factory.h"
 #include "printing/printed_document.h"
 #include "ui/base/l10n/l10n_util.h"
 
 using base::TimeDelta;
 
+namespace {
+
+string16 GenerateRenderSourceName(TabContents* tab_contents) {
+  string16 name(tab_contents->GetTitle());
+  if (name.empty())
+    name = l10n_util::GetStringUTF16(IDS_DEFAULT_PRINT_DOCUMENT_TITLE);
+  return name;
+}
+
+GURL GenerateRenderSourceUrl(TabContents* tab_contents) {
+  NavigationEntry* entry = tab_contents->controller().GetActiveEntry();
+  if (entry)
+    return entry->virtual_url();
+  else

[stuartmorgan, 2011/04/01 22:34:14]

else is unnecessary here.

[Lei Zhang, 2011/04/04 22:21:03]

Copy + Paste. In any case, I'm reverting this particular change.

+    return GURL();
+}
+
+}  // namespace
+
 namespace printing {
 
 PrintViewManager::PrintViewManager(TabContents* tab_contents)
     : TabContentsObserver(tab_contents),
       number_pages_(0),
       waiting_to_print_(false),
       printing_succeeded_(false),
-      inside_inner_message_loop_(false) {
+      inside_inner_message_loop_(false),
+      overridden_title_url_(false) {
 #if defined(OS_POSIX) && !defined(OS_MACOSX)
   expecting_first_page_ = true;
 #endif
 }
 
 PrintViewManager::~PrintViewManager() {
   DisconnectFromCurrentPrintJob();
 }
 
 void PrintViewManager::StopNavigation() {
   // Cancel the current job, wait for the worker to finish.
   TerminatePrintJob(true);
 }
 
 void PrintViewManager::RenderViewGone() {
   if (!print_job_.get())
     return;
 
   scoped_refptr<PrintedDocument> document(print_job_->document());
   if (document) {
     // If IsComplete() returns false, the document isn't completely rendered.
     // Since our renderer is gone, there's nothing to do, cancel it. Otherwise,
     // the print job may finish without problem.
     TerminatePrintJob(!document->IsComplete());
   }
 }
 
+void PrintViewManager::OverrideTitleAndUrl(TabContents* tab_contents) {
+  overridden_title_url_ = true;
+  overridden_title_ = GenerateRenderSourceName(tab_contents);
+  overridden_url_ = GenerateRenderSourceUrl(tab_contents);
+}
+
 string16 PrintViewManager::RenderSourceName() {
-  string16 name(tab_contents()->GetTitle());
-  if (name.empty())
-    name = l10n_util::GetStringUTF16(IDS_DEFAULT_PRINT_DOCUMENT_TITLE);
-  return name;
+  if (overridden_title_url_)
+    return overridden_title_;
+  return GenerateRenderSourceName(tab_contents());
 }
 
 GURL PrintViewManager::RenderSourceUrl() {
-  NavigationEntry* entry = tab_contents()->controller().GetActiveEntry();
-  if (entry)
-    return entry->virtual_url();
-  else
-    return GURL();
+  if (overridden_title_url_)
+    return overridden_url_;
+  return GenerateRenderSourceUrl(tab_contents());
 }
 
 void PrintViewManager::OnDidGetPrintedPagesCount(int cookie, int number_pages) {
   DCHECK_GT(cookie, 0);
   DCHECK_GT(number_pages, 0);
   number_pages_ = number_pages;
   if (!OpportunisticallyCreatePrintJob(cookie))
     return;
 
   PrintedDocument* document = print_job_->document();
@@ skipping 343 matching lines @@
     return false;
   }
 
   // Settings are already loaded. Go ahead. This will set
   // print_job_->is_job_pending() to true.
   print_job_->StartPrinting();
   return true;
 }
 
 }  // namespace printing