String GITHUB_REPOSITORY = 'ows-royalties'
String ECR_ACCOUNT_ID = '086679231553'
List<String> AWS_REGIONS = ['us-east-1']
String SLACK_NOTIFICATIONS_CHANNEL = '#abacus-devs'
String QA_ACCOUNT_ID = '437795906767'
String QA_DEPLOYMENT_ROLE = 'prod-jenkins-aws-pipeline-agent'
String INTEGRATION_TEST_ROLE = 'qa-ows-royalties-test-role'
String SESSION_NAME = 'qa-ows-royalties-integration-test'
String UAT_ACCOUNT_ID = '989790945997'
String UAT_DEPLOYMENT_ROLE = 'uat-jenkins-pipeline-deploy-role'
String PROD_ACCOUNT_ID = '437795906767'
String PROD_DEPLOYMENT_ROLE = 'prod-jenkins-aws-pipeline-agent'
List<String> VULNERABILITIES_TO_IGNORE = [
]

String getEnvironment(String branchName = 'dev', boolean isProdDeploy = false) {
    switch (branchName) {
        case 'main':
        case 'master':
            return isProdDeploy ? 'prod' : 'qa'
        case 'qa':
            return 'qa'
        case 'uat':
            return 'uat'
        default:
            return 'dev'
    }
}

