Implement UiDataProvider along side DataProvider.

Author: rhdunn

src/intellij-compat/src/242/compat/com/intellij/compat/actionSystem/DataSink.kt

@@ -0,0 +1,8 @@
+// Copyright (C) 2025 Reece H. Dunn. SPDX-License-Identifier: Apache-2.0
+package com.intellij.compat.actionSystem
+
+import com.intellij.openapi.actionSystem.DataKey
+
+interface DataSink {
+    operator fun <T : Any> set(key: DataKey<T>, data: T?)
+}

src/intellij-compat/src/242/compat/com/intellij/compat/actionSystem/UiDataProvider.kt

@@ -0,0 +1,6 @@
+// Copyright (C) 2025 Reece H. Dunn. SPDX-License-Identifier: Apache-2.0
+package com.intellij.compat.actionSystem
+
+interface UiDataProvider {
+    fun uiDataSnapshot(sink: DataSink)
+}

src/intellij-compat/src/242/native/com/intellij/compat/actionSystem/TypeAliases.kt

@@ -0,0 +1,6 @@
+// Copyright (C) 2025 Reece H. Dunn. SPDX-License-Identifier: Apache-2.0
+package com.intellij.compat.actionSystem
+
+typealias DataSink = com.intellij.openapi.actionSystem.DataSink
+
+typealias UiDataProvider = com.intellij.openapi.actionSystem.UiDataProvider

src/kotlin-intellij/main/uk/co/reecedunn/intellij/plugin/core/execution/ui/ConsoleViewImpl.kt

@@ -1,20 +1,8 @@
-/*
- * Copyright (C) 2019 Reece H. Dunn
- *
- * 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.
- */
+// Copyright (C) 2019, 2025 Reece H. Dunn. SPDX-License-Identifier: Apache-2.0
 package uk.co.reecedunn.intellij.plugin.core.execution.ui
 
+import com.intellij.compat.actionSystem.DataSink
+import com.intellij.compat.actionSystem.UiDataProvider
 import com.intellij.execution.filters.Filter
 import com.intellij.execution.filters.HyperlinkInfo
 import com.intellij.execution.process.ProcessHandler
@@ -28,7 +16,7 @@ import java.awt.BorderLayout
 import javax.swing.JComponent
 import javax.swing.JPanel
 
-open class ConsoleViewImpl : JPanel(BorderLayout()), ConsoleView, DataProvider {
+open class ConsoleViewImpl : JPanel(BorderLayout()), ConsoleView, UiDataProvider, DataProvider {
     private var helpId: String? = null
 
     // region ConsoleView
@@ -83,6 +71,14 @@ open class ConsoleViewImpl : JPanel(BorderLayout()), ConsoleView, DataProvider {
     override fun scrollTo(offset: Int) {
     }
 
+    // endregion
+    // region UiDataProvider
+
+    override fun uiDataSnapshot(sink: DataSink) {
+        sink[PlatformDataKeys.HELP_ID] = helpId
+        sink[LangDataKeys.CONSOLE_VIEW] = this
+    }
+
     // endregion
     // region DataProvider
 

src/plugin-exquery/main/uk/co/reecedunn/intellij/plugin/exquery/restxq/endpoints/RestXqEndpoint.kt

@@ -1,20 +1,8 @@
-/*
- * Copyright (C) 2020-2023 Reece H. Dunn
- *
- * 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.
- */
+// Copyright (C) 2020-2023, 2025 Reece H. Dunn. SPDX-License-Identifier: Apache-2.0
 package uk.co.reecedunn.intellij.plugin.exquery.restxq.endpoints
 
+import com.intellij.compat.actionSystem.DataSink
+import com.intellij.compat.actionSystem.UiDataProvider
 import com.intellij.navigation.ItemPresentation
 import com.intellij.openapi.actionSystem.CommonDataKeys
 import com.intellij.openapi.actionSystem.DataProvider
@@ -29,6 +17,7 @@ import javax.swing.Icon
 class RestXqEndpoint(private val endpoint: XpmFunctionDeclaration) :
     EndpointMethodPresentation,
     ItemPresentation,
+    UiDataProvider,
     DataProvider {
     // region ItemPresentation
 
@@ -52,6 +41,13 @@ class RestXqEndpoint(private val endpoint: XpmFunctionDeclaration) :
     override val endpointMethods: List<String>
         get() = rest?.methods ?: listOf()
 
+    // endregion
+    // region UiDataProvider
+
+    override fun uiDataSnapshot(sink: DataSink) {
+        sink[CommonDataKeys.PSI_ELEMENT] = endpoint as PsiElement
+    }
+
     // endregion
     // region DataProvider
 

src/plugin-marklogic/main/uk/co/reecedunn/intellij/plugin/marklogic/rewriter/endpoints/RewriterEndpoint.kt

@@ -1,26 +1,14 @@
-/*
- * Copyright (C) 2020-2023 Reece H. Dunn
- *
- * 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.
- */
+// Copyright (C) 2020-2023, 2025 Reece H. Dunn. SPDX-License-Identifier: Apache-2.0
 package uk.co.reecedunn.intellij.plugin.marklogic.rewriter.endpoints
 
+import com.intellij.compat.actionSystem.DataSink
+import com.intellij.compat.actionSystem.UiDataProvider
+import com.intellij.compat.microservices.endpoints.presentation.EndpointMethodPresentation
+import com.intellij.compat.microservices.endpoints.presentation.HttpMethodPresentation
 import com.intellij.navigation.ItemPresentation
 import com.intellij.openapi.actionSystem.CommonDataKeys
 import com.intellij.openapi.actionSystem.DataProvider
 import com.intellij.psi.xml.XmlTag
-import com.intellij.compat.microservices.endpoints.presentation.EndpointMethodPresentation
-import com.intellij.compat.microservices.endpoints.presentation.HttpMethodPresentation
 import uk.co.reecedunn.intellij.plugin.core.xml.psi.ancestor
 import uk.co.reecedunn.intellij.plugin.marklogic.resources.MarkLogicBundle
 import uk.co.reecedunn.intellij.plugin.marklogic.resources.MarkLogicIcons
@@ -31,6 +19,7 @@ import javax.swing.Icon
 class RewriterEndpoint(val endpoint: XmlTag) :
     EndpointMethodPresentation,
     ItemPresentation,
+    UiDataProvider,
     DataProvider {
     // region ItemPresentation
 
@@ -59,6 +48,13 @@ class RewriterEndpoint(val endpoint: XmlTag) :
         methods?.split("\\s+") ?: listOf()
     }
 
+    // endregion
+    // region UiDataProvider
+
+    override fun uiDataSnapshot(sink: DataSink) {
+        sink[CommonDataKeys.PSI_ELEMENT] = endpointTarget?.resolve()
+    }
+
     // endregion
     // region DataProvider