128 lines
4.4 KiB
Java
128 lines
4.4 KiB
Java
package com.ryanmichela.sshd;
|
|
|
|
import org.apache.sshd.common.Factory;
|
|
import org.apache.sshd.server.Command;
|
|
import org.apache.sshd.server.Environment;
|
|
import org.apache.sshd.server.ExitCallback;
|
|
import org.bukkit.Bukkit;
|
|
import org.bukkit.craftbukkit.libs.jline.console.ConsoleReader;
|
|
|
|
import java.io.IOException;
|
|
import java.io.InputStream;
|
|
import java.io.OutputStream;
|
|
import java.util.logging.Formatter;
|
|
import java.util.logging.Logger;
|
|
import java.util.logging.StreamHandler;
|
|
|
|
public class ConsoleShellFactory implements Factory<Command> {
|
|
|
|
public Command create() {
|
|
return new ConsoleShell();
|
|
}
|
|
|
|
public static class ConsoleShell implements Command, Runnable {
|
|
|
|
private InputStream in;
|
|
private OutputStream out;
|
|
private OutputStream err;
|
|
private ExitCallback callback;
|
|
private Environment environment;
|
|
private Thread thread;
|
|
|
|
StreamHandler streamHandler;
|
|
ConsoleReader consoleReader;
|
|
|
|
public InputStream getIn() {
|
|
return in;
|
|
}
|
|
|
|
public OutputStream getOut() {
|
|
return out;
|
|
}
|
|
|
|
public OutputStream getErr() {
|
|
return err;
|
|
}
|
|
|
|
public Environment getEnvironment() {
|
|
return environment;
|
|
}
|
|
|
|
public void setInputStream(InputStream in) {
|
|
this.in = in;
|
|
}
|
|
|
|
public void setOutputStream(OutputStream out) {
|
|
this.out = out;
|
|
}
|
|
|
|
public void setErrorStream(OutputStream err) {
|
|
this.err = err;
|
|
}
|
|
|
|
public void setExitCallback(ExitCallback callback) {
|
|
this.callback = callback;
|
|
}
|
|
|
|
public void start(Environment env) throws IOException {
|
|
|
|
Formatter bukkitFormatter = Bukkit.getLogger().getHandlers()[0].getFormatter();
|
|
|
|
try {
|
|
consoleReader = new ConsoleReader(in, new FlushyOutputStream(out), new SshTerminal());
|
|
consoleReader.setExpandEvents(true);
|
|
consoleReader.addCompleter(new ConsoleCommandCompleter());
|
|
|
|
streamHandler = new FlushyStreamHandler(out, bukkitFormatter, consoleReader);
|
|
Bukkit.getLogger().addHandler(streamHandler);
|
|
Logger.getLogger("").addHandler(streamHandler);
|
|
|
|
environment = env;
|
|
thread = new Thread(this, "EchoShell " + env.getEnv().get(Environment.ENV_USER));
|
|
thread.start();
|
|
} catch (Exception e) {
|
|
throw new IOException("Error starting shell", e);
|
|
}
|
|
}
|
|
|
|
public void destroy() {
|
|
Bukkit.getLogger().removeHandler(streamHandler);
|
|
Logger.getLogger("").removeHandler(streamHandler);
|
|
}
|
|
|
|
public void run() {
|
|
String command;
|
|
try {
|
|
printPreamble(consoleReader);
|
|
while(true) {
|
|
command = consoleReader.readLine("\r>", null);
|
|
if (command != null) {
|
|
if (command.equals("exit")) {
|
|
break;
|
|
}
|
|
SshdPlugin.instance.getLogger().info("[U: " + environment.getEnv().get(Environment.ENV_USER) + "] " + command);
|
|
Bukkit.dispatchCommand(Bukkit.getConsoleSender(), command);
|
|
}
|
|
}
|
|
} catch (IOException e) {
|
|
SshdPlugin.instance.getLogger().severe("Error processing command from SSH");
|
|
} finally {
|
|
callback.onExit(0);
|
|
}
|
|
}
|
|
|
|
private void printPreamble(ConsoleReader consoleReader) throws IOException{
|
|
consoleReader.println(" _____ _____ _ _ _____");
|
|
consoleReader.println(" / ____/ ____| | | | __ \\");
|
|
consoleReader.println("| (___| (___ | |__| | | | |");
|
|
consoleReader.println(" \\___ \\\\___ \\| __ | | | |");
|
|
consoleReader.println(" ____) |___) | | | | |__| |");
|
|
consoleReader.println("|_____/_____/|_| |_|_____/");
|
|
consoleReader.println("Connected to: " + Bukkit.getServer().getName());
|
|
consoleReader.println("- " + Bukkit.getServer().getMotd());
|
|
consoleReader.println();
|
|
consoleReader.println("Type 'exit' to exit the shell.");
|
|
consoleReader.println("===============================================");
|
|
}
|
|
}
|
|
} |