LinShare

/*

 *    This file is part of Linshare.

 *

 *   Linshare is free software: you can redistribute it and/or modify

 *   it under the terms of the GNU Affero General Public License as

 *   published by the Free Software Foundation, either version 3 of

 *   the License, or (at your option) any later version.

 *

 *   Linshare 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 Affero General Public License for more details.

 *

 *   You should have received a copy of the GNU Affero General Public

 *   License along with Foobar.  If not, see

 *                                    <http://www.gnu.org/licenses/>.

 *

 *   (c) 2008 Groupe Linagora - http://linagora.org

 *

*/

package org.linagora.linShare.core.service.impl;

import java.util.Properties;

import javax.activation.DataHandler;

import javax.mail.BodyPart;

import javax.mail.MessagingException;

import javax.mail.Multipart;

import javax.mail.Session;

import javax.mail.Transport;

import javax.mail.internet.InternetAddress;

import javax.mail.internet.MimeBodyPart;

import javax.mail.internet.MimeMessage;

import javax.mail.internet.MimeMultipart;

import javax.mail.util.ByteArrayDataSource;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.linagora.linShare.core.exception.TechnicalErrorCode;

import org.linagora.linShare.core.exception.TechnicalException;

import org.linagora.linShare.core.service.NotifierService;

/**

 *

 */

public class MailNotifierServiceImpl implements NotifierService {

    /** The smtpServer that will send the email. */

    private final String smtpServer;

    /** The smtp user. */

    private final String smtpUser;

    /** The sender identity. */

    private final String smtpSender;

    /** The smtp password. */

    private final String smtpPassword;

    /** The smtp port. */

    private  final int smtpPort;

    /** Is the server needs authentification. */

    private  final boolean needsAuth;

    /** Mail charset. */

    private final String charset;

    private final static Log logger = LogFactory.getLog(MailNotifierServiceImpl.class);

    /**

     * see  http://java.sun.com/developer/EJTechTips/2004/tt0625.html for multipart/alternative

     * @param smtpServer

     * @param smtpSender

     * @param smtpUser

     * @param smtpPassword

     * @param needsAuth

     * @param charset

     */

    public MailNotifierServiceImpl(String smtpServer,int smtpPort,  String smtpSender, String smtpUser, String smtpPassword,

        boolean needsAuth, String charset) {

        this.smtpServer = smtpServer;

        this.smtpPort = smtpPort;

        this.smtpSender = smtpSender;

        this.smtpUser = smtpUser;

        this.smtpPassword = smtpPassword;

        this.needsAuth = needsAuth;

        this.charset = charset;

    }

        /** Send notification to a recipient.

     * @param recipient notification recipient.

     * @param subject subject.

     * @param content content.

     */

    public void sendNotification(String fromUser, String recipient, String subject, String htmlContent, String textContent) {

       // get the mail session

        Session session = getMailSession();

        // Define message

        MimeMessage messageMim = new MimeMessage(session);

                try {

            if(fromUser==null)

            messageMim.setFrom(new InternetAddress(smtpSender));

            else            

            messageMim.setFrom(new InternetAddress(fromUser));

            messageMim.addRecipient(javax.mail.Message.RecipientType.TO,

                    new InternetAddress(recipient));

            messageMim.setSubject(subject,charset);

                // Create a "related" Multipart message

           //content type is multipart/alternative

           //it will contain two part BodyPart 1 and 2

                Multipart mp = new MimeMultipart("alternative");

               //BodyPart 1 is text file

               BodyPart alt_bp1 = new MimeBodyPart();

               alt_bp1.setDataHandler(new DataHandler(new ByteArrayDataSource(textContent,"text/plain")));

              mp.addBodyPart(alt_bp1); 

              //BodyPart 2 

              //content type is multipart/related

              //A multipart/related is used to indicate that message parts should not be considered individually but rather

              //as parts of an aggregate whole. The message consists of a root part (by default, the first) which reference other parts inline,

              //which may in turn reference other parts.

              Multipart html_mp = new MimeMultipart("related");

              //Include an HTML message with images.

              //BodyParts: the HTML file and an image 

               // Get the HTML file

               BodyPart rel_bph = new MimeBodyPart();

               rel_bph.setDataHandler(new DataHandler(new ByteArrayDataSource(htmlContent,"text/html; charset="+charset)));

               html_mp.addBodyPart(rel_bph);

                    //inline image ?

                    String cid = "image.part.1@linshare.org";

                    MimeBodyPart rel_bpi = new MimeBodyPart();

                       // Initialize and add the image file to the html body part

                   rel_bpi.setFileName("mail_logo.png");

                   rel_bpi.setText("linshare");

                   rel_bpi.setDataHandler(new DataHandler(getClass().getResource("/org/linagora/linShare/core/service/mail_logo.png")));

                   rel_bpi.setHeader("Content-ID", "<" + cid + ">");

                   rel_bpi.setDisposition("inline");

                   html_mp.addBodyPart(rel_bpi);

               // Create the second BodyPart of the multipart/alternative,

               // set its content to the html multipart, and add the

               // second bodypart to the main multipart.

               BodyPart alt_bp2 = new MimeBodyPart();

               alt_bp2.setContent(html_mp);

               mp.addBodyPart(alt_bp2);

            messageMim.setContent(mp);

            // Since we used html tags, the content must be marker as text/html

            //messageMim.setContent(content,"text/html; charset="+charset);

            Transport tr = session.getTransport("smtp");

            // Connect to smtp server, if needed

            if (needsAuth) {

                tr.connect(smtpServer, smtpPort ,smtpUser, smtpPassword);

                messageMim.saveChanges();

                tr.sendMessage(messageMim, messageMim.getAllRecipients());

                tr.close();

            } else {

                // Send message

                    Transport.send(messageMim);

            }

        } catch (MessagingException e) {

                logger.error("Error sending notification on "+smtpServer+" port "+smtpPort,e);

            throw new TechnicalException(TechnicalErrorCode.MAIL_EXCEPTION,  "Error sending notification",e);

        }  catch (Exception e) {

                  logger.error("Error sending notification on "+smtpServer+" port "+smtpPort,e);

            throw new TechnicalException(TechnicalErrorCode.MAIL_EXCEPTION, "Error sending notification",e);

        }

    }

    private Session getMailSession(){

             // create some properties and get the default Session

            // Set the host smtp address

        Properties props = new Properties();

        props.put("mail.smtp.host", smtpServer);

        props.put("mail.smtp.port", smtpPort+"");

        if (needsAuth){

            props.put("mail.smtp.auth", "true");

        }else{

            props.put("mail.smtp.auth", "false");

        }

        // create some properties and get the default Session

        Session session = Session.getInstance(props, null);

       if(logger.isDebugEnabled()){

               session.setDebug(true);

       } else {

              session.setDebug(false);

       }

        return session;

    }

}