pipeline {
    agent {
        docker { image 'paweldopierala/ium:2.0.0' }
    }

    parameters{
        buildSelector(
            defaultSelector: lastSuccessful(),
            description: 'Which build to use for copying artifacts (create-dataset, evaluation)',
            name: 'BUILD_SELECTOR_MAIN'
        )
        buildSelector(
            defaultSelector: lastSuccessful(),
            description: 'Which build to use for copying artifacts (training)',
            name: 'BUILD_SELECTOR_TRAINING'
        )
        gitParameter(
            branchFilter: '.*',
            defaultValue: 'main',
            name: 'TRAINING_BRANCH',
            type: 'PT_BRANCH',
            description: 'Branch for training project'
        )
    }

    triggers {
        upstream(upstreamProjects: 's495719-training/' + params.TRAINING_BRANCH, threshold: hudson.model.Result.SUCCESS)
    }

    stages {
        stage('Git') {
            steps {
                git(
                    url: "https://git.wmi.amu.edu.pl/s495719/ium_495719.git",
                    branch: "main"
                )
            }
        }
        stage('CopyArtifacts') {
            steps {
                copyArtifacts fingerprintArtifacts: true, projectName: 'z-s495719-create-dataset', selector: buildParameter('BUILD_SELECTOR_MAIN')
                copyArtifacts fingerprintArtifacts: true, projectName: 's495719-training/' + params.TRAINING_BRANCH, selector: buildParameter('BUILD_SELECTOR_MAIN')
                copyArtifacts fingerprintArtifacts: true, projectName: 's495719-evaluation/main', selector: buildParameter('BUILD_SELECTOR_TRAINING'), optional: true
            }
        }
        stage('Script') {
            steps {
                sh 'chmod 777 ./evaluate.py'
                sh "python3 ./evaluate.py ${currentBuild.number}"
            }
        }
        stage('CreateArtifacts') {
            steps {
                archiveArtifacts artifacts: 'hp_test_predictions.csv,hp_test_metrics.csv,plot_mae.png, plot_r2.png, plot_rmse.png'
            }
        }
    }
}