import configurations.FunctionalTest
import configurations.ParallelizationMethod
import java.util.SortedSet

class TestClassTime(
    val sourceSet: String,
    val buildTimeMs: Int,
)

data class TestCoverageAndBucketSplits(
    val testCoverageUuid: Int,
    val buckets: List<FunctionalTestBucket>,
)

data class FunctionalTestBucket(
    val subprojects: SortedSet<String>,
    val parallelizationMethod: ParallelizationMethod,

    private val buckets: Map<TestCoverage, List<SmallSubprojectBucket>> = buildBuckets(testTimeDataJson, model)

    fun generate(jsonFile: File) {
        val output =
            buckets.map {
                TestCoverageAndBucketSplits(it.key.uuid, it.value.map { it.toJsonBucket() })
            }
        jsonFile.writeText(gson.toJson(output))
    }

    private fun buildBuckets(
        buildClassTimeJson: File,