buildframework/helium/sf/java/internaldata/src/com/nokia/helium/internaldata/ant/listener/EmailDataSender.java
/*
* 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.helium.internaldata.ant.listener;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.Hashtable;
import java.util.Properties;
import java.util.zip.GZIPOutputStream;
import javax.activation.DataHandler;
import javax.mail.BodyPart;
import javax.mail.Message;
import javax.mail.MessagingException;
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 javax.naming.Context;
import javax.naming.NamingEnumeration;
import javax.naming.NamingException;
import javax.naming.directory.DirContext;
import javax.naming.directory.InitialDirContext;
import javax.naming.directory.SearchControls;
import javax.naming.directory.SearchResult;
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 (NamingException e) {
// We are Ignoring the errors as no need to fail the build.
log.debug("Internal data failure: NamingException " + e.getMessage(), e);
} catch (MessagingException e) {
// We are Ignoring the errors as no need to fail the build.
log.debug("Internal data failure: MessagingException " + e.getMessage(), e);
} catch (IOException e) {
// We are Ignoring the errors as no need to fail the build.
log.debug("Internal data failure: IOException " + e.getMessage(), e);
}
}
/**
* 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 javax.naming.NamingException {
String username = System.getProperty("user.name");
// Set up environment for creating initial context
Hashtable<String, String> env = new Hashtable<String, String>(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<SearchResult> en = ctx.search("", "uid=" + username, controls);
if (en.hasMore()) {
SearchResult sr = (SearchResult) en.next();
if (sr.getAttributes().get("mail") != null) {
String email = (String) sr.getAttributes().get("mail").get();
return email;
}
}
throw new NamingException("Could not find user email in LDAP.");
}
}