Merge two listeners.
[idea/community.git] / plugins / maven / src / main / java / org / jetbrains / idea / maven / wizards / MavenArchetypesPanel.java
1 /*
2  * Copyright 2000-2013 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.wizards;
17
18 import com.intellij.ide.wizard.StepAdapter;
19 import com.intellij.openapi.Disposable;
20 import com.intellij.openapi.application.ApplicationManager;
21 import com.intellij.openapi.util.Disposer;
22 import com.intellij.openapi.util.text.StringUtil;
23 import com.intellij.ui.*;
24 import com.intellij.ui.treeStructure.Tree;
25 import com.intellij.util.containers.Convertor;
26 import com.intellij.util.ui.AsyncProcessIcon;
27 import com.intellij.util.ui.UIUtil;
28 import com.intellij.util.ui.tree.TreeUtil;
29 import org.jetbrains.annotations.Nullable;
30 import org.jetbrains.idea.maven.indices.MavenIndicesManager;
31 import org.jetbrains.idea.maven.model.MavenArchetype;
32
33 import javax.swing.*;
34 import javax.swing.event.TreeSelectionEvent;
35 import javax.swing.event.TreeSelectionListener;
36 import javax.swing.tree.*;
37 import java.awt.*;
38 import java.awt.event.ActionEvent;
39 import java.awt.event.ActionListener;
40 import java.util.*;
41 import java.util.List;
42
43 /**
44  * @author Dmitry Avdeev
45  *         Date: 24.09.13
46  */
47 public class MavenArchetypesPanel implements Disposable {
48
49   private JCheckBox myUseArchetypeCheckBox;
50   private JButton myAddArchetypeButton;
51   private JPanel myArchetypesPanel;
52   private final Tree myArchetypesTree;
53   private JScrollPane myArchetypeDescriptionScrollPane;
54   private JPanel myMainPanel;
55   private JTextArea myArchetypeDescriptionField;
56
57   private Object myCurrentUpdaterMarker;
58   private final AsyncProcessIcon myLoadingIcon = new AsyncProcessIcon.Big(getClass() + ".loading");
59
60   private boolean skipUpdateUI;
61   private final MavenModuleBuilder myBuilder;
62   @Nullable private final StepAdapter myStep;
63
64   public MavenArchetypesPanel(MavenModuleBuilder builder, @Nullable StepAdapter step) {
65     myBuilder = builder;
66     myStep = step;
67     Disposer.register(this, myLoadingIcon);
68
69     myArchetypesTree = new Tree();
70     myArchetypesTree.setModel(new DefaultTreeModel(new DefaultMutableTreeNode()));
71     JScrollPane archetypesScrollPane = ScrollPaneFactory.createScrollPane(myArchetypesTree);
72
73     myArchetypesPanel.add(archetypesScrollPane, "archetypes");
74
75     JPanel loadingPanel = new JPanel(new GridBagLayout());
76     JPanel bp = new JPanel(new BorderLayout(10, 10));
77     bp.add(new JLabel("Loading archetype list..."), BorderLayout.NORTH);
78     bp.add(myLoadingIcon, BorderLayout.CENTER);
79
80     loadingPanel.add(bp, new GridBagConstraints());
81
82     myArchetypesPanel.add(ScrollPaneFactory.createScrollPane(loadingPanel), "loading");
83     ((CardLayout)myArchetypesPanel.getLayout()).show(myArchetypesPanel, "archetypes");
84
85
86     myUseArchetypeCheckBox.addActionListener(new ActionListener() {
87       @Override
88       public void actionPerformed(ActionEvent e) {
89         updateComponents();
90         archetypeMayBeChanged();
91       }
92     });
93
94     myAddArchetypeButton.addActionListener(new ActionListener() {
95       public void actionPerformed(ActionEvent e) {
96         doAddArchetype();
97       }
98     });
99
100     myArchetypesTree.setRootVisible(false);
101     myArchetypesTree.setShowsRootHandles(true);
102     myArchetypesTree.setCellRenderer(new MyRenderer());
103     myArchetypesTree.getSelectionModel().setSelectionMode(TreeSelectionModel.SINGLE_TREE_SELECTION);
104
105     myArchetypesTree.getSelectionModel().addTreeSelectionListener(new TreeSelectionListener() {
106       public void valueChanged(TreeSelectionEvent e) {
107         updateArchetypeDescription();
108         archetypeMayBeChanged();
109       }
110     });
111
112     new TreeSpeedSearch(myArchetypesTree, new Convertor<TreePath, String>() {
113       public String convert(TreePath path) {
114         MavenArchetype info = getArchetypeInfoFromPathComponent(path.getLastPathComponent());
115         return info.groupId + ":" + info.artifactId + ":" + info.version;
116       }
117     }).setComparator(new SpeedSearchComparator(false));
118
119     myArchetypeDescriptionField.setEditable(false);
120     myArchetypeDescriptionField.setBackground(UIUtil.getPanelBackground());
121
122     requestUpdate();
123     updateComponents();
124   }
125
126   public JPanel getMainPanel() {
127     return myMainPanel;
128   }
129
130   private void updateComponents() {
131     boolean archetypesEnabled = myUseArchetypeCheckBox.isSelected();
132     myAddArchetypeButton.setEnabled(archetypesEnabled);
133     myArchetypesTree.setEnabled(archetypesEnabled);
134     myArchetypesTree.setBackground(archetypesEnabled ? UIUtil.getListBackground() : UIUtil.getPanelBackground());
135
136   }
137
138   @Nullable
139   public MavenArchetype getSelectedArchetype() {
140     if (!myUseArchetypeCheckBox.isSelected() || myArchetypesTree.isSelectionEmpty()) return null;
141     return getArchetypeInfoFromPathComponent(myArchetypesTree.getLastSelectedPathComponent());
142   }
143
144   private static MavenArchetype getArchetypeInfoFromPathComponent(Object sel) {
145     return (MavenArchetype)((DefaultMutableTreeNode)sel).getUserObject();
146   }
147
148   private void updateArchetypeDescription() {
149     MavenArchetype sel = getSelectedArchetype();
150     String desc = sel == null ? null : sel.description;
151     if (StringUtil.isEmptyOrSpaces(desc)) {
152       myArchetypeDescriptionScrollPane.setVisible(false);
153     }
154     else {
155       myArchetypeDescriptionScrollPane.setVisible(true);
156       myArchetypeDescriptionField.setText(desc);
157     }
158   }
159
160   @Nullable
161   private static TreePath findNodePath(MavenArchetype object, TreeModel model, Object parent) {
162     for (int i = 0; i < model.getChildCount(parent); i++) {
163       DefaultMutableTreeNode each = (DefaultMutableTreeNode)model.getChild(parent, i);
164       if (each.getUserObject().equals(object)) return new TreePath(each.getPath());
165
166       TreePath result = findNodePath(object, model, each);
167       if (result != null) return result;
168     }
169     return null;
170   }
171
172   private static TreeNode groupAndSortArchetypes(Set<MavenArchetype> archetypes) {
173     List<MavenArchetype> list = new ArrayList<MavenArchetype>(archetypes);
174
175     Collections.sort(list, new Comparator<MavenArchetype>() {
176       public int compare(MavenArchetype o1, MavenArchetype o2) {
177         String key1 = o1.groupId + ":" + o1.artifactId;
178         String key2 = o2.groupId + ":" + o2.artifactId;
179
180         int result = key1.compareToIgnoreCase(key2);
181         if (result != 0) return result;
182
183         return o2.version.compareToIgnoreCase(o1.version);
184       }
185     });
186
187     Map<String, List<MavenArchetype>> map = new TreeMap<String, List<MavenArchetype>>();
188
189     for (MavenArchetype each : list) {
190       String key = each.groupId + ":" + each.artifactId;
191       List<MavenArchetype> versions = map.get(key);
192       if (versions == null) {
193         versions = new ArrayList<MavenArchetype>();
194         map.put(key, versions);
195       }
196       versions.add(each);
197     }
198
199     DefaultMutableTreeNode result = new DefaultMutableTreeNode("root", true);
200     for (List<MavenArchetype> each : map.values()) {
201       MavenArchetype eachArchetype = each.get(0);
202       DefaultMutableTreeNode node = new DefaultMutableTreeNode(eachArchetype, true);
203       for (MavenArchetype eachVersion : each) {
204         DefaultMutableTreeNode versionNode = new DefaultMutableTreeNode(eachVersion, false);
205         node.add(versionNode);
206       }
207       result.add(node);
208     }
209
210     return result;
211   }
212
213   public void requestUpdate() {
214
215     MavenArchetype selectedArch = getSelectedArchetype();
216     if (selectedArch == null) {
217       selectedArch = myBuilder.getArchetype();
218     }
219     if (selectedArch != null) myUseArchetypeCheckBox.setSelected(true);
220
221     if (myArchetypesTree.getRowCount() == 0) updateArchetypesList(selectedArch);
222   }
223
224   public void updateArchetypesList(final MavenArchetype selected) {
225     ApplicationManager.getApplication().assertIsDispatchThread();
226
227     myLoadingIcon.setBackground(myArchetypesTree.getBackground());
228
229     ((CardLayout)myArchetypesPanel.getLayout()).show(myArchetypesPanel, "loading");
230
231     final Object currentUpdaterMarker = new Object();
232     myCurrentUpdaterMarker = currentUpdaterMarker;
233
234     ApplicationManager.getApplication().executeOnPooledThread(new Runnable() {
235       public void run() {
236         final Set<MavenArchetype> archetypes = MavenIndicesManager.getInstance().getArchetypes();
237
238         //noinspection SSBasedInspection
239         SwingUtilities.invokeLater(new Runnable() {
240           public void run() {
241             if (currentUpdaterMarker != myCurrentUpdaterMarker) return; // Other updater has been run.
242
243             ((CardLayout)myArchetypesPanel.getLayout()).show(myArchetypesPanel, "archetypes");
244
245             TreeNode root = groupAndSortArchetypes(archetypes);
246             TreeModel model = new DefaultTreeModel(root);
247             myArchetypesTree.setModel(model);
248
249             if (selected != null) {
250               TreePath path = findNodePath(selected, model, model.getRoot());
251               if (path != null) {
252                 myArchetypesTree.expandPath(path.getParentPath());
253                 TreeUtil.selectPath(myArchetypesTree, path, true);
254               }
255             }
256
257             updateArchetypeDescription();
258           }
259         });
260       }
261     });
262   }
263
264   private void archetypeMayBeChanged() {
265     MavenArchetype selectedArchetype = getSelectedArchetype();
266     if (((myBuilder.getArchetype() == null) != (selectedArchetype == null))) {
267       myBuilder.setArchetype(selectedArchetype);
268       skipUpdateUI = true;
269       try {
270         if (myStep != null) {
271           myStep.fireStateChanged();
272         }
273       }
274       finally {
275         skipUpdateUI = false;
276       }
277     }
278   }
279
280
281   private void doAddArchetype() {
282     MavenAddArchetypeDialog dialog = new MavenAddArchetypeDialog(myMainPanel);
283     dialog.show();
284     if (!dialog.isOK()) return;
285
286     MavenArchetype archetype = dialog.getArchetype();
287     MavenIndicesManager.getInstance().addArchetype(archetype);
288     updateArchetypesList(archetype);
289   }
290
291   @Override
292   public void dispose() {
293   }
294
295   private static class MyRenderer extends ColoredTreeCellRenderer {
296     public void customizeCellRenderer(JTree tree,
297                                       Object value,
298                                       boolean selected,
299                                       boolean expanded,
300                                       boolean leaf,
301                                       int row,
302                                       boolean hasFocus) {
303       Object userObject = ((DefaultMutableTreeNode)value).getUserObject();
304       if (!(userObject instanceof MavenArchetype)) return;
305
306       MavenArchetype info = (MavenArchetype)userObject;
307
308       if (leaf) {
309         append(info.artifactId, SimpleTextAttributes.GRAY_ATTRIBUTES);
310         append(":" + info.version, SimpleTextAttributes.REGULAR_ATTRIBUTES);
311       }
312       else {
313         append(info.groupId + ":", SimpleTextAttributes.GRAY_ATTRIBUTES);
314         append(info.artifactId, SimpleTextAttributes.REGULAR_ATTRIBUTES);
315       }
316     }
317   }
318
319 }