package org.jetbrains.plugins.clojure.psi.resolve.completion;
import com.intellij.codeInsight.TailType;
import com.intellij.codeInsight.completion.InsertionContext;
import com.intellij.codeInsight.completion.JavaPsiClassReferenceElement;
import com.intellij.codeInsight.lookup.LookupElementPresentation;
import com.intellij.codeInsight.lookup.LookupItem;
import com.intellij.psi.*;
import com.intellij.psi.util.PsiTreeUtil;
import org.jetbrains.annotations.NonNls;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.plugins.clojure.psi.api.defs.ClDef;
import org.jetbrains.plugins.clojure.psi.api.symbols.ClSymbol;
import org.jetbrains.plugins.clojure.psi.util.ClojurePsiFactory;
/**
* @author Alefas
* @since 15.01.13
*/
public class ClojureLookupItem extends LookupItem<PsiElement> {
@NotNull
private final PsiElement myElement;
private boolean isInImport = false;
private boolean isClassName = false;
public boolean isInImport() {
return isInImport;
}
public void setInImport(boolean inImport) {
isInImport = inImport;
}
public boolean isClassName() {
return isClassName;
}
public void setClassName(boolean className) {
isClassName = className;
}
@NonNls @NotNull
private static String getLookupString(@NotNull PsiElement element) {
if (element instanceof PsiNamedElement) {
return ((PsiNamedElement) element).getName();
} else {
return element.getText();
}
}
public ClojureLookupItem(@NotNull PsiElement element) {
super(element, getLookupString(element));
myElement = element;
}
@Override
public void renderElement(LookupElementPresentation presentation) {
super.renderElement(presentation);
if (myElement instanceof ClDef) {
ClDef def = (ClDef) myElement;
presentation.setTailText(" " + def.getParameterString());
presentation.setTypeText(def.getContainingFile().getName());
} else if (myElement instanceof PsiClass) {
PsiClass clazz = (PsiClass) myElement;
String location = clazz.getPresentation().getLocationString();
JavaPsiClassReferenceElement.renderClassItem(presentation, this, clazz, false);
presentation.appendTailText(" " + location, true);
} //todo: What next?
}
@Override
public void handleInsert(InsertionContext context) {
if (isClassName()) {
PsiDocumentManager.getInstance(context.getProject()).commitDocument(context.getDocument());
final int startOffset = context.getStartOffset();
final ClSymbol symbol = PsiTreeUtil.findElementOfClassAtOffset(context.getFile(), startOffset, ClSymbol.class, false);
if (symbol == null) return;
if (myElement instanceof PsiClass) {
PsiClass clazz = (PsiClass) myElement;
if (symbol.getQualifierSymbol() != null) return;
if (isInImport()) {
ClSymbol newSymbol = ClojurePsiFactory.getInstance(context.getProject()).
createSymbolNodeFromText(clazz.getQualifiedName()).getPsi(ClSymbol.class);
symbol.replace(newSymbol);
} else {
symbol.bindToElement(clazz);
}
}
} else {
super.handleInsert(context);
}
}
}