feat: getProjectEndpoints

Author: BFergerson

marker/jvm-marker/src/main/kotlin/spp/jetbrains/marker/jvm/detect/JVMEndpointDetector.kt

@@ -45,7 +45,7 @@ import spp.jetbrains.marker.source.mark.guide.GuideMark
  */
 class JVMEndpointDetector(project: Project) : EndpointDetector<JVMEndpointNameDetector>(project) {
 
-    override val detectorSet: Set<JVMEndpointNameDetector> = setOf(
+    override val detectorSet = mutableSetOf(
         SkywalkingTraceEndpoint(),
         SpringMVCEndpoint(),
         MicronautEndpoint(),

marker/jvm-marker/src/main/kotlin/spp/jetbrains/marker/jvm/detect/endpoint/MicronautEndpoint.kt

@@ -31,6 +31,7 @@ import org.jetbrains.kotlin.psi.psiUtil.containingClassOrObject
 import org.jetbrains.kotlin.resolve.BindingContext
 import org.jetbrains.kotlin.resolve.lazy.BodyResolveMode
 import spp.jetbrains.marker.jvm.detect.JVMEndpointDetector.JVMEndpointNameDetector
+import spp.jetbrains.marker.service.ArtifactNamingService
 import spp.jetbrains.marker.source.info.EndpointDetector.DetectedEndpoint
 
 /**
@@ -48,7 +49,7 @@ class MicronautEndpoint : JVMEndpointNameDetector {
         val promise = Promise.promise<List<DetectedEndpoint>>()
         ApplicationManager.getApplication().runReadAction {
             val annotation = element.annotations.find {
-                it.qualifiedName?.startsWith(endpointAnnotationPrefix) == true
+                ArtifactNamingService.getFullyQualifiedName(it).identifier.startsWith(endpointAnnotationPrefix)
             }
             if (annotation != null) {
                 val endpointType = annotation.qualifiedName!!.substringAfterLast(".").uppercase()
@@ -61,7 +62,7 @@ class MicronautEndpoint : JVMEndpointNameDetector {
 
                 //get controller
                 val controllerAnnotation = element.containingClass?.annotations?.find {
-                    it.qualifiedName?.startsWith(endpointAnnotationPrefix) == true
+                    ArtifactNamingService.getFullyQualifiedName(it).identifier.startsWith(endpointAnnotationPrefix)
                 }
                 val controllerValue = controllerAnnotation?.let { getAttributeValue(it, "value") }
                 val controller = if (controllerValue is PsiLiteral) {

marker/jvm-marker/src/main/kotlin/spp/jetbrains/marker/jvm/service/utils/JVMMarkerUtils.kt

@@ -52,6 +52,7 @@ object JVMMarkerUtils {
         when {
             element.isKotlin() && element is KtClass -> return getFullyQualifiedName(element)
             element.isKotlin() && element is KtNamedFunction -> return getFullyQualifiedName(element)
+            element is PsiAnnotation -> return getFullyQualifiedName(element)
             element is PsiClass -> return getFullyQualifiedName(element)
             element is PsiMethod -> return getFullyQualifiedName(element)
             else -> Unit
@@ -79,6 +80,35 @@ object JVMMarkerUtils {
         )
     }
 
+    private fun getFullyQualifiedName(annotation: PsiAnnotation): ArtifactQualifiedName {
+        val qualifiedName = getFullyQualifiedName(annotation, annotation.qualifiedName.toString())
+        return ArtifactQualifiedName(
+            qualifiedName,
+            type = ArtifactType.ANNOTATION,
+            lineNumber = annotation.nameReferenceElement?.let { SourceMarkerUtils.getLineNumber(it) }
+        )
+    }
+
+    private fun getFullyQualifiedName(psiElement: PsiElement, simpleName: String): String {
+        var parent = psiElement
+        while (parent !is PsiJavaFile && parent.parent != null) {
+            parent = parent.parent
+        }
+        val javaFile = parent as? PsiJavaFile ?: TODO() //todo: others
+
+        // Loop through imports to find fully qualified name
+        for (importStatement in javaFile.importList?.importStatements ?: emptyArray()) {
+            val qName = importStatement.qualifiedName
+            if (qName?.endsWith(".$simpleName") == true || qName == simpleName) {
+                return qName
+            }
+        }
+
+        // If the simple name wasn't found in the imports, it might be in the same package.
+        val packageName = javaFile.packageStatement?.packageName
+        return if (packageName != null) "$packageName.$simpleName" else simpleName
+    }
+
     private fun getFullyQualifiedName(clazz: PsiClass): ArtifactQualifiedName {
         return ArtifactQualifiedName(
             "${JvmClassUtil.getJvmClassName(clazz)}",