forked from jooby-project/jooby
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathAttachedFile.java
More file actions
145 lines (125 loc) · 3.42 KB
/
AttachedFile.java
File metadata and controls
145 lines (125 loc) · 3.42 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
/**
* 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.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;
/**
* Represent a file attachment response.
*
* @author edgar
* @since 2.0.0
*/
public class AttachedFile {
private static final String CONTENT_DISPOSITION = "attachment;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 String fileName;
private String contentDisposition;
private InputStream content;
/**
* Creates a new file attachment.
*
* @param content File content.
* @param fileName Filename.
* @param fileSize File size or <code>-1</code> if unknown.
*/
public AttachedFile(@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, this.fileName);
} else {
this.contentDisposition = String.format(CONTENT_DISPOSITION, 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 content File content.
* @param fileName Filename.
*/
public AttachedFile(@Nonnull InputStream content, @Nonnull String fileName) {
this(content, fileName, -1);
}
/**
* Creates a new file attachment.
*
* @param file File content.
* @param fileName Filename.
* @throws IOException For IO exception while reading file.
*/
public AttachedFile(@Nonnull Path file, @Nonnull String fileName) throws IOException {
this(new FileInputStream(file.toFile()), fileName, Files.size(file));
}
/**
* Creates a new file attachment.
*
* @param file File content.
* @throws IOException For IO exception while reading file.
*/
public AttachedFile(@Nonnull Path file) throws IOException {
this(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;
}
}