/*
* Copyright 2000-2016 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 com.intellij.codeInsight.completion;
import com.intellij.codeInsight.lookup.LookupElementBuilder;
import com.intellij.lang.xhtml.XHTMLLanguage;
import com.intellij.openapi.util.text.StringUtil;
import com.intellij.openapi.vfs.CharsetToolkit;
import com.intellij.patterns.XmlPatterns;
import com.intellij.psi.PsiElement;
import com.intellij.psi.PsiFile;
import com.intellij.psi.PsiReference;
import com.intellij.psi.html.HtmlTag;
import com.intellij.psi.impl.source.html.dtd.HtmlElementDescriptorImpl;
import com.intellij.psi.util.PsiTreeUtil;
import com.intellij.psi.xml.XmlAttribute;
import com.intellij.psi.xml.XmlAttributeValue;
import com.intellij.psi.xml.XmlTag;
import com.intellij.util.ArrayUtil;
import com.intellij.util.ProcessingContext;
import com.intellij.xml.util.HtmlUtil;
import com.intellij.xml.util.XmlUtil;
import org.jetbrains.annotations.NonNls;
import org.jetbrains.annotations.NotNull;
import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.List;
import static com.intellij.html.impl.util.MicrodataUtil.*;
import static com.intellij.patterns.PlatformPatterns.psiElement;
public class HtmlCompletionContributor extends CompletionContributor {
public HtmlCompletionContributor() {
extend(CompletionType.BASIC, psiElement().inside(XmlPatterns.xmlAttributeValue()), new CompletionProvider<CompletionParameters>() {
@Override
protected void addCompletions(@NotNull CompletionParameters parameters,
ProcessingContext context,
@NotNull CompletionResultSet result) {
final PsiElement position = parameters.getPosition();
if (!hasHtmlAttributesCompletion(position)) {
return;
}
final XmlAttributeValue attributeValue = PsiTreeUtil.getParentOfType(position, XmlAttributeValue.class, false);
if (attributeValue != null && attributeValue.getParent() instanceof XmlAttribute) {
for (String element : addSpecificCompletions((XmlAttribute)attributeValue.getParent())) {
result.addElement(LookupElementBuilder.create(element));
}
}
}
});
}
public static boolean hasHtmlAttributesCompletion(PsiElement position) {
if (PsiTreeUtil.getParentOfType(position, HtmlTag.class, false) != null) {
return true;
}
XmlTag xmlTag = PsiTreeUtil.getParentOfType(position, XmlTag.class, false);
return xmlTag != null && xmlTag.getLanguage() == XHTMLLanguage.INSTANCE;
}
@NotNull
@NonNls
protected static String[] addSpecificCompletions(final XmlAttribute attribute) {
@NonNls String name = attribute.getName();
final XmlTag tag = attribute.getParent();
if (tag == null) return ArrayUtil.EMPTY_STRING_ARRAY;
@NonNls String tagName = tag.getName();
if (tag.getDescriptor() instanceof HtmlElementDescriptorImpl) {
name = name.toLowerCase();
tagName = tagName.toLowerCase();
}
final String namespace = tag.getNamespace();
if (XmlUtil.XHTML_URI.equals(namespace) || XmlUtil.HTML_URI.equals(namespace)) {
if ("target".equals(name)) {
return new String[]{"_blank", "_top", "_self", "_parent"};
}
else if ("enctype".equals(name)) {
return new String[]{"multipart/form-data", "application/x-www-form-urlencoded"};
}
else if ("rel".equals(name) || "rev".equals(name)) {
return new String[]{"alternate", "author", "bookmark", "help", "icon", "license", "next", "nofollow",
"noreferrer", "prefetch", "prev", "search", "stylesheet", "tag", "start", "contents", "index",
"glossary", "copyright", "chapter", "section", "subsection", "appendix", "script", "import",
"apple-touch-icon", "apple-touch-icon-precomposed", "apple-touch-startup-image"};
}
else if ("media".equals(name)) {
return new String[]{ "all", "braille", "embossed", "handheld", "print", "projection", "screen", "speech", "tty", "tv" };
}
else if ("language".equals(name)) {
return new String[]{"JavaScript", "VBScript", "JScript", "JavaScript1.2", "JavaScript1.3", "JavaScript1.4", "JavaScript1.5"};
}
else if ("type".equals(name) && "link".equals(tagName)) {
return new String[]{"text/css", "text/html", "text/plain", "text/xml"};
}
else if ("http-equiv".equals(name) && "meta".equals(tagName)) {
return HtmlUtil.RFC2616_HEADERS;
}
else if("content".equals(name) && "meta".equals(tagName) && tag.getAttribute("name") == null) {
return HtmlUtil.CONTENT_TYPES;
}
else if ("accept".equals(name) && "input".equals(tagName)) {
return HtmlUtil.CONTENT_TYPES;
}
else if("accept-charset".equals(name) || "charset".equals(name)) {
Charset[] charSets = CharsetToolkit.getAvailableCharsets();
String[] names = new String[charSets.length];
for (int i = 0; i < names.length; i++) {
names[i] = charSets[i].toString();
}
return names;
} else if ("itemprop".equals(name)) {
XmlTag scopeTag = findScopeTag(tag);
return scopeTag != null ? findItemProperties(scopeTag) : ArrayUtil.EMPTY_STRING_ARRAY;
}
}
return ArrayUtil.EMPTY_STRING_ARRAY;
}
private static String[] findItemProperties(@NotNull XmlTag tag) {
final XmlAttribute typeAttribute = tag.getAttribute(ITEM_TYPE);
if (typeAttribute != null) {
final XmlAttributeValue valueElement = typeAttribute.getValueElement();
final PsiReference[] references = valueElement != null ? valueElement.getReferences() : PsiReference.EMPTY_ARRAY;
List<String> result = new ArrayList<>();
for (PsiReference reference : references) {
final PsiElement target = reference != null ? reference.resolve() : null;
if (target instanceof PsiFile) {
result.addAll(extractProperties((PsiFile)target, StringUtil.unquoteString(reference.getCanonicalText())));
}
}
return ArrayUtil.toStringArray(result);
}
return ArrayUtil.EMPTY_STRING_ARRAY;
}
}