michael@0: /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
michael@0: /* This Source Code Form is subject to the terms of the Mozilla Public
michael@0:  * License, v. 2.0. If a copy of the MPL was not distributed with this
michael@0:  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
michael@0: 
michael@0: #include "mozilla/ArrayUtils.h"
michael@0: 
michael@0: #include "nsDeviceContextSpecWin.h"
michael@0: #include "prmem.h"
michael@0: 
michael@0: #include <winspool.h>
michael@0: 
michael@0: #include <tchar.h>
michael@0: 
michael@0: #include "nsAutoPtr.h"
michael@0: #include "nsIWidget.h"
michael@0: 
michael@0: #include "nsTArray.h"
michael@0: #include "nsIPrintSettingsWin.h"
michael@0: 
michael@0: #include "nsString.h"
michael@0: #include "nsCRT.h"
michael@0: #include "nsIServiceManager.h"
michael@0: #include "nsReadableUtils.h"
michael@0: #include "nsStringEnumerator.h"
michael@0: 
michael@0: #include "gfxPDFSurface.h"
michael@0: #include "gfxWindowsSurface.h"
michael@0: 
michael@0: #include "nsIFileStreams.h"
michael@0: #include "nsIWindowWatcher.h"
michael@0: #include "nsIDOMWindow.h"
michael@0: #include "mozilla/Services.h"
michael@0: 
michael@0: // For NS_CopyNativeToUnicode
michael@0: #include "nsNativeCharsetUtils.h"
michael@0: 
michael@0: // File Picker
michael@0: #include "nsIFile.h"
michael@0: #include "nsIFilePicker.h"
michael@0: #include "nsIStringBundle.h"
michael@0: #define NS_ERROR_GFX_PRINTER_BUNDLE_URL "chrome://global/locale/printing.properties"
michael@0: 
michael@0: #include "prlog.h"
michael@0: #ifdef PR_LOGGING 
michael@0: PRLogModuleInfo * kWidgetPrintingLogMod = PR_NewLogModule("printing-widget");
michael@0: #define PR_PL(_p1)  PR_LOG(kWidgetPrintingLogMod, PR_LOG_DEBUG, _p1)
michael@0: #else
michael@0: #define PR_PL(_p1)
michael@0: #endif
michael@0: 
michael@0: using namespace mozilla;
michael@0: 
michael@0: //----------------------------------------------------------------------------------
michael@0: // The printer data is shared between the PrinterEnumerator and the nsDeviceContextSpecWin
michael@0: // The PrinterEnumerator creates the printer info
michael@0: // but the nsDeviceContextSpecWin cleans it up
michael@0: // If it gets created (via the Page Setup Dialog) but the user never prints anything
michael@0: // then it will never be delete, so this class takes care of that.
michael@0: class GlobalPrinters {
michael@0: public:
michael@0:   static GlobalPrinters* GetInstance() { return &mGlobalPrinters; }
michael@0:   ~GlobalPrinters() { FreeGlobalPrinters(); }
michael@0: 
michael@0:   void FreeGlobalPrinters();
michael@0: 
michael@0:   bool         PrintersAreAllocated() { return mPrinters != nullptr; }
michael@0:   LPWSTR       GetItemFromList(int32_t aInx) { return mPrinters?mPrinters->ElementAt(aInx):nullptr; }
michael@0:   nsresult     EnumeratePrinterList();
michael@0:   void         GetDefaultPrinterName(nsString& aDefaultPrinterName);
michael@0:   uint32_t     GetNumPrinters() { return mPrinters?mPrinters->Length():0; }
michael@0: 
michael@0: protected:
michael@0:   GlobalPrinters() {}
michael@0:   nsresult EnumerateNativePrinters();
michael@0:   void     ReallocatePrinters();
michael@0: 
michael@0:   static GlobalPrinters    mGlobalPrinters;
michael@0:   static nsTArray<LPWSTR>* mPrinters;
michael@0: };
michael@0: //---------------
michael@0: // static members
michael@0: GlobalPrinters    GlobalPrinters::mGlobalPrinters;
michael@0: nsTArray<LPWSTR>* GlobalPrinters::mPrinters = nullptr;
michael@0: 
michael@0: 
michael@0: //******************************************************
michael@0: // Define native paper sizes
michael@0: //******************************************************
michael@0: typedef struct {
michael@0:   short  mPaperSize; // native enum
michael@0:   double mWidth;
michael@0:   double mHeight;
michael@0:   bool mIsInches;
michael@0: } NativePaperSizes;
michael@0: 
michael@0: // There are around 40 default print sizes defined by Windows
michael@0: const NativePaperSizes kPaperSizes[] = {
michael@0:   {DMPAPER_LETTER,    8.5,   11.0,  true},
michael@0:   {DMPAPER_LEGAL,     8.5,   14.0,  true},
michael@0:   {DMPAPER_A4,        210.0, 297.0, false},
michael@0:   {DMPAPER_B4,        250.0, 354.0, false}, 
michael@0:   {DMPAPER_B5,        182.0, 257.0, false},
michael@0:   {DMPAPER_TABLOID,   11.0,  17.0,  true},
michael@0:   {DMPAPER_LEDGER,    17.0,  11.0,  true},
michael@0:   {DMPAPER_STATEMENT, 5.5,   8.5,   true},
michael@0:   {DMPAPER_EXECUTIVE, 7.25,  10.5,  true},
michael@0:   {DMPAPER_A3,        297.0, 420.0, false},
michael@0:   {DMPAPER_A5,        148.0, 210.0, false},
michael@0:   {DMPAPER_CSHEET,    17.0,  22.0,  true},  
michael@0:   {DMPAPER_DSHEET,    22.0,  34.0,  true},  
michael@0:   {DMPAPER_ESHEET,    34.0,  44.0,  true},  
michael@0:   {DMPAPER_LETTERSMALL, 8.5, 11.0,  true},  
michael@0:   {DMPAPER_A4SMALL,   210.0, 297.0, false}, 
michael@0:   {DMPAPER_FOLIO,     8.5,   13.0,  true},
michael@0:   {DMPAPER_QUARTO,    215.0, 275.0, false},
michael@0:   {DMPAPER_10X14,     10.0,  14.0,  true},
michael@0:   {DMPAPER_11X17,     11.0,  17.0,  true},
michael@0:   {DMPAPER_NOTE,      8.5,   11.0,  true},  
michael@0:   {DMPAPER_ENV_9,     3.875, 8.875, true},  
michael@0:   {DMPAPER_ENV_10,    40.125, 9.5,  true},  
michael@0:   {DMPAPER_ENV_11,    4.5,   10.375, true},  
michael@0:   {DMPAPER_ENV_12,    4.75,  11.0,  true},  
michael@0:   {DMPAPER_ENV_14,    5.0,   11.5,  true},  
michael@0:   {DMPAPER_ENV_DL,    110.0, 220.0, false}, 
michael@0:   {DMPAPER_ENV_C5,    162.0, 229.0, false}, 
michael@0:   {DMPAPER_ENV_C3,    324.0, 458.0, false}, 
michael@0:   {DMPAPER_ENV_C4,    229.0, 324.0, false}, 
michael@0:   {DMPAPER_ENV_C6,    114.0, 162.0, false}, 
michael@0:   {DMPAPER_ENV_C65,   114.0, 229.0, false}, 
michael@0:   {DMPAPER_ENV_B4,    250.0, 353.0, false}, 
michael@0:   {DMPAPER_ENV_B5,    176.0, 250.0, false}, 
michael@0:   {DMPAPER_ENV_B6,    176.0, 125.0, false}, 
michael@0:   {DMPAPER_ENV_ITALY, 110.0, 230.0, false}, 
michael@0:   {DMPAPER_ENV_MONARCH,  3.875,  7.5, true},  
michael@0:   {DMPAPER_ENV_PERSONAL, 3.625,  6.5, true},  
michael@0:   {DMPAPER_FANFOLD_US,   14.875, 11.0, true},  
michael@0:   {DMPAPER_FANFOLD_STD_GERMAN, 8.5, 12.0, true},  
michael@0:   {DMPAPER_FANFOLD_LGL_GERMAN, 8.5, 13.0, true},  
michael@0: };
michael@0: const int32_t kNumPaperSizes = 41;
michael@0: 
michael@0: //----------------------------------------------------------------------------------
michael@0: nsDeviceContextSpecWin::nsDeviceContextSpecWin()
michael@0: {
michael@0:   mDriverName    = nullptr;
michael@0:   mDeviceName    = nullptr;
michael@0:   mDevMode       = nullptr;
michael@0: 
michael@0: }
michael@0: 
michael@0: 
michael@0: //----------------------------------------------------------------------------------
michael@0: 
michael@0: NS_IMPL_ISUPPORTS(nsDeviceContextSpecWin, nsIDeviceContextSpec)
michael@0: 
michael@0: nsDeviceContextSpecWin::~nsDeviceContextSpecWin()
michael@0: {
michael@0:   SetDeviceName(nullptr);
michael@0:   SetDriverName(nullptr);
michael@0:   SetDevMode(nullptr);
michael@0: 
michael@0:   nsCOMPtr<nsIPrintSettingsWin> psWin(do_QueryInterface(mPrintSettings));
michael@0:   if (psWin) {
michael@0:     psWin->SetDeviceName(nullptr);
michael@0:     psWin->SetDriverName(nullptr);
michael@0:     psWin->SetDevMode(nullptr);
michael@0:   }
michael@0: 
michael@0:   // Free them, we won't need them for a while
michael@0:   GlobalPrinters::GetInstance()->FreeGlobalPrinters();
michael@0: }
michael@0: 
michael@0: 
michael@0: //------------------------------------------------------------------
michael@0: // helper
michael@0: static char16_t * GetDefaultPrinterNameFromGlobalPrinters()
michael@0: {
michael@0:   nsAutoString printerName;
michael@0:   GlobalPrinters::GetInstance()->GetDefaultPrinterName(printerName);
michael@0:   return ToNewUnicode(printerName);
michael@0: }
michael@0: 
michael@0: //----------------------------------------------------------------------------------
michael@0: NS_IMETHODIMP nsDeviceContextSpecWin::Init(nsIWidget* aWidget,
michael@0:                                            nsIPrintSettings* aPrintSettings,
michael@0:                                            bool aIsPrintPreview)
michael@0: {
michael@0:   mPrintSettings = aPrintSettings;
michael@0: 
michael@0:   nsresult rv = NS_ERROR_GFX_PRINTER_NO_PRINTER_AVAILABLE;
michael@0:   if (aPrintSettings) {
michael@0:     nsCOMPtr<nsIPrintSettingsWin> psWin(do_QueryInterface(aPrintSettings));
michael@0:     if (psWin) {
michael@0:       char16_t* deviceName;
michael@0:       char16_t* driverName;
michael@0:       psWin->GetDeviceName(&deviceName); // creates new memory (makes a copy)
michael@0:       psWin->GetDriverName(&driverName); // creates new memory (makes a copy)
michael@0: 
michael@0:       LPDEVMODEW devMode;
michael@0:       psWin->GetDevMode(&devMode);       // creates new memory (makes a copy)
michael@0: 
michael@0:       if (deviceName && driverName && devMode) {
michael@0:         // Scaling is special, it is one of the few
michael@0:         // devMode items that we control in layout
michael@0:         if (devMode->dmFields & DM_SCALE) {
michael@0:           double scale = double(devMode->dmScale) / 100.0f;
michael@0:           if (scale != 1.0) {
michael@0:             aPrintSettings->SetScaling(scale);
michael@0:             devMode->dmScale = 100;
michael@0:           }
michael@0:         }
michael@0: 
michael@0:         SetDeviceName(deviceName);
michael@0:         SetDriverName(driverName);
michael@0:         SetDevMode(devMode);
michael@0: 
michael@0:         // clean up
michael@0:         free(deviceName);
michael@0:         free(driverName);
michael@0: 
michael@0:         return NS_OK;
michael@0:       } else {
michael@0:         PR_PL(("***** nsDeviceContextSpecWin::Init - deviceName/driverName/devMode was NULL!\n"));
michael@0:         if (deviceName) free(deviceName);
michael@0:         if (driverName) free(driverName);
michael@0:         if (devMode) ::HeapFree(::GetProcessHeap(), 0, devMode);
michael@0:       }
michael@0:     }
michael@0:   } else {
michael@0:     PR_PL(("***** nsDeviceContextSpecWin::Init - aPrintSettingswas NULL!\n"));
michael@0:   }
michael@0: 
michael@0:   // Get the Print Name to be used
michael@0:   char16_t * printerName = nullptr;
michael@0:   if (mPrintSettings) {
michael@0:     mPrintSettings->GetPrinterName(&printerName);
michael@0:   }
michael@0: 
michael@0:   // If there is no name then use the default printer
michael@0:   if (!printerName || (printerName && !*printerName)) {
michael@0:     printerName = GetDefaultPrinterNameFromGlobalPrinters();
michael@0:   }
michael@0: 
michael@0:   NS_ASSERTION(printerName, "We have to have a printer name");
michael@0:   if (!printerName || !*printerName) return rv;
michael@0:  
michael@0:   return GetDataFromPrinter(printerName, mPrintSettings);
michael@0: }
michael@0: 
michael@0: //----------------------------------------------------------
michael@0: // Helper Function - Free and reallocate the string
michael@0: static void CleanAndCopyString(wchar_t*& aStr, const wchar_t* aNewStr)
michael@0: {
michael@0:   if (aStr != nullptr) {
michael@0:     if (aNewStr != nullptr && wcslen(aStr) > wcslen(aNewStr)) { // reuse it if we can
michael@0:       wcscpy(aStr, aNewStr);
michael@0:       return;
michael@0:     } else {
michael@0:       PR_Free(aStr);
michael@0:       aStr = nullptr;
michael@0:     }
michael@0:   }
michael@0: 
michael@0:   if (nullptr != aNewStr) {
michael@0:     aStr = (wchar_t *)PR_Malloc(sizeof(wchar_t)*(wcslen(aNewStr) + 1));
michael@0:     wcscpy(aStr, aNewStr);
michael@0:   }
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP nsDeviceContextSpecWin::GetSurfaceForPrinter(gfxASurface **surface)
michael@0: {
michael@0:   NS_ASSERTION(mDevMode, "DevMode can't be NULL here");
michael@0: 
michael@0:   nsRefPtr<gfxASurface> newSurface;
michael@0: 
michael@0:   int16_t outputFormat = 0;
michael@0:   if (mPrintSettings) {
michael@0:     mPrintSettings->GetOutputFormat(&outputFormat);
michael@0:   }
michael@0: 
michael@0:   if (outputFormat == nsIPrintSettings::kOutputFormatPDF) {
michael@0:     nsXPIDLString filename;
michael@0:     mPrintSettings->GetToFileName(getter_Copies(filename));
michael@0: 
michael@0:     double width, height;
michael@0:     mPrintSettings->GetEffectivePageSize(&width, &height);
michael@0:     // convert twips to points
michael@0:     width  /= TWIPS_PER_POINT_FLOAT;
michael@0:     height /= TWIPS_PER_POINT_FLOAT;
michael@0: 
michael@0:     nsCOMPtr<nsIFile> file = do_CreateInstance("@mozilla.org/file/local;1");
michael@0:     nsresult rv = file->InitWithPath(filename);
michael@0:     if (NS_FAILED(rv))
michael@0:       return rv;
michael@0: 
michael@0:     nsCOMPtr<nsIFileOutputStream> stream = do_CreateInstance("@mozilla.org/network/file-output-stream;1");
michael@0:     rv = stream->Init(file, -1, -1, 0);
michael@0:     if (NS_FAILED(rv))
michael@0:       return rv;
michael@0: 
michael@0:     newSurface = new gfxPDFSurface(stream, gfxSize(width, height));
michael@0:   } else {
michael@0:     if (mDevMode) {
michael@0:       NS_WARN_IF_FALSE(mDriverName, "No driver!");
michael@0:       HDC dc = ::CreateDCW(mDriverName, mDeviceName, nullptr, mDevMode);
michael@0: 
michael@0:       // have this surface take over ownership of this DC
michael@0:       newSurface = new gfxWindowsSurface(dc, gfxWindowsSurface::FLAG_TAKE_DC | gfxWindowsSurface::FLAG_FOR_PRINTING);
michael@0:     }
michael@0:   }
michael@0: 
michael@0:   if (newSurface) {
michael@0:     *surface = newSurface;
michael@0:     NS_ADDREF(*surface);
michael@0:     return NS_OK;
michael@0:   }
michael@0: 
michael@0:   *surface = nullptr;
michael@0:   return NS_ERROR_FAILURE;
michael@0: }
michael@0: 
michael@0: //----------------------------------------------------------------------------------
michael@0: void nsDeviceContextSpecWin::SetDeviceName(char16ptr_t aDeviceName)
michael@0: {
michael@0:   CleanAndCopyString(mDeviceName, aDeviceName);
michael@0: }
michael@0: 
michael@0: //----------------------------------------------------------------------------------
michael@0: void nsDeviceContextSpecWin::SetDriverName(char16ptr_t aDriverName)
michael@0: {
michael@0:   CleanAndCopyString(mDriverName, aDriverName);
michael@0: }
michael@0: 
michael@0: //----------------------------------------------------------------------------------
michael@0: void nsDeviceContextSpecWin::SetDevMode(LPDEVMODEW aDevMode)
michael@0: {
michael@0:   if (mDevMode) {
michael@0:     ::HeapFree(::GetProcessHeap(), 0, mDevMode);
michael@0:   }
michael@0: 
michael@0:   mDevMode = aDevMode;
michael@0: }
michael@0: 
michael@0: //------------------------------------------------------------------
michael@0: void 
michael@0: nsDeviceContextSpecWin::GetDevMode(LPDEVMODEW &aDevMode)
michael@0: {
michael@0:   aDevMode = mDevMode;
michael@0: }
michael@0: 
michael@0: //----------------------------------------------------------------------------------
michael@0: // Map an incoming size to a Windows Native enum in the DevMode
michael@0: static void 
michael@0: MapPaperSizeToNativeEnum(LPDEVMODEW aDevMode,
michael@0:                          int16_t   aType, 
michael@0:                          double    aW, 
michael@0:                          double    aH)
michael@0: {
michael@0: 
michael@0: #ifdef DEBUG_rods
michael@0:   BOOL doingOrientation = aDevMode->dmFields & DM_ORIENTATION;
michael@0:   BOOL doingPaperSize   = aDevMode->dmFields & DM_PAPERSIZE;
michael@0:   BOOL doingPaperLength = aDevMode->dmFields & DM_PAPERLENGTH;
michael@0:   BOOL doingPaperWidth  = aDevMode->dmFields & DM_PAPERWIDTH;
michael@0: #endif
michael@0: 
michael@0:   for (int32_t i=0;i<kNumPaperSizes;i++) {
michael@0:     if (kPaperSizes[i].mWidth == aW && kPaperSizes[i].mHeight == aH) {
michael@0:       aDevMode->dmPaperSize = kPaperSizes[i].mPaperSize;
michael@0:       aDevMode->dmFields &= ~DM_PAPERLENGTH;
michael@0:       aDevMode->dmFields &= ~DM_PAPERWIDTH;
michael@0:       aDevMode->dmFields |= DM_PAPERSIZE;
michael@0:       return;
michael@0:     }
michael@0:   }
michael@0: 
michael@0:   short width  = 0;
michael@0:   short height = 0;
michael@0:   if (aType == nsIPrintSettings::kPaperSizeInches) {
michael@0:     width  = short(NS_TWIPS_TO_MILLIMETERS(NS_INCHES_TO_TWIPS(float(aW))) / 10);
michael@0:     height = short(NS_TWIPS_TO_MILLIMETERS(NS_INCHES_TO_TWIPS(float(aH))) / 10);
michael@0: 
michael@0:   } else if (aType == nsIPrintSettings::kPaperSizeMillimeters) {
michael@0:     width  = short(aW / 10.0);
michael@0:     height = short(aH / 10.0);
michael@0:   } else {
michael@0:     return; // don't set anything
michael@0:   }
michael@0: 
michael@0:   // width and height is in 
michael@0:   aDevMode->dmPaperSize   = 0;
michael@0:   aDevMode->dmPaperWidth  = width;
michael@0:   aDevMode->dmPaperLength = height;
michael@0: 
michael@0:   aDevMode->dmFields |= DM_PAPERSIZE;
michael@0:   aDevMode->dmFields |= DM_PAPERLENGTH;
michael@0:   aDevMode->dmFields |= DM_PAPERWIDTH;
michael@0: }
michael@0: 
michael@0: //----------------------------------------------------------------------------------
michael@0: // Setup Paper Size & Orientation options into the DevMode
michael@0: // 
michael@0: static void 
michael@0: SetupDevModeFromSettings(LPDEVMODEW aDevMode, nsIPrintSettings* aPrintSettings)
michael@0: {
michael@0:   // Setup paper size
michael@0:   if (aPrintSettings) {
michael@0:     int16_t type;
michael@0:     aPrintSettings->GetPaperSizeType(&type);
michael@0:     if (type == nsIPrintSettings::kPaperSizeNativeData) {
michael@0:       int16_t paperEnum;
michael@0:       aPrintSettings->GetPaperData(&paperEnum);
michael@0:       aDevMode->dmPaperSize = paperEnum;
michael@0:       aDevMode->dmFields &= ~DM_PAPERLENGTH;
michael@0:       aDevMode->dmFields &= ~DM_PAPERWIDTH;
michael@0:       aDevMode->dmFields |= DM_PAPERSIZE;
michael@0:     } else {
michael@0:       int16_t unit;
michael@0:       double width, height;
michael@0:       aPrintSettings->GetPaperSizeUnit(&unit);
michael@0:       aPrintSettings->GetPaperWidth(&width);
michael@0:       aPrintSettings->GetPaperHeight(&height);
michael@0:       MapPaperSizeToNativeEnum(aDevMode, unit, width, height);
michael@0:     }
michael@0: 
michael@0:     // Setup Orientation
michael@0:     int32_t orientation;
michael@0:     aPrintSettings->GetOrientation(&orientation);
michael@0:     aDevMode->dmOrientation = orientation == nsIPrintSettings::kPortraitOrientation?DMORIENT_PORTRAIT:DMORIENT_LANDSCAPE;
michael@0:     aDevMode->dmFields |= DM_ORIENTATION;
michael@0: 
michael@0:     // Setup Number of Copies
michael@0:     int32_t copies;
michael@0:     aPrintSettings->GetNumCopies(&copies);
michael@0:     aDevMode->dmCopies = copies;
michael@0:     aDevMode->dmFields |= DM_COPIES;
michael@0:   }
michael@0: 
michael@0: }
michael@0: 
michael@0: #define DISPLAY_LAST_ERROR 
michael@0: 
michael@0: //----------------------------------------------------------------------------------
michael@0: // Setup the object's data member with the selected printer's data
michael@0: nsresult
michael@0: nsDeviceContextSpecWin::GetDataFromPrinter(char16ptr_t aName, nsIPrintSettings* aPS)
michael@0: {
michael@0:   nsresult rv = NS_ERROR_FAILURE;
michael@0: 
michael@0:   if (!GlobalPrinters::GetInstance()->PrintersAreAllocated()) {
michael@0:     rv = GlobalPrinters::GetInstance()->EnumeratePrinterList();
michael@0:     if (NS_FAILED(rv)) {
michael@0:       PR_PL(("***** nsDeviceContextSpecWin::GetDataFromPrinter - Couldn't enumerate printers!\n"));
michael@0:       DISPLAY_LAST_ERROR
michael@0:     }
michael@0:     NS_ENSURE_SUCCESS(rv, rv);
michael@0:   }
michael@0: 
michael@0:   HANDLE hPrinter = nullptr;
michael@0:   wchar_t *name = (wchar_t*)aName; // Windows APIs use non-const name argument
michael@0:   
michael@0:   BOOL status = ::OpenPrinterW(name, &hPrinter, nullptr);
michael@0:   if (status) {
michael@0: 
michael@0:     LPDEVMODEW   pDevMode;
michael@0:     DWORD       dwNeeded, dwRet;
michael@0: 
michael@0:     // Allocate a buffer of the correct size.
michael@0:     dwNeeded = ::DocumentPropertiesW(nullptr, hPrinter, name, nullptr, nullptr, 0);
michael@0: 
michael@0:     pDevMode = (LPDEVMODEW)::HeapAlloc (::GetProcessHeap(), HEAP_ZERO_MEMORY, dwNeeded);
michael@0:     if (!pDevMode) return NS_ERROR_FAILURE;
michael@0: 
michael@0:     // Get the default DevMode for the printer and modify it for our needs.
michael@0:     dwRet = DocumentPropertiesW(nullptr, hPrinter, name,
michael@0:                                pDevMode, nullptr, DM_OUT_BUFFER);
michael@0: 
michael@0:     if (dwRet == IDOK && aPS) {
michael@0:       SetupDevModeFromSettings(pDevMode, aPS);
michael@0:       // Sets back the changes we made to the DevMode into the Printer Driver
michael@0:       dwRet = ::DocumentPropertiesW(nullptr, hPrinter, name,
michael@0:                                    pDevMode, pDevMode,
michael@0:                                    DM_IN_BUFFER | DM_OUT_BUFFER);
michael@0:     }
michael@0: 
michael@0:     if (dwRet != IDOK) {
michael@0:       ::HeapFree(::GetProcessHeap(), 0, pDevMode);
michael@0:       ::ClosePrinter(hPrinter);
michael@0:       PR_PL(("***** nsDeviceContextSpecWin::GetDataFromPrinter - DocumentProperties call failed code: %d/0x%x\n", dwRet, dwRet));
michael@0:       DISPLAY_LAST_ERROR
michael@0:       return NS_ERROR_FAILURE;
michael@0:     }
michael@0: 
michael@0:     SetDevMode(pDevMode); // cache the pointer and takes responsibility for the memory
michael@0: 
michael@0:     SetDeviceName(aName);
michael@0: 
michael@0:     SetDriverName(L"WINSPOOL");
michael@0: 
michael@0:     ::ClosePrinter(hPrinter);
michael@0:     rv = NS_OK;
michael@0:   } else {
michael@0:     rv = NS_ERROR_GFX_PRINTER_NAME_NOT_FOUND;
michael@0:     PR_PL(("***** nsDeviceContextSpecWin::GetDataFromPrinter - Couldn't open printer: [%s]\n", NS_ConvertUTF16toUTF8(aName).get()));
michael@0:     DISPLAY_LAST_ERROR
michael@0:   }
michael@0:   return rv;
michael@0: }
michael@0: 
michael@0: //----------------------------------------------------------------------------------
michael@0: // Setup Paper Size options into the DevMode
michael@0: // 
michael@0: // When using a data member it may be a HGLOCAL or LPDEVMODE
michael@0: // if it is a HGLOBAL then we need to "lock" it to get the LPDEVMODE
michael@0: // and unlock it when we are done.
michael@0: void 
michael@0: nsDeviceContextSpecWin::SetupPaperInfoFromSettings()
michael@0: {
michael@0:   LPDEVMODEW devMode;
michael@0: 
michael@0:   GetDevMode(devMode);
michael@0:   NS_ASSERTION(devMode, "DevMode can't be NULL here");
michael@0:   if (devMode) {
michael@0:     SetupDevModeFromSettings(devMode, mPrintSettings);
michael@0:   }
michael@0: }
michael@0: 
michael@0: //----------------------------------------------------------------------------------
michael@0: // Helper Function - Free and reallocate the string
michael@0: nsresult 
michael@0: nsDeviceContextSpecWin::SetPrintSettingsFromDevMode(nsIPrintSettings* aPrintSettings, 
michael@0:                                                     LPDEVMODEW         aDevMode)
michael@0: {
michael@0:   if (aPrintSettings == nullptr) {
michael@0:     return NS_ERROR_FAILURE;
michael@0:   }
michael@0:   aPrintSettings->SetIsInitializedFromPrinter(true);
michael@0: 
michael@0:   BOOL doingNumCopies   = aDevMode->dmFields & DM_COPIES;
michael@0:   BOOL doingOrientation = aDevMode->dmFields & DM_ORIENTATION;
michael@0:   BOOL doingPaperSize   = aDevMode->dmFields & DM_PAPERSIZE;
michael@0:   BOOL doingPaperLength = aDevMode->dmFields & DM_PAPERLENGTH;
michael@0:   BOOL doingPaperWidth  = aDevMode->dmFields & DM_PAPERWIDTH;
michael@0: 
michael@0:   if (doingOrientation) {
michael@0:     int32_t orientation  = aDevMode->dmOrientation == DMORIENT_PORTRAIT?
michael@0:       int32_t(nsIPrintSettings::kPortraitOrientation):nsIPrintSettings::kLandscapeOrientation;
michael@0:     aPrintSettings->SetOrientation(orientation);
michael@0:   }
michael@0: 
michael@0:   // Setup Number of Copies
michael@0:   if (doingNumCopies) {
michael@0:     aPrintSettings->SetNumCopies(int32_t(aDevMode->dmCopies));
michael@0:   }
michael@0: 
michael@0:   if (aDevMode->dmFields & DM_SCALE) {
michael@0:     double scale = double(aDevMode->dmScale) / 100.0f;
michael@0:     if (scale != 1.0) {
michael@0:       aPrintSettings->SetScaling(scale);
michael@0:       aDevMode->dmScale = 100;
michael@0:       // To turn this on you must change where the mPrt->mShrinkToFit is being set in the DocumentViewer
michael@0:       //aPrintSettings->SetShrinkToFit(false);
michael@0:     }
michael@0:   }
michael@0: 
michael@0:   if (doingPaperSize) {
michael@0:     aPrintSettings->SetPaperSizeType(nsIPrintSettings::kPaperSizeNativeData);
michael@0:     aPrintSettings->SetPaperData(aDevMode->dmPaperSize);
michael@0:     for (int32_t i=0;i<kNumPaperSizes;i++) {
michael@0:       if (kPaperSizes[i].mPaperSize == aDevMode->dmPaperSize) {
michael@0:         aPrintSettings->SetPaperSizeUnit(kPaperSizes[i].mIsInches
michael@0:           ?int16_t(nsIPrintSettings::kPaperSizeInches):nsIPrintSettings::kPaperSizeMillimeters);
michael@0:         break;
michael@0:       }
michael@0:     }
michael@0: 
michael@0:   } else if (doingPaperLength && doingPaperWidth) {
michael@0:     bool found = false;
michael@0:     for (int32_t i=0;i<kNumPaperSizes;i++) {
michael@0:       if (kPaperSizes[i].mPaperSize == aDevMode->dmPaperSize) {
michael@0:         aPrintSettings->SetPaperSizeType(nsIPrintSettings::kPaperSizeDefined);
michael@0:         aPrintSettings->SetPaperWidth(kPaperSizes[i].mWidth);
michael@0:         aPrintSettings->SetPaperHeight(kPaperSizes[i].mHeight);
michael@0:         aPrintSettings->SetPaperSizeUnit(kPaperSizes[i].mIsInches
michael@0:           ?int16_t(nsIPrintSettings::kPaperSizeInches):nsIPrintSettings::kPaperSizeMillimeters);
michael@0:         found = true;
michael@0:         break;
michael@0:       }
michael@0:     }
michael@0:     if (!found) {
michael@0:       return NS_ERROR_FAILURE;
michael@0:     }
michael@0:   } else {
michael@0:     return NS_ERROR_FAILURE;
michael@0:   }
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: //***********************************************************
michael@0: //  Printer Enumerator
michael@0: //***********************************************************
michael@0: nsPrinterEnumeratorWin::nsPrinterEnumeratorWin()
michael@0: {
michael@0: }
michael@0: 
michael@0: nsPrinterEnumeratorWin::~nsPrinterEnumeratorWin()
michael@0: {
michael@0:   // Do not free printers here
michael@0:   // GlobalPrinters::GetInstance()->FreeGlobalPrinters();
michael@0: }
michael@0: 
michael@0: NS_IMPL_ISUPPORTS(nsPrinterEnumeratorWin, nsIPrinterEnumerator)
michael@0: 
michael@0: //----------------------------------------------------------------------------------
michael@0: // Return the Default Printer name
michael@0: /* readonly attribute wstring defaultPrinterName; */
michael@0: NS_IMETHODIMP 
michael@0: nsPrinterEnumeratorWin::GetDefaultPrinterName(char16_t * *aDefaultPrinterName)
michael@0: {
michael@0:   NS_ENSURE_ARG_POINTER(aDefaultPrinterName);
michael@0: 
michael@0:   *aDefaultPrinterName = GetDefaultPrinterNameFromGlobalPrinters(); // helper
michael@0: 
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: /* void initPrintSettingsFromPrinter (in wstring aPrinterName, in nsIPrintSettings aPrintSettings); */
michael@0: NS_IMETHODIMP 
michael@0: nsPrinterEnumeratorWin::InitPrintSettingsFromPrinter(const char16_t *aPrinterName, nsIPrintSettings *aPrintSettings)
michael@0: {
michael@0:   NS_ENSURE_ARG_POINTER(aPrinterName);
michael@0:   NS_ENSURE_ARG_POINTER(aPrintSettings);
michael@0: 
michael@0:   if (!*aPrinterName) {
michael@0:     return NS_OK;
michael@0:   }
michael@0: 
michael@0:   nsRefPtr<nsDeviceContextSpecWin> devSpecWin = new nsDeviceContextSpecWin();
michael@0:   if (!devSpecWin) return NS_ERROR_OUT_OF_MEMORY;
michael@0: 
michael@0:   if (NS_FAILED(GlobalPrinters::GetInstance()->EnumeratePrinterList())) {
michael@0:     return NS_ERROR_FAILURE;
michael@0:   }
michael@0: 
michael@0:   devSpecWin->GetDataFromPrinter(aPrinterName);
michael@0: 
michael@0:   LPDEVMODEW devmode;
michael@0:   devSpecWin->GetDevMode(devmode);
michael@0:   NS_ASSERTION(devmode, "DevMode can't be NULL here");
michael@0:   if (devmode) {
michael@0:     aPrintSettings->SetPrinterName(aPrinterName);
michael@0:     nsDeviceContextSpecWin::SetPrintSettingsFromDevMode(aPrintSettings, devmode);
michael@0:   }
michael@0: 
michael@0:   // Free them, we won't need them for a while
michael@0:   GlobalPrinters::GetInstance()->FreeGlobalPrinters();
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: 
michael@0: //----------------------------------------------------------------------------------
michael@0: // Enumerate all the Printers from the global array and pass their
michael@0: // names back (usually to script)
michael@0: NS_IMETHODIMP 
michael@0: nsPrinterEnumeratorWin::GetPrinterNameList(nsIStringEnumerator **aPrinterNameList)
michael@0: {
michael@0:   NS_ENSURE_ARG_POINTER(aPrinterNameList);
michael@0:   *aPrinterNameList = nullptr;
michael@0: 
michael@0:   nsresult rv = GlobalPrinters::GetInstance()->EnumeratePrinterList();
michael@0:   if (NS_FAILED(rv)) {
michael@0:     PR_PL(("***** nsDeviceContextSpecWin::GetPrinterNameList - Couldn't enumerate printers!\n"));
michael@0:     return rv;
michael@0:   }
michael@0: 
michael@0:   uint32_t numPrinters = GlobalPrinters::GetInstance()->GetNumPrinters();
michael@0:   nsTArray<nsString> *printers = new nsTArray<nsString>(numPrinters);
michael@0:   if (!printers)
michael@0:     return NS_ERROR_OUT_OF_MEMORY;
michael@0: 
michael@0:   uint32_t printerInx = 0;
michael@0:   while( printerInx < numPrinters ) {
michael@0:     LPWSTR name = GlobalPrinters::GetInstance()->GetItemFromList(printerInx++);
michael@0:     printers->AppendElement(nsDependentString(name));
michael@0:   }
michael@0: 
michael@0:   return NS_NewAdoptingStringEnumerator(aPrinterNameList, printers);
michael@0: }
michael@0: 
michael@0: //----------------------------------------------------------------------------------
michael@0: // Display the AdvancedDocumentProperties for the selected Printer
michael@0: NS_IMETHODIMP nsPrinterEnumeratorWin::DisplayPropertiesDlg(const char16_t *aPrinterName, nsIPrintSettings* aPrintSettings)
michael@0: {
michael@0:   // Implementation removed because it is unused
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: //----------------------------------------------------------------------------------
michael@0: //-- Global Printers
michael@0: //----------------------------------------------------------------------------------
michael@0: 
michael@0: //----------------------------------------------------------------------------------
michael@0: // THe array hold the name and port for each printer
michael@0: void 
michael@0: GlobalPrinters::ReallocatePrinters()
michael@0: {
michael@0:   if (PrintersAreAllocated()) {
michael@0:     FreeGlobalPrinters();
michael@0:   }
michael@0:   mPrinters = new nsTArray<LPWSTR>();
michael@0:   NS_ASSERTION(mPrinters, "Printers Array is NULL!");
michael@0: }
michael@0: 
michael@0: //----------------------------------------------------------------------------------
michael@0: void 
michael@0: GlobalPrinters::FreeGlobalPrinters()
michael@0: {
michael@0:   if (mPrinters != nullptr) {
michael@0:     for (uint32_t i=0;i<mPrinters->Length();i++) {
michael@0:       free(mPrinters->ElementAt(i));
michael@0:     }
michael@0:     delete mPrinters;
michael@0:     mPrinters = nullptr;
michael@0:   }
michael@0: }
michael@0: 
michael@0: //----------------------------------------------------------------------------------
michael@0: nsresult 
michael@0: GlobalPrinters::EnumerateNativePrinters()
michael@0: {
michael@0:   nsresult rv = NS_ERROR_GFX_PRINTER_NO_PRINTER_AVAILABLE;
michael@0:   PR_PL(("-----------------------\n"));
michael@0:   PR_PL(("EnumerateNativePrinters\n"));
michael@0: 
michael@0:   WCHAR szDefaultPrinterName[1024];    
michael@0:   DWORD status = GetProfileStringW(L"devices", 0, L",",
michael@0:                                    szDefaultPrinterName,
michael@0:                                    ArrayLength(szDefaultPrinterName));
michael@0:   if (status > 0) {
michael@0:     DWORD count = 0;
michael@0:     LPWSTR sPtr   = szDefaultPrinterName;
michael@0:     LPWSTR ePtr   = szDefaultPrinterName + status;
michael@0:     LPWSTR prvPtr = sPtr;
michael@0:     while (sPtr < ePtr) {
michael@0:       if (*sPtr == 0) {
michael@0:         LPWSTR name = wcsdup(prvPtr);
michael@0:         mPrinters->AppendElement(name);
michael@0:         PR_PL(("Printer Name:    %s\n", prvPtr));
michael@0:         prvPtr = sPtr+1;
michael@0:         count++;
michael@0:       }
michael@0:       sPtr++;
michael@0:     }
michael@0:     rv = NS_OK;
michael@0:   }
michael@0:   PR_PL(("-----------------------\n"));
michael@0:   return rv;
michael@0: }
michael@0: 
michael@0: //------------------------------------------------------------------
michael@0: // Uses the GetProfileString to get the default printer from the registry
michael@0: void 
michael@0: GlobalPrinters::GetDefaultPrinterName(nsString& aDefaultPrinterName)
michael@0: {
michael@0:   aDefaultPrinterName.Truncate();
michael@0:   WCHAR szDefaultPrinterName[1024];    
michael@0:   DWORD status = GetProfileStringW(L"windows", L"device", 0,
michael@0:                                    szDefaultPrinterName,
michael@0:                                    ArrayLength(szDefaultPrinterName));
michael@0:   if (status > 0) {
michael@0:     WCHAR comma = ',';
michael@0:     LPWSTR sPtr = szDefaultPrinterName;
michael@0:     while (*sPtr != comma && *sPtr != 0) 
michael@0:       sPtr++;
michael@0:     if (*sPtr == comma) {
michael@0:       *sPtr = 0;
michael@0:     }
michael@0:     aDefaultPrinterName = szDefaultPrinterName;
michael@0:   } else {
michael@0:     aDefaultPrinterName = EmptyString();
michael@0:   }
michael@0: 
michael@0:   PR_PL(("DEFAULT PRINTER [%s]\n", aDefaultPrinterName.get()));
michael@0: }
michael@0: 
michael@0: //----------------------------------------------------------------------------------
michael@0: // This goes and gets the list of available printers and puts
michael@0: // the default printer at the beginning of the list
michael@0: nsresult 
michael@0: GlobalPrinters::EnumeratePrinterList()
michael@0: {
michael@0:   // reallocate and get a new list each time it is asked for
michael@0:   // this deletes the list and re-allocates them
michael@0:   ReallocatePrinters();
michael@0: 
michael@0:   // any of these could only fail with an OUT_MEMORY_ERROR
michael@0:   // PRINTER_ENUM_LOCAL should get the network printers on Win95
michael@0:   nsresult rv = EnumerateNativePrinters();
michael@0:   if (NS_FAILED(rv)) return rv;
michael@0: 
michael@0:   // get the name of the default printer
michael@0:   nsAutoString defPrinterName;
michael@0:   GetDefaultPrinterName(defPrinterName);
michael@0: 
michael@0:   // put the default printer at the beginning of list
michael@0:   if (!defPrinterName.IsEmpty()) {
michael@0:     for (uint32_t i=0;i<mPrinters->Length();i++) {
michael@0:       LPWSTR name = mPrinters->ElementAt(i);
michael@0:       if (defPrinterName.Equals(name)) {
michael@0:         if (i > 0) {
michael@0:           LPWSTR ptr = mPrinters->ElementAt(0);
michael@0:           mPrinters->ElementAt(0) = name;
michael@0:           mPrinters->ElementAt(i) = ptr;
michael@0:         }
michael@0:         break;
michael@0:       }
michael@0:     }
michael@0:   }
michael@0: 
michael@0:   // make sure we at least tried to get the printers
michael@0:   if (!PrintersAreAllocated()) {
michael@0:     PR_PL(("***** nsDeviceContextSpecWin::EnumeratePrinterList - Printers aren`t allocated\n"));
michael@0:     return NS_ERROR_FAILURE;
michael@0:   }
michael@0: 
michael@0:   return NS_OK;
michael@0: }
michael@0: