Introduce line marker for wasm_bindgen attributes

Author: avrong

js/src/main/kotlin/org/rust/js/MyReferenceContributor.kt

@@ -0,0 +1,90 @@
+/*
+ * Use of this source code is governed by the MIT license that can be
+ * found in the LICENSE file.
+ */
+
+package org.rust.js
+
+import com.intellij.codeHighlighting.Pass
+import com.intellij.codeInsight.daemon.DefaultGutterIconNavigationHandler
+import com.intellij.codeInsight.daemon.RelatedItemLineMarkerInfo
+import com.intellij.codeInsight.daemon.RelatedItemLineMarkerProvider
+import com.intellij.lang.javascript.psi.JSElement
+import com.intellij.lang.javascript.psi.ecma6.TypeScriptClass
+import com.intellij.lang.javascript.psi.ecma6.TypeScriptEnum
+import com.intellij.lang.javascript.psi.ecma6.TypeScriptFunction
+import com.intellij.navigation.GotoRelatedItem
+import com.intellij.openapi.editor.markup.GutterIconRenderer
+import com.intellij.openapi.vfs.LocalFileSystem
+import com.intellij.psi.PsiElement
+import com.intellij.psi.PsiFile
+import com.intellij.psi.PsiManager
+import com.intellij.psi.util.findDescendantOfType
+import com.intellij.psi.util.nextLeaf
+import com.intellij.util.ConstantFunction
+import org.rust.ide.icons.RsIcons
+import org.rust.lang.core.crate.crateGraph
+import org.rust.lang.core.psi.RsEnumItem
+import org.rust.lang.core.psi.RsFunction
+import org.rust.lang.core.psi.RsImplItem
+import org.rust.lang.core.psi.RsStructItem
+import org.rust.lang.core.psi.ext.RsOuterAttributeOwner
+import org.rust.lang.core.psi.ext.RsVisibilityOwner
+import org.rust.lang.core.psi.ext.containingCargoPackage
+import org.rust.lang.core.psi.ext.findOuterAttr
+import org.rust.lang.core.psi.impl.RsBaseTypeImpl
+import java.nio.file.Paths
+
+class RsWasmBindgenLineMarkerProvider : RelatedItemLineMarkerProvider() {
+    override fun getLineMarkerInfo(element: PsiElement): RelatedItemLineMarkerInfo<*>? {
+        if (element.project.crateGraph.topSortedCrates.none { it.presentableName == "wasm-bindgen" }) return null
+        if (element !is RsOuterAttributeOwner) return null
+        if (element is RsVisibilityOwner && element !is RsImplItem && !element.isPublic) return null
+        val attr = element.findOuterAttr("wasm_bindgen")
+            ?.nextLeaf() ?: return null
+
+        val cargoPackage = element.containingCargoPackage ?: return null
+        val expectedName = cargoPackage.normName
+        val basePath = cargoPackage.contentRoot?.path ?: return null
+
+        val expectedPath = Paths.get(basePath, "pkg/$expectedName.d.ts").toString()
+        val tsFile = LocalFileSystem.getInstance().findFileByPath(expectedPath) ?: return null
+        val tsPsiFile = PsiManager.getInstance(element.project).findFile(tsFile) ?: return null
+
+        val destination = when (element) {
+            is RsImplItem -> findTsElementForImpl(element, tsPsiFile)
+            else -> findTsElement(element, tsPsiFile)
+        } ?: return null
+
+        // BACKCOMPAT: 2020.1
+        @Suppress("DEPRECATION")
+        return RelatedItemLineMarkerInfo(
+            attr,
+            attr.textRange,
+            RsIcons.WASM_PACK,
+            Pass.LINE_MARKERS,
+            ConstantFunction<PsiElement, String>("wasm-bindgen declaration"),
+            DefaultGutterIconNavigationHandler(listOf(destination), "wasm-bindgen declaration"),
+            GutterIconRenderer.Alignment.RIGHT,
+            listOf(GotoRelatedItem(destination))
+        )
+    }
+
+    private fun findTsElement(element: RsOuterAttributeOwner, tsPsiFile: PsiFile): JSElement? =
+        when (element) {
+            is RsFunction -> tsPsiFile.findDescendantOfType<TypeScriptFunction> { it.name == element.name }
+            is RsEnumItem -> tsPsiFile.findDescendantOfType<TypeScriptEnum> { it.name == element.name }
+            is RsStructItem -> tsPsiFile.findDescendantOfType<TypeScriptClass> { it.name == element.name }
+            else -> null
+        }
+
+    private fun findTsElementForImpl(element: RsImplItem, tsPsiFile: PsiFile): TypeScriptClass? {
+        val reference = (element.typeReference as? RsBaseTypeImpl) ?: return null
+        val struct = (reference.path?.reference?.resolve() as? RsStructItem) ?: return null
+        val name = struct.name ?: return null
+
+        return tsPsiFile.findDescendantOfType {
+            it.name == name
+        }
+    }
+}

js/src/main/kotlin/org/rust/js/RsWasmBindgenLineMarkerProvider.kt

@@ -1,9 +1,5 @@
 <idea-plugin>
     <extensions defaultExtensionNs="com.intellij">
-        <psi.referenceContributor language="JSON" implementation="org.rust.js.MyReferenceContributor"/>
+        <codeInsight.lineMarkerProvider language="Rust" implementationClass="org.rust.js.RsWasmBindgenLineMarkerProvider"/>
     </extensions>
 </idea-plugin>
-
-
-
-