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 "nsPrintSettingsX.h"
michael@0: #include "nsObjCExceptions.h"
michael@0: 
michael@0: #include "plbase64.h"
michael@0: #include "plstr.h"
michael@0: 
michael@0: #include "nsCocoaUtils.h"
michael@0: 
michael@0: #include "mozilla/Preferences.h"
michael@0: 
michael@0: using namespace mozilla;
michael@0: 
michael@0: #define MAC_OS_X_PAGE_SETUP_PREFNAME    "print.macosx.pagesetup-2"
michael@0: 
michael@0: NS_IMPL_ISUPPORTS_INHERITED(nsPrintSettingsX, nsPrintSettings, nsPrintSettingsX)
michael@0: 
michael@0: nsPrintSettingsX::nsPrintSettingsX()
michael@0: {
michael@0:   NS_OBJC_BEGIN_TRY_ABORT_BLOCK;
michael@0: 
michael@0:   mPrintInfo = [[NSPrintInfo sharedPrintInfo] copy];
michael@0: 
michael@0:   NS_OBJC_END_TRY_ABORT_BLOCK;
michael@0: }
michael@0: 
michael@0: nsPrintSettingsX::nsPrintSettingsX(const nsPrintSettingsX& src)
michael@0: {
michael@0:   *this = src;
michael@0: }
michael@0: 
michael@0: nsPrintSettingsX::~nsPrintSettingsX()
michael@0: {
michael@0:   NS_OBJC_BEGIN_TRY_ABORT_BLOCK;
michael@0: 
michael@0:   [mPrintInfo release];
michael@0: 
michael@0:   NS_OBJC_END_TRY_ABORT_BLOCK;
michael@0: }
michael@0: 
michael@0: nsPrintSettingsX& nsPrintSettingsX::operator=(const nsPrintSettingsX& rhs)
michael@0: {
michael@0:   NS_OBJC_BEGIN_TRY_ABORT_BLOCK_RETURN;
michael@0: 
michael@0:   if (this == &rhs) {
michael@0:     return *this;
michael@0:   }
michael@0:   
michael@0:   nsPrintSettings::operator=(rhs);
michael@0: 
michael@0:   [mPrintInfo release];
michael@0:   mPrintInfo = [rhs.mPrintInfo copy];
michael@0: 
michael@0:   return *this;
michael@0: 
michael@0:   NS_OBJC_END_TRY_ABORT_BLOCK_RETURN(*this);
michael@0: }
michael@0: 
michael@0: nsresult nsPrintSettingsX::Init()
michael@0: {
michael@0:   NS_OBJC_BEGIN_TRY_ABORT_BLOCK_NSRESULT;
michael@0: 
michael@0:   InitUnwriteableMargin();
michael@0: 
michael@0:   return NS_OK;
michael@0: 
michael@0:   NS_OBJC_END_TRY_ABORT_BLOCK_NSRESULT;
michael@0: }
michael@0: 
michael@0: // Should be called whenever the page format changes.
michael@0: NS_IMETHODIMP nsPrintSettingsX::InitUnwriteableMargin()
michael@0: {
michael@0:   NS_OBJC_BEGIN_TRY_ABORT_BLOCK_NSRESULT;
michael@0: 
michael@0:   PMPaper paper;
michael@0:   PMPaperMargins paperMargin;
michael@0:   PMPageFormat pageFormat = GetPMPageFormat();
michael@0:   ::PMGetPageFormatPaper(pageFormat, &paper);
michael@0:   ::PMPaperGetMargins(paper, &paperMargin);
michael@0:   mUnwriteableMargin.top    = NS_POINTS_TO_INT_TWIPS(paperMargin.top);
michael@0:   mUnwriteableMargin.left   = NS_POINTS_TO_INT_TWIPS(paperMargin.left);
michael@0:   mUnwriteableMargin.bottom = NS_POINTS_TO_INT_TWIPS(paperMargin.bottom);
michael@0:   mUnwriteableMargin.right  = NS_POINTS_TO_INT_TWIPS(paperMargin.right);
michael@0: 
michael@0:   return NS_OK;
michael@0: 
michael@0:   NS_OBJC_END_TRY_ABORT_BLOCK_NSRESULT;  
michael@0: }
michael@0: 
michael@0: void
michael@0: nsPrintSettingsX::SetCocoaPrintInfo(NSPrintInfo* aPrintInfo)
michael@0: {
michael@0:   mPrintInfo = aPrintInfo;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP nsPrintSettingsX::ReadPageFormatFromPrefs()
michael@0: {
michael@0:   NS_OBJC_BEGIN_TRY_ABORT_BLOCK_NSRESULT;
michael@0: 
michael@0:   nsAutoCString encodedData;
michael@0:   nsresult rv =
michael@0:     Preferences::GetCString(MAC_OS_X_PAGE_SETUP_PREFNAME, &encodedData);
michael@0:   if (NS_FAILED(rv)) {
michael@0:     return rv;
michael@0:   }
michael@0: 
michael@0:   // decode the base64
michael@0:   char* decodedData = PL_Base64Decode(encodedData.get(), encodedData.Length(), nullptr);
michael@0:   NSData* data = [NSData dataWithBytes:decodedData length:strlen(decodedData)];
michael@0:   if (!data)
michael@0:     return NS_ERROR_FAILURE;
michael@0: 
michael@0:   PMPageFormat newPageFormat;
michael@0:   OSStatus status = ::PMPageFormatCreateWithDataRepresentation((CFDataRef)data, &newPageFormat);
michael@0:   if (status == noErr) {
michael@0:     SetPMPageFormat(newPageFormat);
michael@0:   }
michael@0:   InitUnwriteableMargin();
michael@0: 
michael@0:   return NS_OK;
michael@0: 
michael@0:   NS_OBJC_END_TRY_ABORT_BLOCK_NSRESULT;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP nsPrintSettingsX::WritePageFormatToPrefs()
michael@0: {
michael@0:   NS_OBJC_BEGIN_TRY_ABORT_BLOCK_NSRESULT;
michael@0: 
michael@0:   PMPageFormat pageFormat = GetPMPageFormat();
michael@0:   if (pageFormat == kPMNoPageFormat)
michael@0:     return NS_ERROR_NOT_INITIALIZED;
michael@0: 
michael@0:   NSData* data = nil;
michael@0:   OSStatus err = ::PMPageFormatCreateDataRepresentation(pageFormat, (CFDataRef*)&data, kPMDataFormatXMLDefault);
michael@0:   if (err != noErr)
michael@0:     return NS_ERROR_FAILURE;
michael@0: 
michael@0:   nsAutoCString encodedData;
michael@0:   encodedData.Adopt(PL_Base64Encode((char*)[data bytes], [data length], nullptr));
michael@0:   if (!encodedData.get())
michael@0:     return NS_ERROR_OUT_OF_MEMORY;
michael@0: 
michael@0:   return Preferences::SetCString(MAC_OS_X_PAGE_SETUP_PREFNAME, encodedData);
michael@0: 
michael@0:   NS_OBJC_END_TRY_ABORT_BLOCK_NSRESULT;
michael@0: }
michael@0: 
michael@0: nsresult nsPrintSettingsX::_Clone(nsIPrintSettings **_retval)
michael@0: {
michael@0:   NS_ENSURE_ARG_POINTER(_retval);
michael@0:   *_retval = nullptr;
michael@0:   
michael@0:   nsPrintSettingsX *newSettings = new nsPrintSettingsX(*this);
michael@0:   if (!newSettings)
michael@0:     return NS_ERROR_FAILURE;
michael@0:   *_retval = newSettings;
michael@0:   NS_ADDREF(*_retval);
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP nsPrintSettingsX::_Assign(nsIPrintSettings *aPS)
michael@0: {
michael@0:   nsPrintSettingsX *printSettingsX = static_cast<nsPrintSettingsX*>(aPS);
michael@0:   if (!printSettingsX)
michael@0:     return NS_ERROR_UNEXPECTED;
michael@0:   *this = *printSettingsX;
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: PMPrintSettings
michael@0: nsPrintSettingsX::GetPMPrintSettings()
michael@0: {
michael@0:   return static_cast<PMPrintSettings>([mPrintInfo PMPrintSettings]);
michael@0: }
michael@0: 
michael@0: PMPrintSession
michael@0: nsPrintSettingsX::GetPMPrintSession()
michael@0: {
michael@0:   return static_cast<PMPrintSession>([mPrintInfo PMPrintSession]);
michael@0: }
michael@0: 
michael@0: PMPageFormat
michael@0: nsPrintSettingsX::GetPMPageFormat()
michael@0: {
michael@0:   return static_cast<PMPageFormat>([mPrintInfo PMPageFormat]);
michael@0: }
michael@0: 
michael@0: void
michael@0: nsPrintSettingsX::SetPMPageFormat(PMPageFormat aPageFormat)
michael@0: {
michael@0:   PMPageFormat oldPageFormat = GetPMPageFormat();
michael@0:   ::PMCopyPageFormat(aPageFormat, oldPageFormat);
michael@0:   [mPrintInfo updateFromPMPageFormat];
michael@0: }
michael@0: