/*

  *   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.daemon.command;

 import java.io.IOException;

 import java.io.ByteArrayInputStream;

 import java.nio.charset.Charset;

 import java.nio.charset.IllegalCharsetNameException;

 import java.nio.charset.UnsupportedCharsetException;

 import java.sql.SQLException;

 import java.util.Locale;

 import javax.mail.MessagingException;

 import javax.mail.internet.AddressException;

 import javax.mail.internet.InternetHeaders;

 import org.sonews.daemon.Config;

 import org.sonews.util.Log;

 import org.sonews.mlgw.Dispatcher;

 import org.sonews.daemon.storage.Article;

 import org.sonews.daemon.storage.Database;

 import org.sonews.daemon.storage.Group;

 import org.sonews.daemon.NNTPConnection;

 import org.sonews.daemon.storage.Headers;

 import org.sonews.feed.FeedManager;

 import org.sonews.util.Stats;

 /**

  * Implementation of the POST command. This command requires multiple lines

  * from the client, so the handling of asynchronous reading is a little tricky

  * to handle.

  * @author Christian Lins

  * @since sonews/0.5.0

  */

 public class PostCommand extends AbstractCommand

 {

   private final Article article   = new Article();

   private int           lineCount = 0;

   private long          bodySize  = 0;

   private InternetHeaders headers = null;

   private long          maxBodySize  = 

     Config.getInstance().get(Config.ARTICLE_MAXSIZE, 128) * 1024L; // Size in bytes

   private PostState     state     = PostState.WaitForLineOne;

   private final StringBuilder strBody   = new StringBuilder();

   private final StringBuilder strHead   = new StringBuilder();

   public PostCommand(final NNTPConnection conn)

   {

     super(conn);

   }

   @Override

   public boolean hasFinished()

   {

     return this.state == PostState.Finished;

   }

   /**

    * Process the given line String. line.trim() was called by NNTPConnection.

    * @param line

    * @throws java.io.IOException

    * @throws java.sql.SQLException

    */

   @Override // TODO: Refactor this method to reduce complexity!

   public void processLine(String line)

     throws IOException, SQLException

   {

     switch(state)

     {

       case WaitForLineOne:

       {

         if(line.equalsIgnoreCase("POST"))

         {

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

           state = PostState.ReadingHeaders;

         }

         else

         {

           printStatus(500, "invalid command usage");

         }

         break;

       }

       case ReadingHeaders:

       {

         strHead.append(line);

         strHead.append(NNTPConnection.NEWLINE);

         if("".equals(line) || ".".equals(line))

         {

           // we finally met the blank line

           // separating headers from body

           try

           {

             // Parse the header using the InternetHeader class from JavaMail API

             headers = new InternetHeaders(

               new ByteArrayInputStream(strHead.toString().trim()

                 .getBytes(connection.getCurrentCharset())));

             // add the header entries for the article

             article.setHeaders(headers);

           }

           catch (MessagingException e)

           {

             e.printStackTrace();

             printStatus(500, "posting failed - invalid header");

             state = PostState.Finished;

             break;

           }

           // Change charset for reading body; 

           // for multipart messages UTF-8 is returned

           connection.setCurrentCharset(article.getBodyCharset());

           state = PostState.ReadingBody;

           if(".".equals(line))

           {

             // Post an article without body

             postArticle(article);

             state = PostState.Finished;

           }

         }

         break;

       }

       case ReadingBody:

       {

         if(".".equals(line))

         {    

           // Set some headers needed for Over command

           headers.setHeader(Headers.LINES, Integer.toString(lineCount));

           headers.setHeader(Headers.BYTES, Long.toString(bodySize));

           if(strBody.length() >= 2)

           {

             strBody.deleteCharAt(strBody.length() - 1); // Remove last newline

             strBody.deleteCharAt(strBody.length() - 1); // Remove last CR

           }

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

           postArticle(article);

           state = PostState.Finished;

         }

         else

         {

           bodySize += line.length() + 1;

           lineCount++;

           // Add line to body buffer

           strBody.append(line);

           strBody.append(NNTPConnection.NEWLINE);

           if(bodySize > maxBodySize)

           {

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

             state = PostState.Finished;

             break;

           }

           // Check if this message is a MIME-multipart message and needs a

           // charset change

           try

           {

             line = line.toLowerCase(Locale.ENGLISH);

             if(line.startsWith(Headers.CONTENT_TYPE))

             {

               int idxStart = line.indexOf("charset=") + "charset=".length();

               int idxEnd   = line.indexOf(";", idxStart);

               if(idxEnd < 0)

               {

                 idxEnd = line.length();

               }

               if(idxStart > 0)

               {

                 String charsetName = line.substring(idxStart, idxEnd);

                 if(charsetName.length() > 0 && charsetName.charAt(0) == '"')

                 {

                   charsetName = charsetName.substring(1, charsetName.length() - 1);

                 }

                 try

                 {

                   connection.setCurrentCharset(Charset.forName(charsetName));

                 }

                 catch(IllegalCharsetNameException ex)

                 {

                   Log.msg("PostCommand: " + ex, false);

                 }

                 catch(UnsupportedCharsetException ex)

                 {

                   Log.msg("PostCommand: " + ex, false);

                 }

               } // if(idxStart > 0)

             }

           }

           catch(Exception ex)

           {

             ex.printStackTrace();

           }

         }

         break;

       }

       default:

         Log.msg("PostCommand::processLine(): already finished...", false);

     }

   }

   /**

    * Article is a control message and needs special handling.

    * @param article

    */

   private void controlMessage(Article article)

     throws IOException

   {

     String[] ctrl = article.getHeader(Headers.CONTROL)[0].split(" ");

     if(ctrl.length == 2) // "cancel <mid>"

     {

       try

       {

         Database.getInstance().delete(ctrl[1]);

         // Move cancel message to "control" group

         article.setHeader(Headers.NEWSGROUPS, "control");

         Database.getInstance().addArticle(article);

         printStatus(240, "article cancelled");

       }

       catch(SQLException ex)

       {

         Log.msg(ex, false);

         printStatus(500, "internal server error");

       }

     }

     else

     {

       printStatus(441, "unknown Control header");

     }

   }

   private void supersedeMessage(Article article)

     throws IOException

   {

     try

     {

       String oldMsg = article.getHeader(Headers.SUPERSEDES)[0];

       Database.getInstance().delete(oldMsg);

       Database.getInstance().addArticle(article);

       printStatus(240, "article replaced");

     }

     catch(SQLException ex)

     {

       Log.msg(ex, false);

       printStatus(500, "internal server error");

     }

   }

   private void postArticle(Article article) 

     throws IOException

   {

     if(article.getHeader(Headers.CONTROL)[0].length() > 0)

     {

       controlMessage(article);

     }

     else if(article.getHeader(Headers.SUPERSEDES)[0].length() > 0)

     {

       supersedeMessage(article);

     }

     else // Post the article regularily

     {

       // Try to create the article in the database or post it to

       // appropriate mailing list

       try

       {

         boolean success = false;

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

         for(String groupname : groupnames)

         {

           Group group = Database.getInstance().getGroup(groupname);

           if(group != null)

           {

             if(group.isMailingList() && !connection.isLocalConnection())

             {

               // Send to mailing list; the Dispatcher writes 

               // statistics to database

               Dispatcher.toList(article);

               success = true;

             }

             else

             {

               // Store in database

               if(!Database.getInstance().isArticleExisting(article.getMessageID()))

               {

                 Database.getInstance().addArticle(article);

                 // Log this posting to statistics

                 Stats.getInstance().mailPosted(

                   article.getHeader(Headers.NEWSGROUPS)[0]);

               }

               success = true;

             }

           }

         } // end for

         if(success)

         {

           printStatus(240, "article posted ok");

           FeedManager.queueForPush(article);

         }

         else

         {

           printStatus(441, "newsgroup not found");

         }

       }

       catch(AddressException ex)

       {

         Log.msg(ex.getMessage(), true);

         printStatus(441, "invalid sender address");

       }

       catch(MessagingException ex)

       {

         // A MessageException is thrown when the sender email address is

         // invalid or something is wrong with the SMTP server.

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

         printStatus(441, ex.getClass().getCanonicalName() + ": " + ex.getLocalizedMessage());

       }

       catch(SQLException ex)

       {

         ex.printStackTrace();

         printStatus(500, "internal server error");

       }

     }

   }

 }