io.takari.orchestra.plugins.oneops.OneOpsTask Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of oneops-tasks
The newest version!
package io.takari.orchestra.plugins.oneops;

import com.oneops.client.OneOpsClient;
import io.takari.bpm.api.BpmnError;
import io.takari.bpm.api.ExecutionContext;
import io.takari.orchestra.common.Task;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.inject.Named;
import java.util.Arrays;

@Named
public class OneOpsTask implements Task {

    private static final Logger log = LoggerFactory.getLogger(OneOpsTask.class);
    private static final long DEFAULT_POLL_FREQUENCY = 3000;

    public static final String API_BASEURL = "api_baseurl";
    public static final String API_TOKEN = "api_token";
    public static final String ORGANIZATION_KEY = "organization";
    public static final String ASSEMBLY_KEY = "assembly";
    public static final String ENVIRONMENT_KEY = "environment";
    public static final String PLATFORM_KEY = "platform";
    public static final String COMPONENT_KEY = "component";
    public static final String VARIABLE_NAME_KEY = "variable_name";
    public static final String VARIABLE_VALUE_KEY = "variable_value";
    public static final String VARIABLE_NAMES_KEY = "variable_names";
    public static final String TO_UPPER_CASE_KEY = "to_upper_case";

    @Override
    public String getKey() {
        return "oneOps";
    }

    public void updateVariable(ExecutionContext ctx) {
        String org = (String) ctx.getVariable(ORGANIZATION_KEY);
        String asm = (String) ctx.getVariable(ASSEMBLY_KEY);
        String env = (String) ctx.getVariable(ENVIRONMENT_KEY);
        String platform = (String) ctx.getVariable(PLATFORM_KEY);
        String component = (String) ctx.getVariable(COMPONENT_KEY);
        String variableName = (String) ctx.getVariable(VARIABLE_NAME_KEY);
        String variableValue = (String) ctx.getVariable(VARIABLE_VALUE_KEY);

        if (variableName == null || variableName.isEmpty() || variableValue == null || variableValue.isEmpty()) {
            return;
        }

        try {
            connect(ctx).updateVariable(org, asm, env, platform, variableName, variableValue, component);
        } catch (BpmnError e) {
            log.error("updateVariable ['{}', '{}', '{}', '{}', '{}', '{}', '{}'] -> error", org, asm, env, platform, component, variableName, variableValue, e);
            throw e;
        } catch (Exception e) {
            log.error("updateVariable ['{}', '{}', '{}', '{}', '{}', '{}', '{}'] -> error", org, asm, env, platform, component, variableName, variableValue, e);
            throw new BpmnError("updateVariableError", e);
        }
    }

    public void updateVariables(ExecutionContext ctx) {
        String org = (String) ctx.getVariable(ORGANIZATION_KEY);
        String asm = (String) ctx.getVariable(ASSEMBLY_KEY);
        String env = (String) ctx.getVariable(ENVIRONMENT_KEY);
        String platform = (String) ctx.getVariable(PLATFORM_KEY);
        String component = (String) ctx.getVariable(COMPONENT_KEY);
        boolean toUpperCase = Boolean.parseBoolean((String) ctx.getVariable(TO_UPPER_CASE_KEY));
        String[] keys = split((String) ctx.getVariable(VARIABLE_NAMES_KEY));

        for (String k : keys) {
            String varVal = (String) ctx.getVariable(k);
            String varName = toUpperCase ? k.toUpperCase() : k;
            try {
                connect(ctx).updateVariable(org, asm, env, platform, varName, varVal, component);
            } catch (BpmnError e) {
                log.error("updateVariables ['{}', '{}', '{}', '{}', '{}'] -> error updating the variable '{}'", org, asm, env, platform, component, varName, e);
                throw e;
            } catch (Exception e) {
                log.error("updateVariable ['{}', '{}', '{}', '{}', '{}'] -> error updating the variable '{}'", org, asm, env, platform, component, varName, e);
                throw new BpmnError("updateVariableError", e);
            }
        }
    }

    private String[] split(String s) {
        if (s == null) {
            return new String[0];
        }

        return s.split(",");
    }

    public void touchComponent(ExecutionContext ctx) {
        String org = (String) ctx.getVariable(ORGANIZATION_KEY);
        String asm = (String) ctx.getVariable(ASSEMBLY_KEY);
        String env = (String) ctx.getVariable(ENVIRONMENT_KEY);
        String platform = (String) ctx.getVariable(PLATFORM_KEY);
        String component = (String) ctx.getVariable(COMPONENT_KEY);

        try {
            connect(ctx).touchComponents(org, asm, env, platform, component);
            log.info("touchComponent ['{}', '{}', '{}', '{}', '{}'] -> done", org, asm, env, platform, component);
        } catch (BpmnError e) {
            log.error("touchComponent ['{}', '{}', '{}', '{}', '{}'] -> error", org, asm, env, platform, component, e);
            throw e;
        } catch (Exception e) {
            log.error("touchComponent ['{}', '{}', '{}', '{}', '{}'] -> error", org, asm, env, platform, component, e);
            throw new BpmnError("touchComponentError", e);
        }
    }

    public void commitAndDeploy(ExecutionContext ctx) {
        String org = (String) ctx.getVariable(ORGANIZATION_KEY);
        String asm = (String) ctx.getVariable(ASSEMBLY_KEY);
        String env = (String) ctx.getVariable(ENVIRONMENT_KEY);
        String platform = (String) ctx.getVariable(PLATFORM_KEY);

        try {
            connect(ctx).commitAndDeploy(org, asm, env, platform, false, false, Arrays.asList(platform), String.valueOf(DEFAULT_POLL_FREQUENCY));
            log.info("commitAndDeploy ['{}', '{}', '{}', '{}'] -> done", org, asm, env, platform);
        } catch (BpmnError e) {
            log.error("commitAndDeploy ['{}', '{}', '{}', '{}'] -> error", org, asm, env, platform, e);
            throw e;
        } catch (Exception e) {
            log.error("commitAndDeploy ['{}', '{}', '{}', '{}'] -> error", org, asm, env, platform, e);
            throw new BpmnError("commitAndDeployError", e);
        }
    }

    private static OneOpsClient connect(ExecutionContext ctx) {
        return OneOpsClient.builder()
                .baseUrl((String) ctx.getVariable(API_BASEURL))
                .apiToken((String) ctx.getVariable(API_TOKEN))
                .build();
    }
}