Fix an alway null warning in CustomConstraintFunctionReference.moduleUriValue.

Author: rhdunn

src/lang-xpm/main/uk/co/reecedunn/intellij/plugin/xpm/optree/item/Items.kt

@@ -1,18 +1,4 @@
-/*
- * Copyright (C) 2021 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) 2021, 2026 Reece H. Dunn. SPDX-License-Identifier: Apache-2.0
 package uk.co.reecedunn.intellij.plugin.xpm.optree.item
 
 import uk.co.reecedunn.intellij.plugin.xdm.types.XdmAttributeNode
@@ -43,6 +29,10 @@ val XdmElementNode.namespaceUri: String?
 val XdmElementNode.localName: String?
     get() = nodeName?.localName?.data
 
+fun XdmElementNode.getAttribute(ns: String, localName: String): XdmAttributeNode? {
+    return attributes.find { it.localName == localName && it.namespaceUri == ns }
+}
+
 fun XdmElementNode.getAttributeValue(ns: String, localName: String): String? {
     return attributes.find { it.localName == localName && it.namespaceUri == ns }?.stringValue
 }

src/plugin-marklogic/main/uk/co/reecedunn/intellij/plugin/marklogic/search/options/reference/CustomConstraintFunctionReference.kt

@@ -1,18 +1,4 @@
-/*
- * Copyright (C) 2021 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) 2021, 2026 Reece H. Dunn. SPDX-License-Identifier: Apache-2.0
 package uk.co.reecedunn.intellij.plugin.marklogic.search.options.reference
 
 import com.intellij.psi.PsiElement
@@ -26,6 +12,7 @@ import uk.co.reecedunn.intellij.plugin.xdm.types.XdmAttributeNode
 import uk.co.reecedunn.intellij.plugin.xdm.types.XdmElementNode
 import uk.co.reecedunn.intellij.plugin.xdm.types.element
 import uk.co.reecedunn.intellij.plugin.xpm.optree.function.XpmFunctionDeclaration
+import uk.co.reecedunn.intellij.plugin.xpm.optree.item.getAttribute
 import uk.co.reecedunn.intellij.plugin.xpm.optree.item.getAttributeValue
 import uk.co.reecedunn.intellij.plugin.xpm.optree.item.localName
 import uk.co.reecedunn.intellij.plugin.xquery.ast.xquery.XQueryDirAttributeValue
@@ -51,15 +38,21 @@ class CustomConstraintFunctionReference(
     val moduleNamespace: String?
         get() = node.getAttributeValue("", "ns")
 
-    @Suppress("MemberVisibilityCanBePrivate")
+    @Suppress("unused")
     val moduleUri: String?
         get() = node.getAttributeValue("", "at")
 
     // endregion
     // region function reference
 
     private val moduleUriValue: PsiElement?
-        get() = (moduleUri as? PsiElement)?.children()?.filterIsInstance<XQueryDirAttributeValue>()?.firstOrNull()
+        get() {
+            val moduleUriNode = node.getAttribute("", "at")
+            return (moduleUriNode as? PsiElement)
+                ?.children()
+                ?.filterIsInstance<XQueryDirAttributeValue>()
+                ?.firstOrNull()
+        }
 
     private val moduleUriReference: PsiReference?
         get() = moduleUriValue?.references?.find { it is ModuleUriReference }