package com.jetbrains.lang.dart.ide.generation;
import com.intellij.openapi.actionSystem.AnAction;
import com.intellij.openapi.actionSystem.AnActionEvent;
import com.intellij.openapi.actionSystem.CommonDataKeys;
import com.intellij.openapi.editor.Editor;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.util.Pair;
import com.intellij.psi.PsiFile;
import com.intellij.psi.util.PsiTreeUtil;
import com.jetbrains.lang.dart.psi.DartClass;
/**
* @author: Fedor.Korotkov
*/
public abstract class BaseDartGenerateAction extends AnAction {
public void actionPerformed(final AnActionEvent e) {
final Project project = e.getData(CommonDataKeys.PROJECT);
assert project != null;
final Pair<Editor, PsiFile> editorAndPsiFile = getEditorAndPsiFile(e);
getGenerateHandler().invoke(project, editorAndPsiFile.first, editorAndPsiFile.second);
}
private static Pair<Editor, PsiFile> getEditorAndPsiFile(final AnActionEvent e) {
final Project project = e.getData(CommonDataKeys.PROJECT);
if (project == null) return Pair.create((Editor)null, null);
Editor editor = e.getData(CommonDataKeys.EDITOR);
PsiFile psiFile = e.getData(CommonDataKeys.PSI_FILE);
return Pair.create(editor, psiFile);
}
protected abstract BaseDartGenerateHandler getGenerateHandler();
@Override
public void update(final AnActionEvent e) {
final Pair<Editor, PsiFile> editorAndPsiFile = getEditorAndPsiFile(e);
final Editor editor = editorAndPsiFile.first;
final PsiFile psiFile = editorAndPsiFile.second;
final int caretOffset = editor == null ? -1 : editor.getCaretModel().getOffset();
final boolean inClass = psiFile != null && PsiTreeUtil.getParentOfType(psiFile.findElementAt(caretOffset), DartClass.class) != null;
e.getPresentation().setEnabled(inClass);
e.getPresentation().setVisible(inClass);
}
}