1 | |
package org.kuali.maven.mojo.s3; |
2 | |
|
3 | |
import java.text.NumberFormat; |
4 | |
import java.util.ArrayList; |
5 | |
import java.util.Collections; |
6 | |
import java.util.Comparator; |
7 | |
import java.util.List; |
8 | |
|
9 | |
import org.apache.commons.lang.StringUtils; |
10 | |
|
11 | |
import com.amazonaws.services.s3.model.ObjectListing; |
12 | |
import com.amazonaws.services.s3.model.S3ObjectSummary; |
13 | |
|
14 | |
|
15 | |
|
16 | |
|
17 | |
public class S3DataConverter { |
18 | 0 | HtmlUtils html = new HtmlUtils(); |
19 | 0 | NumberFormat nf = getNumberFormatInstance(); |
20 | |
S3BucketContext context; |
21 | |
String browseHtml; |
22 | |
|
23 | |
public S3DataConverter() { |
24 | 0 | this(null); |
25 | 0 | } |
26 | |
|
27 | |
public S3DataConverter(final S3BucketContext context) { |
28 | 0 | super(); |
29 | 0 | this.context = context; |
30 | 0 | } |
31 | |
|
32 | |
|
33 | |
|
34 | |
|
35 | |
|
36 | |
protected NumberFormat getNumberFormatInstance() { |
37 | 0 | NumberFormat nf = NumberFormat.getInstance(); |
38 | 0 | nf.setMaximumFractionDigits(1); |
39 | 0 | nf.setMinimumFractionDigits(1); |
40 | 0 | nf.setGroupingUsed(false); |
41 | 0 | return nf; |
42 | |
} |
43 | |
|
44 | |
|
45 | |
|
46 | |
|
47 | |
|
48 | |
protected String getTrimmedPrefix(final String prefix, final String delimiter) { |
49 | 0 | int pos = prefix.lastIndexOf(delimiter); |
50 | 0 | if (pos == -1) { |
51 | 0 | return prefix; |
52 | |
} |
53 | 0 | return prefix.substring(0, pos); |
54 | |
} |
55 | |
|
56 | |
|
57 | |
|
58 | |
|
59 | |
protected void addDisplayRows(final List<DisplayRow> displayRows, final List<String[]> data) { |
60 | 0 | for (DisplayRow displayRow : displayRows) { |
61 | 0 | addDisplayRow(displayRow, data); |
62 | |
} |
63 | 0 | } |
64 | |
|
65 | |
|
66 | |
|
67 | |
|
68 | |
protected void addDisplayRow(final DisplayRow displayRow, final List<String[]> data) { |
69 | 0 | if (displayRow == null) { |
70 | 0 | return; |
71 | |
} |
72 | 0 | String[] row = new String[4]; |
73 | 0 | row[0] = displayRow.getImage(); |
74 | 0 | row[1] = displayRow.getAhref(); |
75 | 0 | row[2] = displayRow.getLastModified(); |
76 | 0 | row[3] = displayRow.getSize(); |
77 | 0 | data.add(row); |
78 | 0 | } |
79 | |
|
80 | |
|
81 | |
|
82 | |
|
83 | |
|
84 | |
protected String getShow(final String key, final String prefix) { |
85 | 0 | if (prefix == null) { |
86 | 0 | return key; |
87 | |
} |
88 | 0 | int index = prefix.length(); |
89 | 0 | String s = key.substring(index); |
90 | 0 | return s; |
91 | |
} |
92 | |
|
93 | |
|
94 | |
|
95 | |
|
96 | |
protected DisplayRow getDisplayRow(final String commonPrefix, final String prefix, final String delimiter) { |
97 | |
|
98 | |
|
99 | 0 | String image = html.getImage(context.getDirectoryImage()); |
100 | 0 | String show = getShow(commonPrefix, prefix); |
101 | 0 | String dest = delimiter + commonPrefix; |
102 | 0 | String ahref = html.getHref(dest, show); |
103 | 0 | String date = "-"; |
104 | 0 | String size = "-"; |
105 | |
|
106 | |
|
107 | 0 | DisplayRow displayRow = new DisplayRow(); |
108 | 0 | displayRow.setImage(image); |
109 | 0 | displayRow.setAhref(ahref); |
110 | 0 | displayRow.setLastModified(date); |
111 | 0 | displayRow.setSize(size); |
112 | 0 | displayRow.setShow(show); |
113 | 0 | return displayRow; |
114 | |
} |
115 | |
|
116 | |
protected List<DisplayRow> getDirectoryDisplayRows(final ObjectListing objectListing, final String prefix, |
117 | |
final String delimiter) { |
118 | 0 | List<DisplayRow> displayRows = new ArrayList<DisplayRow>(); |
119 | 0 | for (String commonPrefix : objectListing.getCommonPrefixes()) { |
120 | 0 | DisplayRow displayRow = getDisplayRow(commonPrefix, prefix, delimiter); |
121 | 0 | if (displayRow == null) { |
122 | 0 | continue; |
123 | |
} |
124 | 0 | displayRows.add(displayRow); |
125 | 0 | } |
126 | 0 | return displayRows; |
127 | |
} |
128 | |
|
129 | |
|
130 | |
|
131 | |
|
132 | |
|
133 | |
public List<String[]> getData(final ObjectListing objectListing, final String prefix, final String delimiter) { |
134 | 0 | DisplayRow upOneDirectory = getUpOneDirectoryDisplayRow(prefix, delimiter); |
135 | 0 | List<DisplayRow> objectDisplayRows = getObjectDisplayRows(objectListing, prefix, delimiter); |
136 | 0 | List<DisplayRow> directoryDisplayRows = getDirectoryDisplayRows(objectListing, prefix, delimiter); |
137 | 0 | for (DisplayRow df : directoryDisplayRows) { |
138 | 0 | System.out.println("before=" + df.getShow()); |
139 | |
} |
140 | 0 | Comparator<DisplayRow> c = new DisplayRowComparator(); |
141 | 0 | Collections.sort(directoryDisplayRows, c); |
142 | 0 | for (DisplayRow df : directoryDisplayRows) { |
143 | 0 | System.out.println("after=" + df.getShow()); |
144 | |
} |
145 | 0 | List<String[]> data = new ArrayList<String[]>(); |
146 | 0 | addDisplayRow(upOneDirectory, data); |
147 | 0 | addDisplayRows(directoryDisplayRows, data); |
148 | 0 | addDisplayRows(objectDisplayRows, data); |
149 | 0 | return data; |
150 | |
} |
151 | |
|
152 | |
protected boolean isDirectory(final S3ObjectSummary summary, final List<String> commonPrefixes, |
153 | |
final String prefix, final String delimiter) { |
154 | 0 | String key = summary.getKey(); |
155 | 0 | if (key.equals(prefix)) { |
156 | 0 | return true; |
157 | |
} |
158 | 0 | for (String commonPrefix : commonPrefixes) { |
159 | 0 | if (key.equals(commonPrefix)) { |
160 | 0 | return true; |
161 | |
} |
162 | 0 | String trimmedPrefix = getTrimmedPrefix(commonPrefix, delimiter); |
163 | 0 | if (key.equals(trimmedPrefix)) { |
164 | 0 | return true; |
165 | |
} |
166 | 0 | } |
167 | 0 | return false; |
168 | |
} |
169 | |
|
170 | |
|
171 | |
|
172 | |
|
173 | |
protected DisplayRow getDisplayRow(final S3ObjectSummary summary, final String prefix, final String delimiter) { |
174 | 0 | String key = summary.getKey(); |
175 | |
|
176 | |
|
177 | 0 | String image = html.getImage(context.getFileImage()); |
178 | 0 | String show = getShow(key, prefix); |
179 | 0 | String dest = delimiter + key; |
180 | 0 | String ahref = html.getHref(dest, show); |
181 | 0 | String date = context.getLastModifiedDateFormatter().format(summary.getLastModified()); |
182 | 0 | String size = nf.format((summary.getSize() / 1024D)) + " KiB"; |
183 | |
|
184 | |
|
185 | 0 | DisplayRow displayRow = new DisplayRow(); |
186 | 0 | displayRow.setShow(show); |
187 | 0 | displayRow.setImage(image); |
188 | 0 | displayRow.setAhref(ahref); |
189 | 0 | displayRow.setLastModified(date); |
190 | 0 | displayRow.setSize(size); |
191 | 0 | return displayRow; |
192 | |
} |
193 | |
|
194 | |
protected List<DisplayRow> getObjectDisplayRows(final ObjectListing objectListing, final String prefix, |
195 | |
final String delimiter) { |
196 | 0 | List<DisplayRow> displayRows = new ArrayList<DisplayRow>(); |
197 | 0 | for (S3ObjectSummary summary : objectListing.getObjectSummaries()) { |
198 | 0 | if (isDirectory(summary, objectListing.getCommonPrefixes(), prefix, delimiter)) { |
199 | 0 | continue; |
200 | |
} |
201 | 0 | DisplayRow displayRow = getDisplayRow(summary, prefix, delimiter); |
202 | 0 | if (displayRow == null) { |
203 | 0 | continue; |
204 | |
} |
205 | 0 | displayRows.add(displayRow); |
206 | 0 | } |
207 | 0 | return displayRows; |
208 | |
} |
209 | |
|
210 | |
|
211 | |
|
212 | |
|
213 | |
protected DisplayRow getUpOneDirectoryDisplayRow(final String prefix, final String delimiter) { |
214 | 0 | if (StringUtils.isEmpty(prefix)) { |
215 | 0 | return null; |
216 | |
} |
217 | |
|
218 | |
|
219 | 0 | String image = ""; |
220 | 0 | String show = ".." + delimiter; |
221 | 0 | String dest = getUpOneDirectoryDest(prefix, delimiter); |
222 | 0 | String ahref = html.getHref(dest, show); |
223 | 0 | String date = ""; |
224 | 0 | String size = ""; |
225 | |
|
226 | |
|
227 | 0 | DisplayRow displayRow = new DisplayRow(); |
228 | 0 | displayRow.setImage(image); |
229 | 0 | displayRow.setAhref(ahref); |
230 | 0 | displayRow.setLastModified(date); |
231 | 0 | displayRow.setSize(size); |
232 | 0 | return displayRow; |
233 | |
} |
234 | |
|
235 | |
|
236 | |
|
237 | |
|
238 | |
|
239 | |
protected String getUpOneDirectoryDest(String prefix, final String delimiter) { |
240 | 0 | if (prefix.endsWith(delimiter)) { |
241 | 0 | prefix = prefix.substring(0, prefix.length() - 1); |
242 | |
} |
243 | 0 | int pos = prefix.lastIndexOf(delimiter); |
244 | 0 | if (pos == -1) { |
245 | 0 | return delimiter + getBrowseHtml(); |
246 | |
} else { |
247 | 0 | return delimiter + prefix.substring(0, pos + 1); |
248 | |
} |
249 | |
} |
250 | |
|
251 | |
public void setContext(final S3BucketContext context) { |
252 | 0 | this.context = context; |
253 | 0 | } |
254 | |
|
255 | |
public S3BucketContext getContext() { |
256 | 0 | return context; |
257 | |
} |
258 | |
|
259 | |
public String getBrowseHtml() { |
260 | 0 | return browseHtml; |
261 | |
} |
262 | |
|
263 | |
public void setBrowseHtml(final String browseHtml) { |
264 | 0 | this.browseHtml = browseHtml; |
265 | 0 | } |
266 | |
|
267 | |
} |