/**
* Saves all attachments to a temp directory, and returns the directory path. Null if no
* attachments.
*/
public File saveAttachments(Message message) throws IOException, MessagingException {
File tmpDir = Files.createTempDir();
boolean foundAttachments = false;
Object content = message.getContent();
if (message.isMimeType(MULTIPART_WILDCARD) && content instanceof Multipart) {
Multipart mp = (Multipart) content;
for (int i = 0; i < mp.getCount(); i++) {
BodyPart bodyPart = mp.getBodyPart(i);
if (bodyPart instanceof MimeBodyPart && isNotBlank(bodyPart.getFileName())) {
MimeBodyPart mimeBodyPart = (MimeBodyPart) bodyPart;
mimeBodyPart.saveFile(new File(tmpDir, mimeBodyPart.getFileName()));
foundAttachments = true;
}
}
}
return foundAttachments ? tmpDir : null;
}
protected void fillContent(Message email, Execution execution, JCRSessionWrapper session)
throws MessagingException {
String text = getTemplate().getText();
String html = getTemplate().getHtml();
List<AttachmentTemplate> attachmentTemplates = getTemplate().getAttachmentTemplates();
try {
if (html != null || !attachmentTemplates.isEmpty()) {
// multipart
MimeMultipart multipart = new MimeMultipart("related");
BodyPart p = new MimeBodyPart();
Multipart alternatives = new MimeMultipart("alternative");
p.setContent(alternatives, "multipart/alternative");
multipart.addBodyPart(p);
// html
if (html != null) {
BodyPart htmlPart = new MimeBodyPart();
html = evaluateExpression(execution, html, session);
htmlPart.setContent(html, "text/html; charset=UTF-8");
alternatives.addBodyPart(htmlPart);
}
// text
if (text != null) {
BodyPart textPart = new MimeBodyPart();
text = evaluateExpression(execution, text, session);
textPart.setContent(text, "text/plain; charset=UTF-8");
alternatives.addBodyPart(textPart);
}
// attachments
if (!attachmentTemplates.isEmpty()) {
addAttachments(execution, multipart);
}
email.setContent(multipart);
} else if (text != null) {
// unipart
text = evaluateExpression(execution, text, session);
email.setText(text);
}
} catch (RepositoryException e) {
logger.error(e.getMessage(), e);
} catch (ScriptException e) {
logger.error(e.getMessage(), e);
}
}
/**
* Split mail message into several messages. Each body and each attachment results in separate
* message invoked on endpoint adapter. Mail message response if any should be sent only once
* within test case. However latest mail response sent by test case is returned, others are
* ignored.
*
* @param bodyPart
* @param messageHeaders
*/
private Message split(BodyPart bodyPart, Map<String, Object> messageHeaders) {
MailMessage mailMessage = createMailMessage(messageHeaders);
mailMessage.setBody(new BodyPart(bodyPart.getContent(), bodyPart.getContentType()));
StringResult result = new StringResult();
Stack<Message> responseStack = new Stack<Message>();
if (bodyPart instanceof AttachmentPart) {
marshaller.marshal(mailMessage, result);
fillStack(
getEndpointAdapter()
.handleMessage(
new DefaultMessage(result.toString(), messageHeaders)
.setHeader(
CitrusMailMessageHeaders.MAIL_CONTENT_TYPE, bodyPart.getContentType())
.setHeader(
CitrusMailMessageHeaders.MAIL_FILENAME,
((AttachmentPart) bodyPart).getFileName())),
responseStack);
} else {
marshaller.marshal(mailMessage, result);
fillStack(
getEndpointAdapter()
.handleMessage(
new DefaultMessage(result.toString(), messageHeaders)
.setHeader(
CitrusMailMessageHeaders.MAIL_CONTENT_TYPE, bodyPart.getContentType())),
responseStack);
}
if (bodyPart.hasAttachments()) {
for (AttachmentPart attachmentPart : bodyPart.getAttachments().getAttachments()) {
fillStack(split(attachmentPart, messageHeaders), responseStack);
}
}
return responseStack.isEmpty() ? null : responseStack.pop();
}
public boolean sendMessage() {
Properties properties = new Properties();
properties.put("mail.smtp.host", mailHost);
properties.put("mail.from", Source);
Session session = Session.getInstance(properties, null);
try {
Message message = new MimeMessage(session);
InternetAddress[] addressTO = null;
if (this.Destination != null && this.Destination.length() != 0) {
StringTokenizer TOList = getTokenizer(this.Destination);
addressTO = new InternetAddress[TOList.countTokens()];
for (int i = 0; i < addressTO.length; i++) {
addressTO[i] = new InternetAddress(TOList.nextToken());
message.addRecipient(Message.RecipientType.TO, addressTO[i]);
}
}
if (this.MsgCC != null && this.MsgCC.length() != 0) {
StringTokenizer CCList = getTokenizer(this.MsgCC);
InternetAddress[] addressCC = new InternetAddress[CCList.countTokens()];
for (int i = 0; i < addressCC.length; i++) {
addressCC[i] = new InternetAddress(CCList.nextToken());
message.addRecipient(Message.RecipientType.CC, addressCC[i]);
}
}
message.setFrom(new InternetAddress(Source));
message.setSubject(Subject);
Content = getHtmlHeader() + Content + getHtmlFooter();
Content = Content.replaceAll("\\{style\\}", MailStyle);
BodyPart messageBodyPart = new MimeBodyPart();
messageBodyPart.setText(Content);
messageBodyPart.setContent(Content, "text/html");
Multipart multipart = new MimeMultipart();
multipart.addBodyPart(messageBodyPart);
Iterator it = this.BinaryAttachments.iterator();
while (it.hasNext()) {
ByteArrayDataSource bads = (ByteArrayDataSource) it.next();
messageBodyPart = new MimeBodyPart();
// messageBodyPart.setDataHandler(new DataHandler(new FileDataSource("c:/test/tom.jpg")));
messageBodyPart.setDataHandler(new DataHandler(bads));
messageBodyPart.setFileName(bads.getName());
multipart.addBodyPart(messageBodyPart);
}
message.setContent(multipart);
Transport transport = session.getTransport(addressTO[0]);
transport.addConnectionListener(new ConnectionHandler());
transport.addTransportListener(new TransportHandler());
transport.connect();
transport.send(message);
return true;
} catch (Exception e) {
e.printStackTrace();
return false;
}
}
public static List<String> getText(BytesWritable value, Boolean tokenizep)
throws InterruptedException {
Session s = Session.getDefaultInstance(new Properties());
InputStream is = new ByteArrayInputStream(value.getBytes());
List<String> out = new ArrayList<String>();
try {
MimeMessage message = new MimeMessage(s, is);
message.getAllHeaderLines();
Analyzer standard_analyzer = new StandardAnalyzer(Version.LUCENE_43);
Analyzer email_analyzer = new UAX29URLEmailAnalyzer(Version.LUCENE_43);
Address[] fromAddrs = message.getFrom();
String fromAddrstr = "";
if (fromAddrs != null) {
for (Address addr : fromAddrs) {
fromAddrstr += (addr.toString() + " ");
}
}
Address[] toAddrs = message.getAllRecipients();
String toAddrstr = "";
if (toAddrs != null) {
for (Address addr : toAddrs) {
toAddrstr += (addr.toString() + " ");
}
}
String subject = message.getSubject();
String body = "";
try {
Object content = message.getContent();
// System.err.println(content.getContentType());
if (content instanceof String) {
body = (String) content;
} else if (content instanceof Multipart) {
Multipart mp = (Multipart) content;
for (int i = 0; i < mp.getCount(); i++) {
BodyPart bp = mp.getBodyPart(i);
// System.err.println(bp.getContentType());
Object c = bp.getContent();
if (c instanceof String) {
body = (String) c;
}
}
}
// people do really evil things with email, we're not sorting through it all now
} catch (DecodingException e) {
System.err.println("DecodingException");
} catch (UnsupportedEncodingException e) {
System.err.println("UnsuportedEncodingException");
} catch (IOException e) {
System.err.println("IOException");
}
if (tokenizep) {
List<String> fromData = new ArrayList<String>();
List<String> toData = new ArrayList<String>();
List<String> subjectData = new ArrayList<String>();
List<String> bodyData = new ArrayList<String>();
if (fromAddrstr != null) {
fromData = tokenizeString(email_analyzer, fromAddrstr);
}
if (toAddrstr != null) {
toData = tokenizeString(email_analyzer, toAddrstr);
}
if (subject != null) {
subjectData = tokenizeString(standard_analyzer, subject);
}
if (body != null) {
bodyData = tokenizeString(standard_analyzer, body);
}
out.add("FROM ");
out.addAll(fromData);
out.add("TO ");
out.addAll(toData);
out.add("SUBJECT ");
out.addAll(subjectData);
out.add("BODY ");
out.addAll(bodyData);
} else {
// if not tokenizep, return list with from and subject fields only
out.add(fromAddrstr);
out.add(subject);
}
} catch (MessagingException e) {
System.err.println("MessagineException");
}
return out;
}