/*

 *   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.mlgw;

import java.io.IOException;

import java.util.ArrayList;

import java.util.List;

import java.util.regex.Matcher;

import java.util.regex.Pattern;

import javax.mail.Address;

import javax.mail.Authenticator;

import javax.mail.Message;

import javax.mail.MessagingException;

import javax.mail.PasswordAuthentication;

import javax.mail.internet.InternetAddress;

import org.sonews.config.Config;

import org.sonews.storage.Article;

import org.sonews.storage.Group;

import org.sonews.storage.Headers;

import org.sonews.storage.StorageBackendException;

import org.sonews.storage.StorageManager;

import org.sonews.util.Log;

import org.sonews.util.Stats;

/**

 * Dispatches messages from mailing list to newsserver or vice versa.

 * @author Christian Lins

 * @since sonews/0.5.0

 */

public class Dispatcher 

{

  static class PasswordAuthenticator extends Authenticator

  {

    @Override

    public PasswordAuthentication getPasswordAuthentication()

    {

      final String username = 

        Config.inst().get(Config.MLSEND_USER, "user");

      final String password = 

        Config.inst().get(Config.MLSEND_PASSWORD, "mysecret");

      return new PasswordAuthentication(username, password);

    }

  }

  /**

   * Chunks out the email address of the full List-Post header field.

   * @param listPostValue

   * @return The matching email address or null

   */

  private static String chunkListPost(String listPostValue)

  {

    // listPostValue is of form "<mailto:dev@openoffice.org>"

    Pattern mailPattern = Pattern.compile("(\\w+[-|.])*\\w+@(\\w+.)+\\w+");

    Matcher mailMatcher = mailPattern.matcher(listPostValue);

    if(mailMatcher.find())

    {

      return listPostValue.substring(mailMatcher.start(), mailMatcher.end());

    }

    else

    {

      return null;

    }

  }

  /**

   * This method inspects the header of the given message, trying

   * to find the most appropriate recipient.

   * @param msg

   * @param fallback If this is false only List-Post and X-List-Post headers

   *                 are examined.

   * @return null or fitting group name for the given message.

   */

  private static List<String> getGroupFor(final Message msg, final boolean fallback)

    throws MessagingException, StorageBackendException

  {

    List<String> groups = null;

    // Is there a List-Post header?

    String[]        listPost = msg.getHeader(Headers.LIST_POST);

    InternetAddress listPostAddr;

    if(listPost == null || listPost.length == 0 || "".equals(listPost[0]))

    {

      // Is there a X-List-Post header?

      listPost = msg.getHeader(Headers.X_LIST_POST);

    }

    if(listPost != null && listPost.length > 0 

      && !"".equals(listPost[0]) && chunkListPost(listPost[0]) != null)

    {

      // listPost[0] is of form "<mailto:dev@openoffice.org>"

      listPost[0]  = chunkListPost(listPost[0]);

      listPostAddr = new InternetAddress(listPost[0], false);

      groups = StorageManager.current().getGroupsForList(listPostAddr.getAddress());

    }

    else if(fallback)

    {

      Log.msg("Using fallback recipient discovery for: " + msg.getSubject(), true);

      groups = new ArrayList<String>();

      // Fallback to TO/CC/BCC addresses

      Address[] to = msg.getAllRecipients();

      for(Address toa : to) // Address can have '<' '>' around

      {

        if(toa instanceof InternetAddress)

        {

          List<String> g = StorageManager.current()

            .getGroupsForList(((InternetAddress)toa).getAddress());

          groups.addAll(g);

        }

      }

    }

    return groups;

  }

  /**

   * Posts a message that was received from a mailing list to the 

   * appropriate newsgroup.

   * If the message already exists in the storage, this message checks

   * if it must be posted in an additional group. This can happen for

   * crosspostings in different mailing lists.

   * @param msg

   */

  public static boolean toGroup(final Message msg)

