xml/dom-impl/src/com/intellij/util/xml/impl/DomServiceImpl.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 package com.intellij.util.xml.impl;
  18
  19 import com.intellij.ide.structureView.StructureViewBuilder;
  20 import com.intellij.openapi.project.Project;
  21 import com.intellij.openapi.vfs.VirtualFile;
  22 import com.intellij.psi.PsiElement;
  23 import com.intellij.psi.PsiFile;
  24 import com.intellij.psi.PsiManager;
  25 import com.intellij.psi.search.GlobalSearchScope;
  26 import com.intellij.psi.xml.XmlFile;
  27 import com.intellij.psi.xml.XmlTag;
  28 import com.intellij.util.Function;
  29 import com.intellij.util.indexing.FileBasedIndex;
  30 import com.intellij.util.xml.*;
  31 import com.intellij.util.xml.structure.DomStructureViewBuilder;
  32 import org.jetbrains.annotations.NotNull;
  33 import org.jetbrains.annotations.Nullable;
  34
  35 import java.util.ArrayList;
  36 import java.util.Collection;
  37 import java.util.List;
  38
  39 /**
  40  * @author Gregory.Shrago
  41  */
  42 public class DomServiceImpl extends DomService {
  43
  44   public ModelMerger createModelMerger() {
  45     return new ModelMergerImpl();
  46   }
  47
  48   @Override
  49   public <T extends DomElement> DomAnchor<T> createAnchor(T domElement) {
  50     return DomAnchorImpl.createAnchor(domElement);
  51   }
  52
  53   @NotNull
  54   public XmlFile getContainingFile(@NotNull DomElement domElement) {
  55     if (domElement instanceof DomFileElement) {
  56       return ((DomFileElement)domElement).getFile();
  57     }
  58     DomInvocationHandler handler = DomManagerImpl.getDomInvocationHandler(domElement);
  59     assert handler != null : domElement;
  60     while (handler != null && !(handler instanceof DomRootInvocationHandler) && handler.getXmlTag() == null) {
  61       handler = handler.getParentHandler();
  62     }
  63     if (handler instanceof DomRootInvocationHandler) {
  64       return ((DomRootInvocationHandler)handler).getParent().getFile();
  65     }
  66     assert handler != null;
  67     XmlTag tag = handler.getXmlTag();
  68     assert tag != null;
  69     while (true) {
  70       final PsiElement parentTag = PhysicalDomParentStrategy.getParentTagCandidate(tag);
  71       if (!(parentTag instanceof XmlTag)) {
  72         return (XmlFile)tag.getContainingFile();
  73       }
  74
  75       tag = (XmlTag)parentTag;
  76     }
  77   }
  78
  79   @NotNull
  80   public EvaluatedXmlName getEvaluatedXmlName(@NotNull final DomElement element) {
  81     return DomManagerImpl.getDomInvocationHandler(element).getXmlName();
  82   }
  83
  84   public Collection<VirtualFile> getDomFileCandidates(Class<? extends DomElement> description, Project project) {
  85     return FileBasedIndex.getInstance().getContainingFiles(DomFileIndex.NAME, description.getName(), GlobalSearchScope.allScope(project));
  86   }
  87
  88   public <T extends DomElement> List<DomFileElement<T>> getFileElements(final Class<T> clazz, final Project project, @Nullable final GlobalSearchScope scope) {
  89     final Collection<VirtualFile> list = scope == null ? getDomFileCandidates(clazz, project) : getDomFileCandidates(clazz, project, scope);
  90     final ArrayList<DomFileElement<T>> result = new ArrayList<DomFileElement<T>>(list.size());
  91     for (VirtualFile file : list) {
  92       final PsiFile psiFile = PsiManager.getInstance(project).findFile(file);
  93       if (psiFile instanceof XmlFile) {
  94         final DomFileElement<T> element = DomManager.getDomManager(project).getFileElement((XmlFile)psiFile, clazz);
  95         if (element != null) {
  96           result.add(element);
  97         }
  98       }
  99     }
 100
 101     return result;
 102   }
 103
 104
 105   public StructureViewBuilder createSimpleStructureViewBuilder(final XmlFile file, final Function<DomElement, StructureViewMode> modeProvider) {
 106     return new DomStructureViewBuilder(file, modeProvider);
 107   }
 108
 109 }