/**
 * Copyright (c) 2005 UPS_SCS NL
 *  
 */
package org.wamblee.crawler.kiss;

import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.IOException;
import java.util.Arrays;
import java.util.Date;

import javax.mail.internet.AddressException;
import javax.mail.internet.InternetAddress;
import javax.xml.transform.TransformerException;

import org.apache.commons.mail.EmailException;
import org.apache.commons.mail.HtmlEmail;
import org.apache.xml.serialize.OutputFormat;
import org.apache.xml.serialize.XMLSerializer;
import org.dom4j.Element;
import org.w3c.dom.Document;
import org.wamblee.io.FileResource;
import org.wamblee.io.InputResource;
import org.wamblee.xml.XSLT;

/**
 * A notifier that uses SMTP to notify users by mail. 
 *
 */
public class MailNotifier implements Notifier {
    
    private String _from;
    private String _to;
    private String _subject;
    private String _htmlXslt;
    private String _textXslt;
    private MailServer _server;
    
    /**
     * Constructs the notifier. 
     * 
     * @param aFrom Sender mail address to use. 
     * @param aTo Recipient mail address to use. 
     * @param aSubject Subject to use in the email. 
     * @param aHtmlXslt XSLT file to transform the report into HTML.
     * @param aTextXslt XSLT file to transform the report into text. 
     * @param aServer Mail server to use. 
     */
    public MailNotifier( String aFrom, String aTo, String aSubject,
            String aHtmlXslt, String aTextXslt, MailServer aServer) {
        _from = aFrom;
        _to = aTo;
        _subject = aSubject;
        _htmlXslt = aHtmlXslt;
        _textXslt = aTextXslt; 
        _server = aServer;
    }

    /*
     *  (non-Javadoc)
     * @see org.wamblee.crawler.kiss.Notifier#send(org.dom4j.Element)
     */
    public void send( Element aReport ) throws NotificationException {
        HtmlEmail mail = new HtmlEmail();
        try {
            mail.setFrom(_from);
            mail.setTo(Arrays.asList(new InternetAddress[] { new InternetAddress(_to) }));
            mail.setSentDate(new Date());
            mail.setSubject(_subject);
            
            String htmlText = transformReport( aReport, new FileResource(new File(_htmlXslt)) );
            String plainText = transformReport( aReport, new FileResource(new File(_textXslt)) );
            
            
            
            mail.setHtmlMsg(htmlText);
            mail.setTextMsg(plainText);
            
            _server.send(mail);
        }
        catch (EmailException e) {
            throw new NotificationException(e.getMessage(), e);
        } catch (TransformerException e) { 
            throw new NotificationException(e.getMessage(), e);
        } catch (IOException e) { 
            throw new NotificationException(e.getMessage(), e);
        } catch (AddressException e) { 
            throw new NotificationException(e.getMessage(), e);
        }
    }

    /**
     * Transforms a report into a destination format.
     * @param aReport Report to transform
     * @param aXslt XSLT to use. 
     * @return Transformed result. 
     * @throws IOException In case of IO problems.
     * @throws TransformerException In case of problems transforming. 
     */
    private String transformReport( Element aReport, InputResource aXslt ) throws IOException, TransformerException {
        String reportXmlText = aReport.asXML();
        return new XSLT().textTransform(reportXmlText.getBytes(), aXslt);
    }
}