wrapping everything

This commit is contained in:
Dmitry Avdeev
2012-11-16 11:05:55 +04:00
parent 7b319d8604
commit 0beb6757eb
2 changed files with 9 additions and 1 deletions
@@ -846,6 +846,7 @@ public class SelectTemplateStep extends ModuleWizardStep implements SettingsStep
return myModuleName.getText().trim();
}
@TestOnly
public boolean setSelectedTemplate(String group, String name) {
final ComparableObject.Impl test = new ComparableObject.Impl(group, name);
return myTemplatesTree.select(myTreeBuilder, new SimpleNodeVisitor() {
@@ -21,6 +21,8 @@ import com.intellij.ide.util.treeView.PresentableNodeDescriptor;
import com.intellij.openapi.util.ActionCallback;
import com.intellij.ui.speedSearch.ElementFilter;
import com.intellij.ui.treeStructure.SimpleNode;
import com.intellij.util.Function;
import com.intellij.util.containers.ContainerUtil;
import org.jetbrains.annotations.NotNull;
import java.util.*;
@@ -237,7 +239,12 @@ public class FilteringTreeStructure extends AbstractTreeStructure {
public SimpleNode[] getChildren() {
List<FilteringNode> nodes = myNodesCache.get(this);
if (nodes == null) {
return myDelegate instanceof SimpleNode ? ((SimpleNode)myDelegate).getChildren() : NO_CHILDREN;
return myDelegate instanceof SimpleNode ? ContainerUtil.map(((SimpleNode)myDelegate).getChildren(), new Function<SimpleNode, SimpleNode>() {
@Override
public SimpleNode fun(SimpleNode node) {
return new FilteringNode(FilteringNode.this, node);
}
}, NO_CHILDREN) : NO_CHILDREN;
}
ArrayList<FilteringNode> result = new ArrayList<FilteringNode>();