node {
    checkout scm
    try {
        docker.image('s444452/ium:1.4').inside('-v /code/mlruns:/code/mlruns') {
        stage('Preparation') {
            properties([
                pipelineTriggers([upstream(threshold: hudson.model.Result.SUCCESS, upstreamProjects: "s444452-create-dataset")]),
                parameters([
                string(
                 defaultValue: ".",
                 description: 'Data path',
                 name: 'DATA_PATH'
                ),
                string(
                 defaultValue: "1",
                 description: 'EPOCHS',
                 name: 'EPOCHS'
                ),
                string(
                 defaultValue: "20000",
                 description: 'Num words',
                 name: 'NUM_WORDS'
                ),
                string(
                 defaultValue: "150",
                 description: 'Batch size',
                 name: 'BATCH_SIZE'
                ),
                string(
                 defaultValue: "300",
                 description: 'Pad length',
                 name: 'PAD_LENGTH'
                )
                ])
            ])
        }
        stage('Copy artifacts') {
            copyArtifacts filter: 'train_data.csv', fingerprintArtifacts: true, projectName: 's444452-create-dataset'
            copyArtifacts filter: 'test_data.csv', fingerprintArtifacts: true, projectName: 's444452-create-dataset'
        }
        stage('Run script') {
            withEnv(["DATA_PATH=${params.DATA_PATH}","EPOCHS=${params.EPOCHS}","NUM_WORDS=${params.NUM_WORDS}",
                "BATCH_SIZE=${params.BATCH_SIZE}","PAD_LENGTH=${params.PAD_LENGTH}"]) {
                sh "chmod +x Scripts/train_neural_network.py"
                sh "python3 Scripts/train_neural_network.py $DATA_PATH $EPOCHS $NUM_WORDS $BATCH_SIZE $PAD_LENGTH"
            }
        }
        stage('Archive artifacts') {
            archiveArtifacts "model/neural_net"
            archiveArtifacts "my_runs/**"
            archiveArtifacts "mlruns/**"
            archiveArtifacts "/mlruns/**"
            archiveArtifacts "/code/mlruns/**"
        }
        }
    } catch (e) {
        currentBuild.result = "FAILED"
        throw e
    } finally {
        notifyBuild(currentBuild.result)
    }
}
def notifyBuild(String buildStatus = 'STARTED') {
    buildStatus = buildStatus ?: 'SUCCESS'

    def subject = "Job: ${env.JOB_NAME}"
    def build_params = "Path: ${params.DATA_PATH}, Epochs: ${params.EPOCHS}, Num_words: ${params.NUM_WORDS}, Batch_size: ${params.BATCH_SIZE}, Pad_length: ${params.PAD_LENGTH}"
    def details = "Build nr: ${env.BUILD_NUMBER}, status: ${buildStatus} \n url: ${env.BUILD_URL} \n build params: ${build_params}"

    if (buildStatus == 'SUCCESS') {
        build (
            job: "s444452-evaluation/${env.BRANCH_NAME}",
            parameters: [
                gitParameter(name: "BRANCH", value: "${env.BRANCH_NAME}"),
                string(name: "BUILD_NR", value: "${env.BUILD_NUMBER}"),
                string(name: "DATA_PATH",  value: "${params.DATA_PATH}"),
                string(name: "EPOCHS",  value: "${params.EPOCHS}"),
                string(name: "NUM_WORDS",  value: "${params.NUM_WORDS}"),
                string(name: "BATCH_SIZE",  value: "${params.BATCH_SIZE}"),
                string(name: "PAD_LENGTH",  value: "${params.PAD_LENGTH}")
            ],
            wait: false
        )
    }

    emailext (
        subject: subject,
        body: details,
        to: 'e19191c5.uam.onmicrosoft.com@emea.teams.ms'
    )
}