src/org/sonews/storage/DrupalMessage.java
author František Kučera <franta-hg@frantovo.cz>
Thu, 20 Oct 2011 10:50:58 +0200
changeset 103 a788bf0e1080
parent 102 d843b4fee5dc
child 104 b4c8a2760d6f
permissions -rw-r--r--
Drupal: zpráva od uživatele se před uložením prožene přes XSLT případně Tidy.
franta-hg@72
     1
/*
franta-hg@72
     2
 *   SONEWS News Server
franta-hg@72
     3
 *   see AUTHORS for the list of contributors
franta-hg@72
     4
 *
franta-hg@72
     5
 *   This program is free software: you can redistribute it and/or modify
franta-hg@72
     6
 *   it under the terms of the GNU General Public License as published by
franta-hg@72
     7
 *   the Free Software Foundation, either version 3 of the License, or
franta-hg@72
     8
 *   (at your option) any later version.
franta-hg@72
     9
 *
franta-hg@72
    10
 *   This program is distributed in the hope that it will be useful,
franta-hg@72
    11
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
franta-hg@72
    12
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
franta-hg@72
    13
 *   GNU General Public License for more details.
franta-hg@72
    14
 *
franta-hg@72
    15
 *   You should have received a copy of the GNU General Public License
franta-hg@72
    16
 *   along with this program.  If not, see <http://www.gnu.org/licenses/>.
franta-hg@72
    17
 */
franta-hg@72
    18
package org.sonews.storage;
franta-hg@72
    19
franta-hg@75
    20
import java.io.BufferedReader;
franta-hg@102
    21
import java.io.ByteArrayInputStream;
franta-hg@72
    22
import java.io.ByteArrayOutputStream;
franta-hg@72
    23
import java.io.IOException;
franta-hg@75
    24
import java.io.InputStream;
franta-hg@75
    25
import java.io.InputStreamReader;
franta-hg@75
    26
import java.io.PrintStream;
franta-hg@74
    27
import java.io.StringReader;
franta-hg@74
    28
import java.io.StringWriter;
franta-hg@72
    29
import java.io.UnsupportedEncodingException;
franta-hg@72
    30
import java.sql.ResultSet;
franta-hg@72
    31
import java.sql.SQLException;
franta-hg@72
    32
import java.util.ArrayList;
franta-hg@72
    33
import java.util.Date;
franta-hg@72
    34
import java.util.Enumeration;
franta-hg@74
    35
import java.util.logging.Level;
franta-hg@74
    36
import java.util.logging.Logger;
franta-hg@72
    37
import javax.mail.Header;
franta-hg@72
    38
import javax.mail.MessagingException;
franta-hg@72
    39
import javax.mail.Multipart;
franta-hg@72
    40
import javax.mail.Session;
franta-hg@72
    41
import javax.mail.internet.InternetAddress;
franta-hg@72
    42
import javax.mail.internet.MimeBodyPart;
franta-hg@72
    43
import javax.mail.internet.MimeMessage;
franta-hg@72
    44
import javax.mail.internet.MimeMultipart;
franta-hg@74
    45
import javax.xml.transform.Transformer;
franta-hg@103
    46
import javax.xml.transform.TransformerException;
franta-hg@74
    47
import javax.xml.transform.TransformerFactory;
franta-hg@74
    48
import javax.xml.transform.stream.StreamResult;
franta-hg@74
    49
import javax.xml.transform.stream.StreamSource;
franta-hg@102
    50
import org.sonews.daemon.NNTPConnection;
franta-hg@74
    51
import org.sonews.util.io.Resource;
franta-hg@72
    52
franta-hg@72
    53
