Add environment variables to every Run caused by StashCause

This should provide the environment variables to pipeline projects.

Author: proski

src/main/java/stashpullrequestbuilder/stashpullrequestbuilder/StashAditionalParameterEnvironmentContributor.java

@@ -16,23 +16,26 @@ public class StashAditionalParameterEnvironmentContributor extends EnvironmentCo
   public void buildEnvironmentFor(
       @Nonnull Run r, @Nonnull EnvVars envs, @Nonnull TaskListener listener)
       throws IOException, InterruptedException {
+    Run<?, ?> rootRun;
     if (r instanceof AbstractBuild) {
-      AbstractBuild build = (AbstractBuild) r;
-      AbstractBuild rootBuild = build.getRootBuild();
+      AbstractBuild<?, ?> build = (AbstractBuild) r;
+      rootRun = build.getRootBuild();
+    } else {
+      rootRun = (Run<?, ?>) r;
+    }
 
-      StashCause cause = (StashCause) rootBuild.getCause(StashCause.class);
-      if (cause != null) {
-        putEnvVar(envs, "sourceBranch", cause.getSourceBranch());
-        putEnvVar(envs, "targetBranch", cause.getTargetBranch());
-        putEnvVar(envs, "sourceRepositoryOwner", cause.getSourceRepositoryOwner());
-        putEnvVar(envs, "sourceRepositoryName", cause.getSourceRepositoryName());
-        putEnvVar(envs, "pullRequestId", cause.getPullRequestId());
-        putEnvVar(envs, "destinationRepositoryOwner", cause.getDestinationRepositoryOwner());
-        putEnvVar(envs, "destinationRepositoryName", cause.getDestinationRepositoryName());
-        putEnvVar(envs, "pullRequestTitle", cause.getPullRequestTitle());
-        putEnvVar(envs, "sourceCommitHash", cause.getSourceCommitHash());
-        putEnvVar(envs, "destinationCommitHash", cause.getDestinationCommitHash());
-      }
+    StashCause cause = (StashCause) rootRun.getCause(StashCause.class);
+    if (cause != null) {
+      putEnvVar(envs, "sourceBranch", cause.getSourceBranch());
+      putEnvVar(envs, "targetBranch", cause.getTargetBranch());
+      putEnvVar(envs, "sourceRepositoryOwner", cause.getSourceRepositoryOwner());
+      putEnvVar(envs, "sourceRepositoryName", cause.getSourceRepositoryName());
+      putEnvVar(envs, "pullRequestId", cause.getPullRequestId());
+      putEnvVar(envs, "destinationRepositoryOwner", cause.getDestinationRepositoryOwner());
+      putEnvVar(envs, "destinationRepositoryName", cause.getDestinationRepositoryName());
+      putEnvVar(envs, "pullRequestTitle", cause.getPullRequestTitle());
+      putEnvVar(envs, "sourceCommitHash", cause.getSourceCommitHash());
+      putEnvVar(envs, "destinationCommitHash", cause.getDestinationCommitHash());
     }
 
     super.buildEnvironmentFor(r, envs, listener);