/
SystemLogger.java
83 lines (69 loc) · 3.28 KB
/
SystemLogger.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
package net.ME1312.Galaxi.Engine.Library.Log;
import jline.console.ConsoleReader;
import net.ME1312.Galaxi.Engine.GalaxiEngine;
import net.ME1312.Galaxi.Engine.PluginManager;
import net.ME1312.Galaxi.Library.Container;
import net.ME1312.Galaxi.Library.Log.LogStream;
import net.ME1312.Galaxi.Library.Log.Logger;
import net.ME1312.Galaxi.Library.NamedContainer;
import net.ME1312.Galaxi.Library.Util;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.io.PrintStream;
import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.util.HashMap;
import java.util.List;
/**
* System.out and System.err Override Class
*/
public final class SystemLogger extends OutputStream {
private HashMap<String, NamedContainer<LogStream, ByteArrayOutputStream>> last = new HashMap<String, NamedContainer<LogStream, ByteArrayOutputStream>>();
private boolean error;
SystemLogger(boolean level) {
this.error = level;
}
private static void start(PrintStream out, PrintStream err, ConsoleReader in) throws Exception {
if (Util.isNull(out, err)) throw new NullPointerException();
Util.<Container<PrintStream>>reflect(Logger.class.getDeclaredField("pso"), null).set(new PrintStream(new FileLogger(new ConsoleStream(in, out)), false, "UTF-8"));
Util.<Container<PrintStream>>reflect(Logger.class.getDeclaredField("pse"), null).set(new PrintStream(new FileLogger(new ConsoleStream(in, err)), false, "UTF-8"));
System.setOut(new PrintStream(new SystemLogger(false), false, "UTF-8"));
System.setErr(new PrintStream(new SystemLogger(true), false, "UTF-8"));
}
@SuppressWarnings("unchecked")
private List<String> getKnownClasses() {
return Util.getDespiteException(() -> Util.reflect(PluginManager.class.getDeclaredField("knownClasses"), GalaxiEngine.getInstance().getPluginManager()), null);
}
@Override
public void write(int c) throws IOException {
int i = 0;
String origin = java.lang.System.class.getCanonicalName();
for (StackTraceElement element : Thread.currentThread().getStackTrace()) {
if (i > 1 && getKnownClasses().contains(element.getClassName())) {
origin = element.getClassName().replaceAll("\\$([^.$\\d]+)", ".$1").replaceAll("\\$[\\d]+", "");
break;
}
i++;
}
if (!last.keySet().contains(origin)) {
Logger log = new Logger(origin);
last.put(origin, new NamedContainer<LogStream, ByteArrayOutputStream>((error)?log.error:log.info, new ByteArrayOutputStream()));
}
NamedContainer<LogStream, ByteArrayOutputStream> log = last.get(origin);
log.get().write(c);
if (c == '\n') {
log.name().print(log.get().toString("UTF-8").replace("\r\n", "\n"));
last.remove(origin);
}
}
private static void stop() throws Exception {
Thread.sleep(125);
Util.reflect(Logger.class.getDeclaredField("running"), null, false);
Thread thread = Util.reflect(Logger.class.getDeclaredField("thread"), null);
if (thread != null) while (thread.isAlive()) {
Thread.sleep(125);
}
Util.reflect(FileLogger.class.getDeclaredMethod("stop"), null);
}
}