/*
* Copyright (c) 2007-2008 Nokia Corporation and/or its subsidiary(-ies).
* All rights reserved.
* This component and the accompanying materials are made available
* under the terms of the License "Eclipse Public License v1.0"
* which accompanies this distribution, and is available
* at the URL "http://www.eclipse.org/legal/epl-v10.html".
*
* Initial Contributors:
* Nokia Corporation - initial contribution.
*
* Contributors:
*
* Description: 
*
*/

package com.nokia.ant.listener.internaldata;

import java.util.Properties;
import java.util.Hashtable;
import java.util.zip.GZIPOutputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import javax.mail.*;
import javax.mail.internet.*;
import javax.naming.*;
import javax.naming.directory.*;

import javax.activation.DataHandler;
import javax.mail.util.ByteArrayDataSource;
import org.apache.log4j.Logger;

/**
 * Sends email of the internal data log in a zipped format.
 *
 */
public class EmailDataSender {

    
    // The target address
    public static final String TO_EMAIL = "helium.internaldata@nokia.com";
    
    // LDAP config
    public static final String LDAP_URL = "ldap://nedi.europe.nokia.com:389/o=Nokia";

    // Default SMTP server
    public static final String SMTP_SERVER = "smtp.nokia.com";
    
    // Configured smtp server address
    private String smtpServer;
    // Logger
    private Logger log = Logger.getLogger(EmailDataSender.class);
    
    /**
     * Set the smtp server address.
     */
    public void setSMTPServer(String address) {
        smtpServer = address;
    }

    /**
     * Get the smtp server address.
     */
    public String getSMTPServer() {
        if (smtpServer != null)
            return smtpServer;
        return SMTP_SERVER;
    }
    
    /**
     * Sending the XML data through email.
     */
    public void sendData(String data) {
        try {
            String email = getUserEmail();
            Properties props = new Properties();
            props.setProperty("mail.smtp.host", getSMTPServer());

            Session mailSession = Session.getDefaultInstance(props, null);

            MimeMessage message = new MimeMessage(mailSession);
            message.setSubject("[HELIUM]: internal data");
            
            MimeMultipart multipart = new MimeMultipart("related");
            
            // first part  (the text html content)
            BodyPart messageBodyPart = new MimeBodyPart();
            String htmlText = "<H1>Internal data</H1>";
            messageBodyPart.setContent(htmlText, "text/html");
            multipart.addBodyPart(messageBodyPart); // add to the multipart

            // second part (the data)
            messageBodyPart = new MimeBodyPart();
            ByteArrayDataSource dataSrc = gzip(data, "data.xml");
            messageBodyPart.setFileName(dataSrc.getName());
            messageBodyPart.setDataHandler(new DataHandler(dataSrc));
            messageBodyPart.setHeader("Content-ID","<data>");
            
            multipart.addBodyPart(messageBodyPart); // add to the multipart
            
            message.setContent(multipart);
            message.setFrom(new InternetAddress(email));
            message.addRecipient(Message.RecipientType.TO, new InternetAddress(TO_EMAIL));
            
            log.debug("Sending data.");
            Transport.send(message);
        } catch (Exception e) {
            log.error("Internal data failure: " + e.getMessage());
        }        
    }

    /**
     * GZipping a string.
     * @param data the content to be gzipped.
     * @param filename the name for the file.
     * @return a ByteArrayDataSource.
     */
    protected ByteArrayDataSource gzip(String data, String filename) throws IOException {
        ByteArrayOutputStream out = new ByteArrayOutputStream();
        GZIPOutputStream gz = new GZIPOutputStream(out);
        gz.write(data.getBytes());
        gz.close();
        out.close();
        ByteArrayDataSource dataSrc = new ByteArrayDataSource(out.toByteArray(), "application/x-gzip");
        dataSrc.setName(filename + ".gz");
        return dataSrc;
    }

    /**
     * Getting user email.
     * @returns the user email.
     */
    protected String getUserEmail() throws Exception {
        String username = System.getProperty("user.name");

        // Set up environment for creating initial context
        Hashtable env = new Hashtable(11);
        env.put(Context.INITIAL_CONTEXT_FACTORY,
                "com.sun.jndi.ldap.LdapCtxFactory");
        env.put(Context.PROVIDER_URL, LDAP_URL);

        // Create initial context
        DirContext ctx = new InitialDirContext(env);
        SearchControls controls = new SearchControls();
        controls.setSearchScope(SearchControls.SUBTREE_SCOPE);
        NamingEnumeration en = ctx.search("", "uid=" + username, controls);
        if (en.hasMore()) {
            SearchResult sr = (SearchResult) en.next();
            String email = (String) sr.getAttributes().get("mail").get();
            return email;
        }
        throw new Exception("Could not find user email in LDAP.");
    }

}
