109 lines
4.5 KiB
Java
109 lines
4.5 KiB
Java
package com.ryanmichela.sshd;
|
|
|
|
/**
|
|
* Copyright 2013 Ryan Michela
|
|
*/
|
|
|
|
import org.bukkit.ChatColor;
|
|
import org.fusesource.jansi.Ansi;
|
|
|
|
import java.io.PrintWriter;
|
|
import java.io.StringWriter;
|
|
import java.text.SimpleDateFormat;
|
|
import java.util.EnumMap;
|
|
import java.util.Map;
|
|
import java.util.logging.Formatter;
|
|
import java.util.logging.LogRecord;
|
|
|
|
public class ConsoleLogFormatter extends Formatter {
|
|
|
|
private SimpleDateFormat dateFormat;
|
|
private static final Map<ChatColor, String> replacements = new EnumMap<ChatColor, String>(ChatColor.class);
|
|
|
|
public ConsoleLogFormatter() {
|
|
this.dateFormat = new SimpleDateFormat("HH:mm:ss");
|
|
}
|
|
|
|
public static String ColorizeString(String str)
|
|
{
|
|
// ORIGINAL CODE FROM org.bukkit.craftbukkit.command.ColouredConsoleSender
|
|
|
|
replacements.put(ChatColor.BLACK, Ansi.ansi().a(Ansi.Attribute.RESET).fg(Ansi.Color.BLACK).boldOff().toString());
|
|
replacements.put(ChatColor.DARK_BLUE, Ansi.ansi().a(Ansi.Attribute.RESET).fg(Ansi.Color.BLUE).boldOff().toString());
|
|
replacements.put(ChatColor.DARK_GREEN, Ansi.ansi().a(Ansi.Attribute.RESET).fg(Ansi.Color.GREEN).boldOff().toString());
|
|
replacements.put(ChatColor.DARK_AQUA, Ansi.ansi().a(Ansi.Attribute.RESET).fg(Ansi.Color.CYAN).boldOff().toString());
|
|
replacements.put(ChatColor.DARK_RED, Ansi.ansi().a(Ansi.Attribute.RESET).fg(Ansi.Color.RED).boldOff().toString());
|
|
replacements.put(ChatColor.DARK_PURPLE, Ansi.ansi().a(Ansi.Attribute.RESET).fg(Ansi.Color.MAGENTA).boldOff().toString());
|
|
replacements.put(ChatColor.GOLD, Ansi.ansi().a(Ansi.Attribute.RESET).fg(Ansi.Color.YELLOW).boldOff().toString());
|
|
replacements.put(ChatColor.GRAY, Ansi.ansi().a(Ansi.Attribute.RESET).fg(Ansi.Color.WHITE).boldOff().toString());
|
|
replacements.put(ChatColor.DARK_GRAY, Ansi.ansi().a(Ansi.Attribute.RESET).fg(Ansi.Color.BLACK).bold().toString());
|
|
replacements.put(ChatColor.BLUE, Ansi.ansi().a(Ansi.Attribute.RESET).fg(Ansi.Color.BLUE).bold().toString());
|
|
replacements.put(ChatColor.GREEN, Ansi.ansi().a(Ansi.Attribute.RESET).fg(Ansi.Color.GREEN).bold().toString());
|
|
replacements.put(ChatColor.AQUA, Ansi.ansi().a(Ansi.Attribute.RESET).fg(Ansi.Color.CYAN).bold().toString());
|
|
replacements.put(ChatColor.RED, Ansi.ansi().a(Ansi.Attribute.RESET).fg(Ansi.Color.RED).bold().toString());
|
|
replacements.put(ChatColor.LIGHT_PURPLE, Ansi.ansi().a(Ansi.Attribute.RESET).fg(Ansi.Color.MAGENTA).bold().toString());
|
|
replacements.put(ChatColor.YELLOW, Ansi.ansi().a(Ansi.Attribute.RESET).fg(Ansi.Color.YELLOW).bold().toString());
|
|
replacements.put(ChatColor.WHITE, Ansi.ansi().a(Ansi.Attribute.RESET).fg(Ansi.Color.WHITE).bold().toString());
|
|
replacements.put(ChatColor.MAGIC, Ansi.ansi().a(Ansi.Attribute.BLINK_SLOW).toString());
|
|
replacements.put(ChatColor.BOLD, Ansi.ansi().a(Ansi.Attribute.UNDERLINE_DOUBLE).toString());
|
|
replacements.put(ChatColor.STRIKETHROUGH, Ansi.ansi().a(Ansi.Attribute.STRIKETHROUGH_ON).toString());
|
|
replacements.put(ChatColor.UNDERLINE, Ansi.ansi().a(Ansi.Attribute.UNDERLINE).toString());
|
|
replacements.put(ChatColor.ITALIC, Ansi.ansi().a(Ansi.Attribute.ITALIC).toString());
|
|
replacements.put(ChatColor.RESET, Ansi.ansi().a(Ansi.Attribute.RESET).toString());
|
|
|
|
String result = str;
|
|
for (ChatColor color : ChatColor.values())
|
|
{
|
|
if (replacements.containsKey(color))
|
|
{
|
|
result = result.replaceAll("(?i)" + color.toString(), replacements.get(color));
|
|
}
|
|
else
|
|
{
|
|
result = result.replaceAll("(?i)" + color.toString(), "");
|
|
}
|
|
}
|
|
result += Ansi.ansi().reset().toString();
|
|
return result;
|
|
}
|
|
|
|
public String format(LogRecord logrecord)
|
|
{
|
|
try
|
|
{
|
|
Class.forName("org.bukkit.craftbukkit.command.ColouredConsoleSender");
|
|
}
|
|
catch (ClassNotFoundException ignored)
|
|
{
|
|
// MEANS WE'RE ON PAPER/TACO/OTHER SHIT
|
|
colorize(logrecord);
|
|
}
|
|
StringBuilder stringbuilder = new StringBuilder();
|
|
|
|
stringbuilder.append(" [");
|
|
stringbuilder.append(this.dateFormat.format(logrecord.getMillis())).append(" ");
|
|
|
|
stringbuilder.append(logrecord.getLevel().getName()).append("]: ");
|
|
stringbuilder.append(this.formatMessage(logrecord));
|
|
stringbuilder.append('\n');
|
|
Throwable throwable = logrecord.getThrown();
|
|
|
|
if (throwable != null)
|
|
{
|
|
StringWriter stringwriter = new StringWriter();
|
|
|
|
throwable.printStackTrace(new PrintWriter(stringwriter));
|
|
stringbuilder.append(stringwriter.toString());
|
|
}
|
|
|
|
return stringbuilder.toString().replace("\n", "\r\n");
|
|
}
|
|
|
|
private void colorize(LogRecord logrecord)
|
|
{
|
|
String result = ColorizeString(logrecord.getMessage());
|
|
logrecord.setMessage(result);
|
|
}
|
|
}
|
|
|