/*
    * 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.maven.slf4j;
  
  import java.io.PrintStream;
  import java.util.function.Consumer;
  
  import org.apache.maven.api.services.MessageBuilder;
  import org.slf4j.simple.ExtSimpleLogger;
  
  import static org.apache.maven.jline.MessageUtils.builder;
  
  /**
   * Logger for Maven, that support colorization of levels and stacktraces. This class implements 2 methods introduced in
   * slf4j-simple provider local copy.
   *
   * @since 3.5.0
   */
  public class MavenSimpleLogger extends ExtSimpleLogger {
  
      private String traceRenderedLevel;
      private String debugRenderedLevel;
      private String infoRenderedLevel;
      private String warnRenderedLevel;
      private String errorRenderedLevel;
  
      static Consumer<String> logSink;
  
      public static void setLogSink(Consumer<String> logSink) {
          MavenSimpleLogger.logSink = logSink;
      }
  
      MavenSimpleLogger(String name) {
          super(name);
      }
  
      @Override
      protected String renderLevel(int level) {
          if (traceRenderedLevel == null) {
              traceRenderedLevel = builder().trace("TRACE").build();
              debugRenderedLevel = builder().debug("DEBUG").build();
              infoRenderedLevel = builder().info("INFO").build();
              warnRenderedLevel = builder().warning("WARNING").build();
              errorRenderedLevel = builder().error("ERROR").build();
          }
          switch (level) {
              case LOG_LEVEL_TRACE:
                  return traceRenderedLevel;
              case LOG_LEVEL_DEBUG:
                  return debugRenderedLevel;
              case LOG_LEVEL_INFO:
                  return infoRenderedLevel;
              case LOG_LEVEL_WARN:
                  return warnRenderedLevel;
              case LOG_LEVEL_ERROR:
              default:
                  return errorRenderedLevel;
          }
      }
  
      @Override
      protected void doWrite(StringBuilder buf, Throwable t) {
          Consumer<String> sink = logSink;
          if (sink != null) {
              sink.accept(buf.toString());
          } else {
              super.doWrite(buf, t);
          }
      }
  
      @Override
      protected void writeThrowable(Throwable t, PrintStream stream) {
          if (t == null) {
              return;
          }
          MessageBuilder builder = builder().failure(t.getClass().getName());
          if (t.getMessage() != null) {
              builder.a(": ").failure(t.getMessage());
          }
          stream.println(builder);
  
          printStackTrace(t, stream, "");
      }
 
     private void printStackTrace(Throwable t, PrintStream stream, String prefix) {
         MessageBuilder builder = builder();
         for (StackTraceElement e : t.getStackTrace()) {
             builder.a(prefix);
             builder.a("    ");
             builder.strong("at");
             builder.a(" ");
             builder.a(e.getClassName());
             builder.a(".");
             builder.a(e.getMethodName());
             builder.a("(");
             builder.strong(getLocation(e));
             builder.a(")");
             stream.println(builder);
             builder.setLength(0);
         }
         for (Throwable se : t.getSuppressed()) {
             writeThrowable(se, stream, "Suppressed", prefix + "    ");
         }
         Throwable cause = t.getCause();
         if (cause != null && t != cause) {
             writeThrowable(cause, stream, "Caused by", prefix);
         }
     }
 
     private void writeThrowable(Throwable t, PrintStream stream, String caption, String prefix) {
         MessageBuilder builder =
                 builder().a(prefix).strong(caption).a(": ").a(t.getClass().getName());
         if (t.getMessage() != null) {
             builder.a(": ").failure(t.getMessage());
         }
         stream.println(builder);
 
         printStackTrace(t, stream, prefix);
     }
 
     protected String getLocation(final StackTraceElement e) {
         assert e != null;
 
         if (e.isNativeMethod()) {
             return "Native Method";
         } else if (e.getFileName() == null) {
             return "Unknown Source";
         } else if (e.getLineNumber() >= 0) {
             return e.getFileName() + ":" + e.getLineNumber();
         } else {
             return e.getFileName();
         }
     }
 }