/* * Copyright 2010 The WicketForge-Team * * 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 wicketforge.psi.hierarchy; import com.intellij.ide.structureView.StructureViewTreeElement; import com.intellij.ide.structureView.TextEditorBasedStructureViewModel; import com.intellij.ide.util.treeView.smartTree.TreeElement; import com.intellij.navigation.ItemPresentation; import com.intellij.psi.PsiElement; import com.intellij.psi.xml.XmlFile; import com.intellij.psi.xml.XmlTag; import com.intellij.util.PsiNavigateUtil; import org.jetbrains.annotations.NotNull; import java.util.List; /** */ public class MarkupStructureTreeModel extends TextEditorBasedStructureViewModel { private StructureViewTreeElement root; private MarkupWicketIdHierarchy hierarchy; public MarkupStructureTreeModel(@NotNull XmlFile xmlFile) { super(xmlFile); hierarchy = MarkupWicketIdHierarchy.create(xmlFile); root = new MarkupTreeElement(hierarchy.getRoot()); } @Override @NotNull public StructureViewTreeElement getRoot() { return root; } @Override protected boolean isSuitable(PsiElement element) { if (element instanceof XmlTag) { for (MarkupWicketIdItem markupWicketIdItem : hierarchy.getWicketIdPathMap().values()) { if (element.equals(markupWicketIdItem.getTag())) { return true; } } } return super.isSuitable(element); } private static class MarkupTreeElement implements StructureViewTreeElement { private MarkupWicketIdItem markupWicketIdItem; private TreeElement[] children; private MarkupTreeElement(MarkupWicketIdItem markupWicketIdItem) { this.markupWicketIdItem = markupWicketIdItem; } @Override public Object getValue() { return markupWicketIdItem.getTag(); } @Override public void navigate(boolean requestFocus) { PsiNavigateUtil.navigate(markupWicketIdItem.getAttributeValue()); } @Override public boolean canNavigate() { return true; } @Override public boolean canNavigateToSource() { return true; } @NotNull @Override public ItemPresentation getPresentation() { return markupWicketIdItem; } @NotNull @Override public TreeElement[] getChildren() { if (children == null) { List<MarkupWicketIdItem> markupWicketIdItemChildren = markupWicketIdItem.getChildren(); children = new TreeElement[markupWicketIdItemChildren.size()]; for (int i = 0, n = markupWicketIdItemChildren.size(); i < n; i++) { children[i] = new MarkupTreeElement(markupWicketIdItemChildren.get(i)); } } return children; } } }