plugins/maven/src/main/java/org/jetbrains/idea/maven/embedder/MavenConsole.java

   1 /*
   2  * Copyright 2000-2009 JetBrains s.r.o.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  * http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16 package org.jetbrains.idea.maven.embedder;
  17
  18 import com.intellij.execution.process.ProcessHandler;
  19 import com.intellij.openapi.util.text.StringUtil;
  20 import gnu.trove.THashMap;
  21 import org.apache.maven.plugin.AbstractMojoExecutionException;
  22
  23 import java.text.MessageFormat;
  24 import java.util.Map;
  25
  26 public abstract class MavenConsole {
  27   public static final int LEVEL_AUTO = -1;
  28   public static final int LEVEL_DEBUG = 0;
  29   public static final int LEVEL_INFO = 1;
  30   public static final int LEVEL_WARN = 2;
  31   public static final int LEVEL_ERROR = 3;
  32   public static final int LEVEL_FATAL = 4;
  33   public static final int LEVEL_DISABLED = 5;
  34   public static final int LEVEL_ALWAYS = 6;
  35
  36   public static final String LINE_SEPARATOR = System.getProperty("line.separator");
  37
  38   public enum OutputType {
  39     NORMAL, SYSTEM, ERROR
  40   }
  41
  42   private final int myOutputLevel;
  43   private final boolean myPrintStrackTrace;
  44   private boolean isFinished;
  45
  46   private static final Map<String, Integer> PREFIX_TO_LEVEL = new THashMap<String, Integer>();
  47   private static final Map<Integer, String> LEVEL_TO_PREFIX = new THashMap<Integer, String>();
  48
  49   static {
  50     map("DEBUG", LEVEL_DEBUG);
  51     map("INFO", LEVEL_INFO);
  52     map("WARNING", LEVEL_WARN);
  53     map("ERROR", LEVEL_ERROR);
  54     map("FATAL_ERROR", LEVEL_FATAL);
  55   }
  56
  57   private static void map(String prefix, int level) {
  58     PREFIX_TO_LEVEL.put(prefix, level);
  59     LEVEL_TO_PREFIX.put(level, prefix);
  60   }
  61
  62   public MavenConsole(MavenExecutionOptions.LoggingLevel outputLevel, boolean printStrackTrace) {
  63     myOutputLevel = outputLevel.getLevel();
  64     myPrintStrackTrace = printStrackTrace;
  65   }
  66
  67   public boolean isSuppressed(int level) {
  68     return level < myOutputLevel;
  69   }
  70
  71   public boolean isSuppressed(String line) {
  72     return isSuppressed(getLevel(line));
  73   }
  74
  75   public abstract boolean canPause();
  76
  77   public abstract boolean isOutputPaused();
  78
  79   public abstract void setOutputPaused(boolean outputPaused);
  80
  81   public boolean isFinished() {
  82     return isFinished;
  83   }
  84
  85   public void finish() {
  86     isFinished = true;
  87   }
  88
  89   public abstract void attachToProcess(ProcessHandler processHandler);
  90
  91   public void systemMessage(int level, String string, Throwable throwable) {
  92     printMessage(level, string, throwable);
  93   }
  94
  95   public void printMessage(int level, String string, Throwable throwable) {
  96     if (isSuppressed(level)) return;
  97
  98     OutputType type = OutputType.NORMAL;
  99     if (throwable != null || level == LEVEL_WARN || level == LEVEL_ERROR || level == LEVEL_FATAL) {
 100       type = OutputType.ERROR;
 101     }
 102
 103     doPrint(composeLine(level, string), type);
 104
 105     if (level == LEVEL_FATAL) {
 106       setOutputPaused(false);
 107     }
 108
 109     if (throwable != null) {
 110       String message = null;
 111
 112       Throwable temp = throwable;
 113       while (temp != null) {
 114         if (temp instanceof AbstractMojoExecutionException) {
 115           message = appendExecutionFailureMessage(message, temp.getMessage());
 116           message = appendExecutionFailureMessage(message, ((AbstractMojoExecutionException)temp).getLongMessage());
 117
 118           if (temp.getCause() != null) {
 119             message = appendExecutionFailureMessage(message, temp.getCause().getMessage());
 120           }
 121           break;
 122         }
 123         temp = temp.getCause();
 124       }
 125
 126       if (message == null) message = throwable.getMessage();
 127
 128       if (message != null) {
 129         message += LINE_SEPARATOR;
 130         doPrint(LINE_SEPARATOR + composeLine(LEVEL_ERROR, message), type);
 131       }
 132
 133       if (myPrintStrackTrace) {
 134         doPrint(LINE_SEPARATOR + StringUtil.getThrowableText(throwable), OutputType.ERROR);
 135         }
 136       else {
 137         doPrint(LINE_SEPARATOR +
 138                 "To view full stack traces, please go to the Settings->Maven and check the 'Print Exception Stack Traces' box." +
 139                 LINE_SEPARATOR,
 140                 type);
 141       }
 142     }
 143   }
 144
 145   private String appendExecutionFailureMessage(String message, String newMessage) {
 146     if (message == null) return newMessage;
 147     if (newMessage == null) return message;
 148     return message + LINE_SEPARATOR + LINE_SEPARATOR + newMessage;
 149   }
 150
 151   protected abstract void doPrint(String text, OutputType type);
 152
 153   private static int getLevel(String line) {
 154     return getLevelByPrefix(extractPrefix(line));
 155   }
 156
 157   private static String extractPrefix(String line) {
 158     if (line.startsWith("[")) {
 159       int closing = line.indexOf("] ", 1);
 160       if (closing > 1) {
 161         return line.substring(1, closing);
 162       }
 163     }
 164     return "";
 165   }
 166
 167   private static int getLevelByPrefix(String prefix) {
 168     Integer level = PREFIX_TO_LEVEL.get(prefix);
 169     return level != null ? level : LEVEL_ALWAYS;
 170   }
 171
 172   private static String composeLine(int level, String message) {
 173     return MessageFormat.format("[{0}] {1}", getPrefixByLevel(level), message);
 174   }
 175
 176   private static String getPrefixByLevel(int level) {
 177     return LEVEL_TO_PREFIX.get(level);
 178   }
 179 }