aboutsummaryrefslogtreecommitdiff
path: root/src/main/java/bitbucketpullrequestbuilder/bitbucketpullrequestbuilder/BitbucketAdditionalParameterEnvironmentContributor.java
blob: a4ff97d01a0fab2d47fd0f50a9850f74369fab2e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
package bitbucketpullrequestbuilder.bitbucketpullrequestbuilder;

import hudson.EnvVars;
import hudson.Extension;
import hudson.model.*;

import java.io.IOException;
import java.util.ArrayList;

@Extension
public class BitbucketAdditionalParameterEnvironmentContributor extends EnvironmentContributor {
    @Override
    public void buildEnvironmentFor(Run run, EnvVars envVars, TaskListener taskListener)
            throws IOException, InterruptedException {

        BitbucketCause cause = (BitbucketCause) run.getCause(BitbucketCause.class);
        if (cause == null) {
            return;
        }

        run.addAction(getDefaultParameters(run));

        putEnvVar(envVars, "sourceBranch", cause.getSourceBranch());
        putEnvVar(envVars, "targetBranch", cause.getTargetBranch());
        putEnvVar(envVars, "repositoryOwner", cause.getRepositoryOwner());
        putEnvVar(envVars, "repositoryName", cause.getRepositoryName());
        putEnvVar(envVars, "pullRequestId", cause.getPullRequestId());
        putEnvVar(envVars, "destinationRepositoryOwner", cause.getDestinationRepositoryOwner());
        putEnvVar(envVars, "destinationRepositoryName", cause.getDestinationRepositoryName());
        putEnvVar(envVars, "pullRequestTitle", cause.getPullRequestTitle());
        putEnvVar(envVars, "pullRequestAuthor", cause.getPullRequestAuthor());

    }

    private static void putEnvVar(EnvVars envs, String name, String value) {
        envs.put(name, getString(value, ""));
    }

    private static String getString(String actual, String d) {
        return actual == null ? d : actual;
    }

    private ParametersAction getDefaultParameters(Run<?, ?> run) {
        ArrayList<ParameterValue> values = new ArrayList<ParameterValue>();
        ParametersDefinitionProperty definitionProperty = run.getParent().getProperty(ParametersDefinitionProperty.class);
        if (definitionProperty != null) {
            for (ParameterDefinition pd : definitionProperty.getParameterDefinitions()) {
                values.add(pd.getDefaultParameterValue());
            }
        }
        return new ParametersAction(values);
    }
}