compose-jb/gradle-plugins/compose/src/test/kotlin/org/jetbrains/compose/test/TestProject.kt

/*
 * Copyright 2020-2021 JetBrains s.r.o. and respective authors and developers.
 * Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE.txt file.
 */

package org.jetbrains.compose.test

import org.gradle.testkit.runner.GradleRunner
import org.jetbrains.compose.desktop.application.internal.ComposeProperties
import java.io.File

data class TestEnvironment(
    val workingDir: File,
    val kotlinVersion: TestKotlinVersion = TestKotlinVersion.Default,
    val composeVersion: String = TestProperties.composeVersion
)

class TestProject(
    private val name: String,
    private val testEnvironment: TestEnvironment
) {
    private val additionalArgs = listOf(
        "--stacktrace",
        "-P${ComposeProperties.VERBOSE}=true"
    )

    init {
        val originalTestRoot = File("src/test/test-projects").resolve(name).also {
            check(it.exists()) { "Test project is not found: ${it.absolutePath}" }
        }
        for (orig in originalTestRoot.walk()) {
            if (!orig.isFile) continue

            val target = testEnvironment.workingDir.resolve(orig.relativeTo(originalTestRoot))
            target.parentFile.mkdirs()

            if (orig.name.endsWith(".gradle") || orig.name.endsWith(".gradle.kts")) {
                val origContent = orig.readText()
                val newContent = origContent
                    .replace("COMPOSE_VERSION_PLACEHOLDER", testEnvironment.composeVersion)
                    .replace("KOTLIN_VERSION_PLACEHOLDER", testEnvironment.kotlinVersion.versionString)
                target.writeText(newContent)
            } else {
                orig.copyTo(target)
            }
        }
    }

    fun gradle(vararg args: String): GradleRunner =
        GradleRunner.create().apply {
            withGradleVersion(TestProperties.gradleVersionForTests)
            withProjectDir(testEnvironment.workingDir)
            withArguments(args.toList() + additionalArgs)
            forwardOutput()
        }

    @Suppress("DeprecatedCallableAddReplaceWith")
    @Deprecated("Do not commit!")
    fun gradleDebug(vararg args: String): GradleRunner =
        gradle(*args).withDebug(true)

    fun file(path: String): File =
        testEnvironment.workingDir.resolve(path)
}