Print Preview: Making margin selection sticky (part 2/2)

chrome/browser/ui/webui/print_preview_handler.h

 // 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.
 
 #ifndef CHROME_BROWSER_UI_WEBUI_PRINT_PREVIEW_HANDLER_H_
 #define CHROME_BROWSER_UI_WEBUI_PRINT_PREVIEW_HANDLER_H_
 #pragma once
 
 #include <string>
 
 #include "base/memory/ref_counted.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/memory/weak_ptr.h"
 #include "build/build_config.h"
 #include "chrome/browser/printing/print_view_manager_observer.h"
 #include "chrome/browser/ui/shell_dialogs.h"
 #include "content/browser/webui/web_ui.h"
 #include "printing/print_job_constants.h"
 
 class FilePath;
 class PrintSystemTaskProxy;
 class TabContentsWrapper;
 
 namespace base {
 class DictionaryValue;
 class StringValue;
 }
 
 namespace printing {
+struct PageSizeMargins;
 class PrintBackend;
 }
 
 // The handler for Javascript messages related to the print preview dialog.
 class PrintPreviewHandler : public WebUIMessageHandler,
                             public base::SupportsWeakPtr<PrintPreviewHandler>,
                             public SelectFileDialog::Listener,
                             public printing::PrintViewManagerObserver {
  public:
   PrintPreviewHandler();
@@ skipping 20 matching lines @@
   void OnTabDestroyed();
 
   // Called when print preview failed.
   void OnPrintPreviewFailed();
 
   // Called when the user press ctrl+shift+p to display the native system
   // dialog.
   void ShowSystemDialog();
 
  private:
+  friend class PrintPreviewHandlerTest;
   friend class PrintSystemTaskProxy;
+  FRIEND_TEST_ALL_PREFIXES(PrintPreviewHandlerTest, StickyMarginsCustom);
+  FRIEND_TEST_ALL_PREFIXES(PrintPreviewHandlerTest, StickyMarginsDefault);
+  FRIEND_TEST_ALL_PREFIXES(PrintPreviewHandlerTest,
+                           StickyMarginsCustomThenDefault);
+  FRIEND_TEST_ALL_PREFIXES(PrintPreviewHandlerTest,
+                           GetLastUsedMarginSettingsCustom);
+  FRIEND_TEST_ALL_PREFIXES(PrintPreviewHandlerTest,
+                           GetLastUsedMarginSettingsDefault);
 
   TabContentsWrapper* preview_tab_wrapper() const;
   TabContents* preview_tab() const;
 
-  // Gets the default printer. |args| is unused.
-  void HandleGetDefaultPrinter(const base::ListValue* args);
-
   // Gets the list of printers. |args| is unused.
   void HandleGetPrinters(const base::ListValue* args);
 
   // Asks the initiator renderer to generate a preview.  First element of |args|
   // is a job settings JSON string.
   void HandleGetPreview(const base::ListValue* args);
 
   // Gets the job settings from Web UI and initiate printing. First element of
   // |args| is a job settings JSON string.
   void HandlePrint(const base::ListValue* args);
@@ skipping 36 matching lines @@
   // Asks the browser to show the native printer management dialog.
   // |args| is unused.
   void HandleManagePrinters(const base::ListValue* args);
 
   // Asks the browser to show the cloud print dialog.
   void HandlePrintWithCloudPrint();
 
   // Asks the browser to close the preview tab. |args| is unused.
   void HandleClosePreviewTab(const base::ListValue* args);
 
-  // Asks the browser for the title of the initiator tab.
-  // |args| is unused.
-  void HandleGetInitiatorTabTitle(const base::ListValue* args);
+  // Asks the browser for several settings that are needed before the first
+  // preview is displayed.
+  void HandleGetInitialSettings(const base::ListValue* args);
 
-  // Asks the browser for the number formatting and measurement system according
-  // to the current locale.
-  void HandleGetNumberFormatAndMeasurementSystem(const base::ListValue* args);
+  void SendInitialSettings(
+      const std::string& default_printer,
+      const std::string& cloud_print_data);
 
   // Sends the printer capabilities to the Web UI. |settings_info| contains
   // printer capabilities information.
   void SendPrinterCapabilities(const base::DictionaryValue& settings_info);
 
-  // Sends the default printer to the Web UI.
-  void SendDefaultPrinter(const base::StringValue& default_printer,
-                          const base::StringValue& cloud_print_data);
-
   // Send the list of printers to the Web UI.
   void SetupPrinterList(const base::ListValue& printers);
 
   // Send whether cloud print integration should be enabled.
   void SendCloudPrintEnabled();
 
   // Send the PDF data to the cloud to print.
   void SendCloudPrintJob(const base::DictionaryValue& settings,
                          std::string print_ticket);
 
@@ skipping 11 matching lines @@
 
   // Hides the preview tab for printing.
   void HidePreviewTab();
 
   // Clears initiator tab details for this preview tab.
   void ClearInitiatorTabDetails();
 
   // Posts a task to save to pdf at |print_to_pdf_path_|.
   void PostPrintToPdfTask();
 
+  // Populates |settings| according to the current locale.
+  void GetNumberFormatAndMeasurementSystem(base::DictionaryValue* settings);
+
+  // Populates |last_used_custom_margins| according to the last used margin
+  // settings.
+  void GetLastUsedMarginSettings(
+      base::DictionaryValue* last_used_custom_margins);
+
   // Pointer to current print system.
   scoped_refptr<printing::PrintBackend> print_backend_;
 
   // The underlying dialog object.
   scoped_refptr<SelectFileDialog> select_file_dialog_;
 
   static FilePath* last_saved_path_;
   static std::string* last_used_printer_cloud_print_data_;
   static std::string* last_used_printer_name_;
   static printing::ColorModels last_used_color_model_;
   static printing::MarginType last_used_margins_type_;
+  static printing::PageSizeMargins* last_used_page_size_margins_;
 
   // A count of how many requests received to regenerate preview data.
   // Initialized to 0 then incremented and emitted to a histogram.
   int regenerate_preview_request_count_;
 
   // A count of how many requests received to show manage printers dialog.
   int manage_printers_dialog_request_count_;
 
   // Whether we have already logged a failed print preview.
   bool reported_failed_preview_;
 
   // Whether we have already logged the number of printers this session.
   bool has_logged_printers_count_;
 
   // Holds the path to the print to pdf request. It is empty if no such request
   // exists.
   scoped_ptr<FilePath> print_to_pdf_path_;
 
   DISALLOW_COPY_AND_ASSIGN(PrintPreviewHandler);
 };
 
 #endif  // CHROME_BROWSER_UI_WEBUI_PRINT_PREVIEW_HANDLER_H_