HaxeFoundation.haxe/std/cs/_std/sys/io/Process.hx

/*
 * Copyright (C)2005-2017 Haxe Foundation
 *
 * 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 sys.io;
import haxe.io.BytesInput;
import cs.system.io.StreamReader;
import cs.system.io.StreamWriter;
import cs.system.diagnostics.Process as NativeProcess;
import cs.system.diagnostics.ProcessStartInfo as NativeStartInfo;

@:coreApi
class Process {

	public var stdout(default,null) : haxe.io.Input;
	public var stderr(default,null) : haxe.io.Input;
	public var stdin(default, null) : haxe.io.Output;

	private var native:NativeProcess;

	public function new( cmd : String, ?args : Array<String> ) : Void
	{
		this.native = createNativeProcess(cmd, args);
		native.Start();

		this.stdout = new cs.io.NativeInput(native.StandardOutput.BaseStream);
		this.stderr = new cs.io.NativeInput(native.StandardError.BaseStream);
		this.stdin = new cs.io.NativeOutput(native.StandardInput.BaseStream);
	}

	@:allow(Sys)
	private static function createNativeProcess( cmd : String, ?args : Array<String> ) : NativeProcess {
		var native = new NativeProcess();
		native.StartInfo.CreateNoWindow = true;
		native.StartInfo.RedirectStandardError = native.StartInfo.RedirectStandardInput = native.StartInfo.RedirectStandardOutput = true;
		if (args != null) {
			// mono 4.2.1 on Windows doesn't support relative path correctly
			if (cmd.indexOf("/") != -1 || cmd.indexOf("\\") != -1)
				cmd = sys.FileSystem.fullPath(cmd);
			native.StartInfo.FileName = cmd;
			native.StartInfo.UseShellExecute = false;
			native.StartInfo.Arguments = buildArgumentsString(args);
		} else {
			switch (Sys.systemName()) {
				case "Windows":
					native.StartInfo.FileName = switch (Sys.getEnv("COMSPEC")) {
						case null: "cmd.exe";
						case comspec: comspec;
					}
					native.StartInfo.Arguments = '/C "$cmd"';
				case _:
					native.StartInfo.FileName = "/bin/sh";
					native.StartInfo.Arguments = buildArgumentsString(["-c", cmd]);
			}
			native.StartInfo.UseShellExecute = false;
		}
		return native;
	}

	private static function buildArgumentsString(args:Array<String>):String {
		return switch (Sys.systemName()) {
			case "Windows":
				[
					for (a in args)
					StringTools.quoteWinArg(a, false)
				].join(" ");
			case _:
				// mono uses a slightly different quoting/escaping rule...
				// https://bugzilla.xamarin.com/show_bug.cgi?id=19296
				[
					for (arg in args) {
						var b = new StringBuf();
						b.add('"');
						for (i in 0...arg.length) {
							var c = arg.charCodeAt(i);
							switch (c) {
								case '"'.code | '\\'.code:
									b.addChar('\\'.code);
								case _: //pass
							}
							b.addChar(c);
						}
						b.add('"');
						b.toString();
					}
				].join(" ");
		}
	}

	public function getPid() : Int
	{
		return native.Id;
	}

	public function exitCode( block : Bool = true ) : Null<Int>
	{
		if( block == false && !native.HasExited )
			return null;
		native.WaitForExit();
		return native.ExitCode;
	}

	public function close() : Void
	{
		native.Close();
	}

	public function kill() : Void
	{
		native.Kill();
	}

}