dom/openapi/src/com/intellij/util/xml/actions/generate/DefaultGenerateElementProvider.java

   1 /*
   2  * Copyright 2000-2007 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.actions.generate;
  18
  19 import com.intellij.openapi.editor.Editor;
  20 import com.intellij.openapi.project.Project;
  21 import com.intellij.psi.PsiFile;
  22 import com.intellij.psi.xml.XmlElement;
  23 import com.intellij.util.ReflectionUtil;
  24 import com.intellij.util.xml.DomElement;
  25 import com.intellij.util.xml.ui.actions.generate.GenerateDomElementProvider;
  26 import com.intellij.util.xml.reflect.DomCollectionChildDescription;
  27 import org.jetbrains.annotations.Nullable;
  28
  29 import java.util.List;
  30
  31 /**
  32  * User: Sergey.Vasiliev
  33  */
  34 public abstract class DefaultGenerateElementProvider<T extends DomElement> extends GenerateDomElementProvider<T> {
  35   private Class<? extends DomElement> myChildElementClass;
  36
  37   public DefaultGenerateElementProvider(final String name, Class<T> childElementClass) {
  38     super(name);
  39
  40     myChildElementClass = childElementClass;
  41   }
  42
  43
  44   @Nullable
  45   public T generate(final Project project, final Editor editor, final PsiFile file) {
  46     return generate(getParentDomElement(project, editor, file), editor);
  47   }
  48
  49   @Nullable
  50   protected abstract DomElement getParentDomElement(final Project project, final Editor editor, final PsiFile file);
  51
  52   @Nullable
  53   public T generate(@Nullable final DomElement parent, final Editor editor) {
  54     if (parent != null) {
  55       final List<? extends DomCollectionChildDescription> list = parent.getGenericInfo().getCollectionChildrenDescriptions();
  56
  57       for (DomCollectionChildDescription childDescription : list) {
  58         if (ReflectionUtil.getRawType(childDescription.getType()).isAssignableFrom(myChildElementClass)) {
  59           int index  = getCollectionIndex(parent, childDescription, editor);
  60
  61           return index < 0 ? (T)childDescription.addValue(parent, myChildElementClass) : (T)childDescription.addValue(parent, myChildElementClass, index) ;
  62         }
  63       }
  64     }
  65     return null;
  66   }
  67
  68   private static int getCollectionIndex(final DomElement parent, final DomCollectionChildDescription childDescription, final Editor editor) {
  69     int offset = editor.getCaretModel().getOffset();
  70
  71     for (int i = 0; i < childDescription.getValues(parent).size(); i++) {
  72       DomElement element = childDescription.getValues(parent).get(i);
  73       XmlElement xmlElement = element.getXmlElement();
  74       if (xmlElement != null && xmlElement.getTextRange().getStartOffset() >= offset) {
  75           return i;
  76       }
  77     }
  78
  79     return -1;
  80   }
  81 }