buildframework/helium/tools/common/java/src/com/nokia/ant/listener/internaldata/EmailDataSender.java
changeset 1 be27ed110b50
child 179 d8ac696cc51f
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/buildframework/helium/tools/common/java/src/com/nokia/ant/listener/internaldata/EmailDataSender.java	Wed Oct 28 14:39:48 2009 +0000
@@ -0,0 +1,156 @@
+/*
+* 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.");
+    }
+
+}