.forEach {
                buildType(FlakyTestQuarantine(model, stage, it))
            }

        model.stages
            .filter { it.stageName <= StageName.READY_FOR_RELEASE }
            .flatMap { stage -> stage.specificBuilds.map { it.create(model, stage, FlakyTestStrategy.ONLY) } }
            .filter { it.os == os }
            .filter { it is SmokeTests || it is SmokeIdeTests }

    },
    BuildLogicTest {
        override fun create(
            model: CIBuildModel,
            stage: Stage,
            flakyTestStrategy: FlakyTestStrategy,
        ): OsAwareBaseGradleBuildType = BuildLogicTest(model, stage)
    },
    BuildDistributions {
        override fun create(
            model: CIBuildModel,
            stage: Stage,
            flakyTestStrategy: FlakyTestStrategy,

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,
    override val failStage: Boolean = true,

    }
}

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)
        name = stage.stageName.stageName + " (Trigger)" + (os?.asName()?.toCapitalized()?.let { "($it)" } ?: "")

package configurations

import common.Os
import common.buildScanTagParam
import common.getBuildScanCustomValueParam
import model.CIBuildModel
import model.Stage

class BuildLogicTest(
    model: CIBuildModel,
    stage: Stage,
) : OsAwareBaseGradleBuildType(os = Os.LINUX, stage = stage, init = {
        id(buildTypeId(model))
        name = "Build-logic checks"
        description = "Run check on all build-logic builds"

import common.Os
import common.buildScanTagParam
import common.getBuildScanCustomValueParam
import common.setArtifactRules
import model.CIBuildModel
import model.Stage

class CompileAll(
    model: CIBuildModel,
    stage: Stage,
) : OsAwareBaseGradleBuildType(os = Os.LINUX, stage = stage, init = {
        id(buildTypeId(model))
        name = "Compile All"
        description = "Compiles all production/test source code and warms up the build cache"

            stage.flameGraphs.map { createFlameGraphs(model, stage, it) }

        val (topLevelCoverage, allCoverage) = stage.functionalTests.partition { it.testType == TestType.SOAK }
        val topLevelFunctionalTests =
            topLevelCoverage

package configurations

import common.Os
import common.buildScanTagParam
import common.getBuildScanCustomValueParam
import model.CIBuildModel
import model.Stage

class SanityCheck(
    model: CIBuildModel,
    stage: Stage,
) : OsAwareBaseGradleBuildType(os = Os.LINUX, stage = stage, init = {
        id(buildTypeId(model))
        name = "Sanity Check"
        description = "Static code analysis, checkstyle, release notes verification, etc."

import common.buildScanTagParam
import common.getBuildScanCustomValueParam
import common.setArtifactRules
import model.CIBuildModel
import model.Stage

class BuildDistributions(
    model: CIBuildModel,
    stage: Stage,
) : OsAwareBaseGradleBuildType(os = LINUX, stage = stage, init = {
        id("${model.projectId}_BuildDistributions")
        name = "Build Distributions"

    override fun createPerformanceTestsFor(
        model: CIBuildModel,
        stage: Stage,
        performanceTestCoverage: PerformanceTestCoverage,
        bucketIndex: Int,
    ): PerformanceTest =
        createPerformanceTest(
            model,
            performanceTestCoverage,
            stage,
            bucketIndex,
            "Performance tests for $testProject",