package com.jetbrains.lang.dart.validation.fixes;
import com.intellij.codeInsight.template.Template;
import com.intellij.openapi.editor.Editor;
import com.intellij.openapi.project.Project;
import com.intellij.psi.PsiElement;
import com.intellij.psi.PsiFile;
import com.intellij.psi.util.PsiTreeUtil;
import com.jetbrains.lang.dart.DartBundle;
import com.jetbrains.lang.dart.psi.DartClass;
import com.jetbrains.lang.dart.psi.DartExecutionScope;
import com.jetbrains.lang.dart.psi.DartReference;
import com.jetbrains.lang.dart.util.DartResolveUtil;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
public class CreateDartGetterSetterAction extends CreateDartFunctionActionBase {
protected final boolean myStatic;
protected final boolean myGetter;
public CreateDartGetterSetterAction(@NotNull String name, boolean isGetter, boolean isStatic) {
super(name);
myGetter = isGetter;
myStatic = isStatic;
}
@NotNull
@Override
public String getName() {
if (myGetter) {
return myStatic ? DartBundle.message("dart.create.static.getter.fix.name", myFunctionName)
: DartBundle.message("dart.create.getter.fix.name", myFunctionName);
}
return myStatic ? DartBundle.message("dart.create.static.setter.fix.name", myFunctionName)
: DartBundle.message("dart.create.setter.fix.name", myFunctionName);
}
@Override
protected boolean buildTemplate(Template template, @NotNull PsiElement element) {
if (myStatic) {
template.addTextSegment("static ");
}
template.addTextSegment(myGetter ? "get " : "set ");
template.addTextSegment(myFunctionName);
if (myGetter) {
template.addTextSegment(" => ");
template.addEndVariable();
template.addTextSegment(";\n");
return true;
}
template.addTextSegment("(");
DartClass dartClass = DartResolveUtil.suggestType(element);
if (dartClass != null) {
template.addTextSegment(dartClass.getName());
template.addTextSegment(" ");
}
template.addTextSegment("value");
template.addTextSegment("){\n");
template.addEndVariable();
template.addTextSegment("\n}\n");
return true;
}
@Override
protected PsiElement getScopeBody(PsiElement element) {
final PsiElement classBody = DartResolveUtil.getBody(PsiTreeUtil.getParentOfType(element, DartClass.class));
return classBody != null ? classBody : PsiTreeUtil.getTopmostParentOfType(element, DartExecutionScope.class);
}
@Nullable
@Override
protected PsiElement findAnchor(PsiElement element) {
DartReference leftReference = DartResolveUtil.getLeftReference(PsiTreeUtil.getParentOfType(element, DartReference.class));
return leftReference == null ? super.findAnchor(element) : DartResolveUtil.getBody(leftReference.resolveDartClass().getDartClass());
}
}