feat: Create a Quarkus Dev Explorer in the IDE

Fixes redhat-developer#1175

Signed-off-by: azerr <[email protected]>

src/main/java/com/redhat/devtools/intellij/lsp4mp4ij/psi/core/project/PsiMicroProfileProject.java

@@ -299,4 +299,7 @@ public String getValue(String key) {
         return provider;
     }
 
+    public Module getJavaProject() {
+        return javaProject;
+    }
 }

src/main/java/com/redhat/devtools/intellij/quarkus/QuarkusBundle.java

@@ -26,20 +26,22 @@
  */
 public final class QuarkusBundle extends DynamicBundle {
 
-  @NonNls public static final String BUNDLE = "messages.QuarkusBundle";
-  private static final QuarkusBundle INSTANCE = new QuarkusBundle();
-
-  private QuarkusBundle() {
-    super(BUNDLE);
-  }
-
-  @NotNull
-  public static @Nls String message(@NotNull @PropertyKey(resourceBundle = BUNDLE) String key, Object @NotNull ... params) {
-    return INSTANCE.getMessage(key, params);
-  }
-
-  @NotNull
-  public static Supplier<@Nls String> messagePointer(@NotNull @PropertyKey(resourceBundle = BUNDLE) String key, Object @NotNull ... params) {
-    return INSTANCE.getLazyMessage(key, params);
-  }
+    @NonNls
+    public static final String BUNDLE = "messages.QuarkusBundle";
+    private static final QuarkusBundle INSTANCE = new QuarkusBundle();
+
+    private QuarkusBundle() {
+        super(BUNDLE);
+    }
+
+    @NotNull
+    public static @Nls String message(@NotNull @PropertyKey(resourceBundle = BUNDLE) String key, Object @NotNull ... params) {
+        return INSTANCE.getMessage(key, params);
+    }
+
+    @NotNull
+    public static Supplier<@Nls String> messagePointer(@NotNull @PropertyKey(resourceBundle = BUNDLE) String key, Object @NotNull ... params) {
+        return INSTANCE.getLazyMessage(key, params);
+    }
+
 }

src/main/java/com/redhat/devtools/intellij/quarkus/explorer/QuarkusActionNode.java

@@ -0,0 +1,66 @@
+package com.redhat.devtools.intellij.quarkus.explorer;
+
+import com.intellij.icons.AllIcons;
+import com.intellij.openapi.application.ApplicationManager;
+import com.intellij.openapi.module.Module;
+import com.intellij.ui.treeStructure.Tree;
+
+import javax.swing.*;
+import javax.swing.tree.DefaultMutableTreeNode;
+import javax.swing.tree.DefaultTreeModel;
+import javax.swing.tree.TreePath;
+
+public abstract class QuarkusActionNode extends DefaultMutableTreeNode {
+
+    private final String name;
+
+    private final QuarkusProjectNode projectNode;
+
+    public QuarkusActionNode(String name, QuarkusProjectNode projectNode) {
+        this.name = name;
+        this.projectNode = projectNode;
+    }
+
+    public String getName() {
+        return name;
+    }
+
+    public Module getModule() {
+        return projectNode.getModule();
+    }
+
+    public QuarkusProjectNode getProjectNode() {
+        return projectNode;
+    }
+
+    public String getDisplayName() {
+        return name;
+    }
+
+    public Icon getIcon() {
+        return AllIcons.Actions.InlayGear;
+    }
+
+    public void refreshNode(boolean nodeStructureChanged) {
+        invokeLater(() -> {
+            var tree = projectNode.getTree();
+            if (nodeStructureChanged) {
+                ((DefaultTreeModel) tree.getModel()).nodeStructureChanged(this);
+            } else {
+                ((DefaultTreeModel) tree.getModel()).nodeChanged(this);
+            }
+            var treePath = new TreePath(this.getPath());
+            tree.expandPath(treePath);
+        });
+    }
+
+    private static void invokeLater(Runnable runnable) {
+        if (ApplicationManager.getApplication().isDispatchThread()) {
+            runnable.run();
+        } else {
+            ApplicationManager.getApplication().invokeLater(runnable);
+        }
+    }
+
+    public abstract String getActionId();
+}

