1
0
mirror of https://github.com/mitb-archive/filebot synced 2024-11-13 21:05:01 -05:00
filebot/source/net/filebot/Execute.java
2019-03-15 14:33:16 +07:00

90 lines
2.4 KiB
Java

package net.filebot;
import static java.nio.charset.StandardCharsets.*;
import static java.util.Arrays.*;
import static java.util.stream.Collectors.*;
import static net.filebot.Logging.*;
import java.io.File;
import java.io.IOException;
import java.lang.ProcessBuilder.Redirect;
import java.util.List;
import net.filebot.util.ByteBufferOutputStream;
public class Execute {
private static final int BUFFER_SIZE = 8 * 1024;
public static CharSequence execute(String... command) throws IOException {
return execute(asList(command), null);
}
public static CharSequence execute(String[] command, File directory) throws IOException {
return execute(asList(command), directory);
}
public static CharSequence execute(String[]... command) throws IOException {
return execute(stream(command).flatMap(a -> stream(a)).collect(toList()), null);
}
public static CharSequence execute(List<String> command, File directory) throws IOException {
ProcessBuilder processBuilder = new ProcessBuilder(command);
processBuilder.redirectError(Redirect.INHERIT);
if (directory != null) {
processBuilder.directory(directory);
}
Process process = processBuilder.start();
try (ByteBufferOutputStream bb = new ByteBufferOutputStream(BUFFER_SIZE)) {
bb.transferFully(process.getInputStream());
int exitCode = process.waitFor();
CharSequence output = UTF_8.decode(bb.getByteBuffer());
// DEBUG
debug.finest(format("Execute %s%n%s", command, output));
if (exitCode != 0) {
throw new ExecuteException(command, exitCode);
}
return output;
} catch (InterruptedException e) {
throw new IOException(String.format("%s timed out", command), e);
}
}
public static void system(String... command) throws IOException {
system(asList(command), null);
}
public static void system(String[] command, File directory) throws IOException {
system(asList(command), directory);
}
public static void system(List<String> command, File directory) throws IOException {
ProcessBuilder processBuilder = new ProcessBuilder(command);
processBuilder.inheritIO();
if (directory != null) {
processBuilder.directory(directory);
}
// DEBUG
debug.finest(format("Execute %s", command));
try {
int exitCode = processBuilder.start().waitFor();
if (exitCode != 0) {
throw new ExecuteException(command, exitCode);
}
} catch (InterruptedException e) {
throw new IOException(String.format("%s timed out", command), e);
}
}
}