/*

  *   StarOffice 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 com.so.news.command;

 import java.io.IOException;

 import java.sql.SQLException;

 import java.util.Date;

 import java.text.SimpleDateFormat;

 import java.util.HashMap;

 import java.util.Locale;

 import com.so.news.Config;

 import com.so.news.Debug;

 import com.so.news.NNTPConnection;

 import com.so.news.storage.Article;

 import com.so.news.storage.Database;

 /**

  * Contains the code for the POST command.

  * @author Christian Lins

  * @author Dennis Schwerdel

  */

 public class PostCommand extends Command

 {

   public PostCommand(NNTPConnection conn)

   {

     super(conn);

   }

   public boolean process(String[] command) throws IOException

   {

     printStatus(340, "send article to be posted. End with <CR-LF>.<CR-LF>");

     // some initialization

     Article article = new Article();

     int lineCount     = 0;

     long bodySize     = 0;

     long maxBodySize  = Config.getInstance().get("n3tpd.article.maxsize", 1024) * 1024; // Size in bytes

     // begin with a stringbuilder body

     StringBuilder body = new StringBuilder();

     HashMap<String, String> header = new HashMap<String, String>();

     boolean isHeader = true; // are we in the header part

     String line = readTextLine();

     while(line != null)

     {

       bodySize += line.length();

       if(bodySize > maxBodySize)

       {

         printStatus(500, "article is too long");

         return false;

       }

       if(!isHeader)

       { // body

         if(line.trim().equals("."))

           break;

         bodySize += line.length() + 1;

         lineCount++;

         body.append(line + NEWLINE);

       }

       if(line.equals(""))

       {

         isHeader = false; // we finally met the blank line

                           // separating headers from body

       }

       if(isHeader)

       { // header

         // split name and value and add the header to the map

         int colon = line.indexOf(':');

         String fieldName = line.substring(0, colon).trim();

         String fieldValue = line.substring(colon + 1).trim();

         header.put(fieldName, fieldValue);

       }

       line = readTextLine(); // read a new line

     } // end of input reading

     article.setBody(body.toString()); // set the article body

     article.setHeader(header);     // add the header entries for the article

     // Read the date header and fall back to the current date if it is not set

     try

     {

       SimpleDateFormat sdf = new SimpleDateFormat("EEE, dd MMM yyyy HH:mm:ss Z", Locale.US);

       String date = header.get("DATE");

       if(date == null)

         article.setDate(new Date());

       else

         article.setDate(new Date(sdf.parse(date).getTime())) ;

     }

     catch (Exception e)

     {

       e.printStackTrace(Debug.getInstance().getStream());

       printStatus(541, "posting failed - invalid date format");

       return true;

     }

     // check for a cancel command

     if ( header.containsKey("Control") ) 

     {

       String[] control = header.get("Control").split(" ") ;

       if ( control.length >= 2 && control[0].equalsIgnoreCase("cancel") ) 

       {

         // this article is a cancel-article, try to delete the old article

         try

         {

           Article.getByMessageID(control[1]).delete();

           printStatus(240, "article posted ok - original article canceled"); // quite

           return true; // quit, do not actually post this article since it

         }

         catch (Exception e)

         {

           e.printStackTrace();

           printStatus(441, "posting failed - original posting not found");

           return true;

         }

       }

     }

     // set some headers

     header.put("Message-ID", article.getMessageID());

     header.put("Lines", "" + lineCount);

     header.put("Bytes", "" + bodySize);

     // if needed, set an empty references header, that means this is

     // a initial posting

     if (!header.containsKey("References"))

       header.put("References", "");

     // try to create the article in the database

     try

     {

       Database.getInstance().addArticle(article);

       printStatus(240, "article posted ok");

     }

     catch(SQLException ex)

     {

       System.err.println(ex.getLocalizedMessage());

       ex.printStackTrace(Debug.getInstance().getStream());

       printStatus(500, "internal server error");

     }

     return true;

   }

 }