Subversion Repositories SmartDukaan

package in.shop2020.utils;

import java.io.BufferedReader;

import java.io.File;

import java.io.FileReader;

import java.security.Security;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import java.util.Properties;

import javax.activation.DataHandler;

import javax.activation.DataSource;

import javax.activation.FileDataSource;

import javax.mail.Message;

import javax.mail.MessagingException;

import javax.mail.Multipart;

import javax.mail.PasswordAuthentication;

import javax.mail.Session;

import javax.mail.Transport;

import javax.mail.internet.AddressException;

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;

public class GmailUtils {

	private static final String SMTP_HOST_NAME = "smtp.gmail.com";

	private static final String SMTP_PORT = "465";

	private static final String SSL_FACTORY = "javax.net.ssl.SSLSocketFactory";

	public GmailUtils(){

	}

	public void sendSSLMessage(String recipients[], String subject, String message, final String from, final String password, String filename) throws MessagingException {

		try {

			int i=0;

			boolean add =false;

			for(String str:recipients){

				if(str.equalsIgnoreCase("vikram.raghav@shop2020.in") || str.equalsIgnoreCase("vikram.raghav@saholic.com")){

					add=true;

				}

				i++;

			}

			if(add){

				recipients[i]="anikendra.das@shop2020.in";

			}

			Message msg = prepareMessage(recipients, subject, from, password);

			//Create the multi-part object to hold the text and attachment parts

			Multipart multipart = new MimeMultipart();

			// create the message part 

			MimeBodyPart messageBodyPart = new MimeBodyPart();

			//Part 1: Text message

			messageBodyPart.setText(message);

			multipart.addBodyPart(messageBodyPart);

			//Part 2: Attachment

			messageBodyPart = new MimeBodyPart();

			DataSource source = new FileDataSource(filename);

			messageBodyPart.setDataHandler(new DataHandler(source));

			String[] names = filename.split("/");

			messageBodyPart.setFileName(names[names.length - 1]);

			multipart.addBodyPart(messageBodyPart);

			// Put parts in message

			msg.setContent(multipart);

			Transport.send(msg);

		} catch (Exception e) {

			// If there is any error with normal mail sending, try diving receipients across domains

			if (hasMultipleDomains(recipients)) {

				for (List<String> subReceipients : getReceipientsPerDomain(recipients).values()) {

					sendSSLMessage(subReceipients.toArray(new String[0]), subject, message, from, password, filename);

				}

				return;

			}

			throw new MessagingException(e.getMessage(), e);

		}

	}

	/**

	 * @param recipients

	 * @return

	 */

	private Map<String, List<String>> getReceipientsPerDomain(String[] recipients) {

		Map<String, List<String>> result = new HashMap<String, List<String>>();

		for (String recipient : recipients) {

			String domain = extractDomainFromEmailAddress(recipient);

			if (!result.containsKey(domain)) {

				result.put(domain, new ArrayList<String>());

			}

			result.get(domain).add(recipient);

		}

		return result;

	}

	/**

	 * @param recipients

	 * @return

	 */

	public void sendSSLMessage(String recipients[], String subject, String message, final String from, final String password, String filename,String htmlFileName) throws MessagingException {

		try {

			BufferedReader bufferedReader = new BufferedReader(new FileReader(htmlFileName));

			StringBuffer htmlContent = new StringBuffer();

			String line = null;

			while((line =bufferedReader.readLine())!=null){

				htmlContent.append(line).append("\n");

			}

			Message msg = prepareMessage(recipients, subject, from, password);

			//Create the multi-part object to hold the text and attachment parts

			Multipart multipart = new MimeMultipart();

			MimeBodyPart htmlPart = new MimeBodyPart();

			htmlPart.setContent(htmlContent.toString(), "text/html; charset=utf-8");

			// create the message part

			MimeBodyPart messageBodyPart = new MimeBodyPart();

			//Part 1: Text message

			messageBodyPart.setText(message);

			multipart.addBodyPart(htmlPart);

			//Part 2: Attachment

			messageBodyPart = new MimeBodyPart();

			DataSource source = new FileDataSource(filename);

			messageBodyPart.setDataHandler(new DataHandler(source));

			String[] names = filename.split("/");

			messageBodyPart.setFileName(names[names.length - 1]);

			multipart.addBodyPart(messageBodyPart);

			// Put parts in message

			msg.setContent(multipart);

			Transport.send(msg);

		} catch (Exception e) {

			// If there is any error with normal mail sending, try diving receipients across domains

			if (hasMultipleDomains(recipients)) {

				for (List<String> subReceipients : getReceipientsPerDomain(recipients).values()) {

					sendSSLMessage(subReceipients.toArray(new String[0]), subject, message, from, password, filename);

				}

				return;

			}

			throw new MessagingException(e.getMessage(), e);

		}

	}

