pipeline {
    agent { 
        docker {
            image 'karolcyganik/ium:latest'
        }
    }

    parameters {
        string(
            defaultValue: 'karol9',
            description: 'Kaggle username',
            name: 'KAGGLE_USERNAME',
            trim: false
        )
        string(
            defaultValue: '100',
            description: 'Dataset cutoff size',
            name: 'CUTOFF'
        )
        password(
            defaultValue: '',
            description: 'Kaggle token taken from kaggle.json file, as described in https://github.com/Kaggle/kaggle-api#api-credentials',
            name: 'KAGGLE_KEY'
        )
    }

    stages {
        stage('Checkout') {
            steps {
                git branch: 'main', url: 'https://git.wmi.amu.edu.pl/s495715/iumKC.git'
            }
        }
        stage('Download Dataset') {
            steps {
                script {
                    def datasetDir = 'football_dataset'
                    def downloadDir = 'football_dataset_download'
                    if (!fileExists(downloadDir)) {
                        sh "mkdir ${downloadDir}"
                    }
                    
                    withEnv(["KAGGLE_USERNAME=${params.KAGGLE_USERNAME}",
                             "KAGGLE_KEY=${params.KAGGLE_KEY}"]) {
                        sh "kaggle datasets download sadhliroomyprime/football-semantic-segmentation -p ${downloadDir}"
                        sh "unzip ${downloadDir}/*.zip -d ${datasetDir}"
                    }
                }
            }
        }
        stage('Copy Artifacts') {
            steps {
                script {
                    copyArtifacts fingerprintArtifacts: true, projectName: 'z-s495715-create-dataset', selector: specific('16')
                }
            }
        }
        stage('Execute Script') {
            steps {
                sh 'python3 main.py'
            }
        }
    }
}