1 | |
|
2 | |
|
3 | |
|
4 | |
|
5 | |
|
6 | |
|
7 | |
|
8 | |
|
9 | |
|
10 | |
|
11 | |
package org.mule.providers.email; |
12 | |
|
13 | |
import org.mule.config.i18n.CoreMessages; |
14 | |
import org.mule.impl.ThreadSafeAccess; |
15 | |
import org.mule.providers.AbstractMessageAdapter; |
16 | |
import org.mule.umo.MessagingException; |
17 | |
import org.mule.umo.provider.MessageTypeNotSupportedException; |
18 | |
import org.mule.util.IOUtils; |
19 | |
import org.mule.util.StringUtils; |
20 | |
import org.mule.util.SystemUtils; |
21 | |
|
22 | |
import java.io.BufferedInputStream; |
23 | |
import java.io.BufferedReader; |
24 | |
import java.io.ByteArrayOutputStream; |
25 | |
import java.io.InputStream; |
26 | |
import java.io.InputStreamReader; |
27 | |
import java.util.Date; |
28 | |
import java.util.Enumeration; |
29 | |
import java.util.LinkedList; |
30 | |
import java.util.List; |
31 | |
|
32 | |
import javax.mail.Header; |
33 | |
import javax.mail.Message; |
34 | |
import javax.mail.Part; |
35 | |
|
36 | |
|
37 | |
|
38 | |
|
39 | |
|
40 | |
|
41 | |
|
42 | |
|
43 | |
|
44 | |
|
45 | |
|
46 | |
|
47 | |
|
48 | |
|
49 | |
|
50 | |
|
51 | |
|
52 | |
|
53 | |
|
54 | |
public class SimpleMailMessageAdapter extends AbstractMessageAdapter |
55 | |
{ |
56 | |
|
57 | |
private static final long serialVersionUID = 8002607243523460556L; |
58 | |
public static final String HEADER_LIST_PREFIX = "List:"; |
59 | |
private Part message; |
60 | 40 | private byte[] cache = null; |
61 | |
|
62 | |
public SimpleMailMessageAdapter(Object object) throws MessagingException |
63 | 40 | { |
64 | 40 | Message message = assertMessageType(object); |
65 | |
|
66 | |
try |
67 | |
{ |
68 | 38 | setMessageDetails(message); |
69 | 38 | handleMessage(message); |
70 | |
} |
71 | 0 | catch (Exception e) |
72 | |
{ |
73 | 0 | throw new MessagingException(CoreMessages.failedToCreate("Message Adapter"), e); |
74 | 38 | } |
75 | 38 | } |
76 | |
|
77 | |
protected SimpleMailMessageAdapter(SimpleMailMessageAdapter template) |
78 | |
{ |
79 | 0 | super(template); |
80 | 0 | message = template.message; |
81 | 0 | cache = template.cache; |
82 | 0 | } |
83 | |
|
84 | |
|
85 | |
|
86 | |
|
87 | |
|
88 | |
protected void handleMessage(Message message) throws Exception |
89 | |
{ |
90 | 4 | setMessage(message); |
91 | 4 | } |
92 | |
|
93 | |
protected void setMessage(Part message) |
94 | |
{ |
95 | 38 | this.message = message; |
96 | 38 | } |
97 | |
|
98 | |
public Object getPayload() { |
99 | 44 | return message; |
100 | |
} |
101 | |
|
102 | |
public byte[] getPayloadAsBytes() throws Exception |
103 | |
{ |
104 | 2 | return buildCache(getEncoding()); |
105 | |
} |
106 | |
|
107 | |
public String getPayloadAsString(String encoding) throws Exception |
108 | |
{ |
109 | |
|
110 | |
|
111 | |
|
112 | |
|
113 | |
|
114 | 2 | return new String(buildCache(encoding), encoding); |
115 | |
} |
116 | |
|
117 | |
private byte[] buildCache(String encoding) throws Exception |
118 | |
{ |
119 | 4 | if (null == cache) |
120 | |
{ |
121 | 2 | if (message.getContentType().startsWith("text/")) |
122 | |
{ |
123 | 2 | cache = textPayload(encoding); |
124 | |
} |
125 | |
else |
126 | |
{ |
127 | 0 | cache = binaryPayload(); |
128 | |
} |
129 | |
} |
130 | 4 | return cache; |
131 | |
} |
132 | |
|
133 | |
private static Message assertMessageType(Object message) throws MessageTypeNotSupportedException |
134 | |
{ |
135 | 40 | if (message instanceof Message) |
136 | |
{ |
137 | 38 | return (Message)message; |
138 | |
} |
139 | |
else |
140 | |
{ |
141 | 4 | throw new MessageTypeNotSupportedException(message, MailMessageAdapter.class); |
142 | |
} |
143 | |
} |
144 | |
|
145 | |
private void setMessageDetails(Message message) throws javax.mail.MessagingException |
146 | |
{ |
147 | 38 | setProperty(MailProperties.INBOUND_TO_ADDRESSES_PROPERTY, |
148 | |
MailUtils.mailAddressesToString(message.getRecipients(Message.RecipientType.TO))); |
149 | 38 | setProperty(MailProperties.INBOUND_CC_ADDRESSES_PROPERTY, |
150 | |
MailUtils.mailAddressesToString(message.getRecipients(Message.RecipientType.CC))); |
151 | 38 | setProperty(MailProperties.INBOUND_BCC_ADDRESSES_PROPERTY, |
152 | |
MailUtils.mailAddressesToString(message.getRecipients(Message.RecipientType.BCC))); |
153 | |
try |
154 | |
{ |
155 | 38 | setProperty(MailProperties.INBOUND_REPLY_TO_ADDRESSES_PROPERTY, |
156 | |
MailUtils.mailAddressesToString(message.getReplyTo())); |
157 | |
} |
158 | 2 | catch (javax.mail.MessagingException me) |
159 | |
{ |
160 | 2 | logger.warn("Invalid address found in ReplyTo header:", me); |
161 | 36 | } |
162 | |
|
163 | |
try |
164 | |
{ |
165 | 38 | setProperty(MailProperties.INBOUND_FROM_ADDRESS_PROPERTY, |
166 | |
MailUtils.mailAddressesToString(message.getFrom())); |
167 | |
} |
168 | 2 | catch (javax.mail.MessagingException me) |
169 | |
{ |
170 | 2 | logger.warn("Invalid address found in From header:", me); |
171 | 36 | } |
172 | |
|
173 | 38 | setProperty(MailProperties.INBOUND_SUBJECT_PROPERTY, StringUtils.defaultIfEmpty( |
174 | |
message.getSubject(),"(no subject)")); |
175 | 38 | setProperty(MailProperties.INBOUND_CONTENT_TYPE_PROPERTY, StringUtils.defaultIfEmpty( |
176 | |
message.getContentType(), "text/plain")); |
177 | |
|
178 | 38 | Date sentDate = message.getSentDate(); |
179 | 38 | if (sentDate == null) |
180 | |
{ |
181 | 38 | sentDate = new Date(); |
182 | |
} |
183 | 38 | setProperty(MailProperties.SENT_DATE_PROPERTY, sentDate); |
184 | |
|
185 | 38 | for (Enumeration e = message.getAllHeaders(); e.hasMoreElements();) |
186 | |
{ |
187 | 96 | Header header = (Header)e.nextElement(); |
188 | 96 | String name = header.getName(); |
189 | 96 | String listName = toListHeader(name); |
190 | 96 | String value = header.getValue(); |
191 | |
|
192 | 96 | if (null == getProperty(name)) |
193 | |
{ |
194 | 94 | setProperty(name, value); |
195 | |
} |
196 | |
|
197 | 96 | if (null == getProperty(listName)) |
198 | |
{ |
199 | 94 | setProperty(listName, new LinkedList()); |
200 | |
} |
201 | 96 | if (getProperty(listName) instanceof List) |
202 | |
{ |
203 | 96 | ((List) getProperty(listName)).add(header.getValue()); |
204 | |
} |
205 | 96 | } |
206 | 38 | } |
207 | |
|
208 | |
|
209 | |
|
210 | |
|
211 | |
|
212 | |
|
213 | |
|
214 | |
|
215 | |
|
216 | |
public static boolean isListHeader(String name) |
217 | |
{ |
218 | 104 | return null != name && name.startsWith(HEADER_LIST_PREFIX); |
219 | |
} |
220 | |
|
221 | |
|
222 | |
|
223 | |
|
224 | |
|
225 | |
|
226 | |
|
227 | |
public static String toHeader(String name) |
228 | |
{ |
229 | 0 | if (isListHeader(name)) |
230 | |
{ |
231 | 0 | return name.substring(HEADER_LIST_PREFIX.length()); |
232 | |
} |
233 | |
else |
234 | |
{ |
235 | 0 | return name; |
236 | |
} |
237 | |
} |
238 | |
|
239 | |
|
240 | |
|
241 | |
|
242 | |
|
243 | |
|
244 | |
|
245 | |
public static String toListHeader(String header) |
246 | |
{ |
247 | 104 | if (isListHeader(header)) |
248 | |
{ |
249 | 0 | return header; |
250 | |
} |
251 | |
else |
252 | |
{ |
253 | 104 | return HEADER_LIST_PREFIX + header; |
254 | |
} |
255 | |
} |
256 | |
|
257 | |
private static InputStream addBuffer(InputStream stream) |
258 | |
{ |
259 | 2 | if (!(stream instanceof BufferedInputStream)) |
260 | |
{ |
261 | 2 | stream = new BufferedInputStream(stream); |
262 | |
} |
263 | 2 | return stream; |
264 | |
} |
265 | |
|
266 | |
private byte[] binaryPayload() throws Exception |
267 | |
{ |
268 | 0 | InputStream stream = addBuffer(message.getInputStream()); |
269 | 0 | ByteArrayOutputStream baos = new ByteArrayOutputStream(32768); |
270 | 0 | IOUtils.copy(stream, baos); |
271 | 0 | return baos.toByteArray(); |
272 | |
} |
273 | |
|
274 | |
private byte[] textPayload(String encoding) throws Exception |
275 | |
{ |
276 | 2 | InputStream stream = addBuffer(message.getInputStream()); |
277 | 2 | BufferedReader reader = new BufferedReader(new InputStreamReader(stream)); |
278 | 2 | StringBuffer buffer = new StringBuffer(32768); |
279 | |
|
280 | |
String line; |
281 | 4 | while ((line = reader.readLine()) != null) |
282 | |
{ |
283 | 2 | buffer.append(line).append(SystemUtils.LINE_SEPARATOR); |
284 | |
} |
285 | |
|
286 | 2 | return buffer.toString().getBytes(encoding); |
287 | |
} |
288 | |
|
289 | |
public ThreadSafeAccess newThreadCopy() |
290 | |
{ |
291 | 0 | return new SimpleMailMessageAdapter(this); |
292 | |
} |
293 | |
} |