/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.apache.openejb.log;

import org.apache.openejb.loader.SystemInstance;
import org.fusesource.jansi.Ansi;

import java.io.PrintWriter;
import java.io.StringWriter;
import java.util.HashMap;
import java.util.Map;
import java.util.logging.Level;
import java.util.logging.LogRecord;

public class ColorFormatter extends SingleLineFormatter {
    public static final String OPENEJB_LOG_COLOR_PREFIX = "openejb.log.color.";

    private final Map<Level, Ansi.Color> colors = new HashMap<Level, Ansi.Color>();

    public ColorFormatter() {
        // until info level  information should be not so import so keep default color
        colors.put(Level.FINEST, color("finest", Ansi.Color.DEFAULT.name()));
        colors.put(Level.FINER, color("finer", Ansi.Color.DEFAULT.name()));
        colors.put(Level.FINE, color("fine", Ansi.Color.DEFAULT.name()));

        // info is the main one so white? (default is often a bit gray so less visible)
        colors.put(Level.INFO, color("info", Ansi.Color.WHITE.name()));

        // warn and more are important information so specify a visible color
        colors.put(Level.WARNING, color("warning", Ansi.Color.YELLOW.name()));
        colors.put(Level.SEVERE, color("severe", Ansi.Color.RED.name()));
    }

    @Override
    public synchronized String format(final LogRecord record) {
        final boolean exception = record.getThrown() != null;
        final Ansi sbuf = prefix(record);
        sbuf.a(record.getLevel().getLocalizedName());
        sbuf.a(" - ");
        sbuf.a(formatMessage(record));
        if (!exception) {
            suffix(sbuf, record);
        }
        sbuf.newline();
        if (exception) {
            try {
                final StringWriter sw = new StringWriter();
                final PrintWriter pw = new PrintWriter(sw);
                record.getThrown().printStackTrace(pw);
                pw.close();
                sbuf.a(sw.toString());
            } catch (final Exception ex) {
                // no-op
            } finally {
                suffix(sbuf, record);
            }
        }
        return sbuf.toString();
    }

    private Ansi.Color color(final String lvl, final String aDefault) {
        try {
            return Ansi.Color.valueOf(SystemInstance.get().getProperty(OPENEJB_LOG_COLOR_PREFIX + lvl, aDefault));
        } catch (final IllegalArgumentException iae) {
            return Ansi.Color.valueOf(aDefault);
        }
    }

    private Ansi color(final Level key) {
        if (colors.containsKey(key)) {
            return Ansi.ansi().fg(colors.get(key));
        }
        return Ansi.ansi().fg(Ansi.Color.DEFAULT);
    }

    protected Ansi prefix(final LogRecord record) {
        return color(record.getLevel());
    }

    protected Ansi suffix(final Ansi ansi, final LogRecord record) {
        return ansi.reset();
    }
}
