Split CPDFXFA_Document apart

fpdfsdk/fpdfxfa/cpdfxfa_docenvironment.cpp

-// Copyright 2014 PDFium Authors. All rights reserved.
+// Copyright 2016 PDFium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
 
-#include "fpdfsdk/fpdfxfa/include/fpdfxfa_doc.h"
+#include "fpdfsdk/fpdfxfa/include/cpdfxfa_docenvironment.h"
 
 #include "core/fpdfapi/fpdf_parser/include/cpdf_array.h"
-#include "core/fpdfapi/fpdf_parser/include/cpdf_document.h"
+#include "core/fpdfapi/fpdf_parser/include/cpdf_string.h"
 #include "core/fpdfapi/fpdf_parser/include/cpdf_stream_acc.h"
-#include "core/fpdfapi/fpdf_parser/include/cpdf_string.h"
-#include "fpdfsdk/fpdfxfa/include/fpdfxfa_app.h"
+#include "fpdfsdk/fpdfxfa/include/fpdfxfa_doc.h"
 #include "fpdfsdk/fpdfxfa/include/fpdfxfa_page.h"
-#include "fpdfsdk/fpdfxfa/include/fpdfxfa_util.h"
-#include "fpdfsdk/include/fsdk_define.h"
 #include "fpdfsdk/include/cpdfsdk_interform.h"
+#include "fpdfsdk/javascript/ijs_runtime.h"
 #include "fpdfsdk/include/fsdk_mgr.h"

[Tom Sepez, 2016/09/08 17:12:37]

nit: header order.

[dsinclair, 2016/09/08 18:07:49]

Sorting is hard.

-#include "fpdfsdk/javascript/ijs_runtime.h"
-#include "public/fpdf_formfill.h"
-#include "xfa/fxfa/include/cxfa_eventparam.h"
-#include "xfa/fxfa/include/xfa_ffapp.h"
-#include "xfa/fxfa/include/xfa_ffdoc.h"
 #include "xfa/fxfa/include/xfa_ffdocview.h"
-#include "xfa/fxfa/include/xfa_ffpageview.h"
 #include "xfa/fxfa/include/xfa_ffwidget.h"
 #include "xfa/fxfa/include/xfa_ffwidgethandler.h"
 
 #define IDS_XFA_Validate_Input                                          \
   "At least one required field was empty. Please fill in the required " \
   "fields\r\n(highlighted) before continuing."
 
 // submit
 #define FXFA_CONFIG 0x00000001
 #define FXFA_TEMPLATE 0x00000010
 #define FXFA_LOCALESET 0x00000100
 #define FXFA_DATASETS 0x00001000
 #define FXFA_XMPMETA 0x00010000
 #define FXFA_XFDF 0x00100000
 #define FXFA_FORM 0x01000000
 #define FXFA_PDF 0x10000000
 
-#ifndef _WIN32
-extern void SetLastError(int err);
-extern int GetLastError();
-#endif
-
 namespace {
 
 CFX_FloatRect FXRect2PDFRect(const CFX_RectF& fxRectF) {
   return CFX_FloatRect(fxRectF.left, fxRectF.bottom(), fxRectF.right(),
                        fxRectF.top);
 }
 
 }  // namespace
 
-CPDFXFA_Document::CPDFXFA_Document(std::unique_ptr<CPDF_Document> pPDFDoc,
-                                   CPDFXFA_App* pProvider)
-    : m_iDocType(DOCTYPE_PDF),
-      m_pPDFDoc(std::move(pPDFDoc)),
-      m_pXFADocView(nullptr),
-      m_pApp(pProvider),
-      m_pJSContext(nullptr),
-      m_nLoadStatus(FXFA_LOADSTATUS_PRELOAD),
-      m_nPageCount(0) {}
+CPDFXFA_DocEnvironment::CPDFXFA_DocEnvironment(CPDFXFA_Document* doc)

[Tom Sepez, 2016/09/08 17:12:37]

any code changes here?  Its just moved up from below unchanged, right?  I didn't
scrutinize it.  Let me know if I have to.

[dsinclair, 2016/09/08 18:07:49]

The code changes were all to change things like m_pPDFDoc to
m_pDocument->GetPDFDoc() and other document member accesses.

+    : m_pDocument(doc), m_pJSContext(nullptr) {}
 