void configureEnv(def envVars, def paramVars) {
    String branch = envVars['BRANCH_NAME']?.trim()
    if (!branch) {
        String gitBranch = envVars['GIT_BRANCH']?.trim()
        branch = gitBranch ? gitBranch.tokenize('/')?.last() : 'dev'
    }
    boolean deployToProd = Boolean.parseBoolean("${paramVars.DEPLOY_TO_PROD}")
    envVars.TEST_ENVIRONMENT = getEnvironment(branch, deployToProd)
}


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: 'SHARED_LIBRARIES_VERSION', defaultValue: 'master', description: 'The version of the Jenkins shared libraries to use. Can be a branch, tag, Git revision or PR ref (e.g. pull/PR_NUMBER/merge).')
    }

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

    stages {
        stage('Load Shared Libraries') {
            steps {
                library "jenkins-global-libraries@${params.SHARED_LIBRARIES_VERSION}"
            }
        }
        stage('Compliance Checks') {
            steps {
                complianceChecks()
            }
        }
        stage('Validate Software Catalog Definition') {
            steps {
                datadogSoftwareCatalogValidate()
            }
        }
        stage('Unit Tests and Style Checks') {
            steps {
                script {
                    configureEnv(env, params)
                }
                withEcr{
                    sh "TEST_ENVIRONMENT=${env.TEST_ENVIRONMENT} make ci_unit_lint"
                }
            }
            post {
                always {
                    xunit tools: [JUnit(pattern: 'build/pyunit.xml', deleteOutputFiles: true, failIfNotNew: true, stopProcessingIfError: true)]
//                     recordCoverage(tools: [[parser: 'COBERTURA', pattern: 'build/coverage.xml']], enabledForFailure: true, failOnError: true, sourceDirectories: [[path: "royalties"]])
                }
                cleanup {
                    sh 'make ci_unit_lint_clean'
                }
            }
        }
        stage('Static Application Security Tests') {
            steps {
                sastTests()
            }
        }
        stage('Sonar Scan and Analysis') {
            when {
                branch 'master'
            }
            steps {
                sonarScan project: GITHUB_REPOSITORY, language: 'py'
            }
        }
        stage('Create a Release') {
            when {
                anyOf {
                    branch 'master'
                    expression { env.GITHUB_COMMENT =~ 'build docker' }
                }
            }
            steps {
                dockerToEcr awsRegions: AWS_REGIONS, ecrAccountId: ECR_ACCOUNT_ID, imageName: GITHUB_REPOSITORY, imageTag: env.GIT_COMMIT,
                    dockerBuildTarget: 'deploy'
            }
        }
        stage('Deploy to QA') {
            when {
                branch 'master'
            }
            steps {
                fargateDeploy environment: 'qa', awsRegions: AWS_REGIONS, gitCommit: env.GIT_COMMIT, serviceName: GITHUB_REPOSITORY,
                    ecrRegistryAccountId: ECR_ACCOUNT_ID, awsDeploymentTargetAccountId: QA_ACCOUNT_ID, awsDeploymentRoleName: QA_DEPLOYMENT_ROLE
            }
        }
        stage('Scan Docker Image') {
            when {
                anyOf {
                    branch 'master'
                    expression { env.GITHUB_COMMENT =~ 'build docker' }
                }
            }
            steps {
                dockerScan awsRegion: AWS_REGIONS[0], ecrAccountId: ECR_ACCOUNT_ID, imageName: GITHUB_REPOSITORY, imageTag: env.GIT_COMMIT, failBuild: false, vulnerabilitiesToIgnore: VULNERABILITIES_TO_IGNORE
            }
        }
        stage('Integration Tests') {
            when {
                anyOf {
                    branch 'master'
                    expression { env.GITHUB_COMMENT =~ 'run integration tests' }
                }
            }
            steps {
                withEcr {
                    withAWS(role: INTEGRATION_TEST_ROLE, roleAccount: QA_ACCOUNT_ID, roleSessionName: SESSION_NAME, useNode: true) {
                        withDatadogTestOptimizationEnv {
                            withEnv(["AWS_BUCKET_OWNER=${QA_ACCOUNT_ID}"]) {
                                sh 'make ci_test_integration'
                            }
                        }
                    }
                }
            }
            post {
                cleanup {
                    sh 'make ci_test_integration_clean'
                }
            }
        }
        stage('E2E Tests') {
            when {
                branch 'master'
            }
            environment {
                TAGS = "@ows_royalties"
            }
            steps {
                sleep(10)
                playwrightTests tags: env.TAGS,
                pnpmSetupScriptName: 'normalize-art-relations-vendor',
                reportSuffix: 'main-e2e-suite',
                slackNotificationChannels: [SLACK_NOTIFICATIONS_CHANNEL]
            }
        }

        stage('E2E Closing Balances Tests') {
            when {
                branch 'master'
            }
            environment {
                TAGS = "@ows_royalties_closing_balances"
            }
            steps {
                lock(resource: 'e2e-tests-documents') {
                    sleep(10)
                    playwrightTests tags: env.TAGS,
                    pnpmSetupScriptName: 'normalize-art-relations-vendor',
                    reportSuffix: 'closing-balances-results',
                    runtimeEnvironment: 'ecs',
                    slackNotificationChannels: [SLACK_NOTIFICATIONS_CHANNEL]
                }
            }
        }
        stage('Deploy to UAT') {
            when {
                branch 'master'
            }
            steps {
                fargateDeploy environment: 'uat', awsRegions: AWS_REGIONS, gitCommit: env.GIT_COMMIT, serviceName: GITHUB_REPOSITORY,
                    ecrRegistryAccountId: ECR_ACCOUNT_ID, awsDeploymentTargetAccountId: UAT_ACCOUNT_ID, awsDeploymentRoleName: UAT_DEPLOYMENT_ROLE
            }
        }
        stage('Deploy to Prod') {
            when {
                allOf {
                    branch 'master'
                    expression { params.DEPLOY_TO_PROD }
                }
            }
            steps {
                fargateDeploy environment: 'prod', awsRegions: AWS_REGIONS, gitCommit: env.GIT_COMMIT, serviceName: GITHUB_REPOSITORY,
                    ecrRegistryAccountId: ECR_ACCOUNT_ID, awsDeploymentTargetAccountId: PROD_ACCOUNT_ID, awsDeploymentRoleName: PROD_DEPLOYMENT_ROLE
                datadogSoftwareCatalogPublish()
            }
        }
    }

    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()
        }
    }
}