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

   1 package com.intellij.util.xml.tree;
   2
   3 import jetbrains.fabrique.ui.treeStructure.SimpleNode;
   4 import com.intellij.openapi.diagnostic.Logger;
   5 import com.intellij.ui.SimpleTextAttributes;
   6 import com.intellij.util.xml.DomElement;
   7 import com.intellij.util.xml.GenericValue;
   8
   9 import java.util.*;
  10 import java.lang.reflect.Method;
  11
  12
  13 public class BaseDomElementNode extends AbstractDomElementNode {
  14   static final private Logger LOG = Logger.getInstance(DomModelTreeStructure.class.getName());
  15
  16   protected DomElement myDomElement;
  17   protected String myTagName;
  18
  19   public BaseDomElementNode(final DomElement modelElement) {
  20     this(modelElement, modelElement.getTagName(), null);
  21   }
  22
  23   public BaseDomElementNode(final DomElement modelElement, final String tagName, SimpleNode parent) {
  24     super(parent);
  25
  26     myDomElement = modelElement;
  27     myTagName = tagName == null ? "unknown" : tagName;
  28    }
  29
  30   public SimpleNode[] getChildren() {
  31     return doGetChildren(myDomElement);
  32   }
  33
  34   protected SimpleNode[] doGetChildren(final DomElement domlElement) {
  35     if (!domlElement.isValid()) return NO_CHILDREN;
  36     List<SimpleNode> children = new ArrayList<SimpleNode>();
  37
  38     final Collection<Method> methods = domlElement.getMethodsInfo().getFixedChildrenGetterMethods();
  39
  40     for (Method method : methods) {
  41       try {
  42         final Object result = method.invoke(domlElement, new Object[0]);
  43
  44         if (result instanceof DomElement) {
  45           final String tagName = domlElement.getMethodsInfo().getTagName(method);
  46
  47           if (showGenericValues() && result instanceof GenericValue) {
  48              children.add(new GenericValueNode((GenericValue)result, tagName, this));
  49           } else {
  50
  51             children.add(getDomElementNode((DomElement)result, tagName, this));
  52           }
  53         }
  54       }
  55       catch (Throwable e) {
  56         LOG.error(e);
  57       }
  58     }
  59
  60     final Collection<Method> collectionMethods = domlElement.getMethodsInfo().getCollectionChildrenGetterMethods();
  61
  62     for (Method method : collectionMethods) {
  63       children.add(getDomElementsGroupNode(domlElement, method));
  64     }
  65
  66     AbstractDomElementNode[] childrenNodes = children.toArray(new AbstractDomElementNode[children.size()]);
  67     if (showOrdered()) {
  68       Arrays.sort(childrenNodes, getComparator());
  69     }
  70
  71     return childrenNodes;
  72   }
  73
  74   protected DomElementsGroupNode getDomElementsGroupNode(final DomElement domElement, final Method method) {
  75     return new DomElementsGroupNode(domElement, method);
  76   }
  77
  78   protected boolean showOrdered() {
  79     return true;
  80   }
  81
  82   protected boolean showGenericValues() {
  83     return true;
  84   }
  85
  86   public Object[] getEqualityObjects() {
  87     return NONE;
  88   }
  89
  90   protected boolean doUpdate() {
  91     setUniformIcon(getNodeIcon());
  92     clearColoredText();
  93     addColoredFragment(getNodeName(), SimpleTextAttributes.REGULAR_ATTRIBUTES);
  94
  95     return super.doUpdate();
  96   }
  97
  98   public String getNodeName() {
  99     return getPropertyName(myTagName);
 100   }
 101
 102   public String getTagName() {
 103     return myTagName;
 104   }
 105
 106   public DomElement getDomElement() {
 107     return myDomElement;
 108   }
 109
 110   protected Comparator<AbstractDomElementNode> getComparator() {
 111     return new Comparator<AbstractDomElementNode>() {
 112       public int compare(final AbstractDomElementNode node1, final AbstractDomElementNode node2) {
 113         return node1.getNodeName().toLowerCase().compareTo(node2.getNodeName().toLowerCase());
 114       }
 115     };
 116   }
 117
 118   public boolean isAutoExpandNode() {
 119     return getParent() == null;
 120   }
 121 }