source/com/intellij/util/xml/tree/DomModelTreeView.java

   1 package com.intellij.util.xml.tree;
   2
   3 import com.intellij.openapi.actionSystem.*;
   4 import com.intellij.openapi.actionSystem.ex.ActionManagerEx;
   5 import com.intellij.openapi.util.IconLoader;
   6 import com.intellij.ui.components.panels.Wrapper;
   7 import com.intellij.util.ui.tree.TreeUtil;
   8 import com.intellij.util.xml.DomElement;
   9 import com.intellij.util.xml.DomEventListener;
  10 import com.intellij.util.xml.DomManager;
  11 import com.intellij.util.xml.events.DomEvent;
  12 import jetbrains.fabrique.ui.treeStructure.*;
  13 import jetbrains.fabrique.ui.treeStructure.actions.CollapseAllAction;
  14 import jetbrains.fabrique.ui.treeStructure.actions.ExpandAllAction;
  15
  16 import javax.swing.*;
  17 import javax.swing.event.TreeExpansionListener;
  18 import javax.swing.event.TreeExpansionEvent;
  19 import javax.swing.tree.DefaultMutableTreeNode;
  20 import javax.swing.tree.DefaultTreeModel;
  21 import java.awt.*;
  22 import java.util.Map;
  23 import java.util.ArrayList;
  24
  25 import org.jetbrains.annotations.Nullable;
  26
  27 public class DomModelTreeView extends Wrapper implements DataProvider {
  28   public static String DOM_MODEL_TREE_VIEW_KEY = "DOM_MODEL_TREE_VIEW_KEY";
  29   public static String DOM_MODEL_TREE_VIEW_POPUP = "DOM_MODEL_TREE_VIEW_POPUP";
  30
  31   private final SimpleTree myTree;
  32   private final SimpleTreeBuilder myBuilder;
  33   private DomManager myDomManager;
  34   private DomEventListener myDomEventListener;
  35   private DomElement myRootElement;
  36
  37   public DomModelTreeView(DomElement rootElement) {
  38     this(rootElement, false);
  39   }
  40
  41   public DomModelTreeView(DomElement rootElement, boolean isRootVisible) {
  42     myRootElement = rootElement;
  43     myTree = new SimpleTree(new DefaultTreeModel(new DefaultMutableTreeNode()));
  44     myTree.setRootVisible(isRootVisible);
  45     myTree.setShowsRootHandles(true);
  46
  47     ToolTipManager.sharedInstance().registerComponent(myTree);
  48     TreeUtil.installActions(myTree);
  49
  50     final SimpleTreeStructure treeStructure = rootElement != null ? getTreeStructure(rootElement) : new SimpleTreeStructure() {
  51       public Object getRootElement() {
  52         return null;
  53       }
  54     };
  55
  56     myBuilder = new SimpleTreeBuilder(myTree, (DefaultTreeModel)myTree.getModel(), treeStructure, WeightBasedComparator.INSTANCE);
  57     myBuilder.setNodeDescriptorComparator(null);
  58     myBuilder.initRoot();
  59
  60     add(myTree, BorderLayout.CENTER);
  61
  62     myTree.addTreeExpansionListener(new TreeExpansionListener() {
  63       public void treeExpanded(TreeExpansionEvent event) {
  64         final SimpleNode simpleNode = myTree.getNodeFor(event.getPath());
  65
  66         if(simpleNode instanceof AbstractDomElementNode) {
  67           ((AbstractDomElementNode)simpleNode).setExpanded(true);
  68         }
  69       }
  70
  71       public void treeCollapsed(TreeExpansionEvent event) {
  72          final SimpleNode simpleNode = myTree.getNodeFor(event.getPath());
  73
  74         if(simpleNode instanceof AbstractDomElementNode) {
  75           ((AbstractDomElementNode)simpleNode).setExpanded(false);
  76           simpleNode.update();
  77         }
  78       }
  79     });
  80
  81     myDomEventListener = new DomEventListener() {
  82       public void eventOccured(DomEvent event) {
  83         myBuilder.updateFromRoot(false);
  84       }
  85     };
  86     myDomManager = rootElement.getManager();
  87     myDomManager.addDomEventListener(myDomEventListener);
  88
  89     myTree.setPopupGroup(getPopupActions(), DOM_MODEL_TREE_VIEW_POPUP);
  90   }
  91
  92   public DomElement getRootElement() {
  93     return myRootElement;
  94   }
  95
  96   protected SimpleTreeStructure getTreeStructure(final DomElement rootDomElement) {
  97     return new DomModelTreeStructure(rootDomElement);
  98   }
  99
 100   public SimpleTreeBuilder getBuilder() {
 101     return myBuilder;
 102   }
 103
 104   public void dispose() {
 105     myDomManager.removeDomEventListener(myDomEventListener);
 106   }
 107
 108   public SimpleTree getTree() {
 109     return myTree;
 110   }
 111
 112   protected ActionGroup getPopupActions() {
 113     DefaultActionGroup group = new DefaultActionGroup();
 114
 115     group.add(ActionManagerEx.getInstance().getAction("DomElementsTreeView.TreePopup"));
 116     group.addSeparator();
 117
 118     group.add(new ExpandAllAction(myTree));
 119     group.add(new CollapseAllAction(myTree));
 120
 121     return group;
 122   }
 123
 124   @Nullable
 125   public Object getData(String dataId) {
 126     if (DOM_MODEL_TREE_VIEW_KEY.equals(dataId)) {
 127       return this;
 128     }
 129     return null;
 130   }
 131
 132   public void setSelectedDomElement(final DomElement domElement) {
 133     if(domElement == null) return;
 134
 135     final java.util.List<SimpleNode> parentsNodes = new ArrayList<SimpleNode>();
 136
 137     SimpleNodeVisitor visitor = new SimpleNodeVisitor() {
 138       public boolean accept(SimpleNode simpleNode) {
 139         if(simpleNode instanceof BaseDomElementNode) {
 140           final DomElement nodeElement = ((AbstractDomElementNode)simpleNode).getDomElement();
 141           if(isParent(nodeElement, domElement)) {
 142              parentsNodes.add(simpleNode);
 143           }
 144         }
 145         return false;
 146       }
 147
 148       private boolean isParent(final DomElement potentialParent, final DomElement domElement) {
 149         DomElement currParent = domElement;
 150         while(currParent != null) {
 151           if(currParent.equals(potentialParent)) return true;
 152
 153           currParent = currParent.getParent();
 154         }
 155         return false;
 156       }
 157     };
 158
 159     getTree().accept(getBuilder(), visitor);
 160     if(parentsNodes.size() > 0) {
 161         getTree().setSelectedNode(getBuilder(), parentsNodes.get(parentsNodes.size()-1), true);
 162     }
 163   };
 164 }
 165