org.kuali.maven.plugins.jenkins.UpdateSlaveAMIMojo Maven / Gradle / Ivy

Go to download
/**
 * Copyright 2011-2012 The Kuali Foundation
 *
 * Licensed under the Educational Community License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.opensource.org/licenses/ecl2.php
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.kuali.maven.plugins.jenkins;

import java.io.File;

import org.apache.commons.io.FileUtils;
import org.apache.commons.lang.StringUtils;
import org.apache.maven.plugin.AbstractMojo;
import org.apache.maven.plugin.MojoExecutionException;
import org.kuali.maven.common.ResourceUtils;

/**
 * Update the slave AMI being used by the Jenkins EC2 plugin
 *
 * @goal updateslaveami
 */
public class UpdateSlaveAMIMojo extends AbstractMojo {
    ResourceUtils utils = new ResourceUtils();

    /**
     * The location of the Jenkins config file
     *
     * @parameter expression="${jenkins.configFile}" default-value="/var/lib/jenkins/config.xml"
     * @required
     */
    private File configFile;

    /**
     * The AMI the Jenkins EC2 plugin should use when launching slaves
     *
     * @parameter expression="${jenkins.newAmi}"
     * @required
     */
    private String newAmi;

    @Override
    public void execute() throws MojoExecutionException {
        try {
            String oldContent = utils.read(configFile.getAbsolutePath());
            String open = "";
            String close = "";
            String slaveTemplate = StringUtils.substringBetween(oldContent, open, close);
            String oldAmi = StringUtils.substringBetween(slaveTemplate, "", "");
            String oldText = "" + oldAmi + "";
            String newText = "" + newAmi + "";
            String newContent = oldContent.replace(oldText, newText);
            File bak = new File(configFile.getAbsolutePath() + ".bak");
            getLog().info("Old AMI: " + oldAmi);
            getLog().info("New AMI: " + newAmi);
            getLog().info("Backing up original to " + bak.getAbsolutePath());
            getLog().info("Rewriting " + configFile.getAbsolutePath());
            FileUtils.write(bak, oldContent);
            FileUtils.write(configFile, newContent);
        } catch (Exception e) {
            throw new MojoExecutionException("Unexpected error", e);
        }
    }

    public File getConfigFile() {
        return configFile;
    }

    public void setConfigFile(File configFile) {
        this.configFile = configFile;
    }

    public String getNewAmi() {
        return newAmi;
    }

    public void setNewAmi(String newAmi) {
        this.newAmi = newAmi;
    }

}