chris@1: /*
chris@1:  *   SONEWS News Server
chris@1:  *   see AUTHORS for the list of contributors
chris@1:  *
chris@1:  *   This program is free software: you can redistribute it and/or modify
chris@1:  *   it under the terms of the GNU General Public License as published by
chris@1:  *   the Free Software Foundation, either version 3 of the License, or
chris@1:  *   (at your option) any later version.
chris@1:  *
chris@1:  *   This program is distributed in the hope that it will be useful,
chris@1:  *   but WITHOUT ANY WARRANTY; without even the implied warranty of
chris@1:  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
chris@1:  *   GNU General Public License for more details.
chris@1:  *
chris@1:  *   You should have received a copy of the GNU General Public License
chris@1:  *   along with this program.  If not, see <http://www.gnu.org/licenses/>.
chris@1:  */
chris@1: 
chris@1: package org.sonews.util.io;
chris@1: 
chris@1: import java.io.BufferedOutputStream;
chris@1: import java.io.BufferedReader;
chris@1: import java.io.IOException;
chris@1: import java.io.InputStreamReader;
chris@1: import java.io.UnsupportedEncodingException;
chris@1: import java.net.Socket;
chris@1: import java.net.UnknownHostException;
chris@1: import org.sonews.daemon.storage.Article;
chris@1: 
chris@1: /**
chris@1:  * Posts an Article to a NNTP server using the POST command.
chris@1:  * @author Christian Lins
chris@1:  * @since sonews/0.5.0
chris@1:  */
chris@1: public class ArticleWriter 
chris@1: {
chris@1:   
chris@1:   private BufferedOutputStream out;
chris@1:   private BufferedReader       inr;
chris@1: 
chris@1:   public ArticleWriter(String host, int port)
chris@1:     throws IOException, UnknownHostException
chris@1:   {
chris@1:     // Connect to NNTP server
chris@1:     Socket socket = new Socket(host, port);
chris@1:     this.out = new BufferedOutputStream(socket.getOutputStream());
chris@1:     this.inr = new BufferedReader(new InputStreamReader(socket.getInputStream()));
chris@1:     String line = inr.readLine();
chris@1:     if(line == null || !line.startsWith("200 "))
chris@1:     {
chris@1:       throw new IOException("Invalid hello from server: " + line);
chris@1:     }
chris@1:   }
chris@1:   
chris@1:   public void close()
chris@1:     throws IOException, UnsupportedEncodingException
chris@1:   {
chris@1:     this.out.write("QUIT\r\n".getBytes("UTF-8"));
chris@1:     this.out.flush();
chris@1:   }
chris@1: 
chris@1:   protected void finishPOST()
chris@1:     throws IOException
chris@1:   {
chris@1:     this.out.write("\r\n.\r\n".getBytes());
chris@1:     this.out.flush();
chris@1:     String line = inr.readLine();
chris@1:     if(line == null || !line.startsWith("240 "))
chris@1:     {
chris@1:       throw new IOException(line);
chris@1:     }
chris@1:   }
chris@1: 
chris@1:   protected void preparePOST()
chris@1:     throws IOException
chris@1:   {
chris@1:     this.out.write("POST\r\n".getBytes("UTF-8"));
chris@1:     this.out.flush();
chris@1: 
chris@1:     String line = this.inr.readLine();
chris@1:     if(line == null || !line.startsWith("340 "))
chris@1:     {
chris@1:       throw new IOException(line);
chris@1:     }
chris@1:   }
chris@1: 
chris@1:   public void writeArticle(Article article)
chris@1:     throws IOException, UnsupportedEncodingException
chris@1:   {
chris@1:     byte[] buf = new byte[512];
chris@1:     ArticleInputStream in = new ArticleInputStream(article);
chris@1: 
chris@1:     preparePOST();
chris@1:     
chris@1:     int len = in.read(buf);
chris@1:     while(len != -1)
chris@1:     {
chris@1:       writeLine(buf, len);
chris@1:       len = in.read(buf);
chris@1:     }
chris@1: 
chris@1:     finishPOST();
chris@1:   }
chris@1: 
chris@1:   /**
chris@1:    * Writes the raw content of an article to the remote server. This method
chris@1:    * does no charset conversion/handling of any kind so its the preferred
chris@1:    * method for sending an article to remote peers.
chris@1:    * @param rawArticle
chris@1:    * @throws IOException
chris@1:    */
chris@1:   public void writeArticle(byte[] rawArticle)
chris@1:     throws IOException
chris@1:   {
chris@1:     preparePOST();
chris@1:     writeLine(rawArticle, rawArticle.length);
chris@1:     finishPOST();
chris@1:   }
chris@1: 
chris@1:   /**
chris@1:    * Writes the given buffer to the connect remote server.
chris@1:    * @param buffer
chris@1:    * @param len
chris@1:    * @throws IOException
chris@1:    */
chris@1:   protected void writeLine(byte[] buffer, int len)
chris@1:     throws IOException
chris@1:   {
chris@1:     this.out.write(buffer, 0, len);
chris@1:     this.out.flush();
chris@1:   }
chris@1: 
chris@1: }