forked from jooby-project/jooby
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathFileDownload.java
More file actions
291 lines (256 loc) · 7.47 KB
/
FileDownload.java
File metadata and controls
291 lines (256 loc) · 7.47 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
/**
* Jooby https://jooby.io
* Apache License Version 2.0 https://jooby.io/LICENSE.txt
* Copyright 2014 Edgar Espina
*/
package io.jooby;
import org.apache.commons.io.FilenameUtils;
import javax.annotation.Nonnull;
import java.io.ByteArrayInputStream;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;
import java.nio.file.Files;
import java.nio.file.Path;
/**
* Represents a file download.
*
* @author edgar
* @author imeszaros
* @since 2.9.0
*/
public class FileDownload {
/**
* Download mode.
*/
public enum Mode {
/** Value indicating that the file can be displayed inside the Web page, or as the Web page. */
INLINE("inline"),
/** Value indicating that the file should be downloaded; most browsers present a 'Save as' dialog. */
ATTACHMENT("attachment");
final String value;
Mode(String value) {
this.value = value;
}
}
private static final String CONTENT_DISPOSITION = "%s;filename=\"%s\"";
private static final String FILENAME_STAR = ";filename*=%s''%s";
private static final String CHARSET = "UTF-8";
private final long fileSize;
private final MediaType contentType;
private final String fileName;
private final String contentDisposition;
private final InputStream content;
/**
* Creates a new file attachment.
*
* @param mode Download mode.
* @param content File content.
* @param fileName Filename.
* @param fileSize File size or <code>-1</code> if unknown.
*/
public FileDownload(Mode mode, @Nonnull InputStream content, @Nonnull String fileName, long fileSize) {
try {
this.fileName = FilenameUtils.getName(fileName);
this.contentType = MediaType.byFile(this.fileName);
String filenameStar = URLEncoder.encode(this.fileName, CHARSET).replaceAll("\\+", "%20");
if (this.fileName.equals(filenameStar)) {
this.contentDisposition = String.format(CONTENT_DISPOSITION, mode.value, this.fileName);
} else {
this.contentDisposition = String.format(CONTENT_DISPOSITION, mode.value, this.fileName) + String
.format(FILENAME_STAR, CHARSET, filenameStar);
}
this.content = content;
this.fileSize = fileSize;
} catch (UnsupportedEncodingException x) {
throw SneakyThrows.propagate(x);
}
}
/**
* Creates a new file attachment.
*
* @param mode Download mode.
* @param content File content.
* @param fileName Filename.
*/
public FileDownload(Mode mode, @Nonnull InputStream content, @Nonnull String fileName) {
this(mode, content, fileName, -1);
}
/**
* Creates a new file attachment.
*
* @param mode Download mode.
* @param content File content.
* @param fileName Filename.
*/
public FileDownload(Mode mode, @Nonnull byte[] content, @Nonnull String fileName) {
this(mode, new ByteArrayInputStream(content), fileName, content.length);
}
/**
* Creates a new file attachment.
*
* @param mode Download mode.
* @param file File content.
* @param fileName Filename.
* @throws IOException For IO exception while reading file.
*/
public FileDownload(Mode mode, @Nonnull Path file, @Nonnull String fileName) throws IOException {
this(mode, new FileInputStream(file.toFile()), fileName, Files.size(file));
}
/**
* Creates a new file attachment.
*
* @param mode Download mode.
* @param file File content.
* @throws IOException For IO exception while reading file.
*/
public FileDownload(Mode mode, @Nonnull Path file) throws IOException {
this(mode, file, file.getFileName().toString());
}
/**
* File size or <code>-1</code> if unknown.
*
* @return File size or <code>-1</code> if unknown.
*/
public long getFileSize() {
return fileSize;
}
/**
* File content type.
*
* @return File content type.
*/
public MediaType getContentType() {
return contentType;
}
/**
* File name.
* @return File name.
*/
public String getFileName() {
return fileName;
}
/**
* Content disposition header.
*
* @return Content disposition header.
*/
public String getContentDisposition() {
return contentDisposition;
}
/**
* File content.
*
* @return File content.
*/
public InputStream stream() {
return content;
}
@Override public String toString() {
return fileName;
}
/**
* Allows creating a {@link FileDownload} with
* the specified {@link Mode}.
*/
public interface Builder {
/**
* Creates a {@link FileDownload} with
* the specified {@link Mode}.
*
* @param mode the {@link Mode}
* @return a {@link FileDownload} with the specified mode
*/
FileDownload build(Mode mode);
/**
* Creates an attached {@link FileDownload}.
*
* @return a {@link FileDownload} with {@link Mode#ATTACHMENT}
*/
default FileDownload attachment() {
return build(Mode.ATTACHMENT);
}
/**
* Creates an inline {@link FileDownload}.
*
* @return a {@link FileDownload} with {@link Mode#INLINE}
*/
default FileDownload inline() {
return build(Mode.INLINE);
}
}
/**
* Creates a builder with the specified content which can be used to create
* a {@link FileDownload} with any {@link Mode}.
*
* @param content File content.
* @param fileName Filename.
* @param fileSize File size or <code>-1</code> if unknown.
*
* @return a {@link Builder} with the specified content
*/
public static Builder build(@Nonnull InputStream content, @Nonnull String fileName, long fileSize) {
return mode -> new FileDownload(mode, content, fileName, fileSize);
}
/**
* Creates a builder with the specified content which can be used to create
* a {@link FileDownload} with any {@link Mode}.
*
* @param content File content.
* @param fileName Filename.
*
* @return a {@link Builder} with the specified content
*/
public static Builder build(@Nonnull InputStream content, @Nonnull String fileName) {
return mode -> new FileDownload(mode, content, fileName);
}
/**
* Creates a builder with the specified content which can be used to create
* a {@link FileDownload} with any {@link Mode}.
*
* @param content File content.
* @param fileName Filename.
*
* @return a {@link Builder} with the specified content
*/
public static Builder build(@Nonnull byte[] content, @Nonnull String fileName) {
return mode -> new FileDownload(mode, content, fileName);
}
/**
* Creates a builder with the specified content which can be used to create
* a {@link FileDownload} with any {@link Mode}.
*
* @param file File content.
* @param fileName Filename.
*
* @return a {@link Builder} with the specified content
*/
public static Builder build(@Nonnull Path file, @Nonnull String fileName) {
return mode -> {
try {
return new FileDownload(mode, file, fileName);
} catch (IOException e) {
throw SneakyThrows.propagate(e);
}
};
}
/**
* Creates a builder with the specified content which can be used to create
* a {@link FileDownload} with any {@link Mode}.
*
* @param file File content.
*
* @return a {@link Builder} with the specified content
*/
public static Builder build(@Nonnull Path file) {
return mode -> {
try {
return new FileDownload(mode, file);
} catch (IOException e) {
throw SneakyThrows.propagate(e);
}
};
}
}