1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
/*
* Copyright 2000-2014 JetBrains s.r.o.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.intellij.lang.properties.projectView;
import com.intellij.ide.projectView.TreeStructureProvider;
import com.intellij.ide.projectView.ViewSettings;
import com.intellij.ide.util.treeView.AbstractTreeNode;
import com.intellij.lang.properties.PropertiesImplUtil;
import com.intellij.lang.properties.ResourceBundle;
import com.intellij.lang.properties.psi.PropertiesFile;
import com.intellij.openapi.actionSystem.PlatformDataKeys;
import com.intellij.openapi.application.ApplicationManager;
import com.intellij.openapi.project.DumbAware;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.util.Computable;
import com.intellij.psi.PsiFile;
import com.intellij.util.SmartList;
import gnu.trove.THashMap;
import org.jetbrains.annotations.NotNull;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Map;
public class ResourceBundleGrouper implements TreeStructureProvider, DumbAware {
private final Project myProject;
public ResourceBundleGrouper(Project project) {
myProject = project;
}
@NotNull
public Collection<AbstractTreeNode> modify(@NotNull AbstractTreeNode parent, @NotNull final Collection<AbstractTreeNode> children, final ViewSettings settings) {
if (parent instanceof ResourceBundleNode) return children;
return ApplicationManager.getApplication().runReadAction(new Computable<Collection<AbstractTreeNode>>() {
@Override
public Collection<AbstractTreeNode> compute() {
Map<ResourceBundle,Collection<PropertiesFile>> childBundles = new THashMap<ResourceBundle, Collection<PropertiesFile>>();
for (AbstractTreeNode child : children) {
Object f = child.getValue();
if (f instanceof PsiFile) {
PropertiesFile propertiesFile = PropertiesImplUtil.getPropertiesFile((PsiFile)f);
if (propertiesFile != null) {
ResourceBundle bundle = propertiesFile.getResourceBundle();
Collection<PropertiesFile> files = childBundles.get(bundle);
if (files == null) {
files = new SmartList<PropertiesFile>();
childBundles.put(bundle, files);
}
files.add(propertiesFile);
}
}
}
List<AbstractTreeNode> result = new ArrayList<AbstractTreeNode>();
for (Map.Entry<ResourceBundle, Collection<PropertiesFile>> entry : childBundles.entrySet()) {
ResourceBundle resourceBundle = entry.getKey();
Collection<PropertiesFile> files = entry.getValue();
if (files.size() != 1) {
result.add(new ResourceBundleNode(myProject, resourceBundle, settings));
}
}
for (AbstractTreeNode child : children) {
Object f = child.getValue();
if (f instanceof PsiFile) {
PropertiesFile propertiesFile = PropertiesImplUtil.getPropertiesFile((PsiFile)f);
if (propertiesFile != null) {
ResourceBundle bundle = propertiesFile.getResourceBundle();
if (childBundles.get(bundle).size() != 1) {
continue;
}
}
}
result.add(child);
}
return result;
}
});
}
public Object getData(Collection<AbstractTreeNode> selected, String dataName) {
if (selected == null) return null;
for (AbstractTreeNode selectedElement : selected) {
Object element = selectedElement.getValue();
if (PlatformDataKeys.DELETE_ELEMENT_PROVIDER.is(dataName)) {
if (element instanceof ResourceBundle) {
return new ResourceBundleDeleteProvider((ResourceBundle)element);
}
}
}
if (ResourceBundle.ARRAY_DATA_KEY.is(dataName)) {
final List<ResourceBundle> selectedElements = new ArrayList<ResourceBundle>();
for (AbstractTreeNode node : selected) {
final Object value = node.getValue();
if (value instanceof ResourceBundle) {
selectedElements.add((ResourceBundle)value);
}
}
return selectedElements.isEmpty() ? null : selectedElements.toArray(new ResourceBundle[selectedElements.size()]);
}
return null;
}
}
|