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 bakFile = new File(configFile.getAbsolutePath() + ".bak");
			File newFile = new File(configFile.getAbsolutePath() + ".new");
			getLog().info("Old AMI: " + oldAmi);
			getLog().info("New AMI: " + newAmi);
			getLog().info("Backing up original to " + bakFile.getAbsolutePath());
			getLog().info("Rewriting " + configFile.getAbsolutePath());
			getLog().info("New file " + newFile.getAbsolutePath());
			FileUtils.write(bakFile, oldContent);
			FileUtils.write(newFile, newContent);
			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;
	}

}