/**
franta-hg@72
    54
 * This is MimeMessage which enables custom Message-ID header
franta-hg@72
    55
 * (this header will not be overwritten by the default one like in MimeMessage).
franta-hg@72
    56
 * 
franta-hg@72
    57
 * Also add header and body separate serialization.
franta-hg@72
    58
 * 
franta-hg@72
    59
 * And can be deserialized from SQL ResultSet
franta-hg@72
    60
 * 
franta-hg@72
    61
 * @author František Kučera (frantovo.cz)
franta-hg@72
    62
 */
franta-hg@72
    63
public class DrupalMessage extends MimeMessage {
franta-hg@72
    64
franta-hg@74
    65
	private static final Logger log = Logger.getLogger(DrupalMessage.class.getName());
franta-hg@72
    66
	private static final String MESSAGE_ID_HEADER = "Message-ID";
franta-hg@72
    67
	private static final String CRLF = "\r\n";
franta-hg@72
    68
	public static final String CHARSET = "UTF-8";
franta-hg@72
    69
	private static final String XHTML_CONTENT_TYPE = "text/html; charset=" + CHARSET;
franta-hg@100
    70
	private static final String ZNAKČKA_KONCE_ŘÁDKU = "◆";
franta-hg@72
    71
	private String messageID;
franta-hg@102
    72
	private Long parentID;
franta-hg@102
    73
	private Long groupID;
franta-hg@72
    74
franta-hg@72
    75
	/**
franta-hg@72
    76
	 * Constructs MIME message from SQL result.
franta-hg@72
    77
	 * @param rs ResultSet containing message data. No {@link ResultSet#next()} will be called, just values from current row will be read.
franta-hg@72
    78
	 * @param constructBody true if whole message should be constructed | false if we need only message headers (body will be dummy).
franta-hg@72
    79
	 */
franta-hg@103
    80
	public DrupalMessage(ResultSet rs, String myDomain, boolean constructBody) throws SQLException, UnsupportedEncodingException, MessagingException, TransformerException, IOException {
franta-hg@72
    81
		super(Session.getDefaultInstance(System.getProperties()));
franta-hg@72
    82
franta-hg@102
    83
		groupID = rs.getLong("group_id");
franta-hg@102
    84
		addHeader("Message-id", constructMessageId(rs.getInt("id"), groupID, rs.getString("group_name"), myDomain));
franta-hg@72
    85
		addHeader("Newsgroups", rs.getString("group_name"));
franta-hg@74
    86
		setFrom(new InternetAddress(rs.getString("sender_email"), rs.getString("sender_name")));
franta-hg@72
    87
		setSubject(rs.getString("subject"));
franta-hg@72
    88
		setSentDate(new Date(rs.getLong("created")));
franta-hg@74
    89
franta-hg@102
    90
		parentID = rs.getLong("parent_id");
franta-hg@74
    91
		if (parentID > 0) {
franta-hg@72
    92
			String parentMessageID = constructMessageId(parentID, rs.getInt("group_id"), rs.getString("group_name"), myDomain);
franta-hg@72
    93
			addHeader("In-Reply-To", parentMessageID);
franta-hg@72
    94
			addHeader("References", parentMessageID);
franta-hg@72
    95
		}
franta-hg@72
    96
franta-hg@72
    97
		if (constructBody) {
franta-hg@72
    98
			Multipart multipart = new MimeMultipart("alternative");
franta-hg@72
    99
			setContent(multipart);
franta-hg@72
   100
franta-hg@82
   101
			/** XHTML part */
franta-hg@82
   102
			MimeBodyPart htmlPart = new MimeBodyPart();
franta-hg@103
   103
			String xhtmlText = readXhtmlText(
franta-hg@103
   104
					rs.getString("text"),
franta-hg@103
   105
					rs.getString("subject"),
franta-hg@103
   106
					rs.getInt("parent_id"),
franta-hg@103
   107
					rs.getString("urlBase"),
franta-hg@103
   108
					rs.getString("wwwRead"),
franta-hg@103
   109
					rs.getString("wwwPost"));
franta-hg@82
   110
			htmlPart.setContent(xhtmlText, XHTML_CONTENT_TYPE);
franta-hg@84
   111
franta-hg@74
   112
			/** Plain text part */
franta-hg@72
   113
			MimeBodyPart textPart = new MimeBodyPart();
franta-hg@89
   114
			String plainText = readPlainText(rs, xhtmlText);
franta-hg@89
   115
			textPart.setText(plainText);
franta-hg@89
   116
			//addHeader("Lines", String.valueOf(plainText.split("\n").length));
franta-hg@87
   117
franta-hg@87
   118
			/**
franta-hg@87
   119
			 * Thunderbirdu záleží, v jakém pořadí části jsou 
franta-hg@87
   120
			 * (když je prostý text druhý, html se nezobrazí),
franta-hg@87
   121
			 * KNode zobrazuje HTML správně, i když je na prvním místě.
franta-hg@87
   122
			 */
franta-hg@72
   123
			multipart.addBodyPart(textPart);
franta-hg@87
   124
			multipart.addBodyPart(htmlPart);
franta-hg@72
   125
		} else {
franta-hg@82
   126
			/** empty body, just headers */
franta-hg@72
   127
			setText("");
franta-hg@72
   128
		}
franta-hg@72
   129
	}
franta-hg@72
   130
franta-hg@102
   131
	/**
franta-hg@102
   132
	 * Constructs MIME message from article posted by user.
franta-hg@102
   133
	 * @param article article that came through NNTP.
franta-hg@102
   134
	 * @throws MessagingException 
franta-hg@102
   135
	 */
franta-hg@102
   136
	public DrupalMessage(Article article) throws MessagingException {
franta-hg@102
   137
		super(Session.getDefaultInstance(System.getProperties()), serializeArticle(article));
franta-hg@102
   138
franta-hg@102
   139
		String[] parentHeaders = getHeader("In-Reply-To");
franta-hg@102
   140
		if (parentHeaders.length == 1) {
franta-hg@102
   141
			String parentMessageID = parentHeaders[0];
franta-hg@102
   142
			parentID = parseArticleID(parentMessageID);
franta-hg@102
   143
			groupID = parseGroupID(parentMessageID);
franta-hg@102
   144
		} else {
franta-hg@102
   145
			throw new MessagingException("Message posted by user must have exactly one In-Reply-To header.");
franta-hg@102
   146
		}
franta-hg@102
   147
	}
franta-hg@102
   148
franta-hg@102
   149
	private static InputStream serializeArticle(Article a) {
franta-hg@102
   150
		byte articleHeaders[] = a.getHeaderSource().getBytes();
franta-hg@102
   151
		byte delimiter[] = (NNTPConnection.NEWLINE + NNTPConnection.NEWLINE).getBytes();
franta-hg@102
   152
		byte body[] = a.getBody();
franta-hg@102
   153
franta-hg@102
   154
		byte message[] = new byte[articleHeaders.length + delimiter.length + body.length];
franta-hg@102
   155
franta-hg@102
   156
		System.arraycopy(articleHeaders, 0, message, 0, articleHeaders.length);
franta-hg@102
   157
		System.arraycopy(delimiter, 0, message, articleHeaders.length, delimiter.length);
franta-hg@102
   158
		System.arraycopy(body, 0, message, articleHeaders.length + delimiter.length, body.length);
franta-hg@102
   159
franta-hg@102
   160
		return new ByteArrayInputStream(message);
franta-hg@102
   161
	}
franta-hg@102
   162
franta-hg@82
   163
	private String readPlainText(ResultSet rs, String xhtmlText) {
franta-hg@89
   164
		try {
franta-hg@89
   165
			TransformerFactory tf = TransformerFactory.newInstance();
franta-hg@89
   166
			Transformer textTransformer = tf.newTransformer(new StreamSource(Resource.getAsStream("helpers/mimeTextPart.xsl")));
franta-hg@89
   167
franta-hg@89
   168
			StringReader input = new StringReader(xhtmlText);
franta-hg@89
   169
			StringWriter output = new StringWriter(xhtmlText.length());
franta-hg@89
   170
			textTransformer.transform(new StreamSource(input), new StreamResult(output));
franta-hg@89
   171
franta-hg@89
   172
			return output.toString();
franta-hg@89
   173
		} catch (Exception e) {
franta-hg@89
   174
			/**
franta-hg@89
   175
			 * TODO: lepší ošetření chyby
franta-hg@89
   176
			 */
franta-hg@89
   177
			log.log(Level.WARNING, "Error while transforming article to plain text", e);
franta-hg@89
   178
			return makeSimpleXHTML("Při transformaci příspěvku bohužel došlo k chybě.");
franta-hg@89
   179
		}
franta-hg@72
   180
	}
franta-hg@72
   181
franta-hg@103
   182
	private String readXhtmlText(String text, String subject, long parentId, String urlBase, String wwwRead, String wwwPost) throws TransformerException, IOException {
franta-hg@72
   183
		/**
franta-hg@82
   184
		 * TODO: 
franta-hg@82
   185
		 *		- znovupoužívat XSL transformér
franta-hg@82
   186
		 *		- používat cache, ukládat si vygenerované články
franta-hg@72
   187
		 */
franta-hg@103
   188
		String inputText = makeSimpleXHTML(text);
franta-hg@103
   189
franta-hg@103
   190
		TransformerFactory tf = TransformerFactory.newInstance();
franta-hg@103
   191
		Transformer paragraphTransformer = tf.newTransformer(new StreamSource(Resource.getAsStream("helpers/mimeXhtmlPart-make-paragraphs.xsl")));
franta-hg@103
   192
franta-hg@103
   193
		String paragraphedText;
franta-hg@103
   194
		boolean tidyWasUsed = false;
franta-hg@74
   195
		try {
franta-hg@103
   196
			StringReader input = new StringReader(inputText);
franta-hg@103
   197
			StringWriter output = new StringWriter(2 * inputText.length());
franta-hg@103
   198
			paragraphTransformer.transform(new StreamSource(input), new StreamResult(output));
franta-hg@103
   199
			paragraphedText = output.toString();
franta-hg@103
   200
		} catch (Exception e) {
franta-hg@103
   201
			log.log(Level.FINER, "HTML input was shitty – Tidy had to be called.", e);
franta-hg@103
   202
			StringReader input = new StringReader(tidyXhtml(inputText));
franta-hg@103
   203
			StringWriter output = new StringWriter(2 * inputText.length());
franta-hg@103
   204
			paragraphTransformer.transform(new StreamSource(input), new StreamResult(output));
franta-hg@103
   205
			paragraphedText = output.toString();
franta-hg@103
   206
			tidyWasUsed = true;
franta-hg@103
   207
		}
franta-hg@75
   208
franta-hg@103
   209
		Transformer xhtmlTransformer = tf.newTransformer(new StreamSource(Resource.getAsStream("helpers/mimeXhtmlPart.xsl")));
franta-hg@103
   210
		xhtmlTransformer.setParameter("isRoot", (parentId == 0));
franta-hg@103
   211
		xhtmlTransformer.setParameter("title", subject);
franta-hg@103
   212
		xhtmlTransformer.setParameter("urlBase", urlBase);
franta-hg@103
   213
		xhtmlTransformer.setParameter("wwwRead", wwwRead);
franta-hg@103
   214
		xhtmlTransformer.setParameter("wwwPost", wwwPost);
franta-hg@103
   215
		xhtmlTransformer.setParameter("headComment", String.format("Drupal-NNTP bridge. Transformed: %1$tc. Tidy had to be used: %2$b", new Date(), tidyWasUsed));
franta-hg@103
   216
		StringReader input = new StringReader(paragraphedText);
franta-hg@103
   217
		StringWriter output = new StringWriter(2 * paragraphedText.length());
franta-hg@103
   218
		xhtmlTransformer.transform(new StreamSource(input), new StreamResult(output));
franta-hg@75
   219
franta-hg@103
   220
		return output.toString();
franta-hg@72
   221
	}
franta-hg@72
   222
franta-hg@103
   223
	/**
franta-hg@103
   224
	 * Does not parse XML works just with text.
franta-hg@103
   225
	 * @param body XHTML fragment that should be put between &lt;body&gt; and &lt;/body&gt;
franta-hg@103
   226
	 * @return simple XHTML document (body wrapped in html and body tags)
franta-hg@103
   227
	 */
franta-hg@84
   228
	private static String makeSimpleXHTML(String body) {
franta-hg@84
   229
		return "<html xmlns=\"http://www.w3.org/1999/xhtml\"><body>" + body + "</body></html>";
franta-hg@84
   230
	}
franta-hg@84
   231
franta-hg@75
   232
	/**
franta-hg@103
   233
	 * Does not parse XML works just with text.
franta-hg@103
   234
	 * @param xhtml whole XHTML page
franta-hg@103
   235
	 * @return content between &lt;body&gt; and &lt;/body&gt; tags.
franta-hg@103
   236
	 */
franta-hg@103
   237
	private static String makeFragmentXHTML(String xhtml) {
franta-hg@103
   238
		final String startTag = "<body>";
franta-hg@103
   239
		final String endTag = "</body>";
franta-hg@103
   240
franta-hg@103
   241
		int start = xhtml.indexOf(startTag) + startTag.length();
franta-hg@103
   242
		int end = xhtml.lastIndexOf(endTag);
franta-hg@103
   243
franta-hg@103
   244
		return xhtml.substring(start, end);
franta-hg@103
   245
	}
franta-hg@103
   246
franta-hg@103
   247
	/**
franta-hg@75
   248
	 * TODO: refaktorovat, přesunout
franta-hg@75
   249
	 */
franta-hg@75
   250
	private static String tidyXhtml(String inputText) throws IOException {
franta-hg@89
   251
		/*
franta-hg@89
   252
		 * Viz https://sourceforge.net/tracker/index.php?func=detail&aid=3424437&group_id=27659&atid=390966
franta-hg@89
   253
		 *
franta-hg@89
   254
		 * TODO:
franta-hg@89
   255
		 *		- použít delší zástupný řetězec, ne jen jeden znak
franta-hg@89
   256
		 *		- umísťovat ho jen tam, kde už nějaký text je (ne mezi >\s*<)
franta-hg@89
   257
		 */
franta-hg@100
   258
		inputText = označKonceŘádků(inputText);
franta-hg@82
   259
franta-hg@75
   260
		Runtime r = Runtime.getRuntime();
franta-hg@82
   261
		Process p = r.exec(new String[]{"tidy", // http://tidy.sourceforge.net
franta-hg@82
   262
					"-asxml", // well formed XHTML
franta-hg@82
   263
					"-numeric", // číselné entity
franta-hg@82
   264
					"-utf8", // kódování
franta-hg@82
   265
					"--show-warnings", "false", // žádná varování nás nezajímají
franta-hg@82
   266
					"--show-errors", "0", // ani chyby
franta-hg@82
   267
					"--doctype", "omit", // doctype nepotřebujeme (doplníme si případně vlastní v XSLT)
franta-hg@82
   268
					"--logical-emphasis", "true", // em a strong místo i a b
franta-hg@82
   269
					"--literal-attributes", "true", // zachovat mezery a konce řádků v atributech
franta-hg@82
   270
					"--force-output", "true" // neznámé značky zahodíme, vložíme jen jejich obsah
franta-hg@82
   271
				});
franta-hg@75
   272
franta-hg@75
   273
		PrintStream vstupProcesu = new PrintStream(p.getOutputStream());
franta-hg@75
   274
		vstupProcesu.print(inputText);
franta-hg@75
   275
		vstupProcesu.close();
franta-hg@75
   276
franta-hg@75
   277
		String outputText = streamToString(p.getInputStream());
franta-hg@75
   278
franta-hg@100
   279
		outputText = vraťKonceŘádků(outputText);
franta-hg@82
   280
franta-hg@75
   281
		return outputText;
franta-hg@75
   282
	}
franta-hg@75
   283
franta-hg@100
   284
	private static String označKonceŘádků(String text) {
franta-hg@100
   285
		text = text.replaceAll(">\\s+<", "> <");
franta-hg@100
   286
		text = text.replaceAll("\\n", ZNAKČKA_KONCE_ŘÁDKU + "\n");
franta-hg@100
   287
		return text;
franta-hg@100
   288
	}
franta-hg@100
   289
franta-hg@100
   290
	private static String vraťKonceŘádků(String text) {
franta-hg@100
   291
		text = text.replaceAll(ZNAKČKA_KONCE_ŘÁDKU + "\\n", "\n");
franta-hg@100
   292
		text = text.replaceAll(ZNAKČKA_KONCE_ŘÁDKU, "\n");
franta-hg@100
   293
		return text;
franta-hg@100
   294
	}
franta-hg@100
   295
franta-hg@75
   296
	/**
franta-hg@75
   297
	 * TODO: refaktorovat, přesunout
franta-hg@75
   298
	 */
franta-hg@75
   299
	private static String streamToString(InputStream proud) throws IOException {
franta-hg@75
   300
		StringBuilder výsledek = new StringBuilder();
franta-hg@75
   301
		BufferedReader buf = new BufferedReader(new InputStreamReader(proud));
franta-hg@75
   302
		while (true) {
franta-hg@75
   303
			String radek = buf.readLine();
franta-hg@75
   304
			if (radek == null) {
franta-hg@75
   305
				break;
franta-hg@75
   306
			} else {
franta-hg@75
   307
				výsledek.append(radek);
franta-hg@75
   308
				výsledek.append("\n");
franta-hg@75
   309
			}
franta-hg@75
   310
		}
franta-hg@75
   311
		return výsledek.toString();
franta-hg@75
   312
	}
franta-hg@75
   313
franta-hg@102
   314
	public static String constructMessageId(long articleID, long groupID, String groupName, String domainName) {
franta-hg@72
   315
		StringBuilder sb = new StringBuilder();
franta-hg@72
   316
		sb.append("<");
franta-hg@72
   317
		sb.append(articleID);
franta-hg@72
   318
		sb.append("-");
franta-hg@72
   319
		sb.append(groupID);
franta-hg@72
   320
		sb.append("-");
franta-hg@72
   321
		sb.append(groupName);
franta-hg@72
   322
		sb.append("@");
franta-hg@72
   323
		sb.append(domainName);
franta-hg@72
   324
		sb.append(">");
franta-hg@72
   325
		return sb.toString();
franta-hg@72
   326
	}
franta-hg@72
   327
franta-hg@102
   328
	/**
franta-hg@102
   329
	 * @return article ID of parent of this message | or null, if this is root article and not reply to another one
franta-hg@102
   330
	 */
franta-hg@102
   331
	public Long getParentID() {
franta-hg@102
   332
		return parentID;
franta-hg@102
   333
	}
franta-hg@102
   334
franta-hg@102
   335
	/**
franta-hg@102
   336
	 * @return group ID of this message | or null, if this message is not reply to any other one – which is wrong because we have to know the group
franta-hg@102
   337
	 */
franta-hg@102
   338
	public Long getGroupID() {
franta-hg@102
   339
		return groupID;
franta-hg@102
   340
	}
franta-hg@102
   341
franta-hg@102
   342
	/**
franta-hg@102
   343
	 * 
franta-hg@102
   344
	 * @param messageID &lt;{0}-{1}-{2}@domain.tld&gt; where {0} is nntp_id and {1} is group_id and {2} is group_name
franta-hg@102
   345
	 * @return array where [0] = nntp_id and [1] = group_id and [2] = group_name or returns null if messageID is invalid
franta-hg@102
   346
	 */
franta-hg@102
   347
	private static String[] parseMessageID(String messageID) {
franta-hg@102
   348
		if (messageID.matches("<[0-9]+\\-[0-9]+\\-[a-z0-9\\.]+@.+>")) {
franta-hg@102
   349
			return messageID.substring(1).split("@")[0].split("\\-");
franta-hg@102
   350
		} else {
franta-hg@102
   351
			return null;
franta-hg@102
   352
		}
franta-hg@102
   353
	}
franta-hg@102
   354
franta-hg@102
   355
	public static Long parseArticleID(String messageID) {
franta-hg@102
   356
		String[] localPart = parseMessageID(messageID);
franta-hg@102
   357
		if (localPart == null) {
franta-hg@102
   358
			return null;
franta-hg@102
   359
		} else {
franta-hg@102
   360
			return Long.parseLong(localPart[0]);
franta-hg@102
   361
		}
franta-hg@102
   362
	}
franta-hg@102
   363
franta-hg@102
   364
	public static Long parseGroupID(String messageID) {
franta-hg@102
   365
		String[] localPart = parseMessageID(messageID);
franta-hg@102
   366
		if (localPart == null) {
franta-hg@102
   367
			return null;
franta-hg@102
   368
		} else {
franta-hg@102
   369
			return Long.parseLong(localPart[1]);
franta-hg@102
   370
			// If needed:
franta-hg@102
   371
			// parseGroupName() will be same as this method, just with:
franta-hg@102
   372
			// return localPart[2];
franta-hg@102
   373
		}
franta-hg@102
   374
	}
franta-hg@102
   375
franta-hg@72
   376
	@Override
franta-hg@72
   377
	public void setHeader(String name, String value) throws MessagingException {
franta-hg@72
   378
		super.setHeader(name, value);
franta-hg@72
   379
franta-hg@72
   380
		if (MESSAGE_ID_HEADER.equalsIgnoreCase(name)) {
franta-hg@72
   381
			messageID = value;
franta-hg@72
   382
		}
franta-hg@72
   383
	}
franta-hg@72
   384
franta-hg@72
   385
	@Override
franta-hg@72
   386
	public final void addHeader(String name, String value) throws MessagingException {
franta-hg@72
   387
		super.addHeader(name, value);
franta-hg@72
   388
franta-hg@72
   389
		if (MESSAGE_ID_HEADER.equalsIgnoreCase(name)) {
franta-hg@72
   390
			messageID = value;
franta-hg@72
   391
		}
franta-hg@72
   392
	}
franta-hg@72
   393
franta-hg@72
   394
	@Override
franta-hg@72
   395
	public void removeHeader(String name) throws MessagingException {
franta-hg@72
   396
		super.removeHeader(name);
franta-hg@72
   397
franta-hg@72
   398
		if (MESSAGE_ID_HEADER.equalsIgnoreCase(name)) {
franta-hg@72
   399
			messageID = null;
franta-hg@72
   400
		}
franta-hg@72
   401
	}
franta-hg@72
   402
franta-hg@72
   403
	public void setMessageID(String messageID) {
franta-hg@72
   404
		this.messageID = messageID;
franta-hg@72
   405
	}
franta-hg@72
   406
franta-hg@72
   407
	@Override
franta-hg@72
   408
	protected void updateMessageID() throws MessagingException {
franta-hg@72
   409
		if (messageID == null) {
franta-hg@72
   410
			super.updateMessageID();
franta-hg@72
   411
		} else {
franta-hg@72
   412
			setHeader(MESSAGE_ID_HEADER, messageID);
franta-hg@72
   413
		}
franta-hg@72
   414
	}
franta-hg@72
   415
franta-hg@72
   416
	/**
franta-hg@72
   417
	 * Call {@link #saveChanges()} before this method, if you want all headers including such ones like:
franta-hg@72
   418
	 * 
franta-hg@72
   419
	 * <pre>MIME-Version: 1.0
franta-hg@72
   420
	 *Content-Type: multipart/alternative;</pre>
franta-hg@72
   421
	 * 
franta-hg@72
   422
	 * @return serialized headers
franta-hg@72
   423
	 * @throws MessagingException if getAllHeaders() fails
franta-hg@72
   424
	 */
franta-hg@72
   425
	public String getHeaders() throws MessagingException {
franta-hg@72
   426
		StringBuilder sb = new StringBuilder();
franta-hg@72
   427
		for (Enumeration eh = getAllHeaderLines(); eh.hasMoreElements();) {
franta-hg@72
   428
			sb.append(eh.nextElement());
franta-hg@72
   429
			sb.append(CRLF);
franta-hg@72
   430
		}
franta-hg@72
   431
		return sb.toString();
franta-hg@72
   432
	}
franta-hg@72
   433
franta-hg@72
   434
	public byte[] getBody() throws IOException, MessagingException {
franta-hg@72
   435
		saveChanges();
franta-hg@72
   436
franta-hg@72
   437
		ArrayList<String> skipHeaders = new ArrayList<String>();
franta-hg@72
   438
		for (Enumeration eh = getAllHeaders(); eh.hasMoreElements();) {
franta-hg@72
   439
			Header h = (Header) eh.nextElement();
franta-hg@72
   440
			skipHeaders.add(h.getName());
franta-hg@72
   441
		}
franta-hg@72
   442
franta-hg@72
   443
		ByteArrayOutputStream baos = new ByteArrayOutputStream(1024);
franta-hg@72
   444
		writeTo(baos, skipHeaders.toArray(new String[skipHeaders.size()]));
franta-hg@72
   445
		return baos.toByteArray();
franta-hg@72
   446
	}
franta-hg@103
   447
franta-hg@103
   448
	/**
franta-hg@103
   449
	 * Transforms message content to valid XHTML and strips html and body tags.
franta-hg@103
   450
	 * When receiving message from user through NNTP 
franta-hg@103
   451
	 * this method is used to get text that should be saved into databse.
franta-hg@103
   452
	 * @return XHTML fragment – content between &lt;body&gt; and &lt;/body&gt; tags.
franta-hg@103
   453
	 */
franta-hg@103
   454
	public String getBodyXhtmlFragment() throws StorageBackendException {
franta-hg@103
   455
		/**
franta-hg@103
   456
		 * TODO: podporovat i zprávy přímo v HTML a multipart.
franta-hg@103
   457
		 */
franta-hg@103
   458
		try {
franta-hg@103
   459
			Object c = getContent();
franta-hg@103
   460
			if (isMimeType("text/plain") && c instanceof String) {
franta-hg@103
   461
				String xhtml = readXhtmlText(
franta-hg@103
   462
						(String) c,
franta-hg@103
   463
						getSubject(),
franta-hg@103
   464
						getParentID(),
franta-hg@103
   465
						null,
franta-hg@103
   466
						null,
franta-hg@103
   467
						null);
franta-hg@103
   468
				return makeFragmentXHTML(xhtml);
franta-hg@103
   469
			} else {
franta-hg@103
   470
				throw new StorageBackendException("Only text/plain messages are supported for now – post it as plain text please.");
franta-hg@103
   471
			}
franta-hg@103
   472
		} catch (Exception e) {
franta-hg@103
   473
			throw new StorageBackendException(e);
franta-hg@103
   474
		}
franta-hg@103
   475
	}
franta-hg@72
   476
}