1 files changed, 140 insertions, 0 deletions
diff --git a/layout/printing/nsPrintData.cpp b/layout/printing/nsPrintData.cpp
new file mode 100644
index 000000000..1914f779b
--- /dev/null
+++ b/layout/printing/nsPrintData.cpp
@@ -0,0 +1,140 @@
+/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#include "nsPrintData.h"
+
+#include "nsIStringBundle.h"
+#include "nsIServiceManager.h"
+#include "nsPrintObject.h"
+#include "nsPrintPreviewListener.h"
+#include "nsIWebProgressListener.h"
+#include "mozilla/Services.h"
+
+//-----------------------------------------------------
+// PR LOGGING
+#include "mozilla/Logging.h"
+
+#define DUMP_LAYOUT_LEVEL 9 // this turns on the dumping of each doucment's layout info
+static mozilla::LazyLogModule gPrintingLog("printing");
+
+#define PR_PL(_p1)  MOZ_LOG(gPrintingLog, mozilla::LogLevel::Debug, _p1);
+
+//---------------------------------------------------
+//-- nsPrintData Class Impl
+//---------------------------------------------------
+nsPrintData::nsPrintData(ePrintDataType aType) :
+  mType(aType), mDebugFilePtr(nullptr), mPrintObject(nullptr), mSelectedPO(nullptr),
+  mPrintDocList(0), mIsIFrameSelected(false),
+  mIsParentAFrameSet(false), mOnStartSent(false),
+  mIsAborted(false), mPreparingForPrint(false), mDocWasToBeDestroyed(false),
+  mShrinkToFit(false), mPrintFrameType(nsIPrintSettings::kFramesAsIs),
+  mNumPrintablePages(0), mNumPagesPrinted(0),
+  mShrinkRatio(1.0), mOrigDCScale(1.0), mPPEventListeners(nullptr),
+  mBrandName(nullptr)
+{
+  nsCOMPtr<nsIStringBundle> brandBundle;
+  nsCOMPtr<nsIStringBundleService> svc =
+    mozilla::services::GetStringBundleService();
+  if (svc) {
+    svc->CreateBundle( "chrome://branding/locale/brand.properties", getter_AddRefs( brandBundle ) );
+    if (brandBundle) {
+      brandBundle->GetStringFromName(u"brandShortName", &mBrandName );
+    }
+  }
+
+  if (!mBrandName) {
+    mBrandName = ToNewUnicode(NS_LITERAL_STRING("Mozilla Document"));
+  }
+
+}
+
+nsPrintData::~nsPrintData()
+{
+  // remove the event listeners
+  if (mPPEventListeners) {
+    mPPEventListeners->RemoveListeners();
+    NS_RELEASE(mPPEventListeners);
+  }
+
+  // Only Send an OnEndPrinting if we have started printing
+  if (mOnStartSent && mType != eIsPrintPreview) {
+    OnEndPrinting();
+  }
+
+  if (mPrintDC && !mDebugFilePtr) {
+    PR_PL(("****************** End Document ************************\n"));
+    PR_PL(("\n"));
+    bool isCancelled = false;
+    mPrintSettings->GetIsCancelled(&isCancelled);
+
+    nsresult rv = NS_OK;
+    if (mType == eIsPrinting) {
+      if (!isCancelled && !mIsAborted) {
+        rv = mPrintDC->EndDocument();
+      } else {
+        rv = mPrintDC->AbortDocument();
+      }
+      if (NS_FAILED(rv)) {
+        // XXX nsPrintData::ShowPrintErrorDialog(rv);
+      }
+    }
+  }
+
+  delete mPrintObject;
+
+  if (mBrandName) {
+    free(mBrandName);
+  }
+}
+
+void nsPrintData::OnStartPrinting()
+{
+  if (!mOnStartSent) {
+    DoOnProgressChange(0, 0, true, nsIWebProgressListener::STATE_START|nsIWebProgressListener::STATE_IS_DOCUMENT|nsIWebProgressListener::STATE_IS_NETWORK);
+    mOnStartSent = true;
+  }
+}
+
+void nsPrintData::OnEndPrinting()
+{
+  DoOnProgressChange(100, 100, true, nsIWebProgressListener::STATE_STOP|nsIWebProgressListener::STATE_IS_DOCUMENT);
+  DoOnProgressChange(100, 100, true, nsIWebProgressListener::STATE_STOP|nsIWebProgressListener::STATE_IS_NETWORK);
+}
+
+void
+nsPrintData::DoOnProgressChange(int32_t      aProgress,
+                                int32_t      aMaxProgress,
+                                bool         aDoStartStop,
+                                int32_t      aFlag)
+{
+  size_t numberOfListeners = mPrintProgressListeners.Length();
+  for (size_t i = 0; i < numberOfListeners; ++i) {
+    nsCOMPtr<nsIWebProgressListener> listener =
+      mPrintProgressListeners.SafeElementAt(i);
+    if (NS_WARN_IF(!listener)) {
+      continue;
+    }
+    listener->OnProgressChange(nullptr, nullptr, aProgress, aMaxProgress,
+                               aProgress, aMaxProgress);
+    if (aDoStartStop) {
+      listener->OnStateChange(nullptr, nullptr, aFlag, NS_OK);
+    }
+  }
+}
+
+void
+nsPrintData::DoOnStatusChange(nsresult aStatus)
+{
+  size_t numberOfListeners = mPrintProgressListeners.Length();
+  for (size_t i = 0; i < numberOfListeners; ++i) {
+    nsCOMPtr<nsIWebProgressListener> listener =
+      mPrintProgressListeners.SafeElementAt(i);
+    if (NS_WARN_IF(!listener)) {
+      continue;
+    }
+    listener->OnStatusChange(nullptr, nullptr, aStatus, nullptr);
+  }
+}
+