Convert local printer capabilites to CDD format to unify with cloud printers and PDF printer. This …

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 27 matching lines @@
 #include "chrome/browser/signin/profile_oauth2_token_service_factory.h"
 #include "chrome/browser/signin/signin_manager_factory.h"
 #include "chrome/browser/ui/browser_finder.h"
 #include "chrome/browser/ui/browser_tabstrip.h"
 #include "chrome/browser/ui/chrome_select_file_policy.h"
 #include "chrome/browser/ui/scoped_tabbed_browser_displayer.h"
 #include "chrome/browser/ui/webui/print_preview/print_preview_ui.h"
 #include "chrome/browser/ui/webui/print_preview/sticky_settings.h"
 #include "chrome/common/chrome_paths.h"
 #include "chrome/common/chrome_switches.h"
+#include "chrome/common/cloud_print/cloud_print_cdd_conversion.h"
 #include "chrome/common/cloud_print/cloud_print_constants.h"
 #include "chrome/common/crash_keys.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/common/print_messages.h"
 #include "components/cloud_devices/common/cloud_device_description.h"
 #include "components/cloud_devices/common/cloud_devices_urls.h"
 #include "components/cloud_devices/common/printer_description.h"
 #include "components/signin/core/browser/profile_oauth2_token_service.h"
 #include "components/signin/core/browser/signin_manager.h"
 #include "components/signin/core/browser/signin_manager_base.h"
@@ skipping 284 matching lines @@
   for (size_t i = 0; i < arraysize(kPdfMedia); ++i) {
     Media media_option(kPdfMedia[i]);
     media.AddDefaultOption(media_option,
                            default_media.type == media_option.type);
   }
   media.SaveTo(&description);
 
   return scoped_ptr<base::DictionaryValue>(description.root().DeepCopy());
 }
 
+scoped_ptr<base::DictionaryValue> GetLocalPrinterCapabilitiesOnFileThread(
+    const std::string& printer_name) {
+  DCHECK_CURRENTLY_ON(BrowserThread::FILE);
+
+  scoped_refptr<printing::PrintBackend> print_backend(
+      printing::PrintBackend::CreateInstance(NULL));
+
+  VLOG(1) << "Get printer capabilities start for " << printer_name;
+  crash_keys::ScopedPrinterInfo crash_key(
+      print_backend->GetPrinterDriverInfo(printer_name));
+
+  if (!print_backend->IsValidPrinter(printer_name)) {
+    LOG(WARNING) << "Invalid printer " << printer_name;
+    return scoped_ptr<base::DictionaryValue>();
+  }
+
+  printing::PrinterSemanticCapsAndDefaults info;
+  if (!print_backend->GetPrinterSemanticCapsAndDefaults(printer_name, &info)) {
+    LOG(WARNING) << "Failed to get capabilities for " << printer_name;
+    return scoped_ptr<base::DictionaryValue>();
+  }
+
+  scoped_ptr<base::DictionaryValue> description(
+      cloud_print::PrinterSemanticCapsAndDefaultsToCdd(info));
+  if (description == NULL) {

[Vitaly Buka (NO REVIEWS), 2014/06/04 00:05:46]

if (!description)

[Aleksey Shlyapnikov, 2014/06/04 02:02:47]

Done.

+    LOG(WARNING) << "Failed to convert capabilities for " << printer_name;
+    return scoped_ptr<base::DictionaryValue>();
+  }
+
+#if defined(USE_CUPS)
+  // TODO(alekseys): Use CUSTOM_COLOR/MONOCHROME instead.
+  description->SetInteger(kCUPSsColorModel, info.color_model);
+  description->SetInteger(kCUPSsBWModel, info.bw_model);
+#endif
+  return description.Pass();
+}
+
 void EnumeratePrintersOnFileThread(base::ListValue* printers) {
   DCHECK_CURRENTLY_ON(BrowserThread::FILE);
 
   scoped_refptr<printing::PrintBackend> print_backend(
       printing::PrintBackend::CreateInstance(NULL));
 
   VLOG(1) << "Enumerate printers start";
   printing::PrinterList printer_list;
   print_backend->EnumeratePrinters(&printer_list);
 
@@ skipping 41 matching lines @@
     GetPrinterCapabilitiesFailureCallback;
 
 void GetPrinterCapabilitiesOnFileThread(
     const std::string& printer_name,
     const std::string& locale,
     const GetPrinterCapabilitiesSuccessCallback& success_cb,
     const GetPrinterCapabilitiesFailureCallback& failure_cb) {
   DCHECK_CURRENTLY_ON(BrowserThread::FILE);
   DCHECK(!printer_name.empty());
 
-  // Special case for PDF printer.
-  if (printer_name == kLocalPdfPrinterId) {
-    scoped_ptr<base::DictionaryValue> printer_info(new base::DictionaryValue);
-    printer_info->SetString(kPrinterId, printer_name);
-    printer_info->Set(kPrinterCapabilities,
-                      GetPdfCapabilitiesOnFileThread(locale).release());
-    BrowserThread::PostTask(
-        BrowserThread::UI, FROM_HERE,
-        base::Bind(success_cb, base::Owned(printer_info.release())));
-    return;
-  }
-
-  scoped_refptr<printing::PrintBackend> print_backend(
-      printing::PrintBackend::CreateInstance(NULL));
-
-  VLOG(1) << "Get printer capabilities start for " << printer_name;
-  crash_keys::ScopedPrinterInfo crash_key(
-      print_backend->GetPrinterDriverInfo(printer_name));
-
-  if (!print_backend->IsValidPrinter(printer_name)) {
-    // TODO(gene): Notify explicitly if printer is not valid, instead of
-    // failed to get capabilities.
+  scoped_ptr<base::DictionaryValue> printer_capabilities(
+      printer_name == kLocalPdfPrinterId ?
+      GetPdfCapabilitiesOnFileThread(locale) :
+      GetLocalPrinterCapabilitiesOnFileThread(printer_name));
+  if (printer_capabilities == NULL) {

[Vitaly Buka (NO REVIEWS), 2014/06/04 00:05:46]

if (!printer_capabilities) {

[Aleksey Shlyapnikov, 2014/06/04 02:02:47]

Done.

     BrowserThread::PostTask(BrowserThread::UI, FROM_HERE,
                             base::Bind(failure_cb, printer_name));
     return;
   }
 
-  printing::PrinterSemanticCapsAndDefaults info;
-  if (!print_backend->GetPrinterSemanticCapsAndDefaults(printer_name, &info)) {
-    LOG(WARNING) << "Failed to get capabilities for " << printer_name;
-    BrowserThread::PostTask(BrowserThread::UI, FROM_HERE,
-                            base::Bind(failure_cb, printer_name));
-    return;
-  }
-
-  scoped_ptr<base::DictionaryValue> settings_info(new base::DictionaryValue);
-  settings_info->SetString(kPrinterId, printer_name);
-  settings_info->SetBoolean(kDisableColorOption, !info.color_changeable);
-  settings_info->SetBoolean(printing::kSettingSetColorAsDefault,
-                            info.color_default);
-#if defined(USE_CUPS)
-  settings_info->SetInteger(kCUPSsColorModel, info.color_model);
-  settings_info->SetInteger(kCUPSsBWModel, info.bw_model);
-#endif
-
-  // TODO(gene): Make new capabilities format for Print Preview
-  // that will suit semantic capabilities better.
-  // Refactor pld API code below
-  bool default_duplex = info.duplex_capable ?
-      (info.duplex_default != printing::SIMPLEX) : false;
-  int duplex_value = info.duplex_capable ?
-      printing::LONG_EDGE : printing::UNKNOWN_DUPLEX_MODE;
-  settings_info->SetBoolean(kSetDuplexAsDefault, default_duplex);
-  settings_info->SetInteger(kPrinterDefaultDuplexValue, duplex_value);
+  scoped_ptr<base::DictionaryValue> printer_info(new base::DictionaryValue);
+  printer_info->SetString(kPrinterId, printer_name);
+  printer_info->Set(kPrinterCapabilities, printer_capabilities.release());
 
   BrowserThread::PostTask(
       BrowserThread::UI, FROM_HERE,
-      base::Bind(success_cb, base::Owned(settings_info.release())));
+      base::Bind(success_cb, base::Owned(printer_info.release())));
 }
 
 base::LazyInstance<printing::StickySettings> g_sticky_settings =
     LAZY_INSTANCE_INITIALIZER;
 
 printing::StickySettings* GetStickySettings() {
   return g_sticky_settings.Pointer();
 }
 
 }  // namespace
@@ skipping 932 matching lines @@
   *data = tmp_data;
   *title = print_preview_ui->initiator_title();
   return true;
 }
 
 #if defined(USE_CUPS)
 void PrintPreviewHandler::SaveCUPSColorSetting(
     const base::DictionaryValue* settings) {
   cups_printer_color_models_.reset(new CUPSPrinterColorModels);
   settings->GetString(kPrinterId, &cups_printer_color_models_->printer_name);
-  settings->GetInteger(
+  const base::DictionaryValue* capabilities = NULL;
+  if (!settings->GetDictionary(kPrinterCapabilities, &capabilities) ||
+      !capabilities) {
+    NOTREACHED();
+    return;
+  }
+  capabilities->GetInteger(
       kCUPSsColorModel,
       reinterpret_cast<int*>(&cups_printer_color_models_->color_model));
-  settings->GetInteger(
+  capabilities->GetInteger(
       kCUPSsBWModel,
       reinterpret_cast<int*>(&cups_printer_color_models_->bw_model));
 }
 
 void PrintPreviewHandler::ConvertColorSettingToCUPSColorModel(
     base::DictionaryValue* settings) const {
   if (!cups_printer_color_models_)
     return;
 
   // Sanity check the printer name.
@@ skipping 218 matching lines @@
   printer_value->SetString("name", description.name);
   printer_value->SetBoolean("hasLocalPrinting", has_local_printing);
   printer_value->SetBoolean(
       "isUnregistered",
       description.id.empty() &&
       command_line->HasSwitch(switches::kEnablePrintPreviewRegisterPromos));
   printer_value->SetString("cloudID", description.id);
 }
 
 #endif