widget/qt/nsDeviceContextSpecQt.cpp

   1 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
   2 /* This Source Code Form is subject to the terms of the Mozilla Public
   3  * License, v. 2.0. If a copy of the MPL was not distributed with this
   4  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   5
   6
   7 #include <QTemporaryFile>
   8 #include <QPrinterInfo>
   9
  10 #define SET_PRINTER_FEATURES_VIA_PREFS 1
  11 #define PRINTERFEATURES_PREF "print.tmp.printerfeatures"
  12
  13 #ifdef MOZ_LOGGING
  14 #define FORCE_PR_LOG 1 /* Allow logging in the release build */
  15 #endif /* MOZ_LOGGING */
  16 #include "prlog.h"
  17
  18 #include "plstr.h"
  19
  20 #include "nsDeviceContextSpecQt.h"
  21
  22 #include "prenv.h" /* for PR_GetEnv */
  23
  24 #include "nsReadableUtils.h"
  25 #include "nsStringEnumerator.h"
  26 #include "nsIServiceManager.h"
  27 #include "nsPrintSettingsQt.h"
  28 #include "nsIFileStreams.h"
  29 #include "nsIFile.h"
  30 #include "nsTArray.h"
  31
  32 #include <unistd.h>
  33 #include <sys/types.h>
  34 #include <sys/stat.h>
  35
  36 #include "gfxPDFSurface.h"
  37
  38 #ifdef PR_LOGGING
  39 static PRLogModuleInfo* DeviceContextSpecQtLM =
  40     PR_NewLogModule("DeviceContextSpecQt");
  41 #endif /* PR_LOGGING */
  42 /* Macro to make lines shorter */
  43 #define DO_PR_DEBUG_LOG(x) PR_LOG(DeviceContextSpecQtLM, PR_LOG_DEBUG, x)
  44
  45 nsDeviceContextSpecQt::nsDeviceContextSpecQt()
  46 {
  47     DO_PR_DEBUG_LOG(("nsDeviceContextSpecQt::nsDeviceContextSpecQt()\n"));
  48 }
  49
  50 nsDeviceContextSpecQt::~nsDeviceContextSpecQt()
  51 {
  52     DO_PR_DEBUG_LOG(("nsDeviceContextSpecQt::~nsDeviceContextSpecQt()\n"));
  53 }
  54
  55 NS_IMPL_ISUPPORTS(nsDeviceContextSpecQt,
  56         nsIDeviceContextSpec)
  57
  58 NS_IMETHODIMP nsDeviceContextSpecQt::GetSurfaceForPrinter(
  59         gfxASurface** aSurface)
  60 {
  61     NS_ENSURE_ARG_POINTER(aSurface);
  62     *aSurface = nullptr;
  63
  64     double width, height;
  65     mPrintSettings->GetEffectivePageSize(&width, &height);
  66
  67     // If we're in landscape mode, we'll be rotating the output --
  68     // need to swap width & height.
  69     int32_t orientation;
  70     mPrintSettings->GetOrientation(&orientation);
  71     if (nsIPrintSettings::kLandscapeOrientation == orientation) {
  72         double tmp = width;
  73         width = height;
  74         height = tmp;
  75     }
  76
  77     // convert twips to points
  78     width  /= TWIPS_PER_POINT_FLOAT;
  79     height /= TWIPS_PER_POINT_FLOAT;
  80
  81     DO_PR_DEBUG_LOG(("\"%s\", %f, %f\n", mPath, width, height));
  82
  83     QTemporaryFile file;
  84     if(!file.open()) {
  85         return NS_ERROR_GFX_PRINTER_COULD_NOT_OPEN_FILE;
  86     }
  87     file.setAutoRemove(false);
  88
  89     nsresult rv = NS_NewNativeLocalFile(
  90             nsDependentCString(file.fileName().toUtf8().constData()),
  91             false,
  92             getter_AddRefs(mSpoolFile));
  93     if (NS_FAILED(rv)) {
  94         file.remove();
  95         return NS_ERROR_GFX_PRINTER_COULD_NOT_OPEN_FILE;
  96     }
  97
  98     mSpoolName = file.fileName().toUtf8().constData();
  99
 100     mSpoolFile->SetPermissions(0600);
 101
 102     nsCOMPtr<nsIFileOutputStream> stream =
 103         do_CreateInstance("@mozilla.org/network/file-output-stream;1");
 104
 105     rv = stream->Init(mSpoolFile, -1, -1, 0);
 106     if (NS_FAILED(rv))
 107         return rv;
 108
 109     int16_t format;
 110     mPrintSettings->GetOutputFormat(&format);
 111
 112     nsRefPtr<gfxASurface> surface;
 113     gfxSize surfaceSize(width, height);
 114
 115     if (format == nsIPrintSettings::kOutputFormatNative) {
 116         if (mIsPPreview) {
 117             // There is nothing to detect on Print Preview, use PS.
 118             // TODO: implement for Qt?
 119             //format = nsIPrintSettings::kOutputFormatPS;
 120             return NS_ERROR_NOT_IMPLEMENTED;
 121         }
 122         format = nsIPrintSettings::kOutputFormatPDF;
 123     }
 124     if (format == nsIPrintSettings::kOutputFormatPDF) {
 125         surface = new gfxPDFSurface(stream, surfaceSize);
 126     } else {
 127         return NS_ERROR_NOT_IMPLEMENTED;
 128     }
 129
 130     NS_ABORT_IF_FALSE(surface, "valid address expected");
 131
 132     surface.swap(*aSurface);
 133     return NS_OK;
 134 }
 135
 136 NS_IMETHODIMP nsDeviceContextSpecQt::Init(nsIWidget* aWidget,
 137         nsIPrintSettings* aPS,
 138         bool aIsPrintPreview)
 139 {
 140     DO_PR_DEBUG_LOG(("nsDeviceContextSpecQt::Init(aPS=%p)\n", aPS));
 141
 142     mPrintSettings = aPS;
 143     mIsPPreview = aIsPrintPreview;
 144
 145     // This is only set by embedders
 146     bool toFile;
 147     aPS->GetPrintToFile(&toFile);
 148
 149     mToPrinter = !toFile && !aIsPrintPreview;
 150
 151     nsCOMPtr<nsPrintSettingsQt> printSettingsQt(do_QueryInterface(aPS));
 152     if (!printSettingsQt)
 153         return NS_ERROR_NO_INTERFACE;
 154     return NS_OK;
 155 }
 156
 157 NS_IMETHODIMP nsDeviceContextSpecQt::GetPath(const char** aPath)
 158 {
 159     *aPath = mPath;
 160     return NS_OK;
 161 }
 162
 163 NS_IMETHODIMP nsDeviceContextSpecQt::BeginDocument(
 164         const nsAString& aTitle,
 165         char16_t* aPrintToFileName,
 166         int32_t aStartPage,
 167         int32_t aEndPage)
 168 {
 169     if (mToPrinter) {
 170         return NS_ERROR_NOT_IMPLEMENTED;
 171     }
 172     return NS_OK;
 173 }
 174
 175 NS_IMETHODIMP nsDeviceContextSpecQt::EndDocument()
 176 {
 177     if (mToPrinter) {
 178         return NS_ERROR_NOT_IMPLEMENTED;
 179     }
 180     // Handle print-to-file ourselves for the benefit of embedders
 181     nsXPIDLString targetPath;
 182     nsCOMPtr<nsIFile> destFile;
 183     mPrintSettings->GetToFileName(getter_Copies(targetPath));
 184
 185     nsresult rv = NS_NewNativeLocalFile(NS_ConvertUTF16toUTF8(targetPath),
 186             false, getter_AddRefs(destFile));
 187     NS_ENSURE_SUCCESS(rv, rv);
 188
 189     nsAutoString destLeafName;
 190     rv = destFile->GetLeafName(destLeafName);
 191     NS_ENSURE_SUCCESS(rv, rv);
 192
 193     nsCOMPtr<nsIFile> destDir;
 194     rv = destFile->GetParent(getter_AddRefs(destDir));
 195     NS_ENSURE_SUCCESS(rv, rv);
 196
 197     rv = mSpoolFile->MoveTo(destDir, destLeafName);
 198     NS_ENSURE_SUCCESS(rv, rv);
 199
 200     // This is the standard way to get the UNIX umask. Ugh.
 201     mode_t mask = umask(0);
 202     umask(mask);
 203     // If you're not familiar with umasks, they contain the bits of what NOT
 204     // to set in the permissions
 205     // (thats because files and directories have different numbers of bits
 206     // for their permissions)
 207     destFile->SetPermissions(0666 & ~(mask));
 208
 209     return NS_OK;
 210 }
 211
 212 //  Printer Enumerator
 213 nsPrinterEnumeratorQt::nsPrinterEnumeratorQt()
 214 {
 215 }
 216
 217 nsPrinterEnumeratorQt::~nsPrinterEnumeratorQt()
 218 {
 219 }
 220
 221 NS_IMPL_ISUPPORTS(nsPrinterEnumeratorQt, nsIPrinterEnumerator)
 222
 223 NS_IMETHODIMP nsPrinterEnumeratorQt::GetPrinterNameList(
 224         nsIStringEnumerator** aPrinterNameList)
 225 {
 226     NS_ENSURE_ARG_POINTER(aPrinterNameList);
 227     *aPrinterNameList = nullptr;
 228
 229     QList<QPrinterInfo> qprinters = QPrinterInfo::availablePrinters();
 230     if (qprinters.size() == 0)
 231         return NS_ERROR_NOT_AVAILABLE;
 232
 233     nsTArray<nsString>* printers =
 234         new nsTArray<nsString>(qprinters.size());
 235
 236     for (int32_t i = 0; i < qprinters.size(); ++i) {
 237         printers->AppendElement(
 238                 nsDependentString(
 239                     (const char16_t*)qprinters[i].printerName().constData()));
 240     }
 241
 242     return NS_NewAdoptingStringEnumerator(aPrinterNameList, printers);
 243 }
 244
 245 NS_IMETHODIMP nsPrinterEnumeratorQt::GetDefaultPrinterName(
 246         char16_t** aDefaultPrinterName)
 247 {
 248     DO_PR_DEBUG_LOG(("nsPrinterEnumeratorQt::GetDefaultPrinterName()\n"));
 249     NS_ENSURE_ARG_POINTER(aDefaultPrinterName);
 250
 251     QString defprinter = QPrinterInfo::defaultPrinter().printerName();
 252     *aDefaultPrinterName = ToNewUnicode(nsDependentString(
 253         (const char16_t*)defprinter.constData()));
 254
 255     DO_PR_DEBUG_LOG(("GetDefaultPrinterName(): default printer='%s'.\n",
 256         NS_ConvertUTF16toUTF8(*aDefaultPrinterName).get()));
 257
 258     return NS_OK;
 259 }
 260
 261 NS_IMETHODIMP nsPrinterEnumeratorQt::InitPrintSettingsFromPrinter(
 262         const char16_t* aPrinterName,
 263         nsIPrintSettings* aPrintSettings)
 264 {
 265     DO_PR_DEBUG_LOG(("nsPrinterEnumeratorQt::InitPrintSettingsFromPrinter()"));
 266     // XXX Leave NS_OK for now
 267     // Probably should use NS_ERROR_NOT_IMPLEMENTED
 268     return NS_OK;
 269 }
 270
 271 NS_IMETHODIMP nsPrinterEnumeratorQt::DisplayPropertiesDlg(
 272         const char16_t* aPrinter,
 273         nsIPrintSettings* aPrintSettings)
 274 {
 275     return NS_ERROR_NOT_IMPLEMENTED;
 276 }
 277