import org.jenkinsci.plugins.workflow.job.*
import org.jenkinsci.plugins.workflow.flow.*
import org.jenkinsci.plugins.workflow.cps.*
import org.jenkinsci.plugins.workflow.graph.FlowNode;
import io.jenkins.blueocean.rest.impl.pipeline.*
import io.jenkins.blueocean.rest.model.*
import io.jenkins.blueocean.rest.model.BlueRun.*
@NonCPS
def on( String job, int buildNumber ) {
[
isBuilding : { ->
isBuilding( job, buildNumber )
}
stageStatus : { String keyword, String type = 'parallel', String parentStage = 'Parallel' ->
stageStatus ( keyword, job, buildNumber, type, parentStage )
}
]
}
Boolean isBuilding( String job, int buildNumber ) {
jenkins.model.Jenkins.instance
.getItemByFullName( job )
.getBuildByNumber( buildNumber )
.isInProgress()
}
def withFlowNodes( String name, int buildNumber, Closure body ) {
WorkflowRun run = jenkins.model.Jenkins.instance
.getItemByFullName( name )
.getBuildByNumber( buildNumber )
PipelineNodeGraphVisitor visitor = new PipelineNodeGraphVisitor( run )
List<FlowNodeWrapper> flowNodes = visitor.getPipelineNodes()
body( flowNodes )
}
def stageStatus( String keyword ,
String job ,
int buildNumber ,
String type = 'parallel' ,
String parentStage = 'Parallel'
) {
if ( ! isBuilding(job, buildNumber) ) {
println( "pipeline ${job} #${buildNumber} haven't started yet" )
return false
}
withFlowNodes ( job, buildNumber ) { flowNodes ->
List<String> parentIds = flowNodes.findAll {
it.displayName.startsWith(keyword) && it.getType() == FlowNodeWrapper.NodeType.valueOf( type.toUpperCase() )
}.collectMany { it.parents.findAll { p -> p.displayName == parentStage }
.collect { p -> p.id }
}.unique()
return parentIds
? flowNodes.findAll { parentIds.contains( it.id ) }.collect { it.status.getState() }
: false
} // withFlowNodes
}
// call
on( BUILD_NAME, BUILD_NUMBER ).stageStatus( stageName )