View Javadoc

1   /**
2    * This file Copyright (c) 2003-2014 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.module.files;
35  
36  import info.magnolia.cms.util.ClasspathResourcesUtil;
37  import org.apache.commons.io.IOUtils;
38  
39  import java.io.File;
40  import java.io.FileOutputStream;
41  import java.io.IOException;
42  import java.io.InputStream;
43  import java.io.OutputStream;
44  
45  /**
46   * A simple FileExtractorOperation which just extracts a resource from the class path
47   * to the filesystem.
48   */
49  public class BasicFileExtractorOperation implements FileExtractorOperation {
50      protected final String resourcePath;
51      protected final String absoluteTargetPath;
52  
53      BasicFileExtractorOperation(String resourcePath, String absoluteTargetPath) {
54          this.resourcePath = resourcePath;
55          this.absoluteTargetPath = absoluteTargetPath;
56      }
57  
58      @Override
59      public void extract() throws IOException {
60          final InputStream resourceStream = checkInput();
61          final File checkedTargetFile = checkOutput();
62          if (checkedTargetFile != null) {
63              final OutputStream out = openOutput(checkedTargetFile);
64  
65              copyAndClose(resourceStream, out);
66          }
67      }
68  
69      protected InputStream checkInput() throws IOException {
70          final InputStream resourceStream = ClasspathResourcesUtil.getStream(resourcePath, false);
71          if (resourceStream == null) {
72              throw new IOException("Can't open " + resourcePath);
73          }
74          return resourceStream;
75      }
76  
77      /**
78       * @return a java.io.File corresponding to the absoluteTargetPath if the path is validated,
79       *         or null if the file should not be extracted. TODO ugly.
80       */
81      protected File checkOutput() throws IOException {
82          final File targetFile = new File(absoluteTargetPath);
83          final File parent = targetFile.getParentFile();
84          if (!parent.exists() && !parent.mkdirs()) {
85              throw new IOException("Can't create directories for " + targetFile.getAbsolutePath());
86          } else if (!parent.canWrite()) {
87              throw new IOException("Can't write to " + targetFile.getAbsolutePath());
88          }
89          return targetFile;
90      }
91  
92      protected OutputStream openOutput(File targetFile) throws IOException {
93          return new FileOutputStream(targetFile);
94      }
95  
96      protected void copyAndClose(InputStream in, OutputStream out) throws IOException {
97          try {
98              IOUtils.copy(in, out);
99          } finally {
100             IOUtils.closeQuietly(in);
101             IOUtils.closeQuietly(out);
102         }
103     }
104 }