xml/impl/src/com/intellij/lang/xml/XmlStructureViewBuilderFactory.java

   1 /*
   2  * Copyright 2000-2009 JetBrains s.r.o.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  * http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 /*
  18  * @author max
  19  */
  20 package com.intellij.lang.xml;
  21
  22 import com.intellij.ide.structureView.StructureViewBuilder;
  23 import com.intellij.ide.structureView.StructureViewModel;
  24 import com.intellij.ide.structureView.TreeBasedStructureViewBuilder;
  25 import com.intellij.ide.structureView.impl.xml.XmlStructureViewTreeModel;
  26 import com.intellij.ide.structureView.xml.XmlStructureViewBuilderProvider;
  27 import com.intellij.lang.Language;
  28 import com.intellij.lang.LanguageStructureViewBuilder;
  29 import com.intellij.lang.PsiStructureViewFactory;
  30 import com.intellij.openapi.extensions.Extensions;
  31 import com.intellij.psi.PsiFile;
  32 import com.intellij.psi.xml.XmlFile;
  33 import org.jetbrains.annotations.NotNull;
  34 import org.jetbrains.annotations.Nullable;
  35
  36 public class XmlStructureViewBuilderFactory implements PsiStructureViewFactory {
  37   @Nullable
  38   public StructureViewBuilder getStructureViewBuilder(final PsiFile psiFile) {
  39     if (psiFile instanceof XmlFile) {
  40       StructureViewBuilder builder = getStructureViewBuilderForExtensions(psiFile);
  41       if (builder != null) {
  42         return builder;
  43       }
  44
  45       for (XmlStructureViewBuilderProvider xmlStructureViewBuilderProvider : getStructureViewBuilderProviders()) {
  46         final StructureViewBuilder structureViewBuilder = xmlStructureViewBuilderProvider.createStructureViewBuilder((XmlFile)psiFile);
  47         if (structureViewBuilder != null) {
  48           return structureViewBuilder;
  49         }
  50       }
  51
  52       return new TreeBasedStructureViewBuilder() {
  53         @NotNull
  54         public StructureViewModel createStructureViewModel() {
  55           return new XmlStructureViewTreeModel((XmlFile)psiFile);
  56         }
  57       };
  58     }
  59     else {
  60       return null;
  61     }
  62   }
  63
  64   private static XmlStructureViewBuilderProvider[] getStructureViewBuilderProviders() {
  65     return (XmlStructureViewBuilderProvider[])Extensions.getExtensions(XmlStructureViewBuilderProvider.EXTENSION_POINT_NAME);
  66   }
  67
  68   private static StructureViewBuilder getStructureViewBuilderForExtensions(final PsiFile psiFile) {
  69     for (Language language : ((XMLLanguage)XMLLanguage.INSTANCE).getLanguageExtensionsForFile(psiFile)) {
  70       final StructureViewBuilder builder = LanguageStructureViewBuilder.INSTANCE.forLanguage(language).getStructureViewBuilder(psiFile);
  71       if (builder != null) {
  72         return builder;
  73       }
  74     }
  75     return null;
  76   }
  77 }