#!groovy
def withCheck(String blockName, Closure closure) {
    script {
        def buildStage = true
        catchError(message: 'check previous build status', stageResult:'SUCCESS', buildResult: 'SUCCESS') {
            unstash name:"${blockName}"
            buildStage = false
        }

        if (buildStage) {
            closure.call()
            writeFile file: "${blockName}", text: "1"
            stash name: "${blockName}", includes: "${blockName}"
        }
    }
}

pipeline {
    agent none

    options {
        preserveStashes()
    }

    stages {
        stage("Build and test") {
            parallel() {
                stage("Build/Test Win64") {
                    agent {
                        label 'master'
                    }
                    steps {
                        withCheck("build-deploy-Win64") {
                            echo "test"
                        }

                        withCheck("test-Win64") {
                            echo "test"
                        }
                    }
                }

                stage("Build/Test Win32") {
                    agent {
                        label 'master'
                    }
                    steps {
                        withCheck("build-deploy-Win32") {
                            echo "test"
                        }

                        withCheck("test-Win32") {
                            echo "test"
                        }
                    }
                }
            }
        }

        stage("Deploy") {
            agent {
                label 'master'
            }
            steps {
                withCheck("build-deploy-win64") {
                    echo "test"
                }

                withCheck("test-win64") {
                    echo "test"
                }
            }
        }
    }
}

 def withCheck(String blockName, Closure closure) {
    script {
        def buildStage = true
        catchError(message: 'check previous build status', stageResult:'SUCCESS', buildResult: 'SUCCESS') {
            copyArtifactsFromRestartedBuild(context.env.JOB_NAME, context.env.BUILD_NUMBER, "stageStates/${blockName}")
            buildStage = false
        }

        if (buildStage) {
            closure.call()
            writeFile file: "${blockName}", text: "1"
            archiveArtifacts artifacts: "stageStates/${blockName}"
        }
    }
}

def getBuildActions(String jobName, int buildNumber) {
   return Jenkins.instance.getItemByFullName(jobName).getBuildByNumber(buildNumber).properties.actions
}

def filterBuildAction(def buildActions, String filter = "RestartFlowFactoryAction") {
   return buildActions.findAll { it.getClass().toString().contains(filter) }
}

String getRestartedBuildData(String jobName, int buildNumber) {
   def buildActions = getBuildActions(jobName, buildNumber)
   def filteredAction = filterBuildAction(buildActions, "RestartFlowFactoryAction")
   if (filteredAction.size() == 1) {
      return filteredAction[0].getOriginRunId()
   } else {
      return null
   }
}

void copyArtifactsFromRestartedBuild(String jobName, String buildNumber, String fileFilter) {
   String restartedBuildData = getRestartedBuildData(jobName, buildNumber as Integer)
   String restartedJobName = restartedBuildData.split("#")[0]
   String restartedBuildNumber = restartedBuildData.split("#")[-1]
   copyArtifacts(filter: fileFilter, projectName: restartedJobName, selector: context.specific("${restartedBuildNumber}"))
}