pl.edu.icm.coansys.commons.shell.JavaShellRunner.java Source code

Introduction

Here is the source code for pl.edu.icm.coansys.commons.shell.JavaShellRunner.java

Source

/*
 * This file is part of CoAnSys project.
 * Copyright (c) 2012-2015 ICM-UW
 * 
 * CoAnSys is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
    
 * CoAnSys is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Affero General Public License for more details.
 * 
 * You should have received a copy of the GNU Affero General Public License
 * along with CoAnSys. If not, see <http://www.gnu.org/licenses/>.
 */

package pl.edu.icm.coansys.commons.shell;

import java.io.BufferedReader;
import java.io.File;
import java.io.IOException;
import java.io.InputStreamReader;
import java.nio.charset.Charset;
import java.util.Map;

import org.apache.commons.io.IOUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 *
 * @author akawa
 */

public final class JavaShellRunner {

    private static Logger logger = LoggerFactory.getLogger(JavaShellRunner.class);

    private JavaShellRunner() {
    }

    private static String[] parseArgs(String[] args) {
        String[] result = new String[2];
        if (args == null || args.length < 2) {
            logger.debug("# of parameters should be greater or equal to 2");
            logger.debug("You need to provide:");
            logger.debug("* the script file");
            logger.debug("* the directory where the script is to run");
            logger.debug("");

            result[0] = "/home/akawa/Documents/codes/scripts/create_testProto.sh";
            result[1] = "/home/akawa/Documents/codes/scripts";

            logger.debug("Default values will be used:");
            logger.debug("* " + result[0]);
            logger.debug("* " + result[1]);
        } else {
            result[0] = args[0];
            result[1] = args[1];
        }

        return result;
    }

    private static void printOutputStream(Process proces) throws IOException {
        BufferedReader in = null;
        try {
            in = new BufferedReader(new InputStreamReader(proces.getInputStream(), Charset.forName("UTF-8")));
            String resultLine;
            while ((resultLine = in.readLine()) != null) {
                System.out.println(resultLine);
            }
        } finally {
            IOUtils.closeQuietly(in);
        }
    }

    private static ProcessBuilder buildProcess(String[] args) {
        ProcessBuilder pb = new ProcessBuilder(args[0]);
        Map<String, String> env = pb.environment();
        for (int i = 3; i < args.length; i = i + 2) {
            env.put(args[i], args[i + 1]);
        }
        pb.directory(new File(args[1]));
        return pb;

    }

    public static void main(String[] args) throws IOException, InterruptedException {

        ProcessBuilder pb = buildProcess(parseArgs(args));
        Process proc = pb.start();
        printOutputStream(proc);
        if (proc.waitFor() == 0) {
            System.out.println("Process terminates normally");
        }
        proc.destroy();
    }
}