package com.tngtech.jgiven.gradle.internal;

import com.tngtech.jgiven.gradle.JGivenReport;
import com.tngtech.jgiven.report.AbstractReportConfig;
import com.tngtech.jgiven.report.AbstractReportGenerator;
import com.tngtech.jgiven.report.ReportGenerator;
import com.tngtech.jgiven.report.asciidoc.AsciiDocReportConfig;
import com.tngtech.jgiven.report.asciidoc.AsciiDocReportGenerator;
import com.tngtech.jgiven.report.html5.Html5ReportConfig;
import com.tngtech.jgiven.report.text.PlainTextReportConfig;
import com.tngtech.jgiven.report.text.PlainTextReportGenerator;
import org.gradle.api.Task;
import org.gradle.api.reporting.internal.TaskGeneratedSingleDirectoryReport;

import java.io.File;

public abstract class AbstractJGivenReportImpl extends TaskGeneratedSingleDirectoryReport implements JGivenReport {

    private File customCssFile;
    private File customJsFile;
    private String title;
    private boolean excludeEmptyScenarios = false;
    private boolean thumbnailsAreShown = true;

    public AbstractJGivenReportImpl( String name, Task task, String relativeEntryPath ) {
        super( name, task, relativeEntryPath );
    }

    public AbstractReportGenerator createGenerator() {
        AbstractReportConfig conf;
        AbstractReportGenerator generator;
        switch( getFormat() ) {
            case ASCIIDOC:
                conf = new AsciiDocReportConfig();
                generator = new AsciiDocReportGenerator();
                break;
            case TEXT:
                conf = new PlainTextReportConfig();
                generator = new PlainTextReportGenerator();
                break;
            case HTML:
            case HTML5:
            default:
                Html5ReportConfig customConf = new Html5ReportConfig();
                customConf.setShowThumbnails( isThumbnailsAreShown() );
                if (getCustomCssFile() != null) {
                    customConf.setCustomCss( getCustomCssFile() );
                }
                if (getCustomJsFile() != null) {
                    customConf.setCustomJs( getCustomJsFile() );
                }
                conf = customConf;
                generator = ReportGenerator.generateHtml5Report();
                break;
        }
        if( getTitle() != null ) {
            conf.setTitle( getTitle() );
        }
        conf.setTargetDir( getDestination() );
        conf.setExcludeEmptyScenarios( isExcludeEmptyScenarios() );
        generator.setConfig( conf );
        return generator;
    }

    public abstract ReportGenerator.Format getFormat();

    @Override public File getCustomCssFile() {
        return customCssFile;
    }

    @Override public void setCustomCssFile( File customCssFile ) {
        this.customCssFile = customCssFile;
    }

    @Override public File getCustomJsFile() {
        return customJsFile;
    }

    @Override public void setCustomJsFile( File customJsFile ) {
        this.customJsFile = customJsFile;
    }

    @Override public String getTitle() {
        return title;
    }

    @Override public void setTitle( String title ) {
        this.title = title;
    }

    @Override public boolean isExcludeEmptyScenarios() {
        return excludeEmptyScenarios;
    }

    @Override public void setExcludeEmptyScenarios( boolean excludeEmptyScenarios ) {
        this.excludeEmptyScenarios = excludeEmptyScenarios;
    }

    @Override public boolean isThumbnailsAreShown() {
        return thumbnailsAreShown;
    }

    @Override public void setThumbnailsAreShown( boolean thumbnailsAreShown ) {
        this.thumbnailsAreShown = thumbnailsAreShown;
    }

}


@@ -666,16 +666,54 @@
 yReport;
+%0Aimport org.gradle.api.tasks.Internal;
 %0A%0Aimport
@@ -2538,32 +2538,46 @@
 nerator;%0A    %7D%0A%0A
+    @Internal%0A
     public abstr

add missing annotation to future proof plugin