-CPDFXFA_Document::~CPDFXFA_Document() {
-  m_nLoadStatus = FXFA_LOADSTATUS_CLOSING;
-
-  if (m_pXFADoc) {
-    CXFA_FFApp* pApp = m_pApp->GetXFAApp();
-    if (pApp) {
-      CXFA_FFDocHandler* pDocHandler = pApp->GetDocHandler();
-      if (pDocHandler)
-        CloseXFADoc(pDocHandler);
-    }
-    m_pXFADoc.reset();
-  }
-  if (m_pJSContext && m_pSDKDoc && m_pSDKDoc->GetEnv())
-    m_pSDKDoc->GetEnv()->GetJSRuntime()->ReleaseContext(m_pJSContext);
-
-  m_nLoadStatus = FXFA_LOADSTATUS_CLOSED;
+CPDFXFA_DocEnvironment::~CPDFXFA_DocEnvironment() {
+  if (m_pJSContext && m_pDocument->GetSDKDoc() &&
+      m_pDocument->GetSDKDoc()->GetEnv())
+    m_pDocument->GetSDKDoc()->GetEnv()->GetJSRuntime()->ReleaseContext(
+        m_pJSContext);
 }
 
-FX_BOOL CPDFXFA_Document::LoadXFADoc() {
-  m_nLoadStatus = FXFA_LOADSTATUS_LOADING;
-
-  if (!m_pPDFDoc)
-    return FALSE;
-
-  m_XFAPageList.RemoveAll();
-
-  CXFA_FFApp* pApp = m_pApp->GetXFAApp();
-  if (!pApp)
-    return FALSE;
-
-  m_pXFADoc.reset(pApp->CreateDoc(this, m_pPDFDoc.get()));
-  if (!m_pXFADoc) {
-    SetLastError(FPDF_ERR_XFALOAD);
-    return FALSE;
-  }
-
-  CXFA_FFDocHandler* pDocHandler = pApp->GetDocHandler();
-  if (!pDocHandler) {
-    SetLastError(FPDF_ERR_XFALOAD);
-    return FALSE;
-  }
-
-  m_pXFADoc->StartLoad();
-  int iStatus = m_pXFADoc->DoLoad(nullptr);
-  if (iStatus != XFA_PARSESTATUS_Done) {
-    CloseXFADoc(pDocHandler);
-    SetLastError(FPDF_ERR_XFALOAD);
-    return FALSE;
-  }
-  m_pXFADoc->StopLoad();
-  m_pXFADoc->GetXFADoc()->InitScriptContext(m_pApp->GetJSERuntime());
-
-  if (m_pXFADoc->GetDocType() == XFA_DOCTYPE_Dynamic)
-    m_iDocType = DOCTYPE_DYNAMIC_XFA;
-  else
-    m_iDocType = DOCTYPE_STATIC_XFA;
-
-  m_pXFADocView = m_pXFADoc->CreateDocView(XFA_DOCVIEW_View);
-  if (m_pXFADocView->StartLayout() < 0) {
-    CloseXFADoc(pDocHandler);
-    SetLastError(FPDF_ERR_XFALAYOUT);
-    return FALSE;
-  }
-
-  m_pXFADocView->DoLayout(nullptr);
-  m_pXFADocView->StopLayout();
-  m_nLoadStatus = FXFA_LOADSTATUS_LOADED;
-
-  return TRUE;
+void CPDFXFA_DocEnvironment::SetChangeMark(CXFA_FFDoc* hDoc) {
+  if (hDoc == m_pDocument->GetXFADoc() && m_pDocument->GetSDKDoc())
+    m_pDocument->GetSDKDoc()->SetChangeMark();
 }
 
-int CPDFXFA_Document::GetPageCount() {
-  if (!m_pPDFDoc && !m_pXFADoc)
-    return 0;
-
-  switch (m_iDocType) {
-    case DOCTYPE_PDF:
-    case DOCTYPE_STATIC_XFA:
-      if (m_pPDFDoc)
-        return m_pPDFDoc->GetPageCount();
-    case DOCTYPE_DYNAMIC_XFA:
-      if (m_pXFADoc)
-        return m_pXFADocView->CountPageViews();
-    default:
-      return 0;
-  }
-}
-
-CPDFXFA_Page* CPDFXFA_Document::GetPage(int page_index) {
-  if (page_index < 0)
-    return nullptr;
-
-  CPDFXFA_Page* pPage = nullptr;
-  int nCount = m_XFAPageList.GetSize();
-  if (nCount > 0 && page_index < nCount) {
-    pPage = m_XFAPageList.GetAt(page_index);
-    if (pPage)
-      pPage->Retain();
-  } else {
-    m_nPageCount = GetPageCount();
-    m_XFAPageList.SetSize(m_nPageCount);
-  }
-  if (pPage)
-    return pPage;
-
-  pPage = new CPDFXFA_Page(this, page_index);
-  if (!pPage->LoadPage()) {
-    pPage->Release();
-    return nullptr;
-  }
-  m_XFAPageList.SetAt(page_index, pPage);
-  return pPage;
-}
-
-CPDFXFA_Page* CPDFXFA_Document::GetPage(CXFA_FFPageView* pPage) {
-  if (!pPage)
-    return nullptr;
-
-  if (!m_pXFADoc)
-    return nullptr;
-
-  if (m_iDocType != DOCTYPE_DYNAMIC_XFA)
-    return nullptr;
-
-  int nSize = m_XFAPageList.GetSize();
-  for (int i = 0; i < nSize; i++) {
-    CPDFXFA_Page* pTempPage = m_XFAPageList.GetAt(i);
-    if (!pTempPage)
-      continue;
-    if (pTempPage->GetXFAPageView() && pTempPage->GetXFAPageView() == pPage)
-      return pTempPage;
-  }
-
-  return nullptr;
-}
-
-void CPDFXFA_Document::DeletePage(int page_index) {
-  // Delete from the document first because, if GetPage was never called for
-  // this |page_index| then |m_XFAPageList| may have size < |page_index| even
-  // if it's a valid page in the document.
-  if (m_pPDFDoc)
-    m_pPDFDoc->DeletePage(page_index);
-
-  if (page_index < 0 || page_index >= m_XFAPageList.GetSize())
+void CPDFXFA_DocEnvironment::InvalidateRect(CXFA_FFPageView* pPageView,
+                                            const CFX_RectF& rt,
+                                            uint32_t dwFlags /* = 0 */) {
+  if (!m_pDocument->GetXFADoc() || !m_pDocument->GetSDKDoc())
     return;
 
-  if (CPDFXFA_Page* pPage = m_XFAPageList.GetAt(page_index))
-    pPage->Release();
-}
-
-void CPDFXFA_Document::RemovePage(CPDFXFA_Page* page) {
-  m_XFAPageList.SetAt(page->GetPageIndex(), nullptr);
-}
-
-CPDFSDK_Document* CPDFXFA_Document::GetSDKDocument(
-    CPDFDoc_Environment* pFormFillEnv) {
-  if (!m_pSDKDoc && pFormFillEnv)
-    m_pSDKDoc.reset(new CPDFSDK_Document(this, pFormFillEnv));
-  return m_pSDKDoc.get();
-}
-
-void CPDFXFA_Document::SetChangeMark(CXFA_FFDoc* hDoc) {
-  if (hDoc == m_pXFADoc.get() && m_pSDKDoc)
-    m_pSDKDoc->SetChangeMark();
-}
-
-void CPDFXFA_Document::InvalidateRect(CXFA_FFPageView* pPageView,
-                                      const CFX_RectF& rt,
-                                      uint32_t dwFlags /* = 0 */) {
-  if (!m_pXFADoc || !m_pSDKDoc)
+  if (m_pDocument->GetDocType() != DOCTYPE_DYNAMIC_XFA)
     return;
 
-  if (m_iDocType != DOCTYPE_DYNAMIC_XFA)
-    return;
-
-  CPDFXFA_Page* pPage = GetPage(pPageView);
+  CPDFXFA_Page* pPage = m_pDocument->GetPage(pPageView);
   if (!pPage)
     return;
 
-  CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
+  CPDFDoc_Environment* pEnv = m_pDocument->GetSDKDoc()->GetEnv();
   if (!pEnv)
     return;
 
   CFX_FloatRect rcPage = FXRect2PDFRect(rt);
   pEnv->FFI_Invalidate((FPDF_PAGE)pPage, rcPage.left, rcPage.bottom,
                        rcPage.right, rcPage.top);
 }
 
-void CPDFXFA_Document::DisplayCaret(CXFA_FFWidget* hWidget,
-                                    FX_BOOL bVisible,
-                                    const CFX_RectF* pRtAnchor) {
-  if (!hWidget || !pRtAnchor || !m_pXFADoc || !m_pSDKDoc || !m_pXFADocView)
+void CPDFXFA_DocEnvironment::DisplayCaret(CXFA_FFWidget* hWidget,
+                                          FX_BOOL bVisible,
+                                          const CFX_RectF* pRtAnchor) {
+  if (!hWidget || !pRtAnchor || !m_pDocument->GetXFADoc() ||

[Tom Sepez, 2016/09/08 17:12:37]

Do we have to worry about null m_pDocuments now that these have moved to a
different class?

[dsinclair, 2016/09/08 18:07:49]

Technically, probably. What about storing as a CPDFXFA_Document& internally
instead of a CPDFXFA_Document*const?

[Tom Sepez, 2016/09/08 18:22:01]

Nah, lets just assert non-nullness in the ctor at line 45, and the m_pDocument
is const for the lifetime of the object, so we're cool.

+      !m_pDocument->GetSDKDoc() || !m_pDocument->GetXFADocView())
     return;
 
-  if (m_iDocType != DOCTYPE_DYNAMIC_XFA)
+  if (m_pDocument->GetDocType() != DOCTYPE_DYNAMIC_XFA)
     return;
 
-  CXFA_FFWidgetHandler* pWidgetHandler = m_pXFADocView->GetWidgetHandler();
+  CXFA_FFWidgetHandler* pWidgetHandler =
+      m_pDocument->GetXFADocView()->GetWidgetHandler();
   if (!pWidgetHandler)
     return;
 
   CXFA_FFPageView* pPageView = hWidget->GetPageView();
   if (!pPageView)
     return;
 
-  CPDFXFA_Page* pPage = GetPage(pPageView);
+  CPDFXFA_Page* pPage = m_pDocument->GetPage(pPageView);
   if (!pPage)
     return;
 
-  CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
+  CPDFDoc_Environment* pEnv = m_pDocument->GetSDKDoc()->GetEnv();
   if (!pEnv)
     return;
 
   CFX_FloatRect rcCaret = FXRect2PDFRect(*pRtAnchor);
   pEnv->FFI_DisplayCaret((FPDF_PAGE)pPage, bVisible, rcCaret.left, rcCaret.top,
                          rcCaret.right, rcCaret.bottom);
 }
 
-FX_BOOL CPDFXFA_Document::GetPopupPos(CXFA_FFWidget* hWidget,
-                                      FX_FLOAT fMinPopup,
-                                      FX_FLOAT fMaxPopup,
-                                      const CFX_RectF& rtAnchor,
-                                      CFX_RectF& rtPopup) {
+FX_BOOL CPDFXFA_DocEnvironment::GetPopupPos(CXFA_FFWidget* hWidget,
+                                            FX_FLOAT fMinPopup,
+                                            FX_FLOAT fMaxPopup,
+                                            const CFX_RectF& rtAnchor,
+                                            CFX_RectF& rtPopup) {
   if (!hWidget)
     return FALSE;
 
   CXFA_FFPageView* pXFAPageView = hWidget->GetPageView();
   if (!pXFAPageView)
     return FALSE;
 
-  CPDFXFA_Page* pPage = GetPage(pXFAPageView);
+  CPDFXFA_Page* pPage = m_pDocument->GetPage(pXFAPageView);
   if (!pPage)
     return FALSE;
 
   CXFA_WidgetAcc* pWidgetAcc = hWidget->GetDataAcc();
   int nRotate = pWidgetAcc->GetRotate();
-  CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
+  CPDFDoc_Environment* pEnv = m_pDocument->GetSDKDoc()->GetEnv();
   if (!pEnv)
     return FALSE;
 
   FS_RECTF pageViewRect = {0.0f, 0.0f, 0.0f, 0.0f};
   pEnv->FFI_GetPageViewRect(pPage, pageViewRect);
 
   int t1;
   int t2;
   CFX_FloatRect rcAnchor = FXRect2PDFRect(rtAnchor);
   switch (nRotate) {
@@ skipping 77 matching lines @@
       }
       break;
     }
     default:
       break;
   }
 
   return TRUE;
 }
 
-FX_BOOL CPDFXFA_Document::PopupMenu(CXFA_FFWidget* hWidget,
-                                    CFX_PointF ptPopup) {
+FX_BOOL CPDFXFA_DocEnvironment::PopupMenu(CXFA_FFWidget* hWidget,
+                                          CFX_PointF ptPopup) {
   if (!hWidget)
     return FALSE;
 
   CXFA_FFPageView* pXFAPageView = hWidget->GetPageView();
   if (!pXFAPageView)
     return FALSE;
 
-  CPDFXFA_Page* pPage = GetPage(pXFAPageView);
+  CPDFXFA_Page* pPage = m_pDocument->GetPage(pXFAPageView);
   if (!pPage)
     return FALSE;
 
-  CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
+  CPDFDoc_Environment* pEnv = m_pDocument->GetSDKDoc()->GetEnv();
   if (!pEnv)
     return FALSE;
 
   int menuFlag = 0;
   if (hWidget->CanUndo())
     menuFlag |= FXFA_MEMU_UNDO;
   if (hWidget->CanRedo())
     menuFlag |= FXFA_MEMU_REDO;
   if (hWidget->CanPaste())
     menuFlag |= FXFA_MEMU_PASTE;
   if (hWidget->CanCopy())
     menuFlag |= FXFA_MEMU_COPY;
   if (hWidget->CanCut())
     menuFlag |= FXFA_MEMU_CUT;
   if (hWidget->CanSelectAll())
     menuFlag |= FXFA_MEMU_SELECTALL;
 
   return pEnv->FFI_PopupMenu(pPage, hWidget, menuFlag, ptPopup, nullptr);
 }
 
-void CPDFXFA_Document::PageViewEvent(CXFA_FFPageView* pPageView,
-                                     uint32_t dwFlags) {
-  CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
+void CPDFXFA_DocEnvironment::PageViewEvent(CXFA_FFPageView* pPageView,
+                                           uint32_t dwFlags) {
+  CPDFDoc_Environment* pEnv = m_pDocument->GetSDKDoc()->GetEnv();
   if (!pEnv)
     return;
 
-  if (m_nLoadStatus == FXFA_LOADSTATUS_LOADING ||
-      m_nLoadStatus == FXFA_LOADSTATUS_CLOSING ||
+  if (m_pDocument->GetLoadStatus() == FXFA_LOADSTATUS_LOADING ||
+      m_pDocument->GetLoadStatus() == FXFA_LOADSTATUS_CLOSING ||
       XFA_PAGEVIEWEVENT_StopLayout != dwFlags)
     return;
 
-  int nNewCount = GetPageCount();
-  if (nNewCount == m_nPageCount)
+  int nNewCount = m_pDocument->GetPageCount();
+  if (nNewCount == m_pDocument->GetOriginalPageCount())
     return;
 
-  CXFA_FFDocView* pXFADocView = GetXFADocView();
+  CXFA_FFDocView* pXFADocView = m_pDocument->GetXFADocView();
   if (!pXFADocView)
     return;
 
-  for (int iPageIter = 0; iPageIter < m_nPageCount; iPageIter++) {
-    CPDFXFA_Page* pPage = m_XFAPageList.GetAt(iPageIter);
+  for (int iPageIter = 0; iPageIter < m_pDocument->GetOriginalPageCount();
+       iPageIter++) {
+    CPDFXFA_Page* pPage = m_pDocument->GetXFAPageList()->GetAt(iPageIter);
     if (!pPage)
       continue;
 
-    m_pSDKDoc->RemovePageView(pPage);
+    m_pDocument->GetSDKDoc()->RemovePageView(pPage);
     pPage->SetXFAPageView(pXFADocView->GetPageView(iPageIter));
   }
 
-  int flag = (nNewCount < m_nPageCount) ? FXFA_PAGEVIEWEVENT_POSTREMOVED
-                                        : FXFA_PAGEVIEWEVENT_POSTADDED;
-  int count = FXSYS_abs(nNewCount - m_nPageCount);
-  m_nPageCount = nNewCount;
-  m_XFAPageList.SetSize(nNewCount);
+  int flag = (nNewCount < m_pDocument->GetOriginalPageCount())
+                 ? FXFA_PAGEVIEWEVENT_POSTREMOVED
+                 : FXFA_PAGEVIEWEVENT_POSTADDED;
+  int count = FXSYS_abs(nNewCount - m_pDocument->GetOriginalPageCount());
+  m_pDocument->SetOriginalPageCount(nNewCount);
   pEnv->FFI_PageEvent(count, flag);
 }
 
-void CPDFXFA_Document::WidgetPostAdd(CXFA_FFWidget* hWidget,
-                                     CXFA_WidgetAcc* pWidgetData) {
-  if (m_iDocType != DOCTYPE_DYNAMIC_XFA || !hWidget)
+void CPDFXFA_DocEnvironment::WidgetPostAdd(CXFA_FFWidget* hWidget,
+                                           CXFA_WidgetAcc* pWidgetData) {
+  if (m_pDocument->GetDocType() != DOCTYPE_DYNAMIC_XFA || !hWidget)
     return;
 
   CXFA_FFPageView* pPageView = hWidget->GetPageView();
   if (!pPageView)
     return;
 
-  CPDFXFA_Page* pXFAPage = GetPage(pPageView);
+  CPDFXFA_Page* pXFAPage = m_pDocument->GetPage(pPageView);
   if (!pXFAPage)
     return;
 
-  m_pSDKDoc->GetPageView(pXFAPage, true)->AddAnnot(hWidget);
+  m_pDocument->GetSDKDoc()->GetPageView(pXFAPage, true)->AddAnnot(hWidget);
 }
 
-void CPDFXFA_Document::WidgetPreRemove(CXFA_FFWidget* hWidget,
-                                       CXFA_WidgetAcc* pWidgetData) {
-  if (m_iDocType != DOCTYPE_DYNAMIC_XFA || !hWidget)
+void CPDFXFA_DocEnvironment::WidgetPreRemove(CXFA_FFWidget* hWidget,
+                                             CXFA_WidgetAcc* pWidgetData) {
+  if (m_pDocument->GetDocType() != DOCTYPE_DYNAMIC_XFA || !hWidget)
     return;
 
   CXFA_FFPageView* pPageView = hWidget->GetPageView();
   if (!pPageView)
     return;
 
-  CPDFXFA_Page* pXFAPage = GetPage(pPageView);
+  CPDFXFA_Page* pXFAPage = m_pDocument->GetPage(pPageView);
   if (!pXFAPage)
     return;
 
-  CPDFSDK_PageView* pSdkPageView = m_pSDKDoc->GetPageView(pXFAPage, true);
+  CPDFSDK_PageView* pSdkPageView =
+      m_pDocument->GetSDKDoc()->GetPageView(pXFAPage, true);
   if (CPDFSDK_Annot* pAnnot = pSdkPageView->GetAnnotByXFAWidget(hWidget))
     pSdkPageView->DeleteAnnot(pAnnot);
 }
 
-int32_t CPDFXFA_Document::CountPages(CXFA_FFDoc* hDoc) {
-  if (hDoc == m_pXFADoc.get() && m_pSDKDoc)
-    return GetPageCount();
+int32_t CPDFXFA_DocEnvironment::CountPages(CXFA_FFDoc* hDoc) {
+  if (hDoc == m_pDocument->GetXFADoc() && m_pDocument->GetSDKDoc())
+    return m_pDocument->GetPageCount();
   return 0;
 }
 
-int32_t CPDFXFA_Document::GetCurrentPage(CXFA_FFDoc* hDoc) {
-  if (hDoc != m_pXFADoc.get() || !m_pSDKDoc)
+int32_t CPDFXFA_DocEnvironment::GetCurrentPage(CXFA_FFDoc* hDoc) {
+  if (hDoc != m_pDocument->GetXFADoc() || !m_pDocument->GetSDKDoc())
     return -1;
-  if (m_iDocType != DOCTYPE_DYNAMIC_XFA)
+  if (m_pDocument->GetDocType() != DOCTYPE_DYNAMIC_XFA)
     return -1;
 
-  CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
+  CPDFDoc_Environment* pEnv = m_pDocument->GetSDKDoc()->GetEnv();
   if (!pEnv)
     return -1;
 
   return pEnv->FFI_GetCurrentPageIndex(this);
 }
 
-void CPDFXFA_Document::SetCurrentPage(CXFA_FFDoc* hDoc, int32_t iCurPage) {
-  if (hDoc != m_pXFADoc.get() || !m_pSDKDoc ||
-      m_iDocType != DOCTYPE_DYNAMIC_XFA || iCurPage < 0 ||
-      iCurPage >= m_pSDKDoc->GetPageCount()) {
+void CPDFXFA_DocEnvironment::SetCurrentPage(CXFA_FFDoc* hDoc,
+                                            int32_t iCurPage) {
+  if (hDoc != m_pDocument->GetXFADoc() || !m_pDocument->GetSDKDoc() ||
+      m_pDocument->GetDocType() != DOCTYPE_DYNAMIC_XFA || iCurPage < 0 ||
+      iCurPage >= m_pDocument->GetSDKDoc()->GetPageCount()) {
     return;
   }
 
-  CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
+  CPDFDoc_Environment* pEnv = m_pDocument->GetSDKDoc()->GetEnv();
   if (!pEnv)
     return;
   pEnv->FFI_SetCurrentPage(this, iCurPage);
 }
 
-FX_BOOL CPDFXFA_Document::IsCalculationsEnabled(CXFA_FFDoc* hDoc) {
-  if (hDoc != m_pXFADoc.get() || !m_pSDKDoc)
+FX_BOOL CPDFXFA_DocEnvironment::IsCalculationsEnabled(CXFA_FFDoc* hDoc) {
+  if (hDoc != m_pDocument->GetXFADoc() || !m_pDocument->GetSDKDoc())
     return FALSE;
-  if (m_pSDKDoc->GetInterForm())
-    return m_pSDKDoc->GetInterForm()->IsXfaCalculateEnabled();
+  if (m_pDocument->GetSDKDoc()->GetInterForm())
+    return m_pDocument->GetSDKDoc()->GetInterForm()->IsXfaCalculateEnabled();
   return FALSE;
 }
 
-void CPDFXFA_Document::SetCalculationsEnabled(CXFA_FFDoc* hDoc,
-                                              FX_BOOL bEnabled) {
-  if (hDoc != m_pXFADoc.get() || !m_pSDKDoc)
+void CPDFXFA_DocEnvironment::SetCalculationsEnabled(CXFA_FFDoc* hDoc,
+                                                    FX_BOOL bEnabled) {
+  if (hDoc != m_pDocument->GetXFADoc() || !m_pDocument->GetSDKDoc())
     return;
-  if (m_pSDKDoc->GetInterForm())
-    m_pSDKDoc->GetInterForm()->XfaEnableCalculate(bEnabled);
+  if (m_pDocument->GetSDKDoc()->GetInterForm())
+    m_pDocument->GetSDKDoc()->GetInterForm()->XfaEnableCalculate(bEnabled);
 }
 
-void CPDFXFA_Document::GetTitle(CXFA_FFDoc* hDoc, CFX_WideString& wsTitle) {
-  if (hDoc != m_pXFADoc.get() || !m_pPDFDoc)
+void CPDFXFA_DocEnvironment::GetTitle(CXFA_FFDoc* hDoc,
+                                      CFX_WideString& wsTitle) {
+  if (hDoc != m_pDocument->GetXFADoc() || !m_pDocument->GetPDFDoc())
     return;
 
-  CPDF_Dictionary* pInfoDict = m_pPDFDoc->GetInfo();
+  CPDF_Dictionary* pInfoDict = m_pDocument->GetPDFDoc()->GetInfo();
   if (!pInfoDict)
     return;
 
   CFX_ByteString csTitle = pInfoDict->GetStringBy("Title");
   wsTitle = wsTitle.FromLocal(csTitle.GetBuffer(csTitle.GetLength()));
   csTitle.ReleaseBuffer(csTitle.GetLength());
 }
 
-void CPDFXFA_Document::SetTitle(CXFA_FFDoc* hDoc,
-                                const CFX_WideString& wsTitle) {
-  if (hDoc != m_pXFADoc.get() || !m_pPDFDoc)
+void CPDFXFA_DocEnvironment::SetTitle(CXFA_FFDoc* hDoc,
+                                      const CFX_WideString& wsTitle) {
+  if (hDoc != m_pDocument->GetXFADoc() || !m_pDocument->GetPDFDoc())
     return;
 
-  if (CPDF_Dictionary* pInfoDict = m_pPDFDoc->GetInfo())
+  if (CPDF_Dictionary* pInfoDict = m_pDocument->GetPDFDoc()->GetInfo())
     pInfoDict->SetAt("Title", new CPDF_String(wsTitle));
 }
 
-void CPDFXFA_Document::ExportData(CXFA_FFDoc* hDoc,
-                                  const CFX_WideString& wsFilePath,
-                                  FX_BOOL bXDP) {
-  if (hDoc != m_pXFADoc.get())
+void CPDFXFA_DocEnvironment::ExportData(CXFA_FFDoc* hDoc,
+                                        const CFX_WideString& wsFilePath,
+                                        FX_BOOL bXDP) {
+  if (hDoc != m_pDocument->GetXFADoc())
     return;
 
-  if (m_iDocType != DOCTYPE_DYNAMIC_XFA && m_iDocType != DOCTYPE_STATIC_XFA)
+  if (m_pDocument->GetDocType() != DOCTYPE_DYNAMIC_XFA &&
+      m_pDocument->GetDocType() != DOCTYPE_STATIC_XFA)
     return;
 
-  CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
+  CPDFDoc_Environment* pEnv = m_pDocument->GetSDKDoc()->GetEnv();
   if (!pEnv)
     return;
 
   int fileType = bXDP ? FXFA_SAVEAS_XDP : FXFA_SAVEAS_XML;
   CFX_ByteString bs = wsFilePath.UTF16LE_Encode();
   if (wsFilePath.IsEmpty()) {
     if (!pEnv->GetFormFillInfo() || !pEnv->GetFormFillInfo()->m_pJsPlatform)
       return;
 
     CFX_WideString filepath = pEnv->JS_fieldBrowse();
     bs = filepath.UTF16LE_Encode();
   }
   int len = bs.GetLength();
   FPDF_FILEHANDLER* pFileHandler =
       pEnv->FFI_OpenFile(bXDP ? FXFA_SAVEAS_XDP : FXFA_SAVEAS_XML,
                          (FPDF_WIDESTRING)bs.GetBuffer(len), "wb");
   bs.ReleaseBuffer(len);
   if (!pFileHandler)
     return;
 
   CFPDF_FileStream fileWrite(pFileHandler);
   CFX_ByteString content;
   if (fileType == FXFA_SAVEAS_XML) {
     content = "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\r\n";
     fileWrite.WriteBlock(content.c_str(), fileWrite.GetSize(),
                          content.GetLength());
-    m_pXFADocView->GetDoc()->SavePackage(XFA_HASHCODE_Data, &fileWrite,
-                                         nullptr);
+    m_pDocument->GetXFADocView()->GetDoc()->SavePackage(XFA_HASHCODE_Data,
+                                                        &fileWrite, nullptr);
   } else if (fileType == FXFA_SAVEAS_XDP) {
-    if (!m_pPDFDoc)
+    if (!m_pDocument->GetPDFDoc())
       return;
 
-    CPDF_Dictionary* pRoot = m_pPDFDoc->GetRoot();
+    CPDF_Dictionary* pRoot = m_pDocument->GetPDFDoc()->GetRoot();
     if (!pRoot)
       return;
 
     CPDF_Dictionary* pAcroForm = pRoot->GetDictBy("AcroForm");
     if (!pAcroForm)
       return;
 
     CPDF_Array* pArray = ToArray(pAcroForm->GetObjectBy("XFA"));
     if (!pArray)
       return;
 
     int size = pArray->GetCount();
     for (int i = 1; i < size; i += 2) {
       CPDF_Object* pPDFObj = pArray->GetObjectAt(i);
       CPDF_Object* pPrePDFObj = pArray->GetObjectAt(i - 1);
       if (!pPrePDFObj->IsString())
         continue;
       if (!pPDFObj->IsReference())
         continue;
 
       CPDF_Stream* pStream = ToStream(pPDFObj->GetDirect());
       if (!pStream)
         continue;
       if (pPrePDFObj->GetString() == "form") {
-        m_pXFADocView->GetDoc()->SavePackage(XFA_HASHCODE_Form, &fileWrite,
-                                             nullptr);
+        m_pDocument->GetXFADocView()->GetDoc()->SavePackage(
+            XFA_HASHCODE_Form, &fileWrite, nullptr);
         continue;
       }
       if (pPrePDFObj->GetString() == "datasets") {
-        m_pXFADocView->GetDoc()->SavePackage(XFA_HASHCODE_Datasets, &fileWrite,
-                                             nullptr);
+        m_pDocument->GetXFADocView()->GetDoc()->SavePackage(
+            XFA_HASHCODE_Datasets, &fileWrite, nullptr);
         continue;
       }
       if (i == size - 1) {
         CFX_WideString wPath = CFX_WideString::FromUTF16LE(
             reinterpret_cast<const unsigned short*>(bs.c_str()),
             bs.GetLength() / sizeof(unsigned short));
         CFX_ByteString bPath = wPath.UTF8Encode();
         const char* szFormat =
             "\n<pdf href=\"%s\" xmlns=\"http://ns.adobe.com/xdp/pdf/\"/>";
         content.Format(szFormat, bPath.c_str());
         fileWrite.WriteBlock(content.c_str(), fileWrite.GetSize(),
                              content.GetLength());
       }
       std::unique_ptr<CPDF_StreamAcc> pAcc(new CPDF_StreamAcc);
       pAcc->LoadAllData(pStream);
       fileWrite.WriteBlock(pAcc->GetData(), fileWrite.GetSize(),
                            pAcc->GetSize());
     }
   }
   if (!fileWrite.Flush()) {
     // Ignoring flush error.
   }
 }
 
-void CPDFXFA_Document::GotoURL(CXFA_FFDoc* hDoc,
-                               const CFX_WideString& bsURL,
-                               FX_BOOL bAppend) {
-  if (hDoc != m_pXFADoc.get())
+void CPDFXFA_DocEnvironment::GotoURL(CXFA_FFDoc* hDoc,
+                                     const CFX_WideString& bsURL,
+                                     FX_BOOL bAppend) {
+  if (hDoc != m_pDocument->GetXFADoc())
     return;
 
-  if (m_iDocType != DOCTYPE_DYNAMIC_XFA)
+  if (m_pDocument->GetDocType() != DOCTYPE_DYNAMIC_XFA)
     return;
 
-  CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
+  CPDFDoc_Environment* pEnv = m_pDocument->GetSDKDoc()->GetEnv();
   if (!pEnv)
     return;
 
   CFX_WideStringC str(bsURL.c_str());
 
   pEnv->FFI_GotoURL(this, str, bAppend);
 }
 
-FX_BOOL CPDFXFA_Document::IsValidationsEnabled(CXFA_FFDoc* hDoc) {
-  if (hDoc != m_pXFADoc.get() || !m_pSDKDoc)
+FX_BOOL CPDFXFA_DocEnvironment::IsValidationsEnabled(CXFA_FFDoc* hDoc) {
+  if (hDoc != m_pDocument->GetXFADoc() || !m_pDocument->GetSDKDoc())
     return FALSE;
-  if (m_pSDKDoc->GetInterForm())
-    return m_pSDKDoc->GetInterForm()->IsXfaValidationsEnabled();
+  if (m_pDocument->GetSDKDoc()->GetInterForm())
+    return m_pDocument->GetSDKDoc()->GetInterForm()->IsXfaValidationsEnabled();
   return TRUE;
 }
 
-void CPDFXFA_Document::SetValidationsEnabled(CXFA_FFDoc* hDoc,
-                                             FX_BOOL bEnabled) {
-  if (hDoc != m_pXFADoc.get() || !m_pSDKDoc)
+void CPDFXFA_DocEnvironment::SetValidationsEnabled(CXFA_FFDoc* hDoc,
+                                                   FX_BOOL bEnabled) {
+  if (hDoc != m_pDocument->GetXFADoc() || !m_pDocument->GetSDKDoc())
     return;
-  if (m_pSDKDoc->GetInterForm())
-    m_pSDKDoc->GetInterForm()->XfaSetValidationsEnabled(bEnabled);
+  if (m_pDocument->GetSDKDoc()->GetInterForm())
+    m_pDocument->GetSDKDoc()->GetInterForm()->XfaSetValidationsEnabled(
+        bEnabled);
 }
 
-void CPDFXFA_Document::SetFocusWidget(CXFA_FFDoc* hDoc,
-                                      CXFA_FFWidget* hWidget) {
-  if (hDoc != m_pXFADoc.get())
+void CPDFXFA_DocEnvironment::SetFocusWidget(CXFA_FFDoc* hDoc,
+                                            CXFA_FFWidget* hWidget) {
+  if (hDoc != m_pDocument->GetXFADoc())
     return;
 
   if (!hWidget) {
-    m_pSDKDoc->SetFocusAnnot(nullptr);
+    m_pDocument->GetSDKDoc()->SetFocusAnnot(nullptr);
     return;
   }
 
-  int pageViewCount = m_pSDKDoc->GetPageViewCount();
+  int pageViewCount = m_pDocument->GetSDKDoc()->GetPageViewCount();
   for (int i = 0; i < pageViewCount; i++) {
-    CPDFSDK_PageView* pPageView = m_pSDKDoc->GetPageView(i);
+    CPDFSDK_PageView* pPageView = m_pDocument->GetSDKDoc()->GetPageView(i);
     if (!pPageView)
       continue;
 
     CPDFSDK_Annot* pAnnot = pPageView->GetAnnotByXFAWidget(hWidget);
     if (pAnnot) {
-      m_pSDKDoc->SetFocusAnnot(pAnnot);
+      m_pDocument->GetSDKDoc()->SetFocusAnnot(pAnnot);
       break;
     }
   }
 }
 
-void CPDFXFA_Document::Print(CXFA_FFDoc* hDoc,
-                             int32_t nStartPage,
-                             int32_t nEndPage,
-                             uint32_t dwOptions) {
-  if (hDoc != m_pXFADoc.get())
+void CPDFXFA_DocEnvironment::Print(CXFA_FFDoc* hDoc,
+                                   int32_t nStartPage,
+                                   int32_t nEndPage,
+                                   uint32_t dwOptions) {
+  if (hDoc != m_pDocument->GetXFADoc())
     return;
 
-  CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
+  CPDFDoc_Environment* pEnv = m_pDocument->GetSDKDoc()->GetEnv();
   if (!pEnv || !pEnv->GetFormFillInfo() ||
       !pEnv->GetFormFillInfo()->m_pJsPlatform ||
       !pEnv->GetFormFillInfo()->m_pJsPlatform->Doc_print) {
     return;
   }
 
   pEnv->GetFormFillInfo()->m_pJsPlatform->Doc_print(
       pEnv->GetFormFillInfo()->m_pJsPlatform,
       dwOptions & XFA_PRINTOPT_ShowDialog, nStartPage, nEndPage,
       dwOptions & XFA_PRINTOPT_CanCancel, dwOptions & XFA_PRINTOPT_ShrinkPage,
       dwOptions & XFA_PRINTOPT_AsImage, dwOptions & XFA_PRINTOPT_ReverseOrder,
       dwOptions & XFA_PRINTOPT_PrintAnnot);
 }
 
-FX_ARGB CPDFXFA_Document::GetHighlightColor(CXFA_FFDoc* hDoc) {
-  if (hDoc != m_pXFADoc.get())
+FX_ARGB CPDFXFA_DocEnvironment::GetHighlightColor(CXFA_FFDoc* hDoc) {
+  if (hDoc != m_pDocument->GetXFADoc())
     return 0;
-  if (!m_pSDKDoc)
+  if (!m_pDocument->GetSDKDoc())
     return 0;
 
-  CPDFSDK_InterForm* pInterForm = m_pSDKDoc->GetInterForm();
+  CPDFSDK_InterForm* pInterForm = m_pDocument->GetSDKDoc()->GetInterForm();
   if (!pInterForm)
     return 0;
 
   FX_COLORREF color = pInterForm->GetHighlightColor(FPDF_FORMFIELD_XFA);
   uint8_t alpha = pInterForm->GetHighlightAlpha();
   FX_ARGB argb = ArgbEncode((int)alpha, color);
   return argb;
 }
 
-FX_BOOL CPDFXFA_Document::NotifySubmit(FX_BOOL bPrevOrPost) {
+FX_BOOL CPDFXFA_DocEnvironment::NotifySubmit(FX_BOOL bPrevOrPost) {
   if (bPrevOrPost)
     return OnBeforeNotifySubmit();
 
   OnAfterNotifySubmit();
   return TRUE;
 }
 
-FX_BOOL CPDFXFA_Document::OnBeforeNotifySubmit() {
-  if (m_iDocType != DOCTYPE_DYNAMIC_XFA && m_iDocType != DOCTYPE_STATIC_XFA)
+FX_BOOL CPDFXFA_DocEnvironment::OnBeforeNotifySubmit() {
+  if (m_pDocument->GetDocType() != DOCTYPE_DYNAMIC_XFA &&
+      m_pDocument->GetDocType() != DOCTYPE_STATIC_XFA)
     return TRUE;
 
-  if (!m_pXFADocView)
+  if (!m_pDocument->GetXFADocView())
     return TRUE;
 
-  CXFA_FFWidgetHandler* pWidgetHandler = m_pXFADocView->GetWidgetHandler();
+  CXFA_FFWidgetHandler* pWidgetHandler =
+      m_pDocument->GetXFADocView()->GetWidgetHandler();
   if (!pWidgetHandler)
     return TRUE;
 
   std::unique_ptr<CXFA_WidgetAccIterator> pWidgetAccIterator(
-      m_pXFADocView->CreateWidgetAccIterator());
+      m_pDocument->GetXFADocView()->CreateWidgetAccIterator());
   if (pWidgetAccIterator) {
     CXFA_EventParam Param;
     Param.m_eType = XFA_EVENT_PreSubmit;
     while (CXFA_WidgetAcc* pWidgetAcc = pWidgetAccIterator->MoveToNext())
       pWidgetHandler->ProcessEvent(pWidgetAcc, &Param);
   }
 
-  pWidgetAccIterator.reset(m_pXFADocView->CreateWidgetAccIterator());
+  pWidgetAccIterator.reset(
+      m_pDocument->GetXFADocView()->CreateWidgetAccIterator());
   if (!pWidgetAccIterator)
     return TRUE;
 
   CXFA_WidgetAcc* pWidgetAcc = pWidgetAccIterator->MoveToNext();
   pWidgetAcc = pWidgetAccIterator->MoveToNext();
   while (pWidgetAcc) {
     int fRet = pWidgetAcc->ProcessValidate(-1);
     if (fRet == XFA_EVENTERROR_Error) {
-      CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
+      CPDFDoc_Environment* pEnv = m_pDocument->GetSDKDoc()->GetEnv();
       if (!pEnv)
         return FALSE;
 
       CFX_WideString ws;
       ws.FromLocal(IDS_XFA_Validate_Input);
       CFX_ByteString bs = ws.UTF16LE_Encode();
       int len = bs.GetLength();
       pEnv->FFI_Alert((FPDF_WIDESTRING)bs.GetBuffer(len), (FPDF_WIDESTRING)L"",
                       0, 1);
       bs.ReleaseBuffer(len);
       return FALSE;
     }
     pWidgetAcc = pWidgetAccIterator->MoveToNext();
   }
-  m_pXFADocView->UpdateDocView();
+  m_pDocument->GetXFADocView()->UpdateDocView();
 
   return TRUE;
 }
 
-void CPDFXFA_Document::OnAfterNotifySubmit() {
-  if (m_iDocType != DOCTYPE_DYNAMIC_XFA && m_iDocType != DOCTYPE_STATIC_XFA)
+void CPDFXFA_DocEnvironment::OnAfterNotifySubmit() {
+  if (m_pDocument->GetDocType() != DOCTYPE_DYNAMIC_XFA &&
+      m_pDocument->GetDocType() != DOCTYPE_STATIC_XFA)
     return;
 
-  if (!m_pXFADocView)
+  if (!m_pDocument->GetXFADocView())
     return;
 
-  CXFA_FFWidgetHandler* pWidgetHandler = m_pXFADocView->GetWidgetHandler();
+  CXFA_FFWidgetHandler* pWidgetHandler =
+      m_pDocument->GetXFADocView()->GetWidgetHandler();
   if (!pWidgetHandler)
     return;
 
   std::unique_ptr<CXFA_WidgetAccIterator> pWidgetAccIterator(
-      m_pXFADocView->CreateWidgetAccIterator());
+      m_pDocument->GetXFADocView()->CreateWidgetAccIterator());
   if (!pWidgetAccIterator)
     return;
 
   CXFA_EventParam Param;
   Param.m_eType = XFA_EVENT_PostSubmit;
   CXFA_WidgetAcc* pWidgetAcc = pWidgetAccIterator->MoveToNext();
   while (pWidgetAcc) {
     pWidgetHandler->ProcessEvent(pWidgetAcc, &Param);
     pWidgetAcc = pWidgetAccIterator->MoveToNext();
   }
-  m_pXFADocView->UpdateDocView();
+  m_pDocument->GetXFADocView()->UpdateDocView();
 }
 
-FX_BOOL CPDFXFA_Document::SubmitData(CXFA_FFDoc* hDoc, CXFA_Submit submit) {
-  if (!NotifySubmit(TRUE) || !m_pXFADocView)
+FX_BOOL CPDFXFA_DocEnvironment::SubmitData(CXFA_FFDoc* hDoc,
+                                           CXFA_Submit submit) {
+  if (!NotifySubmit(TRUE) || !m_pDocument->GetXFADocView())
     return FALSE;
 
-  m_pXFADocView->UpdateDocView();
+  m_pDocument->GetXFADocView()->UpdateDocView();
   FX_BOOL ret = SubmitDataInternal(hDoc, submit);
   NotifySubmit(FALSE);
   return ret;
 }
 
-IFX_FileRead* CPDFXFA_Document::OpenLinkedFile(CXFA_FFDoc* hDoc,
-                                               const CFX_WideString& wsLink) {
-  CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
+IFX_FileRead* CPDFXFA_DocEnvironment::OpenLinkedFile(
+    CXFA_FFDoc* hDoc,
+    const CFX_WideString& wsLink) {
+  CPDFDoc_Environment* pEnv = m_pDocument->GetSDKDoc()->GetEnv();
   if (!pEnv)
     return FALSE;
 
   CFX_ByteString bs = wsLink.UTF16LE_Encode();
   int len = bs.GetLength();
   FPDF_FILEHANDLER* pFileHandler =
       pEnv->FFI_OpenFile(0, (FPDF_WIDESTRING)bs.GetBuffer(len), "rb");
   bs.ReleaseBuffer(len);
 
   if (!pFileHandler)
     return nullptr;
   return new CFPDF_FileStream(pFileHandler);
 }
 
-FX_BOOL CPDFXFA_Document::ExportSubmitFile(FPDF_FILEHANDLER* pFileHandler,
-                                           int fileType,
-                                           FPDF_DWORD encodeType,
-                                           FPDF_DWORD flag) {
-  if (!m_pXFADocView)
+FX_BOOL CPDFXFA_DocEnvironment::ExportSubmitFile(FPDF_FILEHANDLER* pFileHandler,
+                                                 int fileType,
+                                                 FPDF_DWORD encodeType,
+                                                 FPDF_DWORD flag) {
+  if (!m_pDocument->GetXFADocView())
     return FALSE;
 
   CFX_ByteString content;
-  CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
+  CPDFDoc_Environment* pEnv = m_pDocument->GetSDKDoc()->GetEnv();
   if (!pEnv)
     return FALSE;
 
   CFPDF_FileStream fileStream(pFileHandler);
   if (fileType == FXFA_SAVEAS_XML) {
     const char kContent[] = "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\r\n";
     fileStream.WriteBlock(kContent, 0, strlen(kContent));
-    m_pXFADoc->SavePackage(XFA_HASHCODE_Data, &fileStream, nullptr);
+    m_pDocument->GetXFADoc()->SavePackage(XFA_HASHCODE_Data, &fileStream,
+                                          nullptr);
     return TRUE;
   }
 
   if (fileType != FXFA_SAVEAS_XDP)
     return TRUE;
 
   if (!flag) {
     flag = FXFA_CONFIG | FXFA_TEMPLATE | FXFA_LOCALESET | FXFA_DATASETS |
            FXFA_XMPMETA | FXFA_XFDF | FXFA_FORM;
   }
-  if (!m_pPDFDoc) {
+  if (!m_pDocument->GetPDFDoc()) {
     fileStream.Flush();
     return FALSE;
   }
 
-  CPDF_Dictionary* pRoot = m_pPDFDoc->GetRoot();
+  CPDF_Dictionary* pRoot = m_pDocument->GetPDFDoc()->GetRoot();
   if (!pRoot) {
     fileStream.Flush();
     return FALSE;
   }
 
   CPDF_Dictionary* pAcroForm = pRoot->GetDictBy("AcroForm");
   if (!pAcroForm) {
     fileStream.Flush();
     return FALSE;
   }
@@ skipping 24 matching lines @@
       continue;
     if (pPrePDFObj->GetString() == "datasets" && !(flag & FXFA_DATASETS))
       continue;
     if (pPrePDFObj->GetString() == "xmpmeta" && !(flag & FXFA_XMPMETA))
       continue;
     if (pPrePDFObj->GetString() == "xfdf" && !(flag & FXFA_XFDF))
       continue;
     if (pPrePDFObj->GetString() == "form" && !(flag & FXFA_FORM))
       continue;
     if (pPrePDFObj->GetString() == "form") {
-      m_pXFADoc->SavePackage(XFA_HASHCODE_Form, &fileStream, nullptr);
+      m_pDocument->GetXFADoc()->SavePackage(XFA_HASHCODE_Form, &fileStream,
+                                            nullptr);
     } else if (pPrePDFObj->GetString() == "datasets") {
-      m_pXFADoc->SavePackage(XFA_HASHCODE_Datasets, &fileStream, nullptr);
+      m_pDocument->GetXFADoc()->SavePackage(XFA_HASHCODE_Datasets, &fileStream,
+                                            nullptr);
     } else {
       // PDF,creator.
     }
   }
   return TRUE;
 }
 
-void CPDFXFA_Document::ClearChangeMark() {
-  if (m_pSDKDoc)
-    m_pSDKDoc->ClearChangeMark();
-}
-
-void CPDFXFA_Document::ToXFAContentFlags(CFX_WideString csSrcContent,
-                                         FPDF_DWORD& flag) {
+void CPDFXFA_DocEnvironment::ToXFAContentFlags(CFX_WideString csSrcContent,
+                                               FPDF_DWORD& flag) {
   if (csSrcContent.Find(L" config ", 0) != -1)
     flag |= FXFA_CONFIG;
   if (csSrcContent.Find(L" template ", 0) != -1)
     flag |= FXFA_TEMPLATE;
   if (csSrcContent.Find(L" localeSet ", 0) != -1)
     flag |= FXFA_LOCALESET;
   if (csSrcContent.Find(L" datasets ", 0) != -1)
     flag |= FXFA_DATASETS;
   if (csSrcContent.Find(L" xmpmeta ", 0) != -1)
     flag |= FXFA_XMPMETA;
   if (csSrcContent.Find(L" xfdf ", 0) != -1)
     flag |= FXFA_XFDF;
   if (csSrcContent.Find(L" form ", 0) != -1)
     flag |= FXFA_FORM;
   if (flag == 0) {
     flag = FXFA_CONFIG | FXFA_TEMPLATE | FXFA_LOCALESET | FXFA_DATASETS |
            FXFA_XMPMETA | FXFA_XFDF | FXFA_FORM;
   }
 }
 
-FX_BOOL CPDFXFA_Document::MailToInfo(CFX_WideString& csURL,
-                                     CFX_WideString& csToAddress,
-                                     CFX_WideString& csCCAddress,
-                                     CFX_WideString& csBCCAddress,
-                                     CFX_WideString& csSubject,
-                                     CFX_WideString& csMsg) {
+FX_BOOL CPDFXFA_DocEnvironment::MailToInfo(CFX_WideString& csURL,
+                                           CFX_WideString& csToAddress,
+                                           CFX_WideString& csCCAddress,
+                                           CFX_WideString& csBCCAddress,
+                                           CFX_WideString& csSubject,
+                                           CFX_WideString& csMsg) {
   CFX_WideString srcURL = csURL;
   srcURL.TrimLeft();
   if (srcURL.Left(7).CompareNoCase(L"mailto:") != 0)
     return FALSE;
 
   int pos = srcURL.Find(L'?', 0);
   CFX_WideString tmp;
   if (pos == -1) {
     pos = srcURL.Find(L'@', 0);
     if (pos == -1)
@@ skipping 39 matching lines @@
       csMsg += tmp;
     }
     srcURL = (pos == -1) ? L"" : srcURL.Right(csURL.GetLength() - (pos + 1));
   }
   csToAddress.Replace(L",", L";");
   csCCAddress.Replace(L",", L";");
   csBCCAddress.Replace(L",", L";");
   return TRUE;
 }
 
-FX_BOOL CPDFXFA_Document::SubmitDataInternal(CXFA_FFDoc* hDoc,
-                                             CXFA_Submit submit) {
-  CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
+FX_BOOL CPDFXFA_DocEnvironment::SubmitDataInternal(CXFA_FFDoc* hDoc,
+                                                   CXFA_Submit submit) {
+  CPDFDoc_Environment* pEnv = m_pDocument->GetSDKDoc()->GetEnv();
   if (!pEnv)
     return FALSE;
 
   CFX_WideStringC csURLC;
   submit.GetSubmitTarget(csURLC);
   CFX_WideString csURL(csURLC);
   if (csURL.IsEmpty()) {
     CFX_WideString ws;
     ws.FromLocal("Submit cancelled.");
     CFX_ByteString bs = ws.UTF16LE_Encode();
@@ skipping 78 matching lines @@
     CFX_WideString ws;
     CFX_ByteString bs = csURL.UTF16LE_Encode();
     int len = bs.GetLength();
     pEnv->FFI_UploadTo(pFileHandler, fileFlag,
                        (FPDF_WIDESTRING)bs.GetBuffer(len));
     bs.ReleaseBuffer(len);
   }
   return bRet;
 }
 
-FX_BOOL CPDFXFA_Document::SetGlobalProperty(CXFA_FFDoc* hDoc,
-                                            const CFX_ByteStringC& szPropName,
-                                            CFXJSE_Value* pValue) {
-  if (hDoc != m_pXFADoc.get())
+FX_BOOL CPDFXFA_DocEnvironment::SetGlobalProperty(
+    CXFA_FFDoc* hDoc,
+    const CFX_ByteStringC& szPropName,
+    CFXJSE_Value* pValue) {
+  if (hDoc != m_pDocument->GetXFADoc())
     return FALSE;
 
-  if (m_pSDKDoc && m_pSDKDoc->GetEnv()->GetJSRuntime())
-    return m_pSDKDoc->GetEnv()->GetJSRuntime()->SetValueByName(szPropName,
-                                                               pValue);
+  if (m_pDocument->GetSDKDoc() &&
+      m_pDocument->GetSDKDoc()->GetEnv()->GetJSRuntime())
+    return m_pDocument->GetSDKDoc()->GetEnv()->GetJSRuntime()->SetValueByName(
+        szPropName, pValue);
   return FALSE;
 }
 
-FX_BOOL CPDFXFA_Document::GetGlobalProperty(CXFA_FFDoc* hDoc,
-                                            const CFX_ByteStringC& szPropName,
-                                            CFXJSE_Value* pValue) {
-  if (hDoc != m_pXFADoc.get())
+FX_BOOL CPDFXFA_DocEnvironment::GetGlobalProperty(
+    CXFA_FFDoc* hDoc,
+    const CFX_ByteStringC& szPropName,
+    CFXJSE_Value* pValue) {
+  if (hDoc != m_pDocument->GetXFADoc())
     return FALSE;
-  if (!m_pSDKDoc || !m_pSDKDoc->GetEnv()->GetJSRuntime())
+  if (!m_pDocument->GetSDKDoc() ||
+      !m_pDocument->GetSDKDoc()->GetEnv()->GetJSRuntime())
     return FALSE;
 
   if (!m_pJSContext) {
-    m_pSDKDoc->GetEnv()->GetJSRuntime()->SetReaderDocument(m_pSDKDoc.get());
-    m_pJSContext = m_pSDKDoc->GetEnv()->GetJSRuntime()->NewContext();
+    m_pDocument->GetSDKDoc()->GetEnv()->GetJSRuntime()->SetReaderDocument(
+        m_pDocument->GetSDKDoc());
+    m_pJSContext =
+        m_pDocument->GetSDKDoc()->GetEnv()->GetJSRuntime()->NewContext();
   }
 
-  return m_pSDKDoc->GetEnv()->GetJSRuntime()->GetValueByName(szPropName,
-                                                             pValue);
+  return m_pDocument->GetSDKDoc()->GetEnv()->GetJSRuntime()->GetValueByName(
+      szPropName, pValue);
 }