package org.apache.maven.plugin.surefire.booterclient.output;
   
   /*
    * Licensed to the Apache Software Foundation (ASF) under one
    * or more contributor license agreements.  See the NOTICE file
    * distributed with this work for additional information
    * regarding copyright ownership.  The ASF licenses this file
    * to you 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 org.apache.maven.surefire.report.ReportEntry;
  import org.apache.maven.surefire.util.NestedRuntimeException;
  
  import java.io.BufferedWriter;
  import java.io.File;
  import java.io.FileWriter;
  import java.io.IOException;
  import java.io.PrintWriter;
  
  /**
   * Surefire output consumer proxy that writes test output to a {@link File} for each test suite.
   *
   * This class is not threadsafe, but can be encapsulated with a SynchronizedOutputConsumer. It may still be
   * accessed from different threads (serially).
   *
   * @author <a href="mailto:carlos@apache.org">Carlos Sanchez</a>
   * @version $Id: FileOutputConsumerProxy.java 1049843 2010-12-16 09:32:19Z krosenvold $
   * @since 2.1
   */
  public class FileOutputConsumerProxy
      extends OutputConsumerProxy
  {
  
      private static final String LINE_SEPARATOR = System.getProperty( "line.separator" );
  
      private final File reportsDirectory;
  
      private final StringBuffer outputBuffer = new StringBuffer();
  
      private volatile PrintWriter printWriter;
  
      /**
       * Create a consumer that will write to a {@link File} for each test
       *
       * @param outputConsumer   the output consumer
       * @param reportsDirectory directory where files will be saved
       */
      public FileOutputConsumerProxy( OutputConsumer outputConsumer, File reportsDirectory )
      {
          super( outputConsumer );
          this.reportsDirectory = reportsDirectory;
      }
  
      public void testSetStarting( ReportEntry reportEntry )
      {
          if ( printWriter != null )
          {
              throw new IllegalStateException( "testSetStarting called twice" );
          }
  
          if ( !reportsDirectory.exists() )
          {
              //noinspection ResultOfMethodCallIgnored
              reportsDirectory.mkdirs();
          }
  
          File file = new File( reportsDirectory, reportEntry.getName() + "-output.txt" );
          try
          {
              this.printWriter = new PrintWriter( new BufferedWriter( new FileWriter( file ) ) );
          }
          catch ( IOException e )
          {
              throw new NestedRuntimeException( e );
          }
          super.testSetStarting( reportEntry );
      }
  
      public void testSetCompleted()
      {
          if ( printWriter == null )
          {
              throw new IllegalStateException( "testSetCompleted called before testSetStarting" );
          }
          if ( outputBuffer.length() > 0 )
          {
              printWriter.write( outputBuffer.toString() );
              printWriter.write( LINE_SEPARATOR );
             outputBuffer.setLength( 0 );
         }
         printWriter.close();
         this.printWriter = null;
         super.testSetCompleted();
     }
 
     /**
      * Write the output to the current test file
      * <p/>
      */
     public void consumeOutputLine( String line )
     {
         if ( printWriter == null )
         {
             outputBuffer.append( line );
             outputBuffer.append( LINE_SEPARATOR );
             return;
         }
 
         if ( outputBuffer.length() > 0 )
         {
             printWriter.write( outputBuffer.toString() );
             printWriter.write( LINE_SEPARATOR );
             outputBuffer.setLength( 0 );
         }
         printWriter.write( line );
         printWriter.write( LINE_SEPARATOR );
     }
 
 }