/*
* Copyright 2000-2013 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.coldFusion.UI.editorActions;
import com.intellij.codeInsight.lookup.LookupElement;
import com.intellij.coldFusion.model.files.CfmlFile;
import com.intellij.coldFusion.model.info.CfmlFunctionDescription;
import com.intellij.coldFusion.model.info.CfmlLangInfo;
import com.intellij.coldFusion.model.lexer.CfscriptTokenTypes;
import com.intellij.coldFusion.model.psi.*;
import com.intellij.lang.ASTNode;
import com.intellij.lang.parameterInfo.*;
import com.intellij.psi.*;
import com.intellij.psi.util.PsiTreeUtil;
import com.intellij.util.ArrayUtil;
import com.intellij.util.Function;
import com.intellij.util.containers.ContainerUtil;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import java.util.ArrayList;
import java.util.List;
/**
* @author vnikolaenko
*/
public class CfmlParameterInfoHandler implements ParameterInfoHandler<PsiElement, CfmlFunctionDescription> {
// for test purposes
private String myText;
public boolean couldShowInLookup() {
return true;
}
public Object[] getParametersForLookup(LookupElement item, ParameterInfoContext context) {
final Object o = item.getObject();
if (o instanceof PsiElement) {
PsiElement element = (PsiElement)o;
List<PsiElement> methods = new ArrayList<PsiElement>();
if (element instanceof CfmlFunction || element instanceof PsiMethod) {
methods.add(element);
}
return ArrayUtil.toObjectArray(methods);
}
return ArrayUtil.EMPTY_OBJECT_ARRAY;
}
public Object[] getParametersForDocumentation(CfmlFunctionDescription p, ParameterInfoContext context) {
return ArrayUtil.toObjectArray(p.getParameters());
}
private static boolean isEmbraced(@Nullable PsiElement element, int offset) {
if (element == null) {
return false;
}
ASTNode node = element.getNode();
if (node == null) {
return false;
}
final ASTNode lbrace = node.findChildByType(CfscriptTokenTypes.L_BRACKET);
final ASTNode rbrace = node.findChildByType(CfscriptTokenTypes.R_BRACKET);
if (lbrace == null || rbrace == null) {
return false;
}
return lbrace.getStartOffset() < offset && rbrace.getStartOffset() >= offset;
}
@Nullable
private static PsiElement findAnchorElement(int offset, PsiFile file) {
if (offset <= 0) {
return null;
}
if (file instanceof CfmlFile) {
final CfmlArgumentList parametersList = PsiTreeUtil.findElementOfClassAtOffset(file, offset, CfmlArgumentList.class, false);
if (parametersList == null) {
return null;
}
if (isEmbraced(parametersList, offset)) {
final PsiElement element = parametersList.getParent();
return element instanceof CfmlFunctionCallExpression ? ((CfmlFunctionCallExpression)element).getReferenceExpression() : element;
}
}
return null;
}
public PsiElement findElementForParameterInfo(@NotNull CreateParameterInfoContext context) {
return findAnchorElement(context.getEditor().getCaretModel().getOffset(), context.getFile());
}
public PsiElement findElementForUpdatingParameterInfo(@NotNull UpdateParameterInfoContext context) {
return findAnchorElement(context.getEditor().getCaretModel().getOffset(), context.getFile());
}
public void showParameterInfo(@NotNull PsiElement element, @NotNull CreateParameterInfoContext context) {
ResolveResult[] variants = ResolveResult.EMPTY_ARRAY;
if (element instanceof PsiPolyVariantReference) {
variants = ((PsiPolyVariantReference)element).multiResolve(true);
if (variants.length != 0) {
context.setItemsToShow(
ContainerUtil.map2Array(variants, CfmlFunctionDescription.class, new Function<ResolveResult, CfmlFunctionDescription>() {
public CfmlFunctionDescription fun(ResolveResult resolveResult) {
final PsiElement element1 = resolveResult.getElement();
if (CfmlPsiUtil.isFunctionDefinition(element1)) {
CfmlFunction function = CfmlPsiUtil.getFunctionDefinition(element1);
if (function != null) {
return function.getFunctionInfo();
}
}
else if (element1 instanceof PsiMethod) {
PsiMethod function = (PsiMethod)element1;
CfmlFunctionDescription javaMethodDescr =
new CfmlFunctionDescription(function.getName(), function.getReturnType().getPresentableText());
final PsiParameter[] psiParameters = function.getParameterList().getParameters();
final int paramsNum = psiParameters.length;
for (int i = 0; i < paramsNum; i++) {
PsiParameter psiParameter = psiParameters[i];
javaMethodDescr.addParameter(new CfmlFunctionDescription.CfmlParameterDescription(psiParameter.getName(),
psiParameter.getType()
.getPresentableText(), true));
}
return javaMethodDescr;
}
return null;
}
}));
context.showHint(element, element.getTextRange().getStartOffset(), this);
return;
}
}
if (element instanceof CfmlReferenceExpression) {
String functionName = element.getText().toLowerCase();
if (ArrayUtil.find(CfmlLangInfo.getInstance(element.getProject()).getPredefinedFunctionsLowCase(), functionName) != -1) {
context.setItemsToShow(new Object[]{CfmlLangInfo.getInstance(element.getProject()).getFunctionParameters().get(functionName)});
context.showHint(element, element.getTextRange().getStartOffset(), this);
}
}
}
public void updateParameterInfo(@NotNull PsiElement place, @NotNull UpdateParameterInfoContext context) {
if (context.getParameterOwner() == null) {
context.setParameterOwner(place);
}
else if (!context.getParameterOwner().equals(place)) {
context.removeHint();
return;
}
final Object[] objects = context.getObjectsToView();
for (int i = 0; i < objects.length; i++) {
context.setUIComponentEnabled(i, true);
}
}
public String getParameterCloseChars() {
return ",){}";
}
public boolean tracksParameterIndex() {
return false;
}
public void updateUI(CfmlFunctionDescription p, @NotNull ParameterInfoUIContext context) {
if (p == null) {
context.setUIComponentEnabled(false);
return;
}
myText = p.getParametersListPresentableText();
context.setupUIComponentPresentation(
myText,
0,
0,
!context.isUIComponentEnabled(),
false,
false,
context.getDefaultParameterColor()
);
}
public String getText() {
return myText;
}
}