/*

  *   SONEWS News Server

  *   see AUTHORS for the list of contributors

  *

  *   This program is free software: you can redistribute it and/or modify

  *   it under the terms of the GNU General Public License as published by

  *   the Free Software Foundation, either version 3 of the License, or

  *   (at your option) any later version.

  *

  *   This program is distributed in the hope that it will be useful,

  *   but WITHOUT ANY WARRANTY; without even the implied warranty of

  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

  *   GNU General Public License for more details.

  *

  *   You should have received a copy of the GNU General Public License

  *   along with this program.  If not, see <http://www.gnu.org/licenses/>.

  */

 package org.sonews.storage;

 import java.io.ByteArrayOutputStream;

 import java.io.IOException;

 import java.io.StringReader;

 import java.io.StringWriter;

 import java.io.UnsupportedEncodingException;

 import java.sql.ResultSet;

 import java.sql.SQLException;

 import java.util.ArrayList;

 import java.util.Date;

 import java.util.Enumeration;

 import java.util.logging.Level;

 import java.util.logging.Logger;

 import javax.mail.Header;

 import javax.mail.MessagingException;

 import javax.mail.Multipart;

 import javax.mail.Session;

 import javax.mail.internet.InternetAddress;

 import javax.mail.internet.MimeBodyPart;

 import javax.mail.internet.MimeMessage;

 import javax.mail.internet.MimeMultipart;

 import javax.xml.transform.Transformer;

 import javax.xml.transform.TransformerFactory;

 import javax.xml.transform.stream.StreamResult;

 import javax.xml.transform.stream.StreamSource;

 import org.sonews.util.io.Resource;

 /**

  * This is MimeMessage which enables custom Message-ID header

  * (this header will not be overwritten by the default one like in MimeMessage).

  * 

  * Also add header and body separate serialization.

  * 

  * And can be deserialized from SQL ResultSet

  * 

  * @author František Kučera (frantovo.cz)

  */

 public class DrupalMessage extends MimeMessage {

 	private static final Logger log = Logger.getLogger(DrupalMessage.class.getName());

 	private static final String MESSAGE_ID_HEADER = "Message-ID";

 	private static final String CRLF = "\r\n";

 	public static final String CHARSET = "UTF-8";

 	private static final String XHTML_CONTENT_TYPE = "text/html; charset=" + CHARSET;

 	private String messageID;

 	/**

 	 * Constructs MIME message from SQL result.

 	 * @param rs ResultSet containing message data. No {@link ResultSet#next()} will be called, just values from current row will be read.

 	 * @param constructBody true if whole message should be constructed | false if we need only message headers (body will be dummy).

 	 */

 	public DrupalMessage(ResultSet rs, String myDomain, boolean constructBody) throws SQLException, UnsupportedEncodingException, MessagingException {

 		super(Session.getDefaultInstance(System.getProperties()));

 		addHeader("Message-id", constructMessageId(rs.getInt("id"), rs.getInt("group_id"), rs.getString("group_name"), myDomain));

 		addHeader("Newsgroups", rs.getString("group_name"));

 		setFrom(new InternetAddress(rs.getString("sender_email"), rs.getString("sender_name")));

 		setSubject(rs.getString("subject"));

 		setSentDate(new Date(rs.getLong("created")));

 		int parentID = rs.getInt("parent_id");

 		if (parentID > 0) {

 			String parentMessageID = constructMessageId(parentID, rs.getInt("group_id"), rs.getString("group_name"), myDomain);

 			addHeader("In-Reply-To", parentMessageID);

 			addHeader("References", parentMessageID);

 		}

 		if (constructBody) {

 			Multipart multipart = new MimeMultipart("alternative");

 			setContent(multipart);

 			/** Plain text part */

 			MimeBodyPart textPart = new MimeBodyPart();

 			multipart.addBodyPart(textPart);

 			textPart.setText(readPlainText(rs));

 			/** XHTML part */

 			MimeBodyPart htmlPart = new MimeBodyPart();

 			multipart.addBodyPart(htmlPart);

 			htmlPart.setContent(readXhtmlText(rs), XHTML_CONTENT_TYPE);

 		} else {

 			setText("");

 		}

 	}

 	private String readPlainText(ResultSet rs) {

 		/**

 		 * TODO: převést na prostý text

 		 */

 		return "TODO: obyčejný text";

 	}

 	private String readXhtmlText(ResultSet rs) {

 		/**

 		 * TODO: znovupoužívat XSL transformér

 		 */

 		try {

 			String originalText = rs.getString("text");

 			StringReader input = new StringReader("<body>" + originalText + "</body>");

 			StringWriter output = new StringWriter();

 			TransformerFactory tf = TransformerFactory.newInstance();

 			Transformer t = tf.newTransformer(new StreamSource(Resource.getAsStream("helpers/mimeXhtmlPart.xsl")));

 			t.setParameter("isRoot", (rs.getInt("parent_id") == 0));

 			t.setParameter("title", rs.getString("subject"));

 			t.setParameter("urlBase", rs.getString("urlBase"));

 			t.setParameter("wwwRead", rs.getString("wwwRead"));

 			t.setParameter("wwwPost", rs.getString("wwwPost"));

 			t.transform(new StreamSource(input), new StreamResult(output));

 			return output.toString();

 		} catch (Exception e) {

 			/**

 			 * TODO: lepší ošetření chyby

 			 */

 			log.log(Level.WARNING, "Error while transforming article to XHTML", e);

 			return "<html><body><p>Při transformaci příspěvku bohužel došlo k chybě.</p></body></html>";

 		}

 	}

 	private static String constructMessageId(int articleID, int groupID, String groupName, String domainName) {

 		StringBuilder sb = new StringBuilder();

 		sb.append("<");

 		sb.append(articleID);

 		sb.append("-");

 		sb.append(groupID);

 		sb.append("-");

 		sb.append(groupName);

 		sb.append("@");

 		sb.append(domainName);

 		sb.append(">");

 		return sb.toString();

 	}

 	@Override

 	public void setHeader(String name, String value) throws MessagingException {

 		super.setHeader(name, value);

 		if (MESSAGE_ID_HEADER.equalsIgnoreCase(name)) {

 			messageID = value;

 		}

 	}

 	@Override

 	public final void addHeader(String name, String value) throws MessagingException {

 		super.addHeader(name, value);

 		if (MESSAGE_ID_HEADER.equalsIgnoreCase(name)) {

 			messageID = value;

 		}

 	}

 	@Override

 	public void removeHeader(String name) throws MessagingException {

 		super.removeHeader(name);

 		if (MESSAGE_ID_HEADER.equalsIgnoreCase(name)) {

 			messageID = null;

 		}

 	}

 	public void setMessageID(String messageID) {

 		this.messageID = messageID;

 	}

 	@Override

 	protected void updateMessageID() throws MessagingException {

 		if (messageID == null) {

 			super.updateMessageID();

 		} else {

 			setHeader(MESSAGE_ID_HEADER, messageID);

 		}

 	}

 	/**

 	 * Call {@link #saveChanges()} before this method, if you want all headers including such ones like:

 	 * 

 	 * <pre>MIME-Version: 1.0

 	 *Content-Type: multipart/alternative;</pre>

 	 * 

 	 * @return serialized headers

 	 * @throws MessagingException if getAllHeaders() fails

 	 */

 	public String getHeaders() throws MessagingException {

 		StringBuilder sb = new StringBuilder();

 		for (Enumeration eh = getAllHeaderLines(); eh.hasMoreElements();) {

 			sb.append(eh.nextElement());

 			sb.append(CRLF);

 		}

 		return sb.toString();

 	}

 	public byte[] getBody() throws IOException, MessagingException {

 		saveChanges();

 		ArrayList<String> skipHeaders = new ArrayList<String>();

 		for (Enumeration eh = getAllHeaders(); eh.hasMoreElements();) {

 			Header h = (Header) eh.nextElement();

 			skipHeaders.add(h.getName());

 		}

 		ByteArrayOutputStream baos = new ByteArrayOutputStream(1024);

 		writeTo(baos, skipHeaders.toArray(new String[skipHeaders.size()]));

 		return baos.toByteArray();

 	}

 }