binky.reportrunner.engine.utils.impl.EmailHandlerImpl.java Source code

Introduction

Here is the source code for binky.reportrunner.engine.utils.impl.EmailHandlerImpl.java

Source

/*******************************************************************************
 * Copyright (c) 2009 Daniel Grout.
 * 
 * GNU GENERAL PUBLIC LICENSE - Version 3
 * 
 * This file is part of Report Runner (http://code.google.com/p/reportrunner).
 * 
 * Report Runner is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 * 
 * Report Runner is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with Report Runner. If not, see <http://www.gnu.org/licenses/>.
 * 
 * Module: EmailHandler.java
 ******************************************************************************/
package binky.reportrunner.engine.utils.impl;

import java.io.IOException;
import java.util.Date;
import java.util.LinkedList;
import java.util.List;

import org.apache.commons.mail.EmailAttachment;
import org.apache.commons.mail.EmailException;
import org.apache.commons.mail.MultiPartEmail;
import org.apache.log4j.Logger;

import binky.reportrunner.engine.utils.EmailHandler;
import binky.reportrunner.engine.utils.FileSystemHandler;

public class EmailHandlerImpl implements EmailHandler {
    private static final boolean debug = false;
    private static final String subject = "Report Runner Files for: ";
    private static final String message = "Please find attached outputs for the above job.";
    private static Logger logger = Logger.getLogger(EmailHandlerImpl.class);

    /* (non-Javadoc)
     * @see binky.reportrunner.engine.utils.EmailHandler#sendEmail(java.lang.String, java.lang.String, java.lang.String, java.lang.String, java.lang.String, java.lang.String)
     */
    public void sendEmail(String destinationEmail, String fromEmail, String smtpServer, String fileUrl,
            String jobName, String groupName) throws EmailException, IOException {
        logger.debug("sending email to: " + destinationEmail + " on host " + smtpServer);
        FileSystemHandler fs = new FileSystemHandlerImpl();
        MultiPartEmail email = new MultiPartEmail();
        email.setDebug(debug);
        email.setHostName(smtpServer);
        for (String toEmail : destinationEmail.split(",")) {
            email.addTo(toEmail.trim());
        }
        email.setFrom(fromEmail);
        email.setSubject(subject + jobName + " [" + groupName + "]");
        email.setMsg(message);
        List<String> tempFiles = new LinkedList<String>();

        EmailAttachment attachment = new EmailAttachment();

        String temp = copyToTemp(fileUrl);
        tempFiles.add(temp);
        attachment.setURL(fs.getURL(temp));
        attachment.setDisposition(EmailAttachment.ATTACHMENT);
        attachment.setDescription("Report Runner Report");

        // get the file name of the report from the complete path
        String name = fs.getFileName(fileUrl);
        attachment.setName(name);
        logger.debug("File: " + fileUrl);
        logger.debug("Name: " + name);

        email.attach(attachment);

        email.send();

        fs.deleteFile(temp);

    }

    /* (non-Javadoc)
     * @see binky.reportrunner.engine.utils.EmailHandler#sendEmail(java.lang.String, java.lang.String, java.lang.String, java.util.List, java.lang.String, java.lang.String)
     */
    public void sendEmail(String destinationEmail, String fromEmail, String smtpServer, List<String> fileUrls,
            String jobName, String groupName) throws EmailException, IOException {
        // logger.debug("sending email to: " + to + " on host " + server);
        FileSystemHandler fs = new FileSystemHandlerImpl();
        MultiPartEmail email = new MultiPartEmail();
        email.setDebug(debug);
        email.setHostName(smtpServer);
        for (String toEmail : destinationEmail.split(",")) {
            email.addTo(toEmail.trim());
        }
        email.setFrom(fromEmail);
        email.setSubject(subject + jobName + " [" + groupName + "]");
        email.setMsg("Please find attached outputs for the above job.");
        List<String> tempFiles = new LinkedList<String>();
        for (String url : fileUrls) {

            EmailAttachment attachment = new EmailAttachment();

            String temp = copyToTemp(url);
            tempFiles.add(temp);
            attachment.setURL(fs.getURL(temp));
            attachment.setDisposition(EmailAttachment.ATTACHMENT);
            attachment.setDescription(message);

            // get the file name of the report from the complete path
            String name = fs.getFileName(url);
            attachment.setName(name);
            logger.debug("File: " + url);
            logger.debug("Name: " + name);

            email.attach(attachment);
        }

        email.send();
        for (String url : tempFiles) {
            fs.deleteFile(url);
        }
    }

    /* (non-Javadoc)
     * @see binky.reportrunner.engine.utils.EmailHandler#sendAlertEmail(java.lang.String, java.lang.String, java.lang.String, java.lang.String, java.lang.String, boolean, java.util.Date)
     */
    public void sendAlertEmail(String destinationEmail, String fromEmail, String smtpServer, String jobName,
            String groupName, boolean success, Date finishTime) throws EmailException, IOException {
        logger.debug("sending email to: " + destinationEmail + " on host " + smtpServer);
        MultiPartEmail email = new MultiPartEmail();
        email.setHostName(smtpServer);
        email.setDebug(debug);
        for (String toEmail : destinationEmail.split(",")) {
            email.addTo(toEmail.trim());
        }
        email.setFrom(fromEmail);

        if (success) {
            email.setSubject("Success message for: " + jobName + " [" + groupName + "]");

            email.setMsg("Job completed successfully at " + finishTime);

        } else {
            email.setSubject("Failure message for: " + jobName + " [" + groupName + "]");

            email.setMsg("Job completed with error(s) at " + finishTime);

        }

        email.send();

    }

    private String copyToTemp(String url) throws IOException {
        logger.debug("copying file to temp to attach to email: " + url);
        FileSystemHandler fs = new FileSystemHandlerImpl();
        String tempDir = System.getProperty("java.io.tmpdir");

        String dest = "file://" + tempDir + "/email_" + fs.getFileName(url);
        fs.copyFile(url, dest);
        return dest;
    }

}