String GITHUB_REPOSITORY = 'ecs-abacus'
String ECR_ACCOUNT_ID = '086679231553'
List<String> AWS_REGIONS = ['us-east-1']
String SLACK_NOTIFICATIONS_CHANNEL = '#abacus-devs'
String QA_ACCOUNT_ID = '989790945997'
String QA_DEPLOYMENT_ROLE = 'qa-jenkins-pipeline-deploy-role'
String PROD_ACCOUNT_ID = '375914681009'
String PROD_DEPLOYMENT_ROLE = 'prod-jenkins-pipeline-deploy-role'

@groovy.transform.Field
List<String> tasksToBuild = null

pipeline {
    agent any

    options {
        ansiColor('xterm')
        disableConcurrentBuilds()
        timestamps()
    }

    parameters {
        booleanParam(name: 'DEPLOY_TO_PROD', defaultValue: true, description: 'Whether or not to deploy to prod.')
        string(name: 'TASK_NAME', defaultValue: '', description: 'Name of the task to build and deploy')
        string(name: 'SHARED_LIBRARIES_VERSION', defaultValue: 'master', description: 'The version of the Jenkins shared libraries to use. Can be a branch, tag or Git revision.')
    }

    triggers {
        issueCommentTrigger('.*retest this please.*')
    }

    stages {
        stage('Load Shared Libraries') {
            steps {
                library "jenkins-global-libraries@${params.SHARED_LIBRARIES_VERSION}"
            }
        }
        stage('Compliance Checks') {
            steps {
                script {
                    parallel(getTasksToBuild().collectEntries { taskName ->
                        return [
                            (taskName): {
                                echo "Running Compliance Checks for ${taskName}"
                                dir("tasks/${taskName}") {
                                    complianceChecks()
                                }
                            }
                        ]
                    })
                }
            }
        }
        stage('Validate Software Catalog Definition') {
            steps {
                datadogSoftwareCatalogValidate(servicePath: "tasks/extract_sales")
            }
        }
        stage('Unit Tests and Style Checks') {
            steps {
                script {
                    withEcr {
                        parallel(getTasksToBuild().collectEntries { taskName ->
                            return [
                                (taskName): {
                                    echo "Running Unit Tests and Style Checks for ${taskName}"
                                    dir("tasks/${taskName}") {
                                        sh 'make docker_test'
                                    }
                                }
                            ]
                        })
                    }
                }
            }
            post {
                cleanup {
                    script {
                        parallel(getTasksToBuild().collectEntries { taskName ->
                            return [
                                (taskName): {
                                    echo "Cleaning up for ${taskName}"
                                    dir("tasks/${taskName}") {
                                        sh 'make docker_test_down'
                                    }
                                }
                            ]
                        })
                    }
                }
            }
        }
        stage('Static Application Security Tests') {
            steps {
                script {
                    parallel(getTasksToBuild().collectEntries { taskName ->
                        return [
                            (taskName): {
                                echo "Running Static App Security Tests for ${taskName}"
                                sastTests(projectDir: "tasks/${taskName}")
                            }
                        ]
                    })
                }
            }
        }
        stage('Sonar Scan and Analysis') {
            when {
                branch 'master'
            }
            steps {
                script {
                    echo "Running Sonar Scan for ${GITHUB_REPOSITORY}"
                    sonarScan project: GITHUB_REPOSITORY, language: 'py'
                }
            }
        }
        stage('Create a Release') {
            when {
                branch 'master'
            }
            steps {
                script {
                    parallel(getTasksToBuild().collectEntries { taskName ->
                        return [
                            (taskName): {
                                echo "Creating a release for ${taskName}"
                                dockerToEcr awsRegions: AWS_REGIONS,
                                    ecrAccountId: ECR_ACCOUNT_ID,
                                    imageName: "${GITHUB_REPOSITORY}-${taskName.replaceAll("_", "-")}",
                                    imageTag: env.GIT_COMMIT,
                                    dockerBuildFile: "tasks/${taskName}/Dockerfile",
                                    dockerBuildContext: "tasks/${taskName}",
                                    dockerBuildTarget: 'deploy'
                            }
                        ]
                    })
                }
            }
        }
         stage('Deploy to QA') {
            when {
                branch 'master'
            }
            steps {
                script {
                    parallel(getTasksToBuild().collectEntries { taskName ->
                        return [
                            (taskName): {
                                echo "Deploying to QA for ${taskName}"
                                fargateDeploy environment: 'qa',
                                    awsRegions: AWS_REGIONS,
                                    gitCommit: env.GIT_COMMIT,
                                    ecrRegistryAccountId: ECR_ACCOUNT_ID,
                                    awsDeploymentTargetAccountId: QA_ACCOUNT_ID,
                                    awsDeploymentRoleName: QA_DEPLOYMENT_ROLE,
                                    serviceName: "${GITHUB_REPOSITORY}-${taskName.replaceAll("_", "-")}",
                                    deployType: 'CREATE_TASK_DEFINITION'
                            }
                        ]
                    })
                }
            }
        }
        stage('Scan Docker Image') {
            when {
                branch 'master'
            }
            steps {
                script {
                    parallel(getTasksToBuild().collectEntries { taskName ->
                        return [
                            (taskName): {
                                echo "Running Scan Docker Image for ${taskName}"
                                dockerScan awsRegion: AWS_REGIONS[0],
                                    ecrAccountId: ECR_ACCOUNT_ID,
                                    imageName: "${GITHUB_REPOSITORY}-${taskName.replaceAll("_", "-")}",
                                    imageTag: env.GIT_COMMIT,
                                    failBuild: false
                            }
                        ]
                    })
                }
            }
        }
        stage('Deploy to Prod') {
            when {
                allOf {
                    branch 'master'
                    expression { params.DEPLOY_TO_PROD }
                }
            }
            steps {
                script {
                    parallel(getTasksToBuild().collectEntries { taskName ->
                        return [
                            (taskName): {
                                echo "Deploying to Prod for ${taskName}"
                                fargateDeploy environment: 'prod',
                                    awsRegions: AWS_REGIONS,
                                    gitCommit: env.GIT_COMMIT,
                                    ecrRegistryAccountId: ECR_ACCOUNT_ID,
                                    awsDeploymentTargetAccountId: PROD_ACCOUNT_ID,
                                    awsDeploymentRoleName: PROD_DEPLOYMENT_ROLE,
                                    serviceName: "${GITHUB_REPOSITORY}-${taskName.replaceAll("_", "-")}",
                                    deployType: 'CREATE_TASK_DEFINITION'
                            }
                        ]
                    })
                    datadogSoftwareCatalogPublish(servicePath: "tasks/extract_sales")
                }
            }
        }
    }

    post {
        regression {
            script {
                if (env.BRANCH_NAME == 'master') {
                    slackNotify channel: SLACK_NOTIFICATIONS_CHANNEL
                }
            }
        }
        fixed {
            script {
                if (env.BRANCH_NAME == 'master') {
                    slackNotify channel: SLACK_NOTIFICATIONS_CHANNEL
                }
            }
        }
        cleanup {
            cleanWs()
        }
    }
}

def getTasksToBuild() {
    if (this.@tasksToBuild != null) {
        return this.@tasksToBuild
    }

    def taskName = params.TASK_NAME ?: null

    if (taskName) {
        this.@tasksToBuild = [taskName]
    } else {
        def directories = findFiles(glob: '**/Dockerfile')
        this.@tasksToBuild = getModifiedFunctions branchName: env.BRANCH_NAME,
            previousSuccessfulCommit: env.GIT_PREVIOUS_SUCCESSFUL_COMMIT,
            lambdaDirectories: directories
    }

    if (this.@tasksToBuild) {
        currentBuild.description = this.@tasksToBuild.join('<br>')
    }

    return this.@tasksToBuild
}