Gradle (groovy) task definitions to be able to run AWS CDK in a project with a main class

build.part.gradle

/**
 * AWS CDK integration. Changes the mainClassName variable and calls run
 * */
def cdkMainClassName = "com.rideal.infrastructure.aws.RidealApp"
def originalMainClassName = mainClassName

tasks.register("run-cdk") {
    description "Specific run-script for AWS CDK integration"
    group "CDK"

    doLast {
        mainClassName = cdkMainClassName
        println("Running up CDK integration")
    }
}

tasks.register("reset-main-class"){
    description "Resets the main class name after running CDK"
    group "CDK"
    mainClassName = originalMainClassName

    doLast {
        println("Cleaning up CDK integration")
    }
}

tasks.getByName("run-cdk").configure { finalizedBy tasks.getByName("run") }
tasks.getByName("run").configure {
    finalizedBy tasks.getByName("reset-main-class")
}