View Javadoc
1   /**
2    * This file Copyright (c) 2017-2018 Magnolia International
3    * Ltd.  (http://www.magnolia-cms.com). All rights reserved.
4    *
5    *
6    * This file is dual-licensed under both the Magnolia
7    * Network Agreement and the GNU General Public License.
8    * You may elect to use one or the other of these licenses.
9    *
10   * This file is distributed in the hope that it will be
11   * useful, but AS-IS and WITHOUT ANY WARRANTY; without even the
12   * implied warranty of MERCHANTABILITY or FITNESS FOR A
13   * PARTICULAR PURPOSE, TITLE, or NONINFRINGEMENT.
14   * Redistribution, except as permitted by whichever of the GPL
15   * or MNA you select, is prohibited.
16   *
17   * 1. For the GPL license (GPL), you can redistribute and/or
18   * modify this file under the terms of the GNU General
19   * Public License, Version 3, as published by the Free Software
20   * Foundation.  You should have received a copy of the GNU
21   * General Public License, Version 3 along with this program;
22   * if not, write to the Free Software Foundation, Inc., 51
23   * Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
24   *
25   * 2. For the Magnolia Network Agreement (MNA), this file
26   * and the accompanying materials are made available under the
27   * terms of the MNA which accompanies this distribution, and
28   * is available at http://www.magnolia-cms.com/mna.html
29   *
30   * Any modifications to this file must keep this entire header
31   * intact.
32   *
33   */
34  package info.magnolia.importexport.command;
35  
36  import info.magnolia.commands.impl.BaseRepositoryCommand;
37  import info.magnolia.context.Context;
38  import info.magnolia.importexport.DataTransporter;
39  import info.magnolia.module.delta.BootstrapFileUtil;
40  
41  import java.io.InputStream;
42  
43  import javax.jcr.ImportUUIDBehavior;
44  import javax.jcr.Node;
45  
46  import org.apache.commons.io.FilenameUtils;
47  import org.apache.commons.io.IOUtils;
48  import org.apache.commons.lang3.StringUtils;
49  
50  /**
51   * Import command with YAML format support.
52   */
53  public class JcrImportCommand extends BaseRepositoryCommand {
54  
55      private boolean saveAfterImport = true;
56      private int identifierBehavior = ImportUUIDBehavior.IMPORT_UUID_CREATE_NEW;
57      private String fileName;
58      private String importPath;
59      private InputStream stream;
60  
61      @Override
62      public boolean execute(Context context) throws Exception {
63          try {
64              final Node node = context.getJCRSession(getRepository()).getNode(getPath()); //don't get the session from getJCRNode(context), that's a system context with superuser access!
65              final String extension = FilenameUtils.getExtension(getFileName());
66  
67              if (StringUtils.equalsIgnoreCase(JcrExportCommand.Format.YAML.name(), extension)) {
68                  DataTransporter.importYamlStream(getStream(), node, node.getSession().getWorkspace().getName(), node.getPath(), getImportPath(), getFileName(), false, isSaveAfterImport(), true);
69              } else {
70                  final InputStream stream = importPath == null ? getStream() : BootstrapFileUtil.getElementAsStream(fileName, importPath);
71                  DataTransporter.importXmlStream(stream, node.getSession().getWorkspace().getName(), node.getPath(), getFileName(), false, false, getIdentifierBehavior(), isSaveAfterImport(), true);
72              }
73          } finally {
74              if (stream != null) {
75                  IOUtils.closeQuietly(stream);
76              }
77          }
78          return true;
79      }
80  
81      public boolean isSaveAfterImport() {
82          return saveAfterImport;
83      }
84  
85      public void setSaveAfterImport(boolean saveAfterImport) {
86          this.saveAfterImport = saveAfterImport;
87      }
88  
89      public int getIdentifierBehavior() {
90          return identifierBehavior;
91      }
92  
93      public void setIdentifierBehavior(int identifierBehavior) {
94          this.identifierBehavior = identifierBehavior;
95      }
96  
97      public String getFileName() {
98          return fileName;
99      }
100 
101     public void setFileName(String fileName) {
102         this.fileName = fileName;
103     }
104 
105     public String getImportPath() {
106         return importPath;
107     }
108 
109     public void setImportPath(String subPathToImport) {
110         this.importPath = subPathToImport;
111     }
112 
113     public InputStream getStream() {
114         return stream;
115     }
116 
117     public void setStream(InputStream stream) {
118         this.stream = stream;
119     }
120 
121 
122 
123 }