125 lines
3.3 KiB
Java
125 lines
3.3 KiB
Java
package me.totalfreedom.bukkittelnet;
|
|
|
|
import java.io.PrintWriter;
|
|
import java.io.StringWriter;
|
|
import java.text.SimpleDateFormat;
|
|
import java.util.Collections;
|
|
import java.util.Date;
|
|
import java.util.HashSet;
|
|
import java.util.Set;
|
|
import me.totalfreedom.bukkittelnet.session.ClientSession;
|
|
import me.totalfreedom.bukkittelnet.session.FilterMode;
|
|
import org.apache.logging.log4j.LogManager;
|
|
import org.apache.logging.log4j.core.LogEvent;
|
|
import org.apache.logging.log4j.core.Logger;
|
|
import org.apache.logging.log4j.core.appender.AbstractAppender;
|
|
import org.apache.logging.log4j.core.config.Property;
|
|
|
|
public class TelnetLogAppender extends AbstractAppender
|
|
{
|
|
private static final long serialVersionUID = 234234L;
|
|
|
|
private final Set<ClientSession> sessions;
|
|
private final SimpleDateFormat dateFormat;
|
|
|
|
public TelnetLogAppender()
|
|
{
|
|
super("BukkitTelnet", null, null, true, Property.EMPTY_ARRAY);
|
|
|
|
this.sessions = new HashSet<>();
|
|
this.dateFormat = new SimpleDateFormat("HH:mm:ss");
|
|
|
|
super.start();
|
|
}
|
|
|
|
public Set<ClientSession> getSessions()
|
|
{
|
|
return Collections.unmodifiableSet(sessions);
|
|
}
|
|
|
|
public boolean addSession(ClientSession session)
|
|
{
|
|
return sessions.add(session);
|
|
}
|
|
|
|
public boolean removeSession(ClientSession session)
|
|
{
|
|
return sessions.remove(session);
|
|
}
|
|
|
|
public void removeAllSesssions()
|
|
{
|
|
sessions.clear();
|
|
}
|
|
|
|
@Override
|
|
public void append(LogEvent event)
|
|
{
|
|
final String message = event.getMessage().getFormattedMessage();
|
|
|
|
for (ClientSession session : sessions)
|
|
{
|
|
try
|
|
{
|
|
if (!session.syncIsConnected())
|
|
{
|
|
continue;
|
|
}
|
|
|
|
boolean chat = message.startsWith("<")
|
|
|| message.startsWith("[Server")
|
|
|| message.startsWith("[CONSOLE") || message.startsWith("[TotalFreedomMod] [ADMIN]");
|
|
|
|
if (session.getFilterMode() == FilterMode.CHAT_ONLY && !chat)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
if (session.getFilterMode() == FilterMode.NON_CHAT_ONLY && chat)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
session.writeRawLine(formatMessage(message, event));
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
ex.printStackTrace();
|
|
}
|
|
}
|
|
}
|
|
|
|
private String formatMessage(String message, LogEvent event)
|
|
{
|
|
final StringBuilder builder = new StringBuilder();
|
|
final Throwable ex = event.getThrown();
|
|
|
|
builder.append("[");
|
|
builder.append(dateFormat.format(new Date()));
|
|
builder.append(" ");
|
|
builder.append(event.getLevel().name().toUpperCase());
|
|
builder.append("]: ");
|
|
builder.append(message);
|
|
|
|
if (ex != null)
|
|
{
|
|
StringWriter writer = new StringWriter();
|
|
ex.printStackTrace(new PrintWriter(writer));
|
|
builder.append(writer);
|
|
}
|
|
|
|
return builder.toString();
|
|
}
|
|
|
|
public void attach()
|
|
{
|
|
((Logger) LogManager.getRootLogger()).addAppender(this);
|
|
}
|
|
|
|
public void deattach()
|
|
{
|
|
((Logger) LogManager.getRootLogger()).removeAppender(this);
|
|
}
|
|
|
|
}
|