ConsoleModule.java

package com.hypixel.hytale.server.core.console;

import com.hypixel.hytale.common.plugin.PluginManifest;
import com.hypixel.hytale.logger.HytaleLogger;
import com.hypixel.hytale.logger.backend.HytaleConsole;
import com.hypixel.hytale.server.core.Constants;
import com.hypixel.hytale.server.core.HytaleServer;
import com.hypixel.hytale.server.core.ShutdownReason;
import com.hypixel.hytale.server.core.command.system.CommandManager;
import com.hypixel.hytale.server.core.command.system.CommandSender;
import com.hypixel.hytale.server.core.console.command.SayCommand;
import com.hypixel.hytale.server.core.plugin.JavaPlugin;
import com.hypixel.hytale.server.core.plugin.JavaPluginInit;
import java.io.IOError;
import java.io.IOException;
import java.util.logging.Level;
import javax.annotation.Nonnull;
import org.jline.reader.EndOfFileException;
import org.jline.reader.LineReader;
import org.jline.reader.LineReaderBuilder;
import org.jline.reader.UserInterruptException;
import org.jline.terminal.Terminal;
import org.jline.terminal.TerminalBuilder;

public class ConsoleModule extends JavaPlugin {
   public static final PluginManifest MANIFEST = PluginManifest.corePlugin(ConsoleModule.class).build();
   private static ConsoleModule instance;
   private Terminal terminal;
   private ConsoleRunnable consoleRunnable;

   public static ConsoleModule get() {
      return instance;
   }

   public ConsoleModule(@Nonnull JavaPluginInit init) {
      super(init);
   }

   protected void setup() {
      instance = this;
      this.getCommandRegistry().registerCommand(new SayCommand());

      try {
         TerminalBuilder builder = TerminalBuilder.builder();
         if (Constants.SINGLEPLAYER) {
            builder.dumb(true);
         } else {
            builder.color(true);
         }

         this.terminal = builder.build();
         HytaleConsole.INSTANCE.setTerminal(this.terminal.getType());
         LineReader lineReader = LineReaderBuilder.builder().terminal(this.terminal).build();
         this.consoleRunnable = new ConsoleRunnable(lineReader, ConsoleSender.INSTANCE);
         this.getLogger().at(Level.INFO).log("Setup console with type: %s", this.terminal.getType());
      } catch (IOException e) {
         ((HytaleLogger.Api)this.getLogger().at(Level.SEVERE).withCause(e)).log("Failed to start console reader");
      }

   }

   protected void shutdown() {
      this.getLogger().at(Level.INFO).log("Restoring terminal...");

      try {
         this.terminal.close();
      } catch (IOException e) {
         ((HytaleLogger.Api)HytaleLogger.getLogger().at(Level.SEVERE).withCause(e)).log("Failed to restore terminal!");
      }

      this.consoleRunnable.interrupt();
   }

   public Terminal getTerminal() {
      return this.terminal;
   }

   private static class ConsoleRunnable implements Runnable {
      private final LineReader lineReader;
      private final ConsoleSender consoleSender;
      @Nonnull
      private final Thread consoleThread;

      public ConsoleRunnable(LineReader lineReader, ConsoleSender consoleSender) {
         this.lineReader = lineReader;
         this.consoleSender = consoleSender;
         this.consoleThread = new Thread(this, "ConsoleThread");
         this.consoleThread.setDaemon(true);
         this.consoleThread.start();
      }

      public void run() {
         try {
            String terminalType = this.lineReader.getTerminal().getType();

            String command;
            for(boolean isDumb = "dumb".equals(terminalType) || "dumb-color".equals(terminalType); !this.consoleThread.isInterrupted(); CommandManager.get().handleCommand((CommandSender)this.consoleSender, command)) {
               command = this.lineReader.readLine(isDumb ? null : "> ");
               if (command == null) {
                  break;
               }

               command = command.trim();
               if (!command.isEmpty() && command.charAt(0) == '/') {
                  command = command.substring(1);
               }
            }
         } catch (UserInterruptException var4) {
            HytaleServer.get().shutdownServer(ShutdownReason.SIGINT);
         } catch (IOError | EndOfFileException var5) {
         }

      }

      public void interrupt() {
         this.consoleThread.interrupt();
      }
   }
}