Minecraft-SSHD/src/main/java/com/ryanmichela/sshd/ConsoleShellFactory.java

package com.ryanmichela.sshd;

import com.ryanmichela.sshd.ConsoleCommandCompleter;
import com.ryanmichela.sshd.ConsoleLogFormatter;
import com.ryanmichela.sshd.FlushyOutputStream;
import com.ryanmichela.sshd.FlushyStreamHandler;
import com.ryanmichela.sshd.SshTerminal;
import com.ryanmichela.sshd.SshdPlugin;
import com.ryanmichela.sshd.StreamHandlerAppender;
import com.ryanmichela.sshd.implementations.SSHDCommandSender;
import jline.console.ConsoleReader;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.core.Logger;
import org.apache.sshd.common.Factory;
import org.apache.sshd.server.shell.ShellFactory;
import org.apache.sshd.server.command.Command;
import org.apache.sshd.server.channel.ChannelSession;
import org.apache.sshd.server.Environment;
import org.apache.sshd.server.ExitCallback;
import org.bukkit.Bukkit;

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.logging.Level;
import java.util.logging.StreamHandler;

public class ConsoleShellFactory implements ShellFactory {

    static SSHDCommandSender sshdCommandSender = new SSHDCommandSender();

    public Command createShell(ChannelSession cs) {
        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;

        StreamHandlerAppender streamHandlerAppender;
        public static 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;
        }

        @Override
		public void start(ChannelSession cs, Environment env) throws IOException
		  {
			  try
			  {
				  consoleReader = new ConsoleReader(in, new FlushyOutputStream(out), new SshTerminal());
				  consoleReader.setExpandEvents(true);
				  consoleReader.addCompleter(new ConsoleCommandCompleter());

				  StreamHandler streamHandler = new FlushyStreamHandler(out, new ConsoleLogFormatter(), consoleReader);
				  streamHandlerAppender		  = new StreamHandlerAppender(streamHandler);

				  ((Logger)LogManager.getRootLogger()).addAppender(streamHandlerAppender);

				  environment = env;
				  thread	  = new Thread(this, "SSHD ConsoleShell " + env.getEnv().get(Environment.ENV_USER));
				  thread.start();
			  }
			  catch (Exception e)
			  {
				  throw new IOException("Error starting shell", e);
			  }
        }

        @Override
		public void destroy(ChannelSession cs) { ((Logger)LogManager.getRootLogger()).removeAppender(streamHandlerAppender); }

		public void run()
        {
            try
            {
                if (!SshdPlugin.instance.getConfig().getString("mode").equals("RPC"))
                    printPreamble(consoleReader);
                while (true)
                {
                    String command = consoleReader.readLine("\r>", null);
                    if (command == null)
                        continue;
                    if (command.equals("exit") || command.equals("quit"))
                        break;
                    Bukkit.getScheduler().runTask(
                        SshdPlugin.instance, () ->
                        {
                            if (SshdPlugin.instance.getConfig().getString("mode").equals("RPC") && command.startsWith("rpc"))
                            {
                                // NO ECHO NO PREAMBLE AND SHIT
                                String cmd = command.substring("rpc".length() + 1, command.length());
                                Bukkit.dispatchCommand(sshdCommandSender, cmd);
                            }
                            else
                            {
                                SshdPlugin.instance.getLogger().info("<" + environment.getEnv().get(Environment.ENV_USER) + "> "
                                                                    + command);
                                Bukkit.dispatchCommand(Bukkit.getConsoleSender(), command);
                            }
                        });
                }
            }
            catch (IOException e)
            {
                SshdPlugin.instance.getLogger().log(Level.SEVERE, "Error processing command from SSH", e);
            }
            finally
            {
                callback.onExit(0);
            }
        }

		private void printPreamble(ConsoleReader consoleReader) throws IOException
        {
            consoleReader.println("  _____ _____ _    _ _____" + "\r");
            consoleReader.println(" / ____/ ____| |  | |  __ \\" + "\r");
            consoleReader.println("| (___| (___ | |__| | |  | |" + "\r");
            consoleReader.println(" \\___ \\\\___ \\|  __  | |  | |" + "\r");
            consoleReader.println(" ____) |___) | |  | | |__| |" + "\r");
            consoleReader.println("|_____/_____/|_|  |_|_____/" + "\r");
            consoleReader.println("Connected to: " + Bukkit.getServer().getName() + "\r");
            consoleReader.println("- " + Bukkit.getServer().getMotd() + "\r");
            consoleReader.println("\r");
            consoleReader.println("Type 'exit' to exit the shell." + "\r");
            consoleReader.println("===============================================" + "\r");
        }
	}
}