1   /**
2    * Copyright (c) 2000-2009 Liferay, Inc. All rights reserved.
3    *
4    * Permission is hereby granted, free of charge, to any person obtaining a copy
5    * of this software and associated documentation files (the "Software"), to deal
6    * in the Software without restriction, including without limitation the rights
7    * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
8    * copies of the Software, and to permit persons to whom the Software is
9    * furnished to do so, subject to the following conditions:
10   *
11   * The above copyright notice and this permission notice shall be included in
12   * all copies or substantial portions of the Software.
13   *
14   * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15   * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16   * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
17   * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
18   * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
19   * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
20   * SOFTWARE.
21   */
22  
23  package com.liferay.documentlibrary.util;
24  
25  import com.liferay.portal.kernel.util.FileUtil;
26  import com.liferay.portal.kernel.util.StringPool;
27  import com.liferay.portal.kernel.util.StringUtil;
28  
29  import java.io.File;
30  
31  /**
32   * <a href="AdvancedFileSystemHook.java.html"><b><i>View Source</i></b></a>
33   *
34   * @author Jorge Ferrer
35   *
36   */
37  public class AdvancedFileSystemHook extends FileSystemHook {
38  
39      protected void buildPath(StringBuilder sb, String fileNameFragment) {
40          if (fileNameFragment.length() <= 2) {
41              return;
42          }
43  
44          if (getDepth(sb.toString()) > 3) {
45              return;
46          }
47  
48          sb.append(fileNameFragment.substring(0, 2) + StringPool.SLASH);
49  
50          buildPath(sb, fileNameFragment.substring(2));
51      }
52  
53      protected int getDepth(String path) {
54          String[] fragments = StringUtil.split(path, StringPool.SLASH);
55  
56          return fragments.length;
57      }
58  
59      protected File getDirNameDir(
60          long companyId, long repositoryId, String dirName) {
61  
62          File repositoryDir = getRepositoryDir(companyId, repositoryId);
63  
64          return new File(repositoryDir + StringPool.SLASH + dirName);
65      }
66  
67      protected File getFileNameVersionFile(
68          long companyId, long repositoryId, String fileName, double version) {
69  
70          String ext = StringPool.PERIOD + FileUtil.getExtension(fileName);
71  
72          int pos = fileName.lastIndexOf(StringPool.SLASH);
73  
74          if (pos == -1) {
75              StringBuilder sb = new StringBuilder();
76  
77              String fileNameFragment = removeExtension(fileName);
78  
79              if (fileNameFragment.startsWith("DLFE-")) {
80                  fileNameFragment = fileNameFragment.substring(5);
81  
82                  sb.append("DLFE" + StringPool.SLASH);
83              }
84  
85              buildPath(sb, fileNameFragment);
86  
87              File repositoryDir = getRepositoryDir(companyId, repositoryId);
88  
89              return new File(
90                  repositoryDir + StringPool.SLASH + sb.toString() +
91                      StringPool.SLASH + fileNameFragment + ext +
92                          StringPool.SLASH + fileNameFragment +
93                              StringPool.UNDERLINE + version + ext);
94          }
95          else {
96              File fileNameDir = getDirNameDir(companyId, repositoryId, fileName);
97  
98              String fileNameFragment = removeExtension(
99                  fileName.substring(pos + 1));
100 
101             return new File(
102                 fileNameDir + StringPool.SLASH + fileNameFragment +
103                     StringPool.UNDERLINE + version + ext);
104         }
105     }
106 
107     protected String removeExtension(String fileName) {
108         String ext = FileUtil.getExtension(fileName);
109 
110         if (ext != null) {
111             fileName = fileName.substring(
112                 0, fileName.length() - ext.length() - 1);
113         }
114 
115         return fileName;
116     }
117 
118 }