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,

        enabled = enable
    }
}

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)

     */
    private fun BaseGradleBuildType.getSubProjectSplitName() = largeSubProjectRegex.find(this.name)!!.groupValues[1]

    private fun BaseGradleBuildType.getGradleTasks(): String {
        val runnerStep = this.steps.items.find { it.name == "GRADLE_RUNNER" } as GradleBuildStep
        return runnerStep.tasks!!
    }

    private fun BaseGradleBuildType.getGradleParams(): String {

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

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"

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(

                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 {

    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"

import common.Os
import common.PLUGINS_PORTAL_URL_OVERRIDE
import common.VersionedSettingsBranch
import configurations.BaseGradleBuildType
import configurations.applyDefaults
import configurations.applyTestDefaults
import io.mockk.every
import io.mockk.impl.annotations.MockK
import io.mockk.junit5.MockKExtension
import io.mockk.mockk
import io.mockk.slot
import jetbrains.buildServer.configs.kotlin.BuildStep