vagrant.impl.cli.VagrantCli Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of vagrant-java-bindings Show documentation

Wrapper over the CLI Vagrant tool presented in a Java API

There is a newer version: 0.2.0

Show newest version

package vagrant.impl.cli;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.Reader;
import java.util.ArrayList;
import java.util.List;
import java.util.logging.Logger;

public class VagrantCli {
    private static final Logger LOG = Logger.getLogger(VagrantCli.class.getName());

    private File path;
    private List args;

    public VagrantCli(File path) {
        if (!path.exists()) {
            throw new IllegalStateException("Vagrant machine path " + path + " doesn't exist.");
        }
        this.path = path;
        args = new ArrayList();
        args.add("vagrant");
    }
    
    public VagrantCli arg(String arg) {
        args.add(arg);
        return this;
    }
    
    public VagrantCli machineReadable() {
        return arg("--machine-readable");
    }
    
    public String execute() {
        synchronized (getClass()) {
            return executeSerialized();
        }
    }

    public String executeSerialized() {
        long start = System.currentTimeMillis();
        LOG.info(start + " - VagrantCli: execute " + args);
        ProcessBuilder pb = new ProcessBuilder(args);
        pb.directory(path);
        pb.redirectErrorStream();
        try {
            Process p = pb.start();
            p.getOutputStream().close();
            String out = readFully(p.getInputStream());
            if (p.waitFor() != 0) {
                throw new IllegalStateException("Invaid return code " + p.exitValue() + ".\n" + out);
            }
            long end = System.currentTimeMillis();
            LOG.info(end + " (" + (end-start) + ") - \n" + out);
            return out;
        } catch (IOException e) {
            throw new RuntimeException(e);
        } catch (InterruptedException e) {
            throw new RuntimeException(e);
        }
    }

    public static String readFully(InputStream in) throws IOException {
        Reader r = new InputStreamReader(in);
        StringBuilder str = new StringBuilder();
        char[] buff = new char[4096];
        int read;
        while ((read = r.read(buff, 0, buff.length)) != -1) {
            str.append(buff, 0, read);
        }
        r.close();
        return str.toString();
    }
}