/*

  *   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.ByteArrayInputStream;

 import java.io.ByteArrayOutputStream;

 import java.io.IOException;

 import java.security.MessageDigest;

 import java.security.NoSuchAlgorithmException;

 import java.util.UUID;

 import java.util.ArrayList;

 import java.util.Enumeration;

 import java.util.List;

 import javax.mail.Header;

 import javax.mail.Message;

 import javax.mail.MessagingException;

 import javax.mail.internet.InternetHeaders;

 import org.sonews.config.Config;

 /**

  * Represents a newsgroup article.

  * @author Christian Lins

  * @author Denis Schwerdel

  * @since n3tpd/0.1

  */

 public class Article extends ArticleHead

 {

   /**

    * Loads the Article identified by the given ID from the JDBCDatabase.

    * @param messageID

    * @return null if Article is not found or if an error occurred.

    */

   public static Article getByMessageID(final String messageID)

   {

     try

     {

       return StorageManager.current().getArticle(messageID);

     }

     catch(StorageBackendException ex)

     {

       ex.printStackTrace();

       return null;

     }

   }

   private byte[] body       = new byte[0];

   /**

    * Default constructor.

    */

   public Article()

   {

   }

   /**

    * Creates a new Article object using the date from the given

    * raw data.

    */

   public Article(String headers, byte[] body)

   {

     try

     {

       this.body  = body;

       // Parse the header

       this.headers = new InternetHeaders(

         new ByteArrayInputStream(headers.getBytes()));

       this.headerSrc = headers;

     }

     catch(MessagingException ex)

     {

       ex.printStackTrace();

     }

   }

   /**

    * Creates an Article instance using the data from the javax.mail.Message

    * object. This constructor is called by the Mailinglist gateway.

    * @see javax.mail.Message

    * @param msg

    * @throws IOException

    * @throws MessagingException

    */

   public Article(final Message msg)

     throws IOException, MessagingException

   {

     this.headers = new InternetHeaders();

     for(Enumeration e = msg.getAllHeaders() ; e.hasMoreElements();) 

     {

       final Header header = (Header)e.nextElement();

       this.headers.addHeader(header.getName(), header.getValue());

     }

 	// Reads the raw byte body using Message.writeTo(OutputStream out)

 	this.body = readContent(msg);

     // Validate headers

     validateHeaders();

   }

   /**

    * Reads from the given Message into a byte array.

    * @param in

    * @return

    * @throws IOException

    */

   private byte[] readContent(Message in)

     throws IOException, MessagingException

   {

     ByteArrayOutputStream out = new ByteArrayOutputStream();

     in.writeTo(out);

     return out.toByteArray();

   }

   /**

    * Removes the header identified by the given key.

    * @param headerKey

    */

   public void removeHeader(final String headerKey)

   {

     this.headers.removeHeader(headerKey);

     this.headerSrc = null;

   }

   /**

    * Generates a message id for this article and sets it into

    * the header object. You have to update the JDBCDatabase manually to make this

    * change persistent.

    * Note: a Message-ID should never be changed and only generated once.

    */

   private String generateMessageID()

   {

     String randomString;

     MessageDigest md5;

     try

     {

       md5 = MessageDigest.getInstance("MD5");

       md5.reset();

       md5.update(getBody());

       md5.update(getHeader(Headers.SUBJECT)[0].getBytes());

       md5.update(getHeader(Headers.FROM)[0].getBytes());

       byte[] result = md5.digest();

       StringBuffer hexString = new StringBuffer();

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

       {

         hexString.append(Integer.toHexString(0xFF & result[i]));

       }

       randomString = hexString.toString();

     }

     catch (NoSuchAlgorithmException e)

     {

       e.printStackTrace();

       randomString = UUID.randomUUID().toString();

     }

     String msgID = "<" + randomString + "@"

         + Config.inst().get(Config.HOSTNAME, "localhost") + ">";

     this.headers.setHeader(Headers.MESSAGE_ID, msgID);

     return msgID;

   }

   /**

    * Returns the body string.

    */

   public byte[] getBody()

   {

     return body;

   }

   /**

    * @return Numerical IDs of the newsgroups this Article belongs to.

    */

   public List<Group> getGroups()

   {

     String[]         groupnames = getHeader(Headers.NEWSGROUPS)[0].split(",");

     ArrayList<Group> groups     = new ArrayList<Group>();

     try

     {

       for(String newsgroup : groupnames)

       {

         newsgroup = newsgroup.trim();

         Group group = StorageManager.current().getGroup(newsgroup);

         if(group != null &&         // If the server does not provide the group, ignore it

           !groups.contains(group))  // Yes, there may be duplicates

         {

           groups.add(group);

         }

       }

     }

     catch(StorageBackendException ex)

     {

       ex.printStackTrace();

       return null;

     }

     return groups;

   }

   public void setBody(byte[] body)

   {

     this.body = body;

   }

   /**

    * 

    * @param groupname Name(s) of newsgroups

    */

   public void setGroup(String groupname)

   {

     this.headers.setHeader(Headers.NEWSGROUPS, groupname);

   }

   /**

    * Returns the Message-ID of this Article. If the appropriate header

    * is empty, a new Message-ID is created.

    * @return Message-ID of this Article.

    */

   public String getMessageID()

   {

     String[] msgID = getHeader(Headers.MESSAGE_ID);

     return msgID[0].equals("") ? generateMessageID() : msgID[0];

   }

   /**

    * @return String containing the Message-ID.

    */

   @Override

   public String toString()

   {

     return getMessageID();

   }

 }