Show UISelection contents in wizard title

forge · Sep 20, 2014 · 6b5b2ae · 6b5b2ae
1 parent 77b95bd
commit 6b5b2ae
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 12 deletions.
diff --git a/src/main/java/org/jboss/forge/plugin/idea/ui/wizard/ForgeWizardDialog.java b/src/main/java/org/jboss/forge/plugin/idea/ui/wizard/ForgeWizardDialog.java
@@ -44,7 +44,7 @@ public String getTitle()
 
     public void refreshTitle()
     {
-        Object selection = context.getSelection();
+        Object selection = context.getSelection().get();
 
         setTitle(name + (selection != null ? " [" + selection + "]" : ""));
     }

diff --git a/src/main/java/org/jboss/forge/plugin/idea/util/IDEUtil.java b/src/main/java/org/jboss/forge/plugin/idea/util/IDEUtil.java
@@ -52,18 +52,9 @@ public static void openSelection(UIContext context)
     {
         Project project = projectFromContext(context);
 
-        if (context.getSelection() instanceof Iterable)
+        for (Object selection : context.getSelection())
         {
-            Iterable iterable = context.getSelection();
-
-            for (Object selection : iterable)
-            {
-                openSingleSelection(project, selection);
-            }
-        }
-        else
-        {
-            openSingleSelection(project, context.getSelection());
+            openSingleSelection(project, selection);
         }
     }