package com.mossle.ext.mail;
import java.net.UnknownHostException;
import java.util.HashMap;
import java.util.Map;
import javax.mail.internet.InternetAddress;
import javax.mail.internet.MimeMessage;
import com.mossle.core.mail.HostGenerator;
import com.mossle.core.mail.HostGeneratorImpl;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.core.io.InputStreamSource;
import org.springframework.mail.javamail.JavaMailSender;
import org.springframework.mail.javamail.JavaMailSenderImpl;
import org.springframework.mail.javamail.MimeMessageHelper;
import org.springframework.util.Assert;
public class MailHelper {
private static Logger logger = LoggerFactory.getLogger(MailHelper.class);
private HostGenerator hostGenerator = new HostGeneratorImpl();
private MailServerInfoCache mailServerInfoCache = new MemoryMailServerInfoCache();
public MailDTO send(String from, String to, String subject, String content) {
return this.send(from, to, subject, content,
this.getDefaultMailServerInfo());
}
public MailDTO send(String from, String to, String subject, String content,
String mailServerInfoName) {
return this.send(from, to, subject, content,
mailServerInfoCache.getMailServerInfo(mailServerInfoName));
}
public MailDTO send(String from, String to, String subject, String content,
MailServerInfo mailServerInfo) {
MailDTO mailDto = new MailDTO();
mailDto.setFrom(from);
mailDto.setTo(to);
mailDto.setSubject(subject);
mailDto.setContent(content);
MailDTO resultMailDto = this.send(mailDto, mailServerInfo);
return resultMailDto;
}
public MailDTO send(MailDTO mailDto) {
logger.debug("send : {}", mailDto);
return this.send(mailDto, this.getDefaultMailServerInfo());
}
public MailDTO send(MailDTO mailDto, MailServerInfo mailServerInfo) {
try {
if (mailServerInfo == null) {
mailServerInfo = this.getDefaultMailServerInfo();
}
String from = mailDto.getFrom();
String to = mailDto.getTo();
String subject = mailDto.getSubject();
String content = mailDto.getContent();
if (this.isBlank(from)) {
from = mailServerInfo.getDefaultFrom();
mailDto.setFrom(from);
}
logger.debug("{} : {}", from, to);
if (mailServerInfo.isSkip()) {
logger.info("send mail from {} to {}", from, to);
logger.info("subject : {}, content : {}", subject, content);
mailDto.setSuccess(true);
return mailDto;
} else if (mailServerInfo.isTest()) {
return this.sendTestMail(mailDto, mailServerInfo);
} else {
return this.sendRealMail(mailDto, mailServerInfo);
}
} catch (Exception ex) {
logger.error(ex.getMessage(), ex);
return null;
}
}
protected MailDTO sendTestMail(MailDTO mailDto,
MailServerInfo mailServerInfo) {
String from = mailDto.getFrom();
String to = mailDto.getTo();
String subject = mailDto.getSubject();
String content = mailDto.getContent();
String address = "";
try {
address = hostGenerator.generateLocalAddress();
} catch (UnknownHostException ex) {
logger.error(ex.getMessage(), ex);
}
String decoratedContent = "address : " + address + "\nfrom : " + from
+ "\nto : " + to + "\n";
decoratedContent += ("subject : " + subject + "\ncontent : " + content);
String decoratedSubject = "[test]" + subject;
String decoratedFrom = from;
String testMail = mailServerInfo.getTestMail();
logger.info("send mail from {} to {}", decoratedFrom, testMail);
logger.info("subject : {}, content : {}", decoratedSubject,
decoratedContent);
mailDto.setFrom(decoratedFrom);
mailDto.setTo(testMail);
mailDto.setSubject(decoratedSubject);
mailDto.setContent(decoratedContent);
return this.sendRealMail(mailDto, mailServerInfo);
}
protected MailDTO sendRealMail(MailDTO mailDto,
MailServerInfo mailServerInfo) {
String from = mailDto.getFrom();
String to = mailDto.getTo();
String cc = mailDto.getCc();
String bcc = mailDto.getBcc();
String subject = mailDto.getSubject();
String content = mailDto.getContent();
to = to.replaceAll("\n", ",").replaceAll(";", ",");
if (this.isBlank(cc)) {
cc = null;
} else {
cc = cc.replaceAll("\n", ",").replaceAll(";", ",");
}
if (this.isBlank(bcc)) {
bcc = null;
} else {
bcc = bcc.replaceAll("\n", ",").replaceAll(";", ",");
}
logger.debug("from : {}, to : {}", from, to);
logger.debug("cc : {}, bcc : {}", cc, bcc);
logger.debug("subject : {}", subject);
try {
JavaMailSender javaMailSender = mailServerInfo.getJavaMailSender();
MimeMessage msg = javaMailSender.createMimeMessage();
MimeMessageHelper helper = new MimeMessageHelper(msg, true, "UTF-8");
helper.setFrom(from);
helper.setSubject(subject);
helper.setTo(InternetAddress.parse(to));
helper.setText(content, true);
if (cc != null) {
helper.setCc(InternetAddress.parse(cc));
}
if (bcc != null) {
helper.setBcc(InternetAddress.parse(bcc));
}
for (Map.Entry<String, InputStreamSource> entry : mailDto
.getAttachments().entrySet()) {
helper.addAttachment(entry.getKey(), entry.getValue());
}
javaMailSender.send(msg);
logger.debug("send mail from {} to {}", from, to);
mailDto.setSuccess(true);
} catch (Exception e) {
logger.error("send mail error", e);
mailDto.setSuccess(false);
if (e.getCause() != null) {
mailDto.setException(e.getCause());
} else {
mailDto.setException(e);
}
}
return mailDto;
}
public boolean isBlank(String text) {
return (text == null) || "".equals(text.trim());
}
public boolean notBlank(String text) {
return (text != null) && (!"".equals(text.trim()));
}
public MailServerInfo getDefaultMailServerInfo() {
return mailServerInfoCache.getDefaultMailServerInfo();
}
public void setDefaultMailServerInfo(MailServerInfo defaultMailServerInfo) {
Assert.notNull(defaultMailServerInfo);
mailServerInfoCache.setDefaultMailServerInfo(defaultMailServerInfo);
}
}