xml/impl/src/com/intellij/psi/filters/getters/XmlAttributeValueGetter.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 package com.intellij.psi.filters.getters;
  17
  18 import com.intellij.codeInsight.completion.CompletionContext;
  19 import com.intellij.psi.PsiElement;
  20 import com.intellij.psi.filters.ContextGetter;
  21 import com.intellij.psi.util.PsiTreeUtil;
  22 import com.intellij.psi.xml.XmlAttribute;
  23 import com.intellij.psi.xml.XmlAttributeValue;
  24 import com.intellij.psi.xml.XmlElement;
  25 import com.intellij.util.ArrayUtil;
  26 import com.intellij.xml.XmlAttributeDescriptor;
  27 import com.intellij.xml.impl.BasicXmlAttributeDescriptor;
  28 import org.jetbrains.annotations.Nullable;
  29
  30 /**
  31  * Created by IntelliJ IDEA.
  32  * User: ik
  33  * Date: 24.11.2003
  34  * Time: 14:17:59
  35  * To change this template use Options | File Templates.
  36  */
  37 public class XmlAttributeValueGetter implements ContextGetter {
  38   public XmlAttributeValueGetter() {}
  39
  40   public Object[] get(PsiElement context, CompletionContext completionContext) {
  41     return getApplicableAttributeVariants(context, completionContext);
  42   }
  43
  44   private Object[] getApplicableAttributeVariants(PsiElement _context, CompletionContext completionContext) {
  45     PsiElement context = _context;
  46     if(context != null) {
  47       context = PsiTreeUtil.getParentOfType(context, XmlAttribute.class);
  48       if (context == null) {
  49         context = PsiTreeUtil.getParentOfType(_context, XmlAttributeValue.class);
  50       }
  51     }
  52
  53     if(context instanceof XmlAttribute) {
  54       final XmlAttributeDescriptor descriptor = ((XmlAttribute)context).getDescriptor();
  55
  56       if(descriptor != null) {
  57         if (descriptor.isFixed()) {
  58           final String defaultValue = descriptor.getDefaultValue();
  59           return defaultValue == null ? ArrayUtil.EMPTY_OBJECT_ARRAY : new Object[] { defaultValue };
  60         }
  61
  62         String[] values = descriptor instanceof BasicXmlAttributeDescriptor ?
  63                           ((BasicXmlAttributeDescriptor)descriptor).getEnumeratedValues((XmlElement)context):descriptor.getEnumeratedValues();
  64
  65         final String[] strings = addSpecificCompletions(context);
  66
  67         if(values == null || values.length==0) {
  68           values = strings;
  69         } else if (strings != null) {
  70           values = ArrayUtil.mergeArrays(values, strings, String.class);
  71         }
  72
  73         return values == null ? ArrayUtil.EMPTY_OBJECT_ARRAY : values;
  74       }
  75     }
  76
  77     return ArrayUtil.EMPTY_OBJECT_ARRAY;
  78   }
  79
  80   @Nullable
  81   protected String[] addSpecificCompletions(final PsiElement context) {
  82     return null;
  83   }
  84
  85 }