package net.sf.logdistiller.publishers;
   
   /*
    * Licensed under the Apache License, Version 2.0 (the "License");
    * you may not use this file except in compliance with the License.
    * You may obtain a copy of the License at
    *
    *     http://www.apache.org/licenses/LICENSE-2.0
    *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  
  import java.io.*;
  import java.util.Properties;
  import javax.activation.DataHandler;
  import javax.activation.DataSource;
  import javax.activation.FileDataSource;
  import javax.mail.*;
  import javax.mail.internet.*;
  
  import net.sf.logdistiller.*;
  
  /**
   * Send mail report of a LogDistillation. When the report is a group report, a file containing logevents (if saved) is
   * added as attachment. Global output parameters are used as JavaMail session properties: the most important property is
   * <code>mail.smtp.host</code>, but see JavaMail documentation to find all available properties.
   * <p>
   * Parameters:
   * <ul>
   * <li><code>to</code>, <code>cc</code> (at least one): recipients of the mail, comma separated values if multiple
   * recipients
   * <li><code>from</code> (default: <code>smtp.from</code> global output parameter, or
   * <code>logdistiller@sourceforge.net</code>
   * <li><code>attachLogs</code> (default: <code>true</code>): add corresponding logevents as attachment
   * </ul>
   */
  public class MailPublisher
      extends Publisher
  {
      public String getId()
      {
          return "mail";
      }
  
      public void publish( LogDistillation logdistillation, LogDistiller.Report report )
          throws IOException, PublishException
      {
          String content = logdistillation.getContent();
          String subject =
              "[logdistiller:" + logdistillation.getDefinition().getOutput().getId() + "]"
                  + ( ( content == null ) ? "" : ( " " + content ) ) + " global report: "
                  + logdistillation.getEventCount() + " logevents";
          ReportFormat format = ReportFormats.getReportFormat( report.getFormat() );
          StringWriter sw = new StringWriter();
          format.report( logdistillation, sw );
          String body = sw.toString();
          try
          {
              sendMail( logdistillation, report, subject, body, format.getContentType(), null );
          }
          catch ( MessagingException me )
          {
              throw new PublishException( "error while sending global report by mail: " + me.getMessage(), me );
          }
      }
  
      public void publish( LogDistillation.Group group, LogDistiller.Report report )
          throws IOException, PublishException
      {
          LogDistillation logdistillation = group.getLogdistillation();
          String content = logdistillation.getContent();
          String subject =
              "[logdistiller:" + logdistillation.getDefinition().getOutput().getId() + "]"
                  + ( ( content == null ) ? "" : ( " " + content ) ) + " report for group "
                  + group.getDefinition().getId() + ": " + group.getEventCount() + " logevents";
          ReportFormat format = ReportFormats.getReportFormat( report.getFormat() );
          StringWriter sw = new StringWriter();
          format.report( group, sw );
          String body = sw.toString();
          try
          {
              sendMail( logdistillation, report, subject, body, format.getContentType(), group.getLogFile() );
          }
          catch ( MessagingException me )
          {
              throw new PublishException( "error while sending group report by mail: " + me.getMessage(), me );
          }
      }
  
      protected void sendMail( LogDistillation logdistillation, LogDistiller.Report report, String subject,
                               String content, String mimetype, File attach )
          throws MessagingException
      {
          LogDistiller ld = logdistillation.getDefinition();
          String to = report.getParam( "to" );
         String cc = report.getParam( "cc" );
         if ( ( to == null ) && ( cc == null ) )
         {
             // nobody to send to: TODO add warning
             return;
         }
         Properties props = new Properties();
         props.putAll( ld.getOutput().getParams() );
         Session session = Session.getDefaultInstance( props, null );
         MimeMessage msg = new MimeMessage( session );
 
         msg.setSubject( subject );
         Multipart mp = new MimeMultipart();
         BodyPart body = new MimeBodyPart();
         body.setContent( content, mimetype );
         mp.addBodyPart( body );
         msg.setContent( mp );
         if ( to != null )
         {
             msg.setRecipients( Message.RecipientType.TO, InternetAddress.parse( to ) );
         }
         if ( cc != null )
         {
             msg.setRecipients( Message.RecipientType.CC, InternetAddress.parse( cc ) );
         }
         String from = report.getParam( "from", "logdistiller@sourceforge.net" );
         msg.setFrom( new InternetAddress( from ) );
 
         String attachLogs = report.getParam( "attachLogs", "true" );
         if ( ( attach != null ) && ( attach.exists() ) && ( "true".equals( attachLogs ) ) )
         {
             BodyPart attachment = new MimeBodyPart();
             DataSource ds = new FileDataSource( attach );
             attachment.setDataHandler( new DataHandler( ds ) );
             attachment.setFileName( attach.getName() );
             mp.addBodyPart( attachment );
         }
 
         sendMail( msg );
     }
 
     protected void sendMail( Message msg )
         throws MessagingException
     {
         Transport.send( msg );
     }
 }