Skip to content
This repository has been archived by the owner on May 23, 2019. It is now read-only.

Commit

Permalink
Update #69: refactor mail template engine and merge it with EndUserBa…
Browse files Browse the repository at this point in the history 
…seMail.
  • Loading branch information
@hdsdi3g
hdsdi3g committed Oct 28, 2014
1 parent 1cc5801 commit 03353ee
Show file tree
Hide file tree
Showing 6 changed files with 191 additions and 246 deletions.
171 changes: 160 additions & 11 deletions app/hd3gtv/mydmam/mail/EndUserBaseMail.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -16,25 +16,89 @@
*/
package hd3gtv.mydmam.mail;

import groovy.lang.Writable;
import groovy.text.SimpleTemplateEngine;
import hd3gtv.log2.Log2;
import hd3gtv.log2.Log2Dump;
import hd3gtv.log2.Log2Dumpable;
import hd3gtv.mydmam.module.MessagesOutsidePlay;
import hd3gtv.mydmam.module.MyDMAMModulesManager;

import java.io.BufferedReader;
import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.Writer;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.Locale;
import java.util.Map;
import java.util.Map.Entry;

import javax.mail.internet.InternetAddress;

import org.codehaus.groovy.control.CompilationFailedException;

/**
* With Groovy engine front-end (outside Play scope). Ignore Jar files content, but manage Play modules.
*/
public class EndUserBaseMail implements Log2Dumpable {

private static MailCenter mailcenter;
// private static InternetAddress admin_addr;
private static LinkedHashMap<String, File> templates;

public static final String BASE_PATH = "mail-templates";
public static final String TEMPLATE_HTML_FILE = "template.html";
public static final String TEMPLATE_TXT_FILE = "template.txt";
public static final String TEMPLATE_SUBJECT_FILE = "subject.txt";

public static boolean GROOVY_VERBOSE = false;
private static SimpleTemplateEngine template_engine;

static {
try {
mailcenter = MailCenter.getGlobal();
// admin_addr = new InternetAddress(Configuration.global.getValue("service", "administrator_mail", "root@localhost"));
templates = new LinkedHashMap<String, File>();
template_engine = new SimpleTemplateEngine();
template_engine.setVerbose(GROOVY_VERBOSE);

LinkedHashMap<String, File> conf_dirs = MyDMAMModulesManager.getAllConfDirectories();
File templates_directory;

for (Map.Entry<String, File> conf_dir_entry : conf_dirs.entrySet()) {
// entry.getKey() entry.getValue()
// TODO import templates
}

// templates_directory = new File(directoryclass.getAbsolutePath() + File.separator + BASE_PATH);

/**
* Test candidates validity
*/
/*for (int pos_tpl = 0; pos_tpl < templates_dir_to_test.size(); pos_tpl++) {
try {
String tpl_path = templates_dir_to_test.get(pos_tpl).getCanonicalPath();
if (new File(tpl_path + File.separator + TEMPLATE_HTML_FILE).exists() == false) {
throw new FileNotFoundException(tpl_path + File.separator + TEMPLATE_HTML_FILE);
}
if (new File(tpl_path + File.separator + TEMPLATE_TXT_FILE).exists() == false) {
throw new FileNotFoundException(tpl_path + File.separator + TEMPLATE_TXT_FILE);
}
String key = templates_dir_to_test.get(pos_tpl).getName().toLowerCase();
if (templates.containsKey(key) == false) {
templates.put(key, templates_dir_to_test.get(pos_tpl));
} else {
Log2.log.error("Template directory is already added", null, new Log2Dump("path", templates_dir_to_test.get(pos_tpl)));
}
} catch (IOException e) {
Log2.log.error("Can't use template directory: ", e, new Log2Dump("path", templates_dir_to_test.get(pos_tpl)));
}
}*/
System.out.println(templates);// XXX

} catch (Exception e) {
Log2.log.error("Can't init message", e);
}
Expand All @@ -48,20 +112,27 @@ private EndUserBaseMail() {
private String mail_template;
private MailPriority priority;

public static EndUserBaseMail create(Locale locale, InternetAddress to, String mail_template) {
private File template_directory;

public static EndUserBaseMail create(Locale locale, InternetAddress to, String mail_template) throws FileNotFoundException {
EndUserBaseMail mail = new EndUserBaseMail();
mail.locale = locale;
if (locale == null) {
throw new NullPointerException("\"locale\" can't to be null");
}

mail.to = to;
if (to == null) {
throw new NullPointerException("\"to\" can't to be null");
}
mail.mail_template = mail_template;

if (mail_template == null) {
throw new NullPointerException("\"mail_template\" can't to be null");
}
if (templates.containsKey(mail_template.toLowerCase()) == false) {
throw new FileNotFoundException("Not template for this name: " + mail_template.toLowerCase());
}
mail.template_directory = templates.get(mail_template.toLowerCase());

return mail;
}
Expand All @@ -83,19 +154,50 @@ public void send() {
send(new HashMap<String, Object>());
}

/**
* @param mail_vars "message" key is reserved, and used with messages.* files.
*/
public void send(HashMap<String, Object> mail_vars) {
if (mail_vars == null) {
throw new NullPointerException("\"mail_vars\" can't to be null");
}
try {
// TODO add all Messages
MailTemplateEngine mte = new MailTemplateEngine(mail_template);
mte.setLocale(locale);
mte.process(mail_vars);
MailContent mail = mailcenter.prepareMessage(mte.getSubject(), to);
/**
* Prepare template variables and messages
*/
HashMap<String, Object> all_mail_vars = new HashMap<String, Object>();
all_mail_vars.putAll(mail_vars);
all_mail_vars.put("messages", new MessagesOutsidePlay(locale));

/**
* Process templates: subject
*/
ArrayList<String> subject_text_list = process(new File(template_directory.getAbsolutePath() + File.separator + TEMPLATE_SUBJECT_FILE), all_mail_vars);
StringBuffer sb = new StringBuffer();
for (int pos = 0; pos < subject_text_list.size(); pos++) {
sb.append(subject_text_list.get(pos));
sb.append(" ");
}
MailContent mail = mailcenter.prepareMessage(sb.toString().trim(), to);

/**
* Process templates: html text
*/
mail.setHtmltext(process(new File(template_directory.getAbsolutePath() + File.separator + TEMPLATE_HTML_FILE), all_mail_vars));

/**
* Process templates: plain text
* TODO remove plain text ??
*/
ArrayList<String> plain_text_list = process(new File(template_directory.getAbsolutePath() + File.separator + TEMPLATE_TXT_FILE), all_mail_vars);
sb = new StringBuffer();
for (int pos = 0; pos < plain_text_list.size(); pos++) {
sb.append(plain_text_list.get(pos));
sb.append("\r\n");
}
mail.setPlaintext(sb.toString());

mail.setMailPriority(priority);
mail.setHtmltext(mte.getHtml_text());
mail.setPlaintext(mte.getPlain_text());
mail.send();

Log2Dump dump = getLog2Dump();
Expand All @@ -107,4 +209,51 @@ public void send(HashMap<String, Object> mail_vars) {
Log2.log.error("Fail to send an user mail", e, this);
}
}

private static class TemplateWriter extends Writer {
StringBuffer raw;

public TemplateWriter() {
raw = new StringBuffer();
}

public void write(char[] cbuf, int off, int len) throws IOException {
raw.append(cbuf, off, len);
}

public void flush() throws IOException {
}

public void close() throws IOException {
}

public ArrayList<String> getContent() {
ArrayList<String> content = new ArrayList<String>();
ByteArrayInputStream baos = new ByteArrayInputStream(raw.toString().getBytes());
BufferedReader reader = new BufferedReader(new InputStreamReader(baos), 0xFFF);
String line;
try {
while (((line = reader.readLine()) != null)) {
content.add(line.trim());
}
} catch (IOException e) {
Log2.log.error("Can't convert text stream", e);
return null;
}
if (content.size() > 0) {
if (content.get(0).trim().equals("")) {
content.remove(0);
}
}
return content;
}

};

private static ArrayList<String> process(File template_file, HashMap<String, Object> all_mail_vars) throws IOException, CompilationFailedException, ClassNotFoundException {
Writable writable = template_engine.createTemplate(template_file).make(all_mail_vars);
TemplateWriter tw = new TemplateWriter();
writable.writeTo(tw);
return tw.getContent();
}
}
Loading

0 comments on commit 03353ee

Please sign in to comment.