	private boolean hasMultipleDomains(String[] recipients) {

		String domain = extractDomainFromEmailAddress(recipients[0]);

		for (String recipient : recipients) {

			if (!domain.equals(extractDomainFromEmailAddress(recipient))) {

				return true;

			}

		}

		return false;

	}

	private String extractDomainFromEmailAddress(String recipient) {

		return recipient.split("@")[1];

	}

	public void sendSSLMessage(String recipients[], String subject, String message, final String from, final String password, List<File> files) throws MessagingException {

		try {

			Message msg = prepareMessage(recipients, subject, from, password);

			//Create the multi-part object to hold the text and attachment parts

			Multipart multipart = new MimeMultipart();

			// create the message part 

			MimeBodyPart messageBodyPart = new MimeBodyPart();

			//Part 1: Text message

			messageBodyPart.setText(message);

			multipart.addBodyPart(messageBodyPart);

			//Part 2: Attachment

			for (File file : files) {

				DataSource source = new FileDataSource(file);

				String[] names = file.getAbsolutePath().split("/");

				messageBodyPart = new MimeBodyPart();

				messageBodyPart.setDataHandler(new DataHandler(source));

				messageBodyPart.setFileName(names[names.length - 1]);

				multipart.addBodyPart(messageBodyPart);

			}

			// Put parts in message

			msg.setContent(multipart);

			Transport.send(msg);

		} catch (Exception e) {

			// If there is any error with normal mail sending, try diving receipients across domains

			if (hasMultipleDomains(recipients)) {

				for (List<String> subReceipients : getReceipientsPerDomain(recipients).values()) {

					sendSSLMessage(subReceipients.toArray(new String[0]), subject, message, from, password, files);

				}

				return;

			}

			throw new MessagingException(e.getMessage(), e);

		}

	}

	@SuppressWarnings("restriction")

	private Message prepareMessage(String[] recipients, String subject,

			final String from, final String password) throws AddressException,

			MessagingException {

		boolean debug = true;

		Properties props = new Properties();

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

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

		props.put("mail.debug", "true");

		props.put("mail.smtp.port", SMTP_PORT);

		props.put("mail.smtp.socketFactory.port", SMTP_PORT);

		props.put("mail.smtp.socketFactory.class", SSL_FACTORY);

		props.put("mail.smtp.socketFactory.fallback", "false");

		Security.addProvider(new com.sun.net.ssl.internal.ssl.Provider());

		Session session = Session.getDefaultInstance(props,

				new javax.mail.Authenticator() {

			protected PasswordAuthentication getPasswordAuthentication() {

				return new PasswordAuthentication(from, password);

			}

		});

		session.setDebug(debug);

		Message msg = new MimeMessage(session);

		//Set the from address

		InternetAddress addressFrom = new InternetAddress(from);

		msg.setFrom(addressFrom);

		//Set the recipients

		InternetAddress[] addressTo = new InternetAddress[recipients.length];

		for (int i = 0; i < recipients.length; i++) {

			addressTo[i] = new InternetAddress(recipients[i]);

		}

		msg.setRecipients(Message.RecipientType.TO, addressTo);

		//Setting the Subject

		msg.setSubject(subject);

		return msg;

	}

	public void sendSSLMessage(String recipients[], String subject,final String from, final String password,String htmlContent) throws MessagingException {

		try {

			Message msg = prepareMessage(recipients, subject, from, password);

			//Create the multi-part object to hold the text and attachment parts

			Multipart multipart = new MimeMultipart();

			MimeBodyPart htmlPart = new MimeBodyPart();

			htmlPart.setContent(htmlContent, "text/html; charset=utf-8");

			multipart.addBodyPart(htmlPart);

			msg.setContent(multipart);

			Transport.send(msg);

		} catch (Exception e) {

			// If there is any error with normal mail sending, try diving receipients across domains

			if (hasMultipleDomains(recipients)) {

				for (List<String> subReceipients : getReceipientsPerDomain(recipients).values()) {

					sendSSLMessage(subReceipients.toArray(new String[0]), subject, "", from, password, "");

				}

				return;

			}

			throw new MessagingException(e.getMessage(), e);

		}

	}

	public static void main(String args[]) throws Exception {

		String[] sendTo = { "mandeep.dhir@shop2020.in", "mandeep.dhir@gmail.com", "mandeepcse07@yahoo.co.in", "kshitij.sood@shop2020.in" };

		String emailSubjectTxt = "Another test from gmail";

		String emailMsgTxt = "Test Message Contents";

		String emailFromAddress = "mandeep.dhir@shop2020.in";

		String password = "";

		String filename = "/home/mandeep/virtual_warehouse.sql";

		GmailUtils utils = new GmailUtils();

		utils.sendSSLMessage(sendTo, emailSubjectTxt, emailMsgTxt, emailFromAddress, password, filename);

		System.out.println("Sucessfully Sent mail to All Users");

	}

}