@@ -26,12 +26,12 @@ public class UpdateServiceStackReference extends AnAction {
2626
2727 @Override
2828 public void actionPerformed(AnActionEvent anActionEvent) {
29- final PsiJavaFile psiJavaFile = getPsiFile(anActionEvent);
30- if(UpdateServiceStackUtils.containsOptionsHeader(psiJavaFile )) {
29+ final PsiFile psiFile = getPsiFile(anActionEvent);
30+ if(UpdateServiceStackUtils.containsOptionsHeader(psiFile )) {
3131 ApplicationManager.getApplication().runWriteAction(new Runnable() {
3232 @Override
3333 public void run() {
34- UpdateServiceStackUtils.updateServiceStackReference(psiJavaFile );
34+ UpdateServiceStackUtils.updateServiceStackReference(psiFile );
3535 }
3636 });
3737 }
@@ -40,13 +40,19 @@ public void run() {
4040 @Override
4141 public void update(AnActionEvent e) {
4242 Module module = getModule(e);
43- PsiJavaFile psiJavaFile = getPsiFile(e);
44- if (psiJavaFile == null || !isAndroidProject(module)) {
43+ PsiFile psiFile = getPsiFile(e);
44+ if (psiFile == null || !isAndroidProject(module)) {
4545 e.getPresentation().setVisible(false);
4646 return;
4747 }
4848
49- if(!UpdateServiceStackUtils.containsOptionsHeader(psiJavaFile)) {
49+ if(!psiFile.getFileType().getDefaultExtension().equals("java") &&
50+ !psiFile.getFileType().getDefaultExtension().equals("kt")) {
51+ e.getPresentation().setVisible(false);
52+ return;
53+ }
54+
55+ if(!UpdateServiceStackUtils.containsOptionsHeader(psiFile)) {
5056 e.getPresentation().setVisible(false);
5157 return;
5258 }
@@ -74,7 +80,7 @@ private static boolean isAndroidProject(@NotNull Module module) {
7480 return false;
7581 }
7682
77- private static PsiJavaFile getPsiFile(AnActionEvent e) {
83+ private static PsiFile getPsiFile(AnActionEvent e) {
7884
7985 Module module = getModule(e);
8086 if(module == null) {
@@ -95,14 +101,7 @@ private static PsiJavaFile getPsiFile(AnActionEvent e) {
95101 return null;
96102 }
97103
98- if(!isJavaFile(psiFile)) {
99- return null;
100- }
101- return (PsiJavaFile)psiFile;
102- }
103-
104- private static boolean isJavaFile(@Nullable PsiFile psiFile) {
105- return psiFile != null && psiFile instanceof PsiJavaFile;
104+ return psiFile;
106105 }
107106
108107 static Module getModule(AnActionEvent e) {
0 commit comments