From f3f832ff7859c40fef3b2b5789376fbbbc62dbf2 Mon Sep 17 00:00:00 2001
From: George Gastaldi <gegastaldi@gmail.com>
Date: Thu, 21 May 2015 14:46:24 -0300
Subject: [PATCH] FORGE-2336: Showing Warning and Info messages on execute in
 shell

---
 .../addon/shell/aesh/CommandAdapter.java      | 33 ++++++++++---------
 .../addon/ui/impl/output/UIMessageImpl.java   |  2 ++
 2 files changed, 20 insertions(+), 15 deletions(-)

diff --git a/shell/impl/src/main/java/org/jboss/forge/addon/shell/aesh/CommandAdapter.java b/shell/impl/src/main/java/org/jboss/forge/addon/shell/aesh/CommandAdapter.java
index 1f85ed9f05..e210d13c37 100644
--- a/shell/impl/src/main/java/org/jboss/forge/addon/shell/aesh/CommandAdapter.java
+++ b/shell/impl/src/main/java/org/jboss/forge/addon/shell/aesh/CommandAdapter.java
@@ -21,7 +21,6 @@
 import org.jboss.forge.addon.shell.ui.ShellContext;
 import org.jboss.forge.addon.ui.context.UISelection;
 import org.jboss.forge.addon.ui.output.UIMessage;
-import org.jboss.forge.addon.ui.output.UIMessage.Severity;
 import org.jboss.forge.addon.ui.output.UIOutput;
 import org.jboss.forge.addon.ui.result.CompositeResult;
 import org.jboss.forge.addon.ui.result.Failed;
@@ -74,7 +73,24 @@ public CommandResult execute(CommandInvocation commandInvocation) throws IOExcep
       }
       if (!failure)
       {
-         if (interaction.getController().isValid())
+         UIOutput output = shell.getOutput();
+         for (UIMessage message : interaction.getController().validate())
+         {
+            switch (message.getSeverity())
+            {
+            case ERROR:
+               failure = true;
+               output.error(output.err(), message.getDescription());
+               break;
+            case INFO:
+               output.info(output.out(), message.getDescription());
+               break;
+            case WARN:
+               output.warn(output.out(), message.getDescription());
+               break;
+            }
+         }
+         if (!failure)
          {
             Result commandResult = null;
             try
@@ -97,19 +113,6 @@ public CommandResult execute(CommandInvocation commandInvocation) throws IOExcep
                }
             }
          }
-         else
-         {
-            List<UIMessage> messages = interaction.getController().validate();
-            UIOutput output = shell.getOutput();
-            for (UIMessage message : messages)
-            {
-               if (message.getSeverity() == Severity.ERROR)
-               {
-                  failure = true;
-                  output.error(output.err(), message.getDescription());
-               }
-            }
-         }
       }
       return failure ? CommandResult.FAILURE : CommandResult.SUCCESS;
    }
diff --git a/ui/impl/src/main/java/org/jboss/forge/addon/ui/impl/output/UIMessageImpl.java b/ui/impl/src/main/java/org/jboss/forge/addon/ui/impl/output/UIMessageImpl.java
index 55aacf5338..7e552df91d 100644
--- a/ui/impl/src/main/java/org/jboss/forge/addon/ui/impl/output/UIMessageImpl.java
+++ b/ui/impl/src/main/java/org/jboss/forge/addon/ui/impl/output/UIMessageImpl.java
@@ -9,6 +9,7 @@
 
 import org.jboss.forge.addon.ui.input.InputComponent;
 import org.jboss.forge.addon.ui.output.UIMessage;
+import org.jboss.forge.furnace.util.Assert;
 
 /**
  * 
@@ -22,6 +23,7 @@ public class UIMessageImpl implements UIMessage
 
    public UIMessageImpl(Severity severity, String description, InputComponent<?, ?> inputComponent)
    {
+      Assert.notNull(severity, "Severity should never be null");
       this.description = description;
       this.severity = severity;
       this.inputComponent = inputComponent;