Added "Delete" context menu action.

2012-09-05 20:20:20 +04:00
parent 11bba8dd77
commit 2a8687b4a5
2 changed files with 76 additions and 0 deletions
@@ -0,0 +1,73 @@
+/*
+ * Copyright 2010-2012 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 org.jetbrains.jet.plugin.codeInsight.ktSignature;
+
+import com.intellij.codeInsight.ExternalAnnotationsManager;
+import com.intellij.openapi.actionSystem.AnAction;
+import com.intellij.openapi.actionSystem.AnActionEvent;
+import com.intellij.openapi.application.Result;
+import com.intellij.openapi.command.WriteCommandAction;
+import com.intellij.openapi.project.Project;
+import com.intellij.psi.PsiAnnotation;
+import com.intellij.psi.PsiMethod;
+
+import static org.jetbrains.jet.plugin.codeInsight.ktSignature.KotlinSignatureUtil.KOTLIN_SIGNATURE_ANNOTATION;
+import static org.jetbrains.jet.plugin.codeInsight.ktSignature.KotlinSignatureUtil.findKotlinSignatureAnnotation;
+import static org.jetbrains.jet.plugin.codeInsight.ktSignature.KotlinSignatureUtil.refreshMarkers;
+
+/**
+ * @author Evgeny Gerashchenko
+ * @since 5 Sep 12
+ */
+public class DeleteSignatureAction extends AnAction {
+    private final PsiMethod annotationOwner;
+
+    public DeleteSignatureAction(PsiMethod annotationOwner) {
+        super("Delete");
+        this.annotationOwner = annotationOwner;
+    }
+
+    @Override
+    public void actionPerformed(AnActionEvent e) {
+        final Project project = e.getProject();
+        assert project != null;
+
+        final PsiAnnotation annotation = findKotlinSignatureAnnotation(annotationOwner);
+        assert annotation != null;
+
+        if (annotation.getContainingFile() != annotationOwner.getContainingFile()) {
+            // external annotation
+            new WriteCommandAction(project) {
+                @Override
+                protected void run(Result result) throws Throwable {
+                    ExternalAnnotationsManager.getInstance(project).deannotate(annotationOwner, KOTLIN_SIGNATURE_ANNOTATION);
+                }
+            }.execute();
+        }
+        else {
+            new WriteCommandAction(project) {
+                @Override
+                protected void run(Result result) throws Throwable {
+                    annotation.delete();
+                }
+            }.execute();
+        }
+
+
+        refreshMarkers(project);
+    }
+}
@@ -105,7 +105,10 @@ public class KotlinSignatureInJavaMarkerProvider implements LineMarkerProvider {
                @Override
                public ActionGroup getPopupMenuActions() {
                    DefaultActionGroup actionGroup = new DefaultActionGroup();
+
                    actionGroup.add(new EditSignatureAction(getElement()));
+                    actionGroup.add(new DeleteSignatureAction(getElement()));
+
                    return actionGroup;
                }
            };