  {

    try

    {

      // Create new Article object

      Article article = new Article(msg);

      boolean posted  = false;

      // Check if this mail is already existing the storage

      boolean updateReq = 

        StorageManager.current().isArticleExisting(article.getMessageID());

      List<String> newsgroups = getGroupFor(msg, !updateReq);

      List<String> oldgroups  = new ArrayList<String>();

      if(updateReq)

      {

        // Check for duplicate entries of the same group

        Article oldArticle = StorageManager.current().getArticle(article.getMessageID());

        List<Group> oldGroups = oldArticle.getGroups();

        for(Group oldGroup : oldGroups)

        {

          if(!newsgroups.contains(oldGroup.getName()))

          {

            oldgroups.add(oldGroup.getName());

          }

        }

      }

      if(newsgroups.size() > 0)

      {

        newsgroups.addAll(oldgroups);

        StringBuilder groups = new StringBuilder();

        for(int n = 0; n < newsgroups.size(); n++)

        {

          groups.append(newsgroups.get(n));

          if (n + 1 != newsgroups.size())

          {

            groups.append(',');

          }

        }

        Log.msg("Posting to group " + groups.toString(), true);

        article.setGroup(groups.toString());

        //article.removeHeader(Headers.REPLY_TO);

        //article.removeHeader(Headers.TO);

        // Write article to database

        if(updateReq)

        {

          Log.msg("Updating " + article.getMessageID() + " with additional groups", true);

          StorageManager.current().delete(article.getMessageID());

          StorageManager.current().addArticle(article);

        }

        else

        {

          Log.msg("Gatewaying " + article.getMessageID() + " to "

            + article.getHeader(Headers.NEWSGROUPS)[0], true);

          StorageManager.current().addArticle(article);

          Stats.getInstance().mailGatewayed(

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

        }

        posted = true;

      }

      else

      {

        StringBuilder buf = new StringBuilder();

        for (Address toa : msg.getAllRecipients())

        {

          buf.append(' ');

          buf.append(toa.toString());

        }

        buf.append(" " + article.getHeader(Headers.LIST_POST)[0]);

        Log.msg("No group for" + buf.toString(), false);

      }

      return posted;

    }

    catch(Exception ex)

    {

      ex.printStackTrace();

      return false;

    }

  }

  /**

   * Mails a message received through NNTP to the appropriate mailing list.

   * This method MAY be called several times by PostCommand for the same

   * article.

   */

  public static void toList(Article article, String group)

    throws IOException, MessagingException, StorageBackendException

  {

    // Get mailing lists for the group of this article

    List<String> rcptAddresses = StorageManager.current().getListsForGroup(group);

    if(rcptAddresses == null || rcptAddresses.size() == 0)

    {

      Log.msg("No ML-address for " + group + " found.", false);

      return;

    }

    for(String rcptAddress : rcptAddresses)

    {

      // Compose message and send it via given SMTP-Host

      String smtpHost = Config.inst().get(Config.MLSEND_HOST, "localhost");

      int smtpPort = Config.inst().get(Config.MLSEND_PORT, 25);

      String smtpUser = Config.inst().get(Config.MLSEND_USER, "user");

      String smtpPw = Config.inst().get(Config.MLSEND_PASSWORD, "mysecret");

      String smtpFrom = Config.inst().get(

        Config.MLSEND_ADDRESS, article.getHeader(Headers.FROM)[0]);

      // TODO: Make Article cloneable()

      article.getMessageID(); // Make sure an ID is existing

      article.removeHeader(Headers.NEWSGROUPS);

      article.removeHeader(Headers.PATH);

      article.removeHeader(Headers.LINES);

      article.removeHeader(Headers.BYTES);

      article.setHeader("To", rcptAddress);

      //article.setHeader("Reply-To", listAddress);

      if (Config.inst().get(Config.MLSEND_RW_SENDER, false))

      {

        rewriteSenderAddress(article); // Set the SENDER address

      }

      SMTPTransport smtpTransport = new SMTPTransport(smtpHost, smtpPort);

      smtpTransport.send(article, smtpFrom, rcptAddress);

      smtpTransport.close();

      Stats.getInstance().mailGatewayed(group);

      Log.msg("MLGateway: Mail " + article.getHeader("Subject")[0]

        + " was delivered to " + rcptAddress + ".", true);

    }

  }

  /**

   * Sets the SENDER header of the given MimeMessage. This might be necessary

   * for moderated groups that does not allow the "normal" FROM sender.

   * @param msg

   * @throws javax.mail.MessagingException

   */

  private static void rewriteSenderAddress(Article msg)

    throws MessagingException

  {

    String mlAddress = Config.inst().get(Config.MLSEND_ADDRESS, null);

    if(mlAddress != null)

    {

      msg.setHeader(Headers.SENDER, mlAddress);

    }

    else

    {

      throw new MessagingException("Cannot rewrite SENDER header!");

    }

  }

}