/*

  *   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 org.sonews.storage.Article;

 import org.sonews.daemon.NNTPConnection;

 import org.sonews.storage.Channel;

 import org.sonews.storage.StorageBackendException;

 /**

  * Class handling the ARTICLE, BODY and HEAD commands.

  * @author Christian Lins

  * @author Dennis Schwerdel

  * @since n3tpd/0.1

  */

 public class ArticleCommand implements Command

 {

   @Override

   public String[] getSupportedCommandStrings()

   {

     return new String[] {"ARTICLE", "BODY", "HEAD"};

   }

   @Override

   public boolean hasFinished()

   {

     return true;

   }

   @Override

   public String impliedCapability()

   {

     return null;

   }

   @Override

   public boolean isStateful()

   {

     return false;

   }

   // TODO: Refactor this method to reduce its complexity!

   @Override

   public void processLine(NNTPConnection conn, final String line, byte[] raw)

     throws IOException

   {

     final String[] command = line.split(" ");

     Article article  = null;

     long    artIndex = -1;

     if (command.length == 1)

     {

       article = conn.getCurrentArticle();

       if (article == null)

       {

         conn.println("420 no current article has been selected");

         return;

       }

     }

     else if (command[1].matches(NNTPConnection.MESSAGE_ID_PATTERN))

     {

       // Message-ID

       article = Article.getByMessageID(command[1]);

       if (article == null)

       {

         conn.println("430 no such article found");

         return;

       }

     }

     else

     {

       // Message Number

       try

       {

         Channel currentGroup = conn.getCurrentChannel();

         if(currentGroup == null)

         {

           conn.println("400 no group selected");

           return;

         }

         artIndex = Long.parseLong(command[1]);

         article  = currentGroup.getArticle(artIndex);

       }

       catch(NumberFormatException ex)

       {

         ex.printStackTrace();

       }

       catch(StorageBackendException ex)

       {

         ex.printStackTrace();

       }

       if (article == null)

       {

         conn.println("423 no such article number in this group");

         return;

       }

       conn.setCurrentArticle(article);

     }

     if(command[0].equalsIgnoreCase("ARTICLE"))

     {

       conn.println("220 " + artIndex + " " + article.getMessageID()

           + " article retrieved - head and body follow");

       conn.println(article.getHeaderSource());

       conn.println("");

       conn.println(article.getBody());

       conn.println(".");

     }

     else if(command[0].equalsIgnoreCase("BODY"))

     {

       conn.println("222 " + artIndex + " " + article.getMessageID() + " body");

       conn.println(article.getBody());

       conn.println(".");

     }

     /*

      * HEAD: This command is mandatory.

      *

      * Syntax

      *    HEAD message-id

      *    HEAD number

      *    HEAD

      *

      * Responses

      *

      * First form (message-id specified)

      *  221 0|n message-id    Headers follow (multi-line)

      *  430                   No article with that message-id

      *

      * Second form (article number specified)

      *  221 n message-id      Headers follow (multi-line)

      *  412                   No newsgroup selected

      *  423                   No article with that number

      *

      * Third form (current article number used)

      *  221 n message-id      Headers follow (multi-line)

      *  412                   No newsgroup selected

      *  420                   Current article number is invalid

      *

      * Parameters

      *  number        Requested article number

      *  n             Returned article number

      *  message-id    Article message-id

      */

     else if(command[0].equalsIgnoreCase("HEAD"))

     {

       conn.println("221 " + artIndex + " " + article.getMessageID()

           + " Headers follow (multi-line)");

       conn.println(article.getHeaderSource());

       conn.println(".");

     }

   }  

 }