FORGE-2191: Support UIDesktop as a reserved parameter

Author: gastaldi

ui/impl/src/main/java/org/jboss/forge/addon/ui/impl/annotation/ReservedParameters.java

@@ -7,6 +7,7 @@
 
 package org.jboss.forge.addon.ui.impl.annotation;
 
+import org.jboss.forge.addon.ui.UIDesktop;
 import org.jboss.forge.addon.ui.UIProvider;
 import org.jboss.forge.addon.ui.context.UIContext;
 import org.jboss.forge.addon.ui.context.UIExecutionContext;
@@ -28,7 +29,8 @@ public static boolean isReservedParameter(Class<?> type)
                || type == UIOutput.class
                || type == UIProgressMonitor.class
                || type == UIProvider.class
-               || type == UIExecutionContext.class;
+               || type == UIExecutionContext.class
+               || type == UIDesktop.class;
    }
 
    public static Object getReservedParameter(UIExecutionContext context, Class<?> type)
@@ -53,10 +55,14 @@ else if (type == UIProgressMonitor.class)
       {
          return context.getProgressMonitor();
       }
-      else if (type == UIExecutionContext.class) 
+      else if (type == UIExecutionContext.class)
       {
          return context;
       }
+      else if (type == UIDesktop.class)
+      {
+         return context.getUIContext().getProvider().getDesktop();
+      }
       else
       {
          return null;

ui/impl/src/test/java/org/jboss/forge/addon/ui/impl/annotation/ReservedParametersTest.java

@@ -0,0 +1,38 @@
+/**
+ * Copyright 2015 Red Hat, Inc. and/or its affiliates.
+ *
+ * Licensed under the Eclipse Public License version 1.0, available at
+ * http://www.eclipse.org/legal/epl-v10.html
+ */
+
+package org.jboss.forge.addon.ui.impl.annotation;
+
+import org.jboss.forge.addon.ui.UIDesktop;
+import org.jboss.forge.addon.ui.UIProvider;
+import org.jboss.forge.addon.ui.context.UIContext;
+import org.jboss.forge.addon.ui.context.UIExecutionContext;
+import org.jboss.forge.addon.ui.input.UIPrompt;
+import org.jboss.forge.addon.ui.output.UIOutput;
+import org.jboss.forge.addon.ui.progress.UIProgressMonitor;
+import org.junit.Assert;
+import org.junit.Test;
+
+/**
+ * Test class for {@link ReservedParameters}
+ * 
+ * @author <a href="ggastald@redhat.com">George Gastaldi</a>
+ */
+public class ReservedParametersTest
+{
+   @Test
+   public void testIsReservedParameter()
+   {
+      Assert.assertTrue(ReservedParameters.isReservedParameter(UIContext.class));
+      Assert.assertTrue(ReservedParameters.isReservedParameter(UIPrompt.class));
+      Assert.assertTrue(ReservedParameters.isReservedParameter(UIOutput.class));
+      Assert.assertTrue(ReservedParameters.isReservedParameter(UIProgressMonitor.class));
+      Assert.assertTrue(ReservedParameters.isReservedParameter(UIProvider.class));
+      Assert.assertTrue(ReservedParameters.isReservedParameter(UIExecutionContext.class));
+      Assert.assertTrue(ReservedParameters.isReservedParameter(UIDesktop.class));
+   }
+}