MarkupStructureTreeModel.java

/*
 * 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;
        }
    }
}