pipeline {
    parameters {
        string(
            defaultValue: '64',
            description: 'Batch size used in ADAM',
            name: 'BATCHSIZE',
            trim: true
        )
        string(
            defaultValue: '5',
            description: 'Number of iterations',
            name: 'EPOCHS',
            trim: true
        )
        gitParameter branchFilter: 'origin/(.*)', defaultValue: 'main', name: 'BRANCH', type: 'PT_BRANCH'
        buildSelector(
			defaultSelector: lastSuccessful(),
			description: 'Which build to use for copying artifacts',
			name: 'BUILD_SELECTOR'
		)
    }

    agent {
        docker {
            image 's444409-create-dataset'
        }
    }
    
    stages {
        stage('Get dataset from artifact') {
            steps {
                copyArtifacts projectName: 's444409-create-dataset', selector: lastSuccessful(), optional: true
            }
        }
        stage('Train model') {
            steps {
                sh "python train_model.py with 'epochs=${params.EPOCHS}' 'batch_size=${params.BATCHSIZE}'"
            }
        }
        stage('Archive model and evaluate it') {
            steps {
               archiveArtifacts artifacts: 'model_out', onlyIfSuccessful: true
               archiveArtifacts artifacts: 'sacred_runs/**', onlyIfSuccessful: true
               build job: "s444409-evaluation/${params.BRANCH}/", parameters: [string(name: 'BRANCH', value: "${params.BRANCH}")]
            }
        }
    }

    environment {
        NOTIFICATION_ADDRESS = 'e19191c5.uam.onmicrosoft.com@emea.teams.ms'
    }

    post {
        success {
            emailext body: 'SUCCESS', subject: "${env.JOB_NAME}", to: "${env.NOTIFICATION_ADDRESS}"
        }

        failure {
            emailext body: 'FAILURE', subject: "${env.JOB_NAME}", to: "${env.NOTIFICATION_ADDRESS}"
        }

        unstable {
            emailext body: 'UNSTABLE', subject: "${env.JOB_NAME}", to: "${env.NOTIFICATION_ADDRESS}"
        }

        changed {
            emailext body: 'CHANGED', subject: "${env.JOB_NAME}", to: "${env.NOTIFICATION_ADDRESS}"
        }	
    }
}