package org.apache.sling.performance;

import java.io.File;
import java.io.IOException;
import java.io.PrintWriter;
import java.text.DateFormat;
import java.text.SimpleDateFormat;
import java.util.Date;

import org.apache.commons.io.output.FileWriterWithEncoding;
import org.apache.commons.math.stat.descriptive.DescriptiveStatistics;

public class ReportLogger {

	public enum ReportType{
		TXT, XML
	}
	
	
	public static void writeReport(String test, String testSuiteName, String name, DescriptiveStatistics statistics, ReportType reportType) throws Exception{
		switch(reportType){
		case TXT:
			writeReportTxt(test, testSuiteName, name, statistics);
			break;
		case XML:
			throw new Exception("The XML reporting format is not yet supported");
		default:
			throw new Exception("The specified reporting format is not yet supported");
		}
	}
	
	/**
     * Method the writes the performance report after a test is run
     * @param test the test name
     * @param name the name that will be listed in the report
     * @param statistics the statistics data to be written
     * @throws IOException
     */
    public static void writeReportTxt(String test, String testSuiteName, String name, DescriptiveStatistics statistics)
   		    throws IOException {
    	
	        	String className=test;  
	        	className=className.substring(className.lastIndexOf(".")+1);
    	
    			File reportDir = new File("target/performance-reports");
    			if (!reportDir.exists()){
   		        	boolean test1 = reportDir.mkdir();	 
   		        }
    			
    			File report = new File("target/performance-reports", className + ".txt");
   		           		        
   		        // need this in the case a user wants to set the suite name from the command line
   		        // useful if we run the test cases from the command line for example 
   		        // by using maven
   		        if (testSuiteName.equals(ParameterizedTestList.TEST_CASE_ONLY)){
	   		        if (System.getenv("testsuitename") != null){
	   		        	testSuiteName = System.getenv("testsuitename");
	   		        }
   		        }
   		        
   		        boolean needsPrefix = !report.exists();
   		        PrintWriter writer = new PrintWriter(
   		                new FileWriterWithEncoding(report, "UTF-8", true));   
   		     try {
   	            if (needsPrefix) {
   	                writer.format(
   	                        "# %-34.34s     min     10%%     50%%     90%%     max%n",
   	                        className);
   	            }

   	            writer.format(
   	                    "%-36.36s  %6.0f  %6.0f  %6.0f  %6.0f  %6.0f%n",
   	                    testSuiteName,
   	                    statistics.getMin(),
   	                    statistics.getPercentile(10.0),
   	                    statistics.getPercentile(50.0),
   	                    statistics.getPercentile(90.0),
   	                    statistics.getMax());
   	        } finally {
   	            writer.close();
   	        }
    }
    
  
    /**
     * Get the date that will be written into the result file
     * @return
     */
    private static String getDate(){
		DateFormat dateFormat = new SimpleDateFormat("yyyy/MM/dd HH:mm:ss");
		Date date = new Date();

		return dateFormat.format(date);
	}
	
}