ca.canucksoftware.systoolsmgr.CommandLine.java Source code

Introduction

Here is the source code for ca.canucksoftware.systoolsmgr.CommandLine.java

Source

/**
 * Copyright (c) 2010 Jason Robitaille
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */

package ca.canucksoftware.systoolsmgr;

import java.io.BufferedReader;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.Arrays;
import org.json.JSONArray;
import org.json.JSONObject;
import com.palm.luna.service.ServiceMessage;

public class CommandLine extends Thread {
    private ServiceMessage message;
    private String[] command;
    private String response;
    private int returnCode;

    public CommandLine(String cmd) {
        this(null, cmd.split(" "));
    }

    public CommandLine(ServiceMessage msg, String cmd) {
        this(msg, cmd.split(" "));
    }

    public CommandLine(String[] cmd) {
        this(null, cmd);
    }

    public CommandLine(ServiceMessage msg, String[] cmd) {
        super();
        message = msg;
        command = cmd;
        returnCode = -1;
    }

    public String[] getCommand() {
        return command;
    }

    public void setCommand(String[] cmd) {
        command = cmd;
    }

    public String getResponse() {
        return response;
    }

    public boolean doCmd() {
        try {
            response = null;
            ProcessBuilder pb = new ProcessBuilder(command);
            pb.redirectErrorStream(false);
            Process p = pb.start();
            String stdout = getTextFromStream(p.getInputStream());
            String stderr = getTextFromStream(p.getErrorStream());
            if (p.waitFor() != 0) {
                returnCode = p.exitValue();
            } else {
                returnCode = 0;
            }
            if (returnCode == 0) {
                response = stdout;
            } else {
                response = stderr;
            }
        } catch (Exception e) {
            response = e.getMessage();
            returnCode = -1;
        }
        return (returnCode == 0);
    }

    private String getTextFromStream(InputStream is) {
        String result = "";
        try {
            BufferedReader br = new BufferedReader(new InputStreamReader(is));
            String line = null;
            line = br.readLine();
            while (line != null) {
                if (line.trim().length() != 0) {
                    result += line.trim();
                }
                line = br.readLine();
                if (line != null) {
                    if (line.trim().length() != 0) {
                        result += " ";
                    }
                }
            }
            br.close();
        } catch (Exception e) {
            result = "";
        }
        return result;
    }

    public void run() {
        if (message != null) {
            try {
                if (doCmd()) {
                    JSONObject reply = new JSONObject();
                    reply.put("output", response);
                    JSONArray cmds = new JSONArray(Arrays.asList(command));
                    reply.put("commands", cmds);
                    message.respond(reply.toString());
                } else {
                    message.respondError(String.valueOf(returnCode), response);
                }
            } catch (Exception e) {
                try {
                    message.respondError("-1", e.getMessage());
                } catch (Exception e2) {
                }
            }
        }
    }
}