package prog8tests

import org.junit.jupiter.api.DynamicTest
import org.junit.jupiter.api.DynamicTest.dynamicTest
import org.junit.jupiter.api.TestFactory
import org.junit.jupiter.api.TestInstance
import prog8.compiler.compileProgram
import prog8.compiler.target.C64Target
import prog8.compiler.target.Cx16Target
import prog8.compilerinterface.ICompilationTarget
import prog8tests.ast.helpers.assumeDirectory
import prog8tests.ast.helpers.mapCombinations
import prog8tests.ast.helpers.outputDir
import prog8tests.ast.helpers.workingDir
import prog8tests.helpers.assertSuccess
import kotlin.io.path.absolute
import kotlin.io.path.exists


/**
 * ATTENTION: this is just kludge!
 * They are not really unit tests, but rather tests of the whole process,
 * from source file loading all the way through to running 64tass.
 */
// @Disabled("disable to save some time")
@TestInstance(TestInstance.Lifecycle.PER_CLASS)
class TestCompilerOnExamples {
    private val examplesDir = assumeDirectory(workingDir, "../examples")

    private fun makeDynamicCompilerTest(name: String, platform: ICompilationTarget, optimize: Boolean) : DynamicTest {
        val searchIn = mutableListOf(examplesDir)
        if (platform == Cx16Target) {
            searchIn.add(0, assumeDirectory(examplesDir, "cx16"))
        }
        val filepath = searchIn
            .map { it.resolve("$name.p8") }
            .map { it.normalize().absolute() }
            .map { workingDir.relativize(it) }
            .first { it.exists() }
        val displayName = "${examplesDir.relativize(filepath.absolute())}: ${platform.name}, optimize=$optimize"
        return dynamicTest(displayName) {
            compileProgram(
                filepath,
                optimize,
                optimizeFloatExpressions = false,
                writeAssembly = true,
                slowCodegenWarnings = false,
                quietAssembler = true,
                compilationTarget = platform.name,
                sourceDirs = listOf(),
                outputDir
            ).assertSuccess("; $displayName")
        }
    }

    @TestFactory
// @Disabled("disable to save some time")
    fun bothCx16AndC64() = mapCombinations(
        dim1 = listOf(
            "animals",
            "balls",
            "cube3d",
            "cube3d-float",
            "cube3d-gfx",
            "cxlogo",
            "dirlist",
            "fibonacci",
            "line-circle-gfx",
            "line-circle-txt",
            "mandelbrot",
            "mandelbrot-gfx",
            "numbergame",
            "primes",
            "rasterbars",
            "screencodes",
            "sorting",
            "swirl",
            "swirl-float",
            "tehtriz",
            "textelite",
        ),
        dim2 = listOf(Cx16Target, C64Target),
        dim3 = listOf(false, true),
        combine3 = ::makeDynamicCompilerTest
    )

    @TestFactory
// @Disabled("disable to save some time")
    fun onlyC64() = mapCombinations(
        dim1 = listOf(
            "balloonflight",
            "bdmusic",
            "bdmusic-irq",
            "charset",
            "cube3d-sprites",
            "plasma",
            "sprites",
            "turtle-gfx",
            "wizzine",
        ),
        dim2 = listOf(C64Target),
        dim3 = listOf(false, true),
        combine3 = ::makeDynamicCompilerTest
    )

    @TestFactory
// @Disabled("disable to save some time")
    fun onlyCx16() = mapCombinations(
        dim1 = listOf(
            "vtui/testvtui",
            "amiga",
            "bobs",
            "cobramk3-gfx",
            "colorbars",
            "datetime",
            "highresbitmap",
            "kefrenbars",
            "mandelbrot-gfx-colors",
            "multipalette",
            "testgfx2",
        ),
        dim2 = listOf(Cx16Target),
        dim3 = listOf(false, true),
        combine3 = ::makeDynamicCompilerTest
    )
}