chris@3: /*
chris@3:  *   SONEWS News Server
chris@3:  *   see AUTHORS for the list of contributors
chris@3:  *
chris@3:  *   This program is free software: you can redistribute it and/or modify
chris@3:  *   it under the terms of the GNU General Public License as published by
chris@3:  *   the Free Software Foundation, either version 3 of the License, or
chris@3:  *   (at your option) any later version.
chris@3:  *
chris@3:  *   This program is distributed in the hope that it will be useful,
chris@3:  *   but WITHOUT ANY WARRANTY; without even the implied warranty of
chris@3:  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
chris@3:  *   GNU General Public License for more details.
chris@3:  *
chris@3:  *   You should have received a copy of the GNU General Public License
chris@3:  *   along with this program.  If not, see <http://www.gnu.org/licenses/>.
chris@3:  */
chris@3: 
chris@3: package org.sonews.storage;
chris@3: 
chris@3: import java.io.ByteArrayInputStream;
chris@3: import java.io.ByteArrayOutputStream;
chris@3: import java.io.IOException;
chris@3: import java.security.MessageDigest;
chris@3: import java.security.NoSuchAlgorithmException;
chris@3: import java.util.UUID;
chris@3: import java.util.ArrayList;
chris@3: import java.util.Enumeration;
chris@3: import java.util.List;
chris@3: import javax.mail.Header;
chris@3: import javax.mail.Message;
chris@3: import javax.mail.MessagingException;
chris@3: import javax.mail.internet.InternetHeaders;
chris@3: import org.sonews.config.Config;
chris@3: 
chris@3: /**
chris@3:  * Represents a newsgroup article.
chris@3:  * @author Christian Lins
chris@3:  * @author Denis Schwerdel
chris@3:  * @since n3tpd/0.1
chris@3:  */
chris@3: public class Article extends ArticleHead
chris@3: {
chris@3: 
cli@37: 	/**
cli@37: 	 * Loads the Article identified by the given ID from the JDBCDatabase.
cli@37: 	 * @param messageID
cli@37: 	 * @return null if Article is not found or if an error occurred.
cli@37: 	 */
cli@37: 	public static Article getByMessageID(final String messageID)
cli@37: 	{
cli@37: 		try {
cli@37: 			return StorageManager.current().getArticle(messageID);
cli@37: 		} catch (StorageBackendException ex) {
cli@37: 			ex.printStackTrace();
cli@37: 			return null;
cli@37: 		}
cli@37: 	}
cli@37: 	private byte[] body = new byte[0];
chris@3: 
cli@37: 	/**
cli@37: 	 * Default constructor.
cli@37: 	 */
cli@37: 	public Article()
cli@37: 	{
cli@37: 	}
chris@3: 
cli@37: 	/**
cli@37: 	 * Creates a new Article object using the date from the given
cli@37: 	 * raw data.
cli@37: 	 */
cli@37: 	public Article(String headers, byte[] body)
cli@37: 	{
cli@37: 		try {
cli@37: 			this.body = body;
cli@33: 
cli@37: 			// Parse the header
cli@37: 			this.headers = new InternetHeaders(
cli@37: 				new ByteArrayInputStream(headers.getBytes()));
chris@3: 
cli@37: 			this.headerSrc = headers;
cli@37: 		} catch (MessagingException ex) {
cli@37: 			ex.printStackTrace();
cli@37: 		}
cli@37: 	}
chris@3: 
cli@37: 	/**
cli@37: 	 * Creates an Article instance using the data from the javax.mail.Message
cli@37: 	 * object. This constructor is called by the Mailinglist gateway.
cli@37: 	 * @see javax.mail.Message
cli@37: 	 * @param msg
cli@37: 	 * @throws IOException
cli@37: 	 * @throws MessagingException
cli@37: 	 */
cli@37: 	public Article(final Message msg)
cli@37: 		throws IOException, MessagingException
cli@37: 	{
cli@37: 		this.headers = new InternetHeaders();
chris@3: 
cli@37: 		for (Enumeration e = msg.getAllHeaders(); e.hasMoreElements();) {
cli@37: 			final Header header = (Header) e.nextElement();
cli@37: 			this.headers.addHeader(header.getName(), header.getValue());
cli@37: 		}
chris@3: 
cli@37: 		// Reads the raw byte body using Message.writeTo(OutputStream out)
cli@37: 		this.body = readContent(msg);
chris@3: 
cli@37: 		// Validate headers
cli@37: 		validateHeaders();
cli@37: 	}
chris@3: 
cli@37: 	/**
cli@37: 	 * Reads from the given Message into a byte array.
cli@37: 	 * @param in
cli@37: 	 * @return
cli@37: 	 * @throws IOException
cli@37: 	 */
cli@37: 	private byte[] readContent(Message in)
cli@37: 		throws IOException, MessagingException
cli@37: 	{
cli@37: 		ByteArrayOutputStream out = new ByteArrayOutputStream();
cli@37: 		in.writeTo(out);
cli@37: 		return out.toByteArray();
cli@37: 	}
chris@3: 
cli@37: 	/**
cli@37: 	 * Removes the header identified by the given key.
cli@37: 	 * @param headerKey
cli@37: 	 */
cli@37: 	public void removeHeader(final String headerKey)
cli@37: 	{
cli@37: 		this.headers.removeHeader(headerKey);
cli@37: 		this.headerSrc = null;
cli@37: 	}
chris@3: 
cli@37: 	/**
cli@37: 	 * Generates a message id for this article and sets it into
cli@37: 	 * the header object. You have to update the JDBCDatabase manually to make this
cli@37: 	 * change persistent.
cli@37: 	 * Note: a Message-ID should never be changed and only generated once.
cli@37: 	 */
cli@37: 	private String generateMessageID()
cli@37: 	{
cli@37: 		String randomString;
cli@37: 		MessageDigest md5;
cli@37: 		try {
cli@37: 			md5 = MessageDigest.getInstance("MD5");
cli@37: 			md5.reset();
cli@37: 			md5.update(getBody());
cli@37: 			md5.update(getHeader(Headers.SUBJECT)[0].getBytes());
cli@37: 			md5.update(getHeader(Headers.FROM)[0].getBytes());
cli@37: 			byte[] result = md5.digest();
cli@37: 			StringBuffer hexString = new StringBuffer();
cli@37: 			for (int i = 0; i < result.length; i++) {
cli@37: 				hexString.append(Integer.toHexString(0xFF & result[i]));
cli@37: 			}
cli@37: 			randomString = hexString.toString();
cli@37: 		} catch (NoSuchAlgorithmException e) {
cli@37: 			e.printStackTrace();
cli@37: 			randomString = UUID.randomUUID().toString();
cli@37: 		}
cli@37: 		String msgID = "<" + randomString + "@"
cli@37: 			+ Config.inst().get(Config.HOSTNAME, "localhost") + ">";
cli@37: 
cli@37: 		this.headers.setHeader(Headers.MESSAGE_ID, msgID);
cli@37: 
cli@37: 		return msgID;
cli@37: 	}
cli@37: 
cli@37: 	/**
cli@37: 	 * Returns the body string.
cli@37: 	 */
cli@37: 	public byte[] getBody()
cli@37: 	{
cli@37: 		return body;
cli@37: 	}
cli@37: 
cli@37: 	/**
cli@37: 	 * @return Numerical IDs of the newsgroups this Article belongs to.
cli@37: 	 */
cli@37: 	public List<Group> getGroups()
cli@37: 	{
cli@37: 		String[] groupnames = getHeader(Headers.NEWSGROUPS)[0].split(",");
cli@37: 		ArrayList<Group> groups = new ArrayList<Group>();
cli@37: 
cli@37: 		try {
cli@37: 			for (String newsgroup : groupnames) {
cli@37: 				newsgroup = newsgroup.trim();
cli@37: 				Group group = StorageManager.current().getGroup(newsgroup);
cli@37: 				if (group != null && // If the server does not provide the group, ignore it
cli@37: 					!groups.contains(group)) // Yes, there may be duplicates
cli@37: 				{
cli@37: 					groups.add(group);
cli@37: 				}
cli@37: 			}
cli@37: 		} catch (StorageBackendException ex) {
cli@37: 			ex.printStackTrace();
cli@37: 			return null;
cli@37: 		}
cli@37: 		return groups;
cli@37: 	}
cli@37: 
cli@37: 	public void setBody(byte[] body)
cli@37: 	{
cli@37: 		this.body = body;
cli@37: 	}
cli@37: 
cli@37: 	/**
cli@37: 	 *
cli@37: 	 * @param groupname Name(s) of newsgroups
cli@37: 	 */
cli@37: 	public void setGroup(String groupname)
cli@37: 	{
cli@37: 		this.headers.setHeader(Headers.NEWSGROUPS, groupname);
cli@37: 	}
cli@37: 
cli@37: 	/**
cli@37: 	 * Returns the Message-ID of this Article. If the appropriate header
cli@37: 	 * is empty, a new Message-ID is created.
cli@37: 	 * @return Message-ID of this Article.
cli@37: 	 */
cli@37: 	public String getMessageID()
cli@37: 	{
cli@37: 		String[] msgID = getHeader(Headers.MESSAGE_ID);
cli@37: 		return msgID[0].equals("") ? generateMessageID() : msgID[0];
cli@37: 	}
cli@37: 
cli@37: 	/**
cli@37: 	 * @return String containing the Message-ID.
cli@37: 	 */
cli@37: 	@Override
cli@37: 	public String toString()
cli@37: 	{
cli@37: 		return getMessageID();
cli@37: 	}
chris@3: }