Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
package co.escapeideas.maven.ansible;
import org.apache.maven.plugin.AbstractMojo;
import org.apache.maven.plugin.MojoExecutionException;
import org.apache.maven.plugin.MojoFailureException;
import org.apache.maven.plugins.annotations.Parameter;
import java.io.*;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
/**
* Abstract mojo for common configuration parameters and Runtime processing
*
* Created by tmullender on 22/10/14.
*/
public abstract class AbstractAnsibleMojo extends AbstractMojo {
private static final String NEW_LINE_SEPARATOR = System.getProperty("line.separator");;
private static final String STDERR_LOG = "stderr.log";
private static final String STDOUT_LOG = "stdout.log";
/**
* Connection type to use
*/
@Parameter( property = "ansible.connection" )
private String connection;
/**
* The number of parallel processes to use
*/
@Parameter( property = "ansible.forks" )
private Integer forks;
/**
* The inventory host file
*/
@Parameter( property = "ansible.inventory" )
private File inventory;
/**
* Limit selected hosts to an additional pattern
*/
@Parameter( property = "ansible.limit" )
private String limit;
/**
* The path to the ansible module library
*/
@Parameter( property = "ansible.modulePath" )
private File modulePath;
/**
* Use this file to authenticate the connection
*/
@Parameter( property = "ansible.privateKey" )
private File privateKey;
/**
* Override the SSH timeout in seconds
*/
@Parameter( property = "ansible.timeout" )
private Integer timeout;
/**
* Connect as this user
*/
@Parameter( property = "ansible.remoteUser" )
private String remoteUser;
/**
* Vault password file
*/
@Parameter( property = "ansible.vaultPasswordFile" )
private File vaultPasswordFile;
/**
* If further options are required to the ansible command they can be added here
* @since 1.1.0
*/
@Parameter( property = "ansible.options" )
private List options;
/**
* The directory in which to run, defaults to the projects build directory
*/
@Parameter( defaultValue = "${project.build.directory}", required = true, property = "ansible.workingDirectory" )
private File workingDirectory;
/**
* If true, the build will fail when the ansible command returns an error(a non zero exit status),
* defaults to false
* @since 1.1.0
*/
@Parameter( defaultValue = "false", property = "ansible.failOnAnsibleError" )
private boolean failOnAnsibleError;
/**
* If present the plugin will log the output of the execution to files in this directory
* @since 1.1.0
*/
@Parameter( property = "ansible.logDirectory" )
private File logDirectory;
/**
* Gets the value for inventory
* @return
*/
protected File getInventory() {
return inventory;
}
/**
* Get the value for the vault password file
* @return
*/
protected File getVaultPasswordFile() {
return vaultPasswordFile;
}
/**
* Constructs a command from the configured parameters and executes it, logging output at debug
*
* @throws MojoExecutionException
*/
public void execute() throws MojoExecutionException, MojoFailureException {
try {
final List command = createCommand();
checkWorkingDirectory();
final int status = execute(command);
if (failOnAnsibleError && status != 0){
throw new MojoFailureException("Non-zero exit status returned");
}
} catch (IOException e) {
throw new MojoExecutionException("Unable to run playbook", e);
} catch (InterruptedException e) {
throw new MojoExecutionException("Run interrupted", e);
}
}
private List createCommand() throws IOException {
final List command = new ArrayList();
command.add(getExecutable());
addOptions(command);
command.addAll(options);
final String argument = getArgument();
if (argument != null) {
command.add(argument);
}
getLog().info("Command: " + command);
return command;
}
private void checkWorkingDirectory() {
if (!workingDirectory.exists()){
workingDirectory = new File(System.getProperty("java.io.tmpdir"));
getLog().info("Updated working directory: " + workingDirectory);
} else {
getLog().info("Working directory: " + workingDirectory);
}
}
private int execute(final List command) throws IOException, InterruptedException {
final ProcessBuilder builder = new ProcessBuilder(command);
builder.directory(workingDirectory);
final Process process = builder.start();
logOutput(process);
process.waitFor();
return process.exitValue();
}
private void logOutput(final Process process) throws IOException {
final BufferedReader output = new BufferedReader(new InputStreamReader(process.getInputStream()));
final BufferedReader error = new BufferedReader(new InputStreamReader(process.getErrorStream()));
final Writer outputFile = createFileWriter(STDOUT_LOG);
final Writer errorFile = createFileWriter(STDERR_LOG);
try {
processOutput(output, error, outputFile, errorFile);
} finally {
outputFile.close();
errorFile.close();
}
}
private void processOutput(final BufferedReader output, final BufferedReader error,
final Writer outputFile, final Writer errorFile) throws IOException {
String outputLine, errorLine = null;
while ((outputLine = output.readLine()) != null || (errorLine = error.readLine()) != null) {
if (outputLine != null) {
getLog().debug(outputLine);
outputFile.write(outputLine);
outputFile.write(NEW_LINE_SEPARATOR);
outputFile.flush();
}
if (errorLine != null) {
getLog().warn(errorLine);
errorFile.write(errorLine);
errorFile.write(NEW_LINE_SEPARATOR);
errorFile.flush();
}
}
}
private Writer createFileWriter(final String fileName) throws IOException {
if (logDirectory == null){
return new NoopWriter();
} else if (!logDirectory.exists()) {
logDirectory.mkdirs();
}
final File output = new File(logDirectory, fileName);
return new FileWriter(output, true);
}
/**
* The argument to be used for the command.
* For ansible this is the hosts
* For ansible-playbook this is the playbook
* @return the argument to pass
* @throws IOException
*/
protected abstract String getArgument() throws IOException;
/**
* The executable for the command
* @return the name of the command to run
*/
protected abstract String getExecutable();
/**
* Adds the configured options to the list of command strings
* @param command
*/
protected void addOptions(final List command) {
command.addAll(createOption("-c", connection));
command.addAll(createOption("-f", forks));
command.addAll(createOption("-i", inventory));
command.addAll(createOption("-l", limit));
command.addAll(createOption("-M", modulePath));
command.addAll(createOption("--private-key", privateKey));
command.addAll(createOption("-T", timeout));
command.addAll(createOption("-u", remoteUser));
command.addAll(createOption("--vault-password-file", vaultPasswordFile));
}
/**
* Creates a list for the given option, an empty list if the option's value is null
* @param option
* @param value
* @return a list of the strings for the given option, empty if the option should not be used
*/
protected List createOption(final String option, final Object value) {
if (value == null){
return new ArrayList();
}
return Arrays.asList(option, String.valueOf(value));
}
/**
* Creates a list for the given option, an empty list if the option's value is null
* @param option
* @param value
* @return a list of the strings for the given option, empty if the option should not be used
*/
protected List createOption(final String option, final boolean value) {
if (value){
return Arrays.asList(option);
}
return new ArrayList();
}
/**
* Checks whether the given file is an absolute path or a classpath file
* @param path
* @return
* @throws IOException
*/
protected String findClasspathFile(final String path) throws IOException {
if (path == null){
return null;
}
final File file = new File(path);
if (file.exists()){
return file.getAbsolutePath();
}
return createTmpFile(path).getAbsolutePath();
}
/**
* Copies a classpath resource to the tmp directory to allow it to be run by ansible
* @param path
* @return
* @throws IOException if the path is not found
*/
private File createTmpFile(final String path) throws IOException {
getLog().debug("Creating temporary file for: " + path);
final File output = new File(System.getProperty("java.io.tmpdir"), "ansible-maven-plugin." + System.nanoTime());
final FileOutputStream outputStream = new FileOutputStream(output);
final InputStream inputStream = getClass().getResourceAsStream("/" + path);
if (inputStream == null){
throw new FileNotFoundException("Unable to locate: " + path);
}
copy(inputStream, outputStream);
return output;
}
/**
* Copies the input stream to the output stream using a 4K buffer
* @param inputStream
* @param outputStream
* @throws IOException
*/
private void copy(final InputStream inputStream, final FileOutputStream outputStream) throws IOException {
final byte[] buffer = new byte[1024*4];
int n;
try {
while (-1 != (n = inputStream.read(buffer))) {
outputStream.write(buffer, 0, n);
}
} finally {
inputStream.close();
outputStream.close();
}
}
}
