| /* |
| * Copyright 2000-2010 JetBrains s.r.o. |
| * |
| * 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 org.jetbrains.android; |
| |
| import com.android.SdkConstants; |
| import com.intellij.codeInsight.completion.*; |
| import com.intellij.codeInsight.lookup.LookupElement; |
| import com.intellij.codeInsight.lookup.LookupElementBuilder; |
| import com.intellij.codeInsight.lookup.LookupElementDecorator; |
| import com.intellij.lang.ASTNode; |
| import com.intellij.openapi.util.TextRange; |
| import com.intellij.psi.PsiClass; |
| import com.intellij.psi.PsiElement; |
| import com.intellij.psi.PsiFile; |
| import com.intellij.psi.PsiReference; |
| import com.intellij.psi.xml.*; |
| import com.intellij.util.Consumer; |
| import com.intellij.util.containers.HashSet; |
| import com.intellij.util.xml.Converter; |
| import com.intellij.util.xml.DomElement; |
| import com.intellij.util.xml.DomManager; |
| import com.intellij.util.xml.GenericAttributeValue; |
| import com.intellij.util.xml.converters.DelimitedListConverter; |
| import org.jetbrains.android.dom.AndroidDomElementDescriptorProvider; |
| import org.jetbrains.android.dom.animation.AndroidAnimationUtils; |
| import org.jetbrains.android.dom.animation.AnimationDomFileDescription; |
| import org.jetbrains.android.dom.animator.AndroidAnimatorUtil; |
| import org.jetbrains.android.dom.animator.AnimatorDomFileDescription; |
| import org.jetbrains.android.dom.color.ColorDomFileDescription; |
| import org.jetbrains.android.dom.converters.FlagConverter; |
| import org.jetbrains.android.dom.drawable.AndroidDrawableDomUtil; |
| import org.jetbrains.android.dom.drawable.DrawableStateListDomFileDescription; |
| import org.jetbrains.android.dom.layout.AndroidLayoutUtil; |
| import org.jetbrains.android.dom.layout.LayoutDomFileDescription; |
| import org.jetbrains.android.dom.layout.LayoutElement; |
| import org.jetbrains.android.dom.manifest.ManifestDomFileDescription; |
| import org.jetbrains.android.dom.transition.TransitionDomFileDescription; |
| import org.jetbrains.android.dom.transition.TransitionDomUtil; |
| import org.jetbrains.android.dom.xml.AndroidXmlResourcesUtil; |
| import org.jetbrains.android.dom.xml.PreferenceElement; |
| import org.jetbrains.android.dom.xml.XmlResourceDomFileDescription; |
| import org.jetbrains.android.facet.AndroidFacet; |
| import org.jetbrains.android.facet.SimpleClassMapConstructor; |
| import org.jetbrains.android.util.AndroidUtils; |
| import org.jetbrains.annotations.NotNull; |
| |
| import javax.swing.*; |
| import java.util.*; |
| |
| /** |
| * @author coyote |
| */ |
| public class AndroidCompletionContributor extends CompletionContributor { |
| |
| private static void addAll(Collection<String> collection, CompletionResultSet set) { |
| for (String s : collection) { |
| set.addElement(LookupElementBuilder.create(s)); |
| } |
| } |
| |
| private static boolean completeTagNames(@NotNull AndroidFacet facet, @NotNull XmlFile xmlFile, @NotNull CompletionResultSet resultSet) { |
| if (ManifestDomFileDescription.isManifestFile(xmlFile, facet)) { |
| resultSet.addElement(LookupElementBuilder.create("manifest")); |
| return false; |
| } |
| else if (LayoutDomFileDescription.isLayoutFile(xmlFile)) { |
| final Map<String,PsiClass> classMap = facet.getClassMap( |
| AndroidUtils.VIEW_CLASS_NAME, SimpleClassMapConstructor.getInstance()); |
| |
| for (String rootTag : AndroidLayoutUtil.getPossibleRoots(facet)) { |
| final PsiClass aClass = classMap.get(rootTag); |
| LookupElementBuilder builder = aClass != null |
| ? LookupElementBuilder.create(aClass, rootTag) |
| : LookupElementBuilder.create(rootTag); |
| final Icon icon = AndroidDomElementDescriptorProvider.getIconForViewTag(rootTag); |
| |
| if (icon != null) { |
| builder = builder.withIcon(icon); |
| } |
| resultSet.addElement(builder); |
| } |
| return false; |
| } |
| else if (AnimationDomFileDescription.isAnimationFile(xmlFile)) { |
| addAll(AndroidAnimationUtils.getPossibleChildren(facet), resultSet); |
| return false; |
| } |
| else if (AnimatorDomFileDescription.isAnimatorFile(xmlFile)) { |
| addAll(AndroidAnimatorUtil.getPossibleChildren(), resultSet); |
| return false; |
| } |
| else if (XmlResourceDomFileDescription.isXmlResourceFile(xmlFile)) { |
| addAll(AndroidXmlResourcesUtil.getPossibleRoots(facet), resultSet); |
| return false; |
| } |
| else if (AndroidDrawableDomUtil.isDrawableResourceFile(xmlFile)) { |
| addAll(AndroidDrawableDomUtil.getPossibleRoots(facet), resultSet); |
| return false; |
| } |
| else if (TransitionDomFileDescription.isTransitionFile(xmlFile)) { |
| addAll(TransitionDomUtil.getPossibleRoots(), resultSet); |
| return false; |
| } |
| else if (ColorDomFileDescription.isColorResourceFile(xmlFile)) { |
| addAll(Arrays.asList(DrawableStateListDomFileDescription.SELECTOR_TAG_NAME), resultSet); |
| return false; |
| } |
| return true; |
| } |
| |
| @Override |
| public void fillCompletionVariants(@NotNull CompletionParameters parameters, @NotNull CompletionResultSet resultSet) { |
| PsiElement position = parameters.getPosition(); |
| PsiElement originalPosition = parameters.getOriginalPosition(); |
| AndroidFacet facet = AndroidFacet.getInstance(position); |
| |
| if (facet == null) { |
| return; |
| } |
| PsiElement parent = position.getParent(); |
| PsiElement originalParent = originalPosition != null ? originalPosition.getParent() : null; |
| |
| if (parent instanceof XmlTag) { |
| XmlTag tag = (XmlTag)parent; |
| |
| if (tag.getParentTag() != null) { |
| return; |
| } |
| final ASTNode startTagName = XmlChildRole.START_TAG_NAME_FINDER.findChild(tag.getNode()); |
| |
| if (startTagName == null || startTagName.getPsi() != position) { |
| return; |
| } |
| final PsiFile file = tag.getContainingFile(); |
| if (!(file instanceof XmlFile)) { |
| return; |
| } |
| final PsiReference reference = file.findReferenceAt(parameters.getOffset()); |
| if (reference != null) { |
| final PsiElement element = reference.getElement(); |
| if (element != null) { |
| final int refOffset = element.getTextRange().getStartOffset() + |
| reference.getRangeInElement().getStartOffset(); |
| if (refOffset != position.getTextRange().getStartOffset()) { |
| // do not provide completion if we're inside some reference starting in the middle of tag name |
| return; |
| } |
| } |
| } |
| |
| if (!completeTagNames(facet, (XmlFile)file, resultSet)) { |
| resultSet.stopHere(); |
| } |
| } |
| else if (parent instanceof XmlAttribute) { |
| final ASTNode attrName = XmlChildRole.ATTRIBUTE_NAME_FINDER.findChild(parent.getNode()); |
| |
| if (attrName == null || |
| attrName.getPsi() != position) { |
| return; |
| } |
| addAndroidPrefixElement(position, parent, resultSet); |
| moveLayoutAttributeUp(parameters, (XmlAttribute)parent, resultSet); |
| } |
| else if (originalParent instanceof XmlAttributeValue) { |
| completeTailsInFlagAttribute(parameters, resultSet, (XmlAttributeValue)originalParent); |
| } |
| } |
| |
| private static void addAndroidPrefixElement(PsiElement position, PsiElement parent, CompletionResultSet resultSet) { |
| if (position.getText().startsWith("android:")) { |
| return; |
| } |
| |
| final PsiElement gp = parent.getParent(); |
| if (!(gp instanceof XmlTag)) { |
| return; |
| } |
| |
| final DomElement element = DomManager.getDomManager(gp.getProject()).getDomElement((XmlTag)gp); |
| if (!(element instanceof LayoutElement) && |
| !(element instanceof PreferenceElement)) { |
| return; |
| } |
| |
| final String prefix = ((XmlTag)gp).getPrefixByNamespace(SdkConstants.NS_RESOURCES); |
| if (prefix == null || prefix.length() < 3) { |
| return; |
| } |
| final LookupElementBuilder e = LookupElementBuilder.create(prefix + ":").withTypeText("[Namespace Prefix]", true); |
| resultSet.addElement(PrioritizedLookupElement.withPriority(e, Double.MAX_VALUE)); |
| } |
| |
| private static void moveLayoutAttributeUp(CompletionParameters parameters, |
| XmlAttribute attribute, |
| final CompletionResultSet resultSet) { |
| final PsiElement gp = attribute.getParent(); |
| |
| if (gp == null) { |
| return; |
| } |
| final XmlTag tag = (XmlTag)gp; |
| final DomElement element = DomManager.getDomManager(gp.getProject()).getDomElement(tag); |
| |
| if (!(element instanceof LayoutElement)) { |
| return; |
| } |
| final boolean localNameCompletion; |
| |
| if (attribute.getName().contains(":")) { |
| final String nsPrefix = attribute.getNamespacePrefix(); |
| |
| if (nsPrefix.length() == 0) { |
| return; |
| } |
| if (!SdkConstants.NS_RESOURCES.equals(tag.getNamespaceByPrefix(nsPrefix))) { |
| return; |
| } |
| else { |
| localNameCompletion = true; |
| } |
| } |
| else { |
| localNameCompletion = false; |
| } |
| final Map<String, String> prefix2ns = new HashMap<String, String>(); |
| |
| resultSet.runRemainingContributors(parameters, new Consumer<CompletionResult>() { |
| @Override |
| public void consume(CompletionResult result) { |
| LookupElement lookupElement = result.getLookupElement(); |
| final Object obj = lookupElement.getObject(); |
| |
| if (obj instanceof String) { |
| final String s = (String)obj; |
| final int idx = s.indexOf(':'); |
| |
| if (idx > 0) { |
| final String prefix = s.substring(0, idx); |
| String ns = prefix2ns.get(prefix); |
| |
| if (ns == null) { |
| ns = tag.getNamespaceByPrefix(prefix); |
| prefix2ns.put(prefix, ns); |
| } |
| if (SdkConstants.NS_RESOURCES.equals(ns)) { |
| result = customizeLayoutAttributeLookupElement(s.substring(idx + 1), lookupElement, result); |
| } |
| } |
| else if (localNameCompletion) { |
| result = customizeLayoutAttributeLookupElement(s.substring(idx + 1), lookupElement, result); |
| } |
| } |
| resultSet.passResult(result); |
| } |
| }); |
| } |
| |
| private static CompletionResult customizeLayoutAttributeLookupElement(String localName, |
| LookupElement lookupElement, |
| CompletionResult result) { |
| final String layoutPrefix = "layout_"; |
| |
| if (!localName.startsWith(layoutPrefix)) { |
| return result; |
| } |
| final String localSuffix = localName.substring(layoutPrefix.length()); |
| |
| if (localSuffix.length() > 0) { |
| final HashSet<String> lookupStrings = new HashSet<String>(lookupElement.getAllLookupStrings()); |
| lookupStrings.add(localSuffix); |
| |
| lookupElement = new LookupElementDecorator<LookupElement>(lookupElement) { |
| @Override |
| public Set<String> getAllLookupStrings() { |
| return lookupStrings; |
| } |
| }; |
| } |
| return result.withLookupElement(PrioritizedLookupElement.withPriority(lookupElement, 100.0)); |
| } |
| |
| private static void completeTailsInFlagAttribute(CompletionParameters parameters, |
| CompletionResultSet resultSet, |
| XmlAttributeValue parent) { |
| final String currentValue = parent.getValue(); |
| |
| if (currentValue == null || currentValue.length() == 0 || currentValue.endsWith("|")) { |
| return; |
| } |
| final PsiElement gp = parent.getParent(); |
| |
| if (!(gp instanceof XmlAttribute)) { |
| return; |
| } |
| final GenericAttributeValue domValue = DomManager.getDomManager(gp.getProject()).getDomElement((XmlAttribute)gp); |
| final Converter converter = domValue != null ? domValue.getConverter() : null; |
| |
| if (!(converter instanceof FlagConverter)) { |
| return; |
| } |
| final TextRange valueRange = parent.getValueTextRange(); |
| |
| if (valueRange != null && valueRange.getEndOffset() == parameters.getOffset()) { |
| final Set<String> valueSet = ((FlagConverter)converter).getValues(); |
| |
| if (valueSet.size() > 0) { |
| final String prefix = resultSet.getPrefixMatcher().getPrefix(); |
| |
| if (valueSet.contains(prefix)) { |
| final ArrayList<String> filteredValues = new ArrayList<String>(valueSet); |
| //noinspection unchecked |
| DelimitedListConverter.filterVariants(filteredValues, domValue); |
| |
| for (String variant : filteredValues) { |
| resultSet.addElement(LookupElementBuilder.create(prefix + "|" + variant)); |
| } |
| } |
| } |
| } |
| } |
| } |