1// Copyright 2016 PDFium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5// Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
6
7#include "xfa/fxfa/parser/cxfa_dataimporter.h"
8
9#include <memory>
10
11#include "core/fxcrt/fx_stream.h"
12#include "core/fxcrt/xml/cfx_xmlnode.h"
13#include "third_party/base/ptr_util.h"
14#include "xfa/fxfa/fxfa.h"
15#include "xfa/fxfa/fxfa_basic.h"
16#include "xfa/fxfa/parser/cxfa_document.h"
17#include "xfa/fxfa/parser/cxfa_node.h"
18#include "xfa/fxfa/parser/cxfa_simple_parser.h"
19
20CXFA_DataImporter::CXFA_DataImporter(CXFA_Document* pDocument)
21    : m_pDocument(pDocument) {
22  ASSERT(m_pDocument);
23}
24
25CXFA_DataImporter::~CXFA_DataImporter() {}
26
27bool CXFA_DataImporter::ImportData(
28    const RetainPtr<IFX_SeekableStream>& pDataDocument) {
29  auto pDataDocumentParser =
30      pdfium::MakeUnique<CXFA_SimpleParser>(m_pDocument.Get());
31  if (pDataDocumentParser->StartParse(
32          pDataDocument, XFA_PacketType::Datasets) != XFA_PARSESTATUS_Ready) {
33    return false;
34  }
35  if (pDataDocumentParser->DoParse() < XFA_PARSESTATUS_Done)
36    return false;
37
38  CXFA_Node* pImportDataRoot = pDataDocumentParser->GetRootNode();
39  if (!pImportDataRoot)
40    return false;
41
42  CXFA_Node* pDataModel =
43      ToNode(m_pDocument->GetXFAObject(XFA_HASHCODE_Datasets));
44  if (!pDataModel)
45    return false;
46
47  CXFA_Node* pDataNode = ToNode(m_pDocument->GetXFAObject(XFA_HASHCODE_Data));
48  if (pDataNode)
49    pDataModel->RemoveChild(pDataNode, true);
50
51  if (pImportDataRoot->GetElementType() == XFA_Element::DataModel) {
52    while (CXFA_Node* pChildNode = pImportDataRoot->GetFirstChild()) {
53      pImportDataRoot->RemoveChild(pChildNode, true);
54      pDataModel->InsertChild(pChildNode, nullptr);
55    }
56  } else {
57    CFX_XMLNode* pXMLNode = pImportDataRoot->GetXMLMappingNode();
58    CFX_XMLNode* pParentXMLNode = pXMLNode->GetNodeItem(CFX_XMLNode::Parent);
59    if (pParentXMLNode)
60      pParentXMLNode->RemoveChildNode(pXMLNode);
61    pDataModel->InsertChild(pImportDataRoot, nullptr);
62  }
63  m_pDocument->DoDataRemerge(false);
64  return true;
65}
66