pipeline {
    agent any

    environment {
        REGISTRY = 'atl.vultrcr.com/vllm'
        IMAGE_NAME = 'vllm'
        IMAGE = "${REGISTRY}/${IMAGE_NAME}:latest"
    }

    parameters {
        string(name: 'IMAGE_TAG', defaultValue: 'custom', description: 'Docker image tag')
        string(name: 'GIT_REPO', defaultValue: '', description: 'Git repository URL (optional, uses workspace if empty)')
        string(name: 'GIT_BRANCH', defaultValue: 'master', description: 'Git branch to build')
    }

    stages {
        stage('Checkout') {
            steps {
                script {
                    if (params.GIT_REPO) {
                        git url: params.GIT_REPO, branch: params.GIT_BRANCH
                    }
                }
            }
        }

        stage('Build') {
            steps {
                script {
                    docker.withRegistry("https://${REGISTRY}", 'ATL_VCR_VLLM') {
                        sh """
                            docker build -t ${REGISTRY}/${IMAGE_NAME}:${params.IMAGE_TAG} .
                        """
                    }
                }
            }
        }

        stage('Push') {
            steps {
                script {
                    docker.withRegistry("https://${REGISTRY}", 'ATL_VCR_VLLM') {
                        docker.image("${REGISTRY}/${IMAGE_NAME}:${params.IMAGE_TAG}").push()
                    }
                }
            }
        }
    }

    post {
        always {
            sh "docker rmi ${IMAGE} || true"
        }
    }
}