Add support for starting an extension print job to print preview

chrome/browser/ui/webui/print_preview/print_preview_handler.cc

 // Copyright (c) 2012 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/ui/webui/print_preview/print_preview_handler.h"
 
 #include <ctype.h>
 
 #include <map>
 #include <string>
@@ skipping 828 matching lines @@
     return;
 
   ReportPrintSettingsStats(*settings);
 
   // Never try to add headers/footers here. It's already in the generated PDF.
   settings->SetBoolean(printing::kSettingHeaderFooterEnabled, false);
 
   bool print_to_pdf = false;
   bool is_cloud_printer = false;
   bool print_with_privet = false;
+  bool print_with_extension = false;
 
   bool open_pdf_in_preview = false;
 #if defined(OS_MACOSX)
   open_pdf_in_preview = settings->HasKey(printing::kSettingOpenPDFInPreview);
 #endif
 
   if (!open_pdf_in_preview) {
     settings->GetBoolean(printing::kSettingPrintToPDF, &print_to_pdf);
     settings->GetBoolean(printing::kSettingPrintWithPrivet, &print_with_privet);
+    settings->GetBoolean(printing::kSettingPrintWithExtension,
+                         &print_with_extension);
     is_cloud_printer = settings->HasKey(printing::kSettingCloudPrintId);
   }
 
   int page_count = 0;
   settings->GetInteger(printing::kSettingPreviewPageCount, &page_count);
 
   if (print_to_pdf) {
     UMA_HISTOGRAM_COUNTS("PrintPreview.PageCount.PrintToPDF", page_count);
     ReportUserActionHistogram(PRINT_TO_PDF);
     PrintToPdf();
@@ skipping 21 matching lines @@
       web_ui()->CallJavascriptFunction("onPrivetPrintFailed", http_code_value);
       return;
     }
 
     PrintToPrivetPrinter(
         printer_name, print_ticket, capabilities, gfx::Size(width, height));
     return;
   }
 #endif
 
+  if (print_with_extension) {
+    // TODO(tbarzic): Implement this and record UMA stats.
+    OnExtensionPrintResult(false, "NOT_IMPLEMENTED");
+    return;
+  }
+
   scoped_refptr<base::RefCountedBytes> data;
   base::string16 title;
   if (!GetPreviewDataAndTitle(&data, &title)) {
     // Nothing to print, no preview available.
     return;
   }
 
   if (is_cloud_printer) {
     UMA_HISTOGRAM_COUNTS("PrintPreview.PageCount.PrintToCloudPrint",
                          page_count);
@@ skipping 740 matching lines @@
   if (capabilities.empty()) {
     web_ui()->CallJavascriptFunction("failedToGetExtensionPrinterCapabilities",
                                      base::StringValue(printer_id));
     return;
   }
 
   web_ui()->CallJavascriptFunction("onExtensionCapabilitiesSet",
                                    base::StringValue(printer_id), capabilities);
 }
 
+void PrintPreviewHandler::OnExtensionPrintResult(bool success,
+                                                 const std::string& status) {
+  if (success) {
+    ClosePreviewDialog();
+    return;
+  }
+
+  // TODO(tbarzic): This function works for extension printers case too, but it
+  // should be renamed to something more generic.
+  web_ui()->CallJavascriptFunction("onPrivetPrintFailed",
+                                   base::StringValue(status));
+}
+
 void PrintPreviewHandler::RegisterForMergeSession() {
   DCHECK(!reconcilor_);
   Profile* profile = Profile::FromWebUI(web_ui());
   if (switches::IsEnableAccountConsistency() && !profile->IsOffTheRecord()) {
     reconcilor_ = AccountReconcilorFactory::GetForProfile(profile);
     if (reconcilor_)
       reconcilor_->AddMergeSessionObserver(this);
   }
 }
 
 void PrintPreviewHandler::UnregisterForMergeSession() {
   if (reconcilor_)
     reconcilor_->RemoveMergeSessionObserver(this);
 }
 
 void PrintPreviewHandler::SetPdfSavedClosureForTesting(
     const base::Closure& closure) {
   pdf_file_saved_closure_ = closure;
 }