/**

 * SQL-DK

 * Copyright © 2013 František Kučera (frantovo.cz)

 *

 * This program is free software: you can redistribute it and/or modify

 * it under the terms of the GNU General Public License as published by

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program. If not, see <http://www.gnu.org/licenses/>.

 */

package info.globalcode.sql.dk.logging;

import info.globalcode.sql.dk.ColorfulPrintWriter;

import static info.globalcode.sql.dk.ColorfulPrintWriter.TerminalColor;

import static info.globalcode.sql.dk.ColorfulPrintWriter.TerminalStyle;

import static info.globalcode.sql.dk.Functions.rpad;

import java.io.StringWriter;

import java.util.logging.Formatter;

import java.util.logging.Level;

import java.util.logging.LogRecord;

/**

 * For console/terminal log output. Log messages are printed in brief and colorful form.

 *

 * @author Ing. František Kučera (frantovo.cz)

 */

public class ColorfulConsoleFormatter extends Formatter {

	private boolean printStacktrace = false;

	@Override

	public String format(LogRecord r) {

		StringWriter sw = new StringWriter();

		try (ColorfulPrintWriter out = new ColorfulPrintWriter(sw)) {

			printLevel(out, r.getLevel());

			printMessage(out, r);

			printThrowable(out, r);

			out.println();

		}

		return sw.toString();

	}

	private void printLevel(ColorfulPrintWriter out, Level l) {

		TerminalColor color = TerminalColor.Magenta;

		if (l == Level.SEVERE) {

			color = TerminalColor.Red;

		} else if (l == Level.WARNING) {

			color = TerminalColor.Yellow;

		}

		out.print(color, rpad(l.getLocalizedName() + ": ", 10));

	}

	private void printMessage(ColorfulPrintWriter out, LogRecord r) {

		out.print(formatMessage(r));

	}

	private void printThrowable(ColorfulPrintWriter out, LogRecord r) {

		Throwable t = r.getThrown();

		if (t != null) {

			out.print(": ");

			out.print(TerminalColor.Red, t.getClass().getSimpleName());

			String message = t.getLocalizedMessage();

			if (message != null) {

				out.print(": ");

				if (printStacktrace) {

					out.print(message);

				} else {

					out.print(message.replaceAll("\\n", " "));

				}

			}

			if (printStacktrace) {

				out.println();

				out.setForegroundColor(TerminalColor.Yellow);

				out.setStyle(TerminalStyle.Dim);

				t.printStackTrace(out);

				out.resetAll();

			}

		}

	}

	public boolean isPrintStacktrace() {

		return printStacktrace;

	}

	public void setPrintStacktrace(boolean printStacktrace) {

		this.printStacktrace = printStacktrace;

	}

}