package configurations

import common.Os
import jetbrains.buildServer.configs.kotlin.BuildType
import model.Stage

open class BaseGradleBuildType(
    open val stage: Stage? = null,
    open val failStage: Boolean = true,
    init: BaseGradleBuildType.() -> Unit = {},
) : BuildType() {
    init {
        this.init()
    }
}

open class OsAwareBaseGradleBuildType(
    val os: Os,
    override val stage: Stage? = null,

class StageTrigger(
    model: CIBuildModel,
    stage: Stage,
    prevStage: Stage?,
    os: Os?,
    dependencies: List<BaseGradleBuildType>,
    generateTriggers: Boolean = true,
) : BaseGradleBuildType(init = {
        id(stageTriggerId(model, stage, os))
        uuid = stageTriggerUuid(model, stage, os)

package projects

import common.HIDDEN_ARTIFACT_DESTINATION
import common.cleanupRule
import common.isSecurityFork
import configurations.BaseGradleBuildType
import configurations.GitHubMergeQueueCheckPass
import configurations.PerformanceTestsPass
import configurations.StageTriggers
import jetbrains.buildServer.configs.kotlin.ParameterDisplay
import jetbrains.buildServer.configs.kotlin.Project
import model.CIBuildModel

package projects

import configurations.BaseGradleBuildType
import jetbrains.buildServer.configs.kotlin.Project
import model.CIBuildModel
import model.FunctionalTestBucketProvider
import model.Stage
import model.TestCoverage

const val DEFAULT_FUNCTIONAL_TEST_BUCKET_SIZE = 50
const val DEFAULT_LINUX_FUNCTIONAL_TEST_BUCKET_SIZE = 20
const val DEFAULT_MACOS_FUNCTIONAL_TEST_BUCKET_SIZE = 20

class FunctionalTestProject(

package projects

import common.HIDDEN_ARTIFACT_DESTINATION
import common.Os
import common.uuidPrefix
import configurations.BaseGradleBuildType
import configurations.DocsTestProject
import configurations.DocsTestTrigger
import configurations.FlakyTestQuarantineProject
import configurations.FlakyTestQuarantineTrigger
import configurations.FunctionalTest
import configurations.FunctionalTestsPass
import configurations.OsAwareBaseGradleBuildType

                filterAuthorRole = PullRequests.GitHubRoleFilter.EVERYBODY
                filterTargetBranch = "+:refs/heads/${VersionedSettingsBranch.fromDslContext().branchName}"
            }
    }
}

fun BaseGradleBuildType.tcParallelTests(numberOfBatches: Int) {
    if (numberOfBatches > 1) {
        params {
            param("env.TEAMCITY_PARALLEL_TESTS_ENABLED", "1")
        }
        features {
            parallelTests {

import common.uuidPrefix
import jetbrains.buildServer.configs.kotlin.DslContext
import model.CIBuildModel

class PartialTrigger<T : BaseGradleBuildType>(
    triggerName: String,
    triggerId: String,
    model: CIBuildModel,
    dependencies: Iterable<T>,
) : BaseGradleBuildType(init = {
        id("${model.projectId}_${triggerId}_Trigger")
        uuid = "${DslContext.uuidPrefix}_${model.projectId}_${triggerId}_Trigger"

    val testProjects: List<String>,
    val bucketIndex: Int,
    extraParameters: String = "",
    performanceTestTaskSuffix: String = "PerformanceTest",
    preBuildSteps: BuildSteps.() -> Unit = {},
) : BaseGradleBuildType(
        stage = stage,
        init = {
            this.id(performanceTestBuildSpec.asConfigurationId(model, "bucket${bucketIndex + 1}"))
            this.name = description

import jetbrains.buildServer.configs.kotlin.triggers.vcs
import model.CIBuildModel
import model.StageName

class GitHubMergeQueueCheckPass(
    model: CIBuildModel,
) : BaseGradleBuildType(init = {
        id("${model.projectId}_GitHubMergeQueueCheckPass")
        uuid = "${DslContext.uuidPrefix}_${model.projectId}_GitHubMergeQueueCheckPass"
        name = "GitHub Merge Queue Check Pass"

        type = Type.COMPOSITE

        applyDefaultSettings()

        dependencies {
            snapshotDependencies(
                flakyTestQuarantineProject.buildTypes.map {
                    it as BaseGradleBuildType
                },
            )
        }
    })

class FlakyTestQuarantineProject(
    model: CIBuildModel,
    stage: Stage,
    val os: Os,
) : Project({