package org.ssdt_ohio.gradle.plugins

import org.gradle.api.Plugin
import org.gradle.api.Project
import org.gradle.api.plugins.ExtraPropertiesExtension.UnknownPropertyException
import org.gradle.api.plugins.GroovyPlugin

class VersionClassPlugin implements Plugin<Project> {

    def void apply(Project project) {
        project.getPlugins().apply(GroovyPlugin.class)
        def genSrc = 'generated-src/version'
        def generatedSrcDir = new File(project.buildDir, genSrc)

        def makeVersionClassTask = project.task('makeVersionClass') << {

            def now = new Date().format('yyyy-MM-dd hh:mm:ss')

            def buildNumber = -1
            if (project.rootProject?.file('build-number.txt').exists()) {
                def props = new Properties()
                project.rootProject?.file('build-number.txt').withInputStream {stream ->
                    props.load(stream)
                }
                buildNumber = props['build.number']
            }

            def targetPackage = getTargetClass(project)

            if (!targetPackage) {
                return
            }

            def outFile = new File(generatedSrcDir, "groovy/" + targetPackage.replaceAll('\\.', "/") + "/ProjectVersion.groovy")

            outFile.getParentFile().mkdirs()
            logger.info("creating $targetPackage in $outFile")
            def f = new FileWriter(outFile)
            f.write("""
package $targetPackage
/**
 * Generated by gradle build. $now ($buildNumber)
 * @version ${project.version}
 */
public class ProjectVersion {

    final String name = "${project.name}"
    final String group = "${project.group}"
    final String version = "${project.version}"
    final String buildDate = "$now"
    final int buildNumber = $buildNumber

    String getFullVersion() {
        "\$version \$buildDate (#\$buildNumber)"
    }

    String getId() {
        "\$group:\$name:\$version"
    }

    String toString() {
        "\$id \$buildDate (#\$buildNumber)"
    }
}
""")
            f.close()
        }

        project.sourceSets {
            main {
                groovy {
                    srcDir project.buildDir.name + '/' + genSrc + '/groovy'
                }
            }
        }

        makeVersionClassTask.getInputs().files(project.sourceSets.main.getAllSource().findAll{ !it.name.startsWith('ProjectVersion')}
        )
        makeVersionClassTask.getOutputs().file( project.buildDir.name + '/' + genSrc)

        if (project.getBuildFile() != null && project.getBuildFile().exists()) {
            makeVersionClassTask.getInputs().files(project.getBuildFile())
        }

        project.getTasks().getByName('compileGroovy').dependsOn('makeVersionClass')


    }



    private getTargetClass(Project project) {
        try {
            return project.extensions.extraProperties.get('versionPackage')
        } catch (UnknownPropertyException ex) {}

        def source = project.sourceSets.main.groovy.getSrcDirTrees()

        def files = project.sourceSets.main.groovy as File[]
        if (!files ) {
            return null
        }

        def targetDir = files.sort {
            it.parentFile.toString().replaceAll('(\\\\|/)','.').split(/\./).size()
        }.first().parentFile.toString()

        def targetPackage = targetDir
        source.each {
            targetPackage -= it.getDir().getPath().toString()
        }
        targetPackage = targetPackage.replaceAll("\\\\", '.').replaceAll('/', '.')
        targetPackage = targetPackage.startsWith(".") ? targetPackage - "." : targetPackage

        return targetPackage

    }

}