Added line marker for Java methods which have alternative Kotlin signature.

2012-08-03 23:25:43 +04:00
parent 6a77ffbfe2
commit 3b865f0121
2 changed files with 86 additions and 0 deletions
@@ -172,6 +172,7 @@
      <configurationProducer implementation="org.jetbrains.jet.plugin.run.JetRunConfigurationProducer"/>
      <configurationProducer implementation="org.jetbrains.jet.plugin.run.JetJUnitConfigurationProducer"/>
      <codeInsight.lineMarkerProvider language="jet" implementationClass="org.jetbrains.jet.plugin.highlighter.JetLineMarkerProvider"/>
+      <codeInsight.lineMarkerProvider language="JAVA" implementationClass="org.jetbrains.jet.plugin.codeInsight.KotlinSignatureInJavaMarkerProvider"/>
      <iconProvider implementation="org.jetbrains.jet.plugin.JetIconProvider"/>
      <itemPresentationProvider implementationClass="org.jetbrains.jet.plugin.presentation.JetFunctionPresenter" forClass="org.jetbrains.jet.lang.psi.JetNamedFunction"/>
      <itemPresentationProvider implementationClass="org.jetbrains.jet.plugin.presentation.JetClassPresenter" forClass="org.jetbrains.jet.lang.psi.JetClass"/>
@@ -0,0 +1,85 @@
+/*
+ * 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;
+
+import com.intellij.codeHighlighting.Pass;
+import com.intellij.codeInsight.daemon.GutterIconNavigationHandler;
+import com.intellij.codeInsight.daemon.LineMarkerInfo;
+import com.intellij.codeInsight.daemon.LineMarkerProvider;
+import com.intellij.psi.*;
+import com.intellij.util.Function;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+import org.jetbrains.jet.lang.resolve.java.JavaDescriptorResolver;
+import org.jetbrains.jet.lang.resolve.java.JvmStdlibNames;
+import org.jetbrains.jet.plugin.JetIcons;
+
+import java.awt.event.MouseEvent;
+import java.util.Collection;
+import java.util.List;
+
+/**
+ * @author Evgeny Gerashchenko
+ * @since 3 August 2012
+ */
+public class KotlinSignatureInJavaMarkerProvider implements LineMarkerProvider {
+    private static final Function<PsiElement,String> TOOLTIP_PROVIDER = new Function<PsiElement, String>() {
+        @Override
+        public String fun(PsiElement element) {
+            PsiAnnotation annotation = findKotlinSignatureAnnotation(element);
+            assert annotation != null;
+            PsiAnnotationMemberValue value = annotation.getParameterList().getAttributes()[0].getValue();
+            String signature = value == null ? "null" : value instanceof PsiLiteralExpression
+                                                        ? ((PsiLiteralExpression) value).getValue().toString()
+                                                        : value.getText();
+            return "Alternative Kotlin signature is available for this method:\n"
+                   + signature;
+        }
+    };
+    private static final GutterIconNavigationHandler<PsiMethod> NAVIGATION_HANDLER = new GutterIconNavigationHandler<PsiMethod>() {
+        @Override
+        public void navigate(MouseEvent e, PsiMethod elt) {
+        }
+    };
+
+    @Nullable
+    private static PsiAnnotation findKotlinSignatureAnnotation(@NotNull PsiElement element) {
+        if (!(element instanceof PsiMethod)) return null;
+        PsiMethod annotationOwner = element.getOriginalElement() instanceof PsiMethod
+                                    ? (PsiMethod) element.getOriginalElement()
+                                    : (PsiMethod) element;
+        PsiAnnotation annotation =
+                JavaDescriptorResolver.findAnnotation(annotationOwner, JvmStdlibNames.KOTLIN_SIGNATURE.getFqName().getFqName());
+        if (annotation == null) return null;
+        if (annotation.getParameterList().getAttributes().length == 0) return null;
+        return annotation;
+    }
+
+    @Override
+    @Nullable
+    public LineMarkerInfo getLineMarkerInfo(@NotNull PsiElement element) {
+        if (findKotlinSignatureAnnotation(element) != null) {
+            return new LineMarkerInfo<PsiMethod>((PsiMethod) element, element.getTextOffset(), JetIcons.SMALL_LOGO, Pass.UPDATE_ALL,
+                                                 TOOLTIP_PROVIDER, NAVIGATION_HANDLER);
+        }
+        return null;
+    }
+
+    @Override
+    public void collectSlowLineMarkers(@NotNull List<PsiElement> elements, @NotNull Collection<LineMarkerInfo> result) {
+    }
+}