1.1 --- /dev/null	Thu Jan 01 00:00:00 1970 +0000
     1.2 +++ b/java/sql-dk/src/main/java/info/globalcode/sql/dk/logging/ColorfulConsoleFormatter.java	Mon Mar 04 20:15:24 2019 +0100
     1.3 @@ -0,0 +1,97 @@
     1.4 +/**
     1.5 + * SQL-DK
     1.6 + * Copyright © 2013 František Kučera (frantovo.cz)
     1.7 + *
     1.8 + * This program is free software: you can redistribute it and/or modify
     1.9 + * it under the terms of the GNU General Public License as published by
    1.10 + * the Free Software Foundation, either version 3 of the License, or
    1.11 + * (at your option) any later version.
    1.12 + *
    1.13 + * This program is distributed in the hope that it will be useful,
    1.14 + * but WITHOUT ANY WARRANTY; without even the implied warranty of
    1.15 + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
    1.16 + * GNU General Public License for more details.
    1.17 + *
    1.18 + * You should have received a copy of the GNU General Public License
    1.19 + * along with this program. If not, see <http://www.gnu.org/licenses/>.
    1.20 + */
    1.21 +package info.globalcode.sql.dk.logging;
    1.22 +
    1.23 +import info.globalcode.sql.dk.ColorfulPrintWriter;
    1.24 +import static info.globalcode.sql.dk.ColorfulPrintWriter.TerminalColor;
    1.25 +import static info.globalcode.sql.dk.ColorfulPrintWriter.TerminalStyle;
    1.26 +import static info.globalcode.sql.dk.Functions.rpad;
    1.27 +import java.io.StringWriter;
    1.28 +import java.util.logging.Formatter;
    1.29 +import java.util.logging.Level;
    1.30 +import java.util.logging.LogRecord;
    1.31 +
    1.32 +/**
    1.33 + * For console/terminal log output. Log messages are printed in brief and colorful form.
    1.34 + *
    1.35 + * @author Ing. František Kučera (frantovo.cz)
    1.36 + */
    1.37 +public class ColorfulConsoleFormatter extends Formatter {
    1.38 +
    1.39 +	private boolean printStacktrace = false;
    1.40 +
    1.41 +	@Override
    1.42 +	public String format(LogRecord r) {
    1.43 +		StringWriter sw = new StringWriter();
    1.44 +		try (ColorfulPrintWriter out = new ColorfulPrintWriter(sw)) {
    1.45 +			printLevel(out, r.getLevel());
    1.46 +			printMessage(out, r);
    1.47 +			printThrowable(out, r);
    1.48 +			out.println();
    1.49 +		}
    1.50 +		return sw.toString();
    1.51 +	}
    1.52 +
    1.53 +	private void printLevel(ColorfulPrintWriter out, Level l) {
    1.54 +		TerminalColor color = TerminalColor.Magenta;
    1.55 +
    1.56 +		if (l == Level.SEVERE) {
    1.57 +			color = TerminalColor.Red;
    1.58 +		} else if (l == Level.WARNING) {
    1.59 +			color = TerminalColor.Yellow;
    1.60 +		}
    1.61 +
    1.62 +		out.print(color, rpad(l.getLocalizedName() + ": ", 10));
    1.63 +	}
    1.64 +
    1.65 +	private void printMessage(ColorfulPrintWriter out, LogRecord r) {
    1.66 +		out.print(formatMessage(r));
    1.67 +	}
    1.68 +
    1.69 +	private void printThrowable(ColorfulPrintWriter out, LogRecord r) {
    1.70 +		Throwable t = r.getThrown();
    1.71 +		if (t != null) {
    1.72 +			out.print(": ");
    1.73 +			out.print(TerminalColor.Red, t.getClass().getSimpleName());
    1.74 +			String message = t.getLocalizedMessage();
    1.75 +			if (message != null) {
    1.76 +				out.print(": ");
    1.77 +				if (printStacktrace) {
    1.78 +					out.print(message);
    1.79 +				} else {
    1.80 +					out.print(message.replaceAll("\\n", " "));
    1.81 +				}
    1.82 +			}
    1.83 +			if (printStacktrace) {
    1.84 +				out.println();
    1.85 +				out.setForegroundColor(TerminalColor.Yellow);
    1.86 +				out.setStyle(TerminalStyle.Dim);
    1.87 +				t.printStackTrace(out);
    1.88 +				out.resetAll();
    1.89 +			}
    1.90 +		}
    1.91 +	}
    1.92 +
    1.93 +	public boolean isPrintStacktrace() {
    1.94 +		return printStacktrace;
    1.95 +	}
    1.96 +
    1.97 +	public void setPrintStacktrace(boolean printStacktrace) {
    1.98 +		this.printStacktrace = printStacktrace;
    1.99 +	}
   1.100 +}