org/sonews/util/io/ArticleInputStream.java
author cli
Sun Aug 29 17:03:21 2010 +0200 (2010-08-29)
changeset 33 f9bf183447d1
parent 3 2fdc9cc89502
permissions -rw-r--r--
Article(javax.mail.Message) now has a safe method to read the byte body from the given message object (fixes #16).
     1 /*
     2  *   SONEWS News Server
     3  *   see AUTHORS for the list of contributors
     4  *
     5  *   This program is free software: you can redistribute it and/or modify
     6  *   it under the terms of the GNU General Public License as published by
     7  *   the Free Software Foundation, either version 3 of the License, or
     8  *   (at your option) any later version.
     9  *
    10  *   This program is distributed in the hope that it will be useful,
    11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of
    12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    13  *   GNU General Public License for more details.
    14  *
    15  *   You should have received a copy of the GNU General Public License
    16  *   along with this program.  If not, see <http://www.gnu.org/licenses/>.
    17  */
    18 
    19 package org.sonews.util.io;
    20 
    21 import java.io.ByteArrayOutputStream;
    22 import java.io.IOException;
    23 import java.io.InputStream;
    24 import java.io.UnsupportedEncodingException;
    25 import org.sonews.storage.Article;
    26 
    27 /**
    28  * Capsulates an Article to provide a raw InputStream.
    29  * @author Christian Lins
    30  * @since sonews/0.5.0
    31  */
    32 public class ArticleInputStream extends InputStream
    33 {
    34 
    35   private byte[] buf;
    36   private int    pos = 0;
    37   
    38   public ArticleInputStream(final Article art)
    39     throws IOException, UnsupportedEncodingException
    40   {
    41     final ByteArrayOutputStream out = new ByteArrayOutputStream();
    42     out.write(art.getHeaderSource().getBytes("UTF-8"));
    43     out.write("\r\n\r\n".getBytes());
    44     out.write(art.getBody()); // Without CRLF
    45     out.flush();
    46     this.buf = out.toByteArray();
    47   }
    48 
    49   /**
    50    * This method reads one byte from the stream.  The <code>pos</code>
    51    * counter is advanced to the next byte to be read.  The byte read is
    52    * returned as an int in the range of 0-255.  If the stream position
    53    * is already at the end of the buffer, no byte is read and a -1 is
    54    * returned in order to indicate the end of the stream.
    55    *
    56    * @return The byte read, or -1 if end of stream
    57    */
    58   @Override
    59   public synchronized int read()
    60   {
    61     if(pos < buf.length)
    62     {
    63       return ((int)buf[pos++]) & 0xFF;
    64     }
    65     else
    66     {
    67       return -1;
    68     }
    69   }
    70   
    71 }