1 files changed, 112 insertions, 0 deletions
diff --git a/buildSrc/src/main/java/deployment/DeploymentRequestProcessStepTask.java b/buildSrc/src/main/java/deployment/DeploymentRequestProcessStepTask.java
new file mode 100644
index 000000000..b7c2769a7
--- /dev/null
+++ b/buildSrc/src/main/java/deployment/DeploymentRequestProcessStepTask.java
@@ -0,0 +1,112 @@
+package deployment;
+
+import github.DeploymentCreate;
+import java.util.Collections;
+import java.util.List;
+import java.util.Locale;
+import java.util.Map;
+import java.util.stream.Collectors;
+import javax.inject.Inject;
+import org.gradle.api.DefaultTask;
+import org.gradle.api.tasks.Input;
+import org.gradle.api.tasks.TaskAction;
+
+public class DeploymentRequestProcessStepTask extends DefaultTask {
+
+    static class DeploymentCommandLineArgs extends RequestCommandLineArgs {
+        static final String PROP_KEY_SHA = "Request.sha";
+        final String sha;
+
+        DeploymentCommandLineArgs(Map<String, ?> properties) {
+            super(properties);
+            this.sha = properties.get(PROP_KEY_SHA).toString();
+        }
+    }
+
+    private final DeploymentProcessConfiguration mConfiguration;
+    private final int mStepIndex;
+
+    @Input
+    public String getEnvironmentKey() {
+        return getEnvironmentName(mConfiguration, mStepIndex);
+    }
+
+    @Inject
+    public DeploymentRequestProcessStepTask(
+            DeploymentProcessConfiguration configuration, int stepIndex) {
+        mConfiguration = configuration;
+        mStepIndex = stepIndex;
+        setGroup("Publishing");
+        setDescription("Request deployment of " + getEnvironmentName(configuration, stepIndex));
+    }
+
+    @TaskAction
+    public void deploymentRequestAction() {
+        try {
+            deploymentRequest(
+                    new DeploymentCommandLineArgs(getProject().getProperties()),
+                    mConfiguration,
+                    mStepIndex);
+        } catch (Exception e) {
+            throw new RuntimeException(e);
+        }
+    }
+
+    private static void deploymentRequest(
+            DeploymentCommandLineArgs data,
+            DeploymentProcessConfiguration configuration,
+            int stepIndex)
+            throws Exception {
+
+        DeploymentCreate deploymentCreate =
+                new DeploymentCreate(data.apiUsername, data.apiUserToken);
+        requestDeploymentAction(deploymentCreate, data, configuration, stepIndex);
+    }
+
+    private static void requestDeploymentAction(
+            DeploymentCreate deploymentCreate,
+            DeploymentCommandLineArgs data,
+            DeploymentProcessConfiguration environment,
+            int stepIndex)
+            throws Exception {
+        final String environmentToDeploy = getEnvironmentName(environment, stepIndex);
+        final List<String> environmentsToKill =
+                environment.environmentSteps.stream()
+                        .map(name -> getEnvironmentName(environment.name, name))
+                        .filter(env -> !env.equals(environmentToDeploy))
+                        .collect(Collectors.toList());
+
+        final DeploymentCreate.Response response =
+                deploymentCreate.request(
+                        new DeploymentCreate.Request(
+                                data.sha,
+                                stepIndex == 0 ? "deploy" : "deploy:migration",
+                                false,
+                                environmentToDeploy,
+                                String.format(
+                                        Locale.ROOT,
+                                        "Deployment for '%s' request by '%s'.",
+                                        environmentToDeploy,
+                                        data.apiUsername),
+                                Collections.singletonList("master-green-requirement"),
+                                new DeploymentCreate.RequestPayloadField(environmentsToKill)));
+
+        System.out.println(
+                String.format(
+                        Locale.ROOT,
+                        "Deploy request response: id %s, sha %s, environment %s, task %s.",
+                        response.id,
+                        response.sha,
+                        response.environment,
+                        response.task));
+    }
+
+    private static String getEnvironmentName(String environmentName, String stepName) {
+        return String.format(Locale.ROOT, "%s_%s", environmentName, stepName);
+    }
+
+    private static String getEnvironmentName(
+            DeploymentProcessConfiguration environment, int index) {
+        return getEnvironmentName(environment.name, environment.environmentSteps.get(index));
+    }
+}