Make ARN contributor work for JSON again (#3150)

Author: rli

.changes/next-release/bugfix-780681e7-6ee9-4712-aeaa-9a1d34930656.json

@@ -0,0 +1,4 @@
+{
+  "type" : "bugfix",
+  "description" : "Workaround regression with ARN console navigation in JSON files"
+}

jetbrains-core/resources/META-INF/ext-json.xml

@@ -0,0 +1,8 @@
+<!-- Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. -->
+<!-- SPDX-License-Identifier: Apache-2.0 -->
+
+<idea-plugin>
+    <extensions defaultExtensionNs="com.intellij">
+        <psi.referenceContributor implementation="software.aws.toolkits.jetbrains.services.federation.psireferences.JsonArnPsiReferenceContributor"/>
+    </extensions>
+</idea-plugin>

jetbrains-core/resources/META-INF/plugin.xml

@@ -73,6 +73,7 @@ with what features/services are supported.
     <depends optional="true" config-file="ext-docker.xml">Docker</depends>
     <depends optional="true" config-file="ext-java.xml">com.intellij.modules.java</depends>
     <depends optional="true" config-file="ext-python.xml">com.intellij.modules.python</depends>
+    <depends optional="true" config-file="ext-json.xml">com.intellij.modules.json</depends>
 
     <xi:include href="/META-INF/IC-221.xml" xpointer="xpointer(/idea-plugin/*)">
         <xi:fallback/>

jetbrains-core/src/software/aws/toolkits/jetbrains/services/federation/psireferences/JsonArnPsiReferenceContributor.kt

@@ -0,0 +1,19 @@
+// Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+package software.aws.toolkits.jetbrains.services.federation.psireferences
+
+import com.intellij.json.psi.JsonStringLiteral
+import com.intellij.patterns.PlatformPatterns
+import com.intellij.psi.PsiReferenceContributor
+import com.intellij.psi.PsiReferenceRegistrar
+
+class JsonArnPsiReferenceContributor : PsiReferenceContributor() {
+    override fun registerReferenceProviders(registrar: PsiReferenceRegistrar) {
+        registrar.registerReferenceProvider(
+            PlatformPatterns.psiElement(JsonStringLiteral::class.java),
+            ArnPsiReferenceProvider(),
+            PsiReferenceRegistrar.LOWER_PRIORITY
+        )
+    }
+}

jetbrains-core/tst/software/aws/toolkits/jetbrains/services/federation/psireferences/ArnPsiReferenceProviderTest.kt

@@ -156,4 +156,34 @@ class ArnPsiReferenceProviderTest {
             }
         }
     }
+
+    @Test
+    fun `attaches reference to JSON values`() {
+        // language=JSON
+        val contents = """
+            {
+              "hello": "arn:aws:lambda::123456789012:function"
+            }
+        """.trimIndent()
+
+        val file = runInEdtAndGet {
+            projectRule.fixture.configureByText("json.json", contents)
+        }.virtualFile
+
+        val elements = mutableListOf<ArnReference>()
+        runReadAction {
+            PsiTreeUtil.processElements(
+                file.toPsi(projectRule.project),
+                PsiElementProcessor { child ->
+                    elements.addAll(child.references.filterIsInstance<ArnReference>())
+
+                    return@PsiElementProcessor true
+                }
+            )
+
+            assertThat(elements).hasSize(1).allSatisfy {
+                assertThat(it).isInstanceOf(ArnReference::class.java)
+            }
+        }
+    }
 }