feat: 支持读取 pipeline 中定义的环境变量

close #195

Author: liuweiGL

src/main/java/io/jenkins/plugins/DingTalkRunListener.java

@@ -13,6 +13,7 @@
 import hudson.model.TaskListener;
 import hudson.model.User;
 import hudson.model.listeners.RunListener;
+import io.jenkins.plugins.context.PipelineEnvContext;
 import io.jenkins.plugins.enums.BuildStatusEnum;
 import io.jenkins.plugins.enums.MsgTypeEnum;
 import io.jenkins.plugins.enums.NoticeOccasionEnum;
@@ -158,16 +159,17 @@ private Map<String, String> getUser(Run<?, ?> run, TaskListener listener) {
   }
 
   private EnvVars getEnvVars(Run<?, ?> run, TaskListener listener) {
-    EnvVars envVars;
+    EnvVars jobEnvVars;
     try {
-      envVars = run.getEnvironment(listener);
+      jobEnvVars = run.getEnvironment(listener);
     } catch (InterruptedException | IOException e) {
-      envVars = new EnvVars();
+      jobEnvVars = new EnvVars();
       log.error(e);
-      log(listener, "获取环境变量时发生异常，将只使用 jenkins 默认的环境变量。");
-      log(listener, ExceptionUtils.getStackTrace(e));
+      DingTalkUtils.log(listener, "获取环境变量时发生异常，将只使用 jenkins 默认的环境变量。");
+      DingTalkUtils.log(listener, ExceptionUtils.getStackTrace(e));
     }
-    return envVars;
+    EnvVars pipelineEnvVars = PipelineEnvContext.get();
+    return jobEnvVars.overrideAll(pipelineEnvVars);
   }
 
   private boolean skip(TaskListener listener, NoticeOccasionEnum noticeOccasion,

src/main/java/io/jenkins/plugins/DingTalkStepListener.java

@@ -0,0 +1,24 @@
+package io.jenkins.plugins;
+
+import hudson.EnvVars;
+import hudson.Extension;
+import io.jenkins.plugins.context.PipelineEnvContext;
+import lombok.extern.slf4j.Slf4j;
+import org.jenkinsci.plugins.workflow.flow.StepListener;
+import org.jenkinsci.plugins.workflow.steps.Step;
+import org.jenkinsci.plugins.workflow.steps.StepContext;
+import org.springframework.lang.NonNull;
+
+@Slf4j
+@Extension
+public class DingTalkStepListener implements StepListener {
+  @Override
+  public void notifyOfNewStep(@NonNull Step step, @NonNull StepContext context) {
+    try {
+      EnvVars vars = context.get(EnvVars.class);
+      PipelineEnvContext.merge(vars);
+    } catch (Exception e) {
+      log.error("钉钉插件在获取 pipeline 中的环境变量时异常", e);
+    }
+  }
+}

src/main/java/io/jenkins/plugins/context/PipelineEnvContext.java

@@ -0,0 +1,23 @@
+package io.jenkins.plugins.context;
+
+import hudson.EnvVars;
+
+public class PipelineEnvContext {
+
+  private final static ThreadLocal<EnvVars> store = new ThreadLocal<>();
+
+  public static void merge(EnvVars value) {
+    EnvVars current = store.get();
+
+    if (current == null) {
+      store.set(value);
+    } else {
+      current.overrideAll(value);
+    }
+  }
+
+  public static EnvVars get() {
+    return store.get();
+  }
+
+}

src/main/java/io/jenkins/plugins/sdk/DingTalkSender.java

@@ -123,7 +123,7 @@ private String call(DingTalkRobotRequest request) {
         return body;
       }
     } catch (IOException e) {
-      log.error("钉钉消息发送失败：", e);
+      log.error("钉钉消息发送失败", e);
       return ExceptionUtils.getStackTrace(e);
     }
     return null;