src/main/java/com/redhat/devtools/intellij/quarkus/explorer/QuarkusExplorer.java

@@ -0,0 +1,210 @@
+package com.redhat.devtools.intellij.quarkus.explorer;
+
+import com.intellij.execution.ExecutionListener;
+import com.intellij.execution.ExecutionManager;
+import com.intellij.execution.RunManagerListener;
+import com.intellij.execution.RunnerAndConfigurationSettings;
+import com.intellij.execution.process.ProcessHandler;
+import com.intellij.execution.runners.ExecutionEnvironment;
+import com.intellij.ide.DataManager;
+import com.intellij.openapi.Disposable;
+import com.intellij.openapi.actionSystem.*;
+import com.intellij.openapi.application.ReadAction;
+import com.intellij.openapi.module.Module;
+import com.intellij.openapi.module.ModuleManager;
+import com.intellij.openapi.project.DumbService;
+import com.intellij.openapi.project.Project;
+import com.intellij.openapi.ui.SimpleToolWindowPanel;
+import com.intellij.ui.AnimatedIcon;
+import com.intellij.ui.DoubleClickListener;
+import com.intellij.ui.treeStructure.Tree;
+import com.intellij.util.concurrency.AppExecutorUtil;
+import com.intellij.util.messages.MessageBusConnection;
+import com.redhat.devtools.intellij.quarkus.explorer.actions.QuarkusDevStartAction;
+import com.redhat.devtools.intellij.quarkus.run.QuarkusRunConfiguration;
+import com.redhat.microprofile.psi.quarkus.PsiQuarkusUtils;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+
+import javax.swing.tree.DefaultMutableTreeNode;
+import javax.swing.tree.DefaultTreeModel;
+import javax.swing.tree.TreeNode;
+import javax.swing.tree.TreePath;
+import java.awt.event.KeyAdapter;
+import java.awt.event.KeyEvent;
+import java.awt.event.MouseEvent;
+import java.util.Enumeration;
+
+import static com.redhat.devtools.intellij.quarkus.run.QuarkusRunConfiguration.QUARKUS_CONFIGURATION;
+
+public class QuarkusExplorer extends SimpleToolWindowPanel implements Disposable {
+
+    private final Tree tree;
+    private final Project project;
+
+    public QuarkusExplorer(@NotNull Project project) {
+        super(true, true);
+        this.project = project;
+        tree = buildTree();
+        this.setContent(tree);
+        load();
+    }
+
+    /**
+     * Builds the Language server tree
+     *
+     * @return Tree object of all language servers
+     */
+    private Tree buildTree() {
+
+        DefaultMutableTreeNode top = new DefaultMutableTreeNode("Quarkus projects");
+
+        Tree tree = new Tree(top);
+        tree.setRootVisible(false);
+        tree.setCellRenderer(new QuarkusTreeRenderer());
+
+        tree.putClientProperty(AnimatedIcon.ANIMATION_IN_RENDERER_ALLOWED, true);
+
+        ((DefaultTreeModel) tree.getModel()).reload(top);
+
+
+        var doubleClickListener = new DoubleClickListener() {
+            @Override
+            protected boolean onDoubleClick(@NotNull MouseEvent event) {
+                executeAction(tree);
+                return false;
+            }
+        };
+        doubleClickListener.installOn(tree);
+
+        tree.addKeyListener(new KeyAdapter() {
+            @Override
+            public void keyPressed(KeyEvent e) {
+                if (e.getKeyCode() == KeyEvent.VK_ENTER) {
+                    executeAction(tree);
+                }
+            }
+        });
+
+        MessageBusConnection connection = project.getMessageBus().connect(project);
+        connection.subscribe(RunManagerListener.TOPIC, new RunManagerListener() {
+            @Override
+            public void runConfigurationSelected(@Nullable RunnerAndConfigurationSettings settings) {
+                // Do nothing
+            }
+
+            @Override
+            public void runConfigurationAdded(@NotNull RunnerAndConfigurationSettings settings) {
+                // TODO: refresh tree
+            }
+        });
+        connection.subscribe(ExecutionManager.EXECUTION_TOPIC, new ExecutionListener() {
+
+            @Override
+            public void processNotStarted(@NotNull String executorId, @NotNull ExecutionEnvironment env) {
+                QuarkusRunApplicationNode application = findQuarkusApplication(env);
+                if (application != null) {
+                    application.setApplicationStatus(QuarkusRunApplicationNode.QuarkusApplicationStatus.stopped);
+                }
+            }
+
+            @Override
+            public void processStarted(@NotNull String executorId, @NotNull ExecutionEnvironment env, final @NotNull ProcessHandler handler) {
+                QuarkusRunApplicationNode application = findQuarkusApplication(env);
+                if (application != null) {
+                    application.setApplicationStatus(QuarkusRunApplicationNode.QuarkusApplicationStatus.starting);
+                }
+            }
+
+            @Override
+            public void processTerminating(@NotNull String executorId, @NotNull ExecutionEnvironment env, @NotNull ProcessHandler handler) {
+                QuarkusRunApplicationNode application = findQuarkusApplication(env);
+                if (application != null) {
+                    application.setApplicationStatus(QuarkusRunApplicationNode.QuarkusApplicationStatus.stopping);
+                }
+            }
+
+            @Override
+            public void processTerminated(@NotNull String executorId, @NotNull ExecutionEnvironment env, @NotNull ProcessHandler handler, int exitCode) {
+                QuarkusRunApplicationNode application = findQuarkusApplication(env);
+                if (application != null) {
+                    application.setApplicationStatus(QuarkusRunApplicationNode.QuarkusApplicationStatus.stopped);
+                }
+            }
+
+            private @Nullable QuarkusRunApplicationNode findQuarkusApplication(@NotNull ExecutionEnvironment env) {
+                QuarkusRunConfiguration runConfiguration = env.getDataContext() != null ? (QuarkusRunConfiguration) env.getDataContext().getData(QUARKUS_CONFIGURATION) : null;
+                if (runConfiguration == null) {
+                    return null;
+                }
+
+                Module module = runConfiguration.getModule();
+                DefaultMutableTreeNode root = (DefaultMutableTreeNode) tree.getModel().getRoot();
+                Enumeration<TreeNode> children = root.children();
+                while (children.hasMoreElements()) {
+                    TreeNode node = children.nextElement();
+                    if (node instanceof QuarkusProjectNode && module.equals(((QuarkusProjectNode) node).getModule())) {
+                        QuarkusProjectNode project = (QuarkusProjectNode) node;
+                        Enumeration<TreeNode> children2 = project.children();
+                        while (children2.hasMoreElements()) {
+                            TreeNode node2 = children2.nextElement();
+                            if (node2 instanceof QuarkusRunApplicationNode && runConfiguration.equals(((QuarkusRunApplicationNode) node2).getConfiguration())) {
+                                return (QuarkusRunApplicationNode) node2;
+                            }
+                        }
+                        break;
+                    }
+                }
+                return null;
+            }
+        });
+        return tree;
+    }
+
+    private void load() {
+        var action = ReadAction.nonBlocking(() -> {
+            DefaultMutableTreeNode root = (DefaultMutableTreeNode) ((DefaultTreeModel) tree.getModel()).getRoot();
+            Module[] modules = ModuleManager.getInstance(project).getModules();
+            for (Module javaProject : modules) {
+                if (PsiQuarkusUtils.isQuarkusProject(javaProject)) {
+                    QuarkusProjectNode projectNode = new QuarkusProjectNode(javaProject, tree);
+                    root.add(projectNode);
+                    // Fill Quarkus actions
+                    projectNode.add(new QuarkusRunApplicationNode(projectNode));
+                }
+            }
+            ((DefaultTreeModel) tree.getModel()).reload(root);
+        });
+        var executeInSmartMode = DumbService.getInstance(project).isDumb();
+        if (executeInSmartMode) {
+            action = action.inSmartMode(project);
+        }
+        action
+                .submit(AppExecutorUtil.getAppExecutorService());
+    }
+
+    private static void executeAction(Tree tree) {
+        final TreePath path = tree.getSelectionPath();
+        Object node = path.getLastPathComponent();
+        if (node instanceof QuarkusActionNode) {
+            ActionManager am = ActionManager.getInstance();
+            String actionId = ((QuarkusActionNode) node).getActionId();
+            if (actionId == null) {
+                return;
+            }
+            AnAction action = am.getAction(actionId);
+            if (action != null) {
+                action.actionPerformed(new AnActionEvent(null,
+                        DataManager.getInstance().getDataContext(tree),
+                        ActionPlaces.UNKNOWN, new Presentation(),
+                        am, 0));
+            }
+
+        }
+    }
+
+    @Override
+    public void dispose() {
+
+    }
+}

src/main/java/com/redhat/devtools/intellij/quarkus/explorer/QuarkusOpenDevUINode.java

@@ -0,0 +1,18 @@
+package com.redhat.devtools.intellij.quarkus.explorer;
+
+import com.intellij.openapi.module.Module;
+import com.intellij.ui.treeStructure.Tree;
+import com.redhat.devtools.intellij.quarkus.run.QuarkusOpenDevUIAction;
+
+import javax.swing.tree.DefaultTreeModel;
+
+public class QuarkusOpenDevUINode extends QuarkusActionNode {
+    public QuarkusOpenDevUINode(QuarkusProjectNode projectNode) {
+        super("Open Dev UI", projectNode);
+    }
+
+    @Override
+    public String getActionId() {
+        return QuarkusOpenDevUIAction.ACTION_ID;
+    }
+}

src/main/java/com/redhat/devtools/intellij/quarkus/explorer/QuarkusProjectNode.java

@@ -0,0 +1,43 @@
+package com.redhat.devtools.intellij.quarkus.explorer;
+
+import com.intellij.openapi.module.Module;
+import com.intellij.ui.treeStructure.Tree;
+import com.redhat.devtools.intellij.lsp4mp4ij.psi.core.project.PsiMicroProfileProject;
+import com.redhat.devtools.intellij.lsp4mp4ij.psi.core.project.PsiMicroProfileProjectManager;
+import com.redhat.devtools.intellij.quarkus.lang.QuarkusIcons;
+import com.redhat.devtools.intellij.quarkus.run.QuarkusRunContext;
+
+import javax.swing.*;
+import javax.swing.tree.DefaultMutableTreeNode;
+
+public class QuarkusProjectNode extends DefaultMutableTreeNode {
+
+    private final QuarkusRunContext runContext;
+
+    private final Tree tree;
+
+    public QuarkusProjectNode(Module module,Tree tree) {
+        this.runContext = new QuarkusRunContext(module);
+        this.tree = tree;
+    }
+
+    public String getDisplayName() {
+        return getModule().getName();
+    }
+
+    public Icon getIcon() {
+        return QuarkusIcons.Quarkus;
+    }
+
+    public Module getModule() {
+        return runContext.getMicroProfileProject().getJavaProject();
+    }
+
+    public QuarkusRunContext getRunContext() {
+        return runContext;
+    }
+
+    public Tree getTree() {
+        return tree;
+    }
+}