/***
    * 
    */
   package net.sf.mailservice.mailhttp;
   
   import java.io.ByteArrayOutputStream;
   import java.io.IOException;
   import java.io.InputStream;
   import java.io.OutputStream;
  import java.io.PrintStream;
  import java.net.HttpURLConnection;
  import java.net.MalformedURLException;
  import java.net.URL;
  import java.net.URLConnection;
  import java.net.URLDecoder;
  import java.util.Enumeration;
  
  import javax.mail.Address;
  import javax.mail.Header;
  import javax.mail.Message;
  import javax.mail.MessagingException;
  import javax.mail.Session;
  import javax.mail.Transport;
  import javax.mail.URLName;
  import javax.mail.event.TransportEvent;
  import javax.mail.internet.InternetAddress;
  import javax.mail.internet.MimeMessage;
  import javax.mail.internet.MimeUtility;
  
  /***
   * An HTTP protocol provider for the JavaMail API that provides access to an
   * SMTP server via HTTP or HTTPS.
   *
   * <table>
   * <tr>
   * <td>mail.http.reportsuccess</td>
   * <td>Throw a HttpTransportSuccessException on success.</td>
   * </tr>
   * <tr>
   * <td>mail.http.url</td>
   * <td>URL of the restful mail service.</td>
   * </tr>
   * </table>
   *
   * If a username and password is added to the URL, it will be added to the
   * request using basic authentication.
   *
   * Both http and https URLs are supported.
   *
   * @author drewyrr
   * @version $Id: $
   */
  public class HttpTransport extends Transport {
  
  	private PrintStream out;
  	private MimeMessage message;
  	private Address[] addresses;
  	private String name;
  	private boolean reportSuccess;
  	private URLConnection conn;
  
  	/***
  	 * Create a new transport.
  	 *
  	 * @param session
  	 *            a {@link javax.mail.Session} object.
  	 * @param urlname
  	 *            a {@link javax.mail.URLName} object.
  	 */
  	public HttpTransport(Session session, URLName urlname) {
  		super(session, urlname);
  		String name;
  		if (urlname != null) {
  			name = urlname.getProtocol();
  		} else {
  			name = "http";
  		}
  		this.name = name;
  
  		out = session.getDebugOut();
  
  		String reportSuccessString = session.getProperty("mail." + name
  				+ ".reportsuccess");
  		reportSuccess = reportSuccessString != null
  				&& reportSuccessString.equalsIgnoreCase("true");
  
  	}
  
  	/***
  	 * {@inheritDoc}
  	 *
  	 * Send the given message.
  	 * @see javax.mail.Transport#sendMessage(javax.mail.Message,
  	 *      javax.mail.Address[])
  	 */
  	@Override
  	public void sendMessage(Message message, Address[] addresses)
  			throws MessagingException {
  
 		String location;
 
 		if (!(message instanceof MimeMessage)) {
 			if (debug) {
 				out.println("DEBUG HTTP: Can only send RFC822 msgs");
 			}
 			throw new MessagingException("HTTP can only send RFC822 messages");
 		}
 		for (int i = 0; i < addresses.length; i++) {
 			if (!(addresses[i] instanceof InternetAddress)) {
 				throw new MessagingException(addresses[i]
 						+ " is not an InternetAddress");
 			}
 		}
 
 		this.message = (MimeMessage) message;
 		this.addresses = addresses;
 		Address[] empty = new Address[0];
 
 		try {
 			Enumeration<?> enumeration = message.getAllHeaders();
 			while (enumeration.hasMoreElements()) {
 				Header header = (Header) enumeration.nextElement();
 				if (debug) {
 					out.println("DEBUG HTTP: Header " + header.getName() + ":"
 							+ MimeUtility.unfold(header.getValue()));
 				}
 				conn.addRequestProperty(header.getName(), MimeUtility
 						.unfold(header.getValue()));
 			}
 
 			/*
 			 * suppress the headers by not using writeTo.
 			 */
 			InputStream is = this.message.getInputStream();
 			OutputStream os = conn.getOutputStream();
 			int c;
 			while ((c = is.read()) != -1) {
 				os.write(c);
 			}
 
 			if (debug) {
 				out.print("DEBUG HTTP: Response: ");
 			}
 			is = conn.getInputStream();
 			while ((c = is.read()) != -1) {
 				if (debug) {
 					out.write(c);
 				}
 			}
 			is.close();
 			if (debug) {
 				out.println();
 			}
 
 			location = conn.getHeaderField("Location");
 			this.message.addHeader("Location", location);
 
 			notifyTransportListeners(TransportEvent.MESSAGE_DELIVERED,
 					this.addresses, empty, empty, this.message);
 		} catch (MessagingException mex) {
 			if (debug)
 				mex.printStackTrace(out);
 			notifyTransportListeners(TransportEvent.MESSAGE_NOT_DELIVERED,
 					empty, this.addresses, empty, this.message);
 
 			throw mex;
 		} catch (IOException ioe) {
 			int response = 0;
 			if (debug)
 				ioe.printStackTrace(out);
 			try {
 				if (conn instanceof HttpURLConnection) {
 					response = ((HttpURLConnection) conn).getResponseCode();
 					InputStream es = ((HttpURLConnection) conn)
 							.getErrorStream();
 					int c;
 					while ((c = es.read()) != -1) {
 						if (debug) {
 							out.write(c);
 						}
 					}
 					es.close();
 					if (debug) {
 						out.println();
 					}
 				}
 			} catch (IOException e) {
 				/* we're doomed at this point */
 			}
 			notifyTransportListeners(TransportEvent.MESSAGE_NOT_DELIVERED,
 					empty, this.addresses, empty, this.message);
 
 			throw new MessagingException(
 					"IOException while sending message, with HTTP response code "
 							+ response, ioe);
 		} finally {
 			/* no reason to keep this data around */
 			this.addresses = null;
 			this.message = null;
 		}
 
 		if (reportSuccess) {
 			throw new HttpTransportSuccessException(location);
 		}
 
 	}
 
 	/***
 	 * {@inheritDoc}
 	 *
 	 * Set up the connection to the mail service.
 	 * @see javax.mail.Service#protocolConnect(java.lang.String, int,
 	 *      java.lang.String, java.lang.String)
 	 */
 	@Override
 	protected boolean protocolConnect(String host, int port, String user,
 			String password) throws MessagingException {
 
 		URL url;
 		String p = session.getProperty("mail." + name + ".url");
 		try {
 			if (p != null) {
 				url = new URL(p);
 			} else {
 				url = new URL(name + "://localhost/");
 			}
 		} catch (MalformedURLException e) {
 			throw new MessagingException(e.getMessage());
 		}
 
 		try {
 			conn = url.openConnection();
 			conn.setDoOutput(true);
 			if (conn instanceof HttpURLConnection) {
 				HttpURLConnection httpUrlConnection = (HttpURLConnection) conn;
 				httpUrlConnection.setInstanceFollowRedirects(false);
 			}
 
 			/*
 			 * Add basic auth from provided user and password, otherwise from
 			 * the URL, manually
 			 */
 			String authority = url.getAuthority();
 			int offset;
 			if (user != null && password != null) {
 				String userpassword = user.concat(":").concat(password);
 				ByteArrayOutputStream baos = new ByteArrayOutputStream();
 				OutputStream stream = MimeUtility.encode(baos, "base64");
 				stream.write(userpassword.getBytes());
 
 				conn.setRequestProperty("Authorization", "Basic "
 						+ baos.toString("UTF-8"));
 			} else if ((offset = authority.indexOf('@')) != -1) {
 				String userpassword = URLDecoder.decode(authority.substring(0,
 						offset), "UTF-8");
 				ByteArrayOutputStream baos = new ByteArrayOutputStream();
 				OutputStream stream = MimeUtility.encode(baos, "base64");
 				stream.write(userpassword.getBytes());
 
 				conn.setRequestProperty("Authorization", "Basic "
 						+ baos.toString("UTF-8"));
 			}
 
 		} catch (MalformedURLException e) {
 			throw new MessagingException(e.getLocalizedMessage());
 		} catch (IOException e) {
 			throw new MessagingException(e.getLocalizedMessage());
 		}
 
 		return true;
 	}
 
 }