feat: add image vector icon support and refactor completion contributor to share icon creation utility

Author: t-regbs

tools/idea-plugin/src/main/kotlin/io/github/composegears/valkyrie/completion/ImageVectorCompletionContributor.kt

@@ -8,11 +8,8 @@ import com.intellij.codeInsight.lookup.LookupElementDecorator
 import com.intellij.codeInsight.lookup.LookupElementPresentation
 import com.intellij.psi.util.CachedValueProvider
 import com.intellij.psi.util.CachedValuesManager
-import io.github.composegears.valkyrie.psi.imagevector.ImageVectorPsiParser
 import io.github.composegears.valkyrie.sdk.core.extensions.safeAs
-import io.github.composegears.valkyrie.sdk.ir.util.aspectRatio
-import io.github.composegears.valkyrie.sdk.ir.util.dominantShadeColor
-import io.github.composegears.valkyrie.sdk.ir.xml.toVectorXmlString
+import io.github.composegears.valkyrie.util.createImageVectorIcon
 import javax.swing.Icon
 import org.jetbrains.kotlin.psi.KtFile
 import org.jetbrains.kotlin.psi.KtProperty
@@ -54,22 +51,11 @@ class ImageVectorCompletionContributor : CompletionContributor() {
         val cachedValuesManager = CachedValuesManager.getManager(ktFile.project)
 
         return cachedValuesManager.getCachedValue(ktFile) {
-            val icon = createIconFromKtFile(ktFile)
+            val icon = ktFile.createImageVectorIcon()
             CachedValueProvider.Result.create(icon, ktFile)
         }
     }
 
-    private fun createIconFromKtFile(ktFile: KtFile): Icon? {
-        val irImageVector = ImageVectorPsiParser.parseToIrImageVector(ktFile)
-        val vectorXml = irImageVector?.toVectorXmlString() ?: return null
-
-        return ImageVectorIcon(
-            vectorXml = vectorXml,
-            aspectRatio = irImageVector.aspectRatio,
-            dominantShade = irImageVector.dominantShadeColor,
-        )
-    }
-
     private class ComposeColorLookupElementDecorator(
         delegate: LookupElement,
         private val icon: Icon,

tools/idea-plugin/src/main/kotlin/io/github/composegears/valkyrie/gutter/ImageVectorGutterProvider.kt

@@ -16,6 +16,7 @@ import io.github.composegears.valkyrie.sdk.ir.core.IrImageVector
 import io.github.composegears.valkyrie.sdk.ir.util.aspectRatio
 import io.github.composegears.valkyrie.sdk.ir.util.dominantShadeColor
 import io.github.composegears.valkyrie.sdk.ir.xml.toVectorXmlString
+import io.github.composegears.valkyrie.util.isImageVector
 import javax.swing.Icon
 import org.jetbrains.kotlin.psi.KtNameReferenceExpression
 import org.jetbrains.kotlin.psi.KtProperty
@@ -135,12 +136,5 @@ class ImageVectorGutterProvider : LineMarkerProvider {
 
     private companion object {
         private const val COMPILED_CODE_MARKER = "/* compiled code */"
-
-        private val IMAGE_VECTOR_TYPES = setOf(
-            "ImageVector",
-            "androidx.compose.ui.graphics.vector.ImageVector",
-        )
-
-        private fun KtProperty.isImageVector(): Boolean = typeReference?.text in IMAGE_VECTOR_TYPES
     }
 }

tools/idea-plugin/src/main/kotlin/io/github/composegears/valkyrie/icon/ImageVectorIconProvider.kt

@@ -0,0 +1,34 @@
+package io.github.composegears.valkyrie.icon
+
+import com.intellij.ide.IconProvider
+import com.intellij.openapi.project.DumbAware
+import com.intellij.psi.PsiElement
+import com.intellij.psi.PsiFile
+import com.intellij.psi.util.CachedValueProvider
+import com.intellij.psi.util.CachedValuesManager
+import io.github.composegears.valkyrie.util.createImageVectorIcon
+import io.github.composegears.valkyrie.util.hasImageVectorProperties
+import javax.swing.Icon
+import org.jetbrains.kotlin.psi.KtFile
+
+class ImageVectorIconProvider :
+    IconProvider(),
+    DumbAware {
+
+    override fun getIcon(element: PsiElement, flags: Int): Icon? {
+        val ktFile = when (element) {
+            is KtFile -> element
+            is PsiFile -> null
+            else -> element.containingFile as? KtFile
+        } ?: return null
+
+        if (!ktFile.hasImageVectorProperties()) {
+            return null
+        }
+
+        return CachedValuesManager.getCachedValue(ktFile) {
+            val icon = ktFile.createImageVectorIcon()
+            CachedValueProvider.Result.create(icon, ktFile)
+        }
+    }
+}

tools/idea-plugin/src/main/kotlin/io/github/composegears/valkyrie/util/ImageVectorPsiUtil.kt

@@ -0,0 +1,53 @@
+package io.github.composegears.valkyrie.util
+
+import io.github.composegears.valkyrie.completion.ImageVectorIcon
+import io.github.composegears.valkyrie.psi.imagevector.ImageVectorPsiParser
+import io.github.composegears.valkyrie.sdk.ir.util.aspectRatio
+import io.github.composegears.valkyrie.sdk.ir.util.dominantShadeColor
+import io.github.composegears.valkyrie.sdk.ir.xml.toVectorXmlString
+import javax.swing.Icon
+import org.jetbrains.kotlin.psi.KtFile
+import org.jetbrains.kotlin.psi.KtProperty
+
+val IMAGE_VECTOR_TYPES = setOf(
+    "ImageVector",
+    "androidx.compose.ui.graphics.vector.ImageVector",
+)
+
+/**
+ * Checks if this property is an ImageVector based on its type reference.
+ */
+fun KtProperty.isImageVector(): Boolean {
+    val typeText = typeReference?.text ?: return false
+    return typeText in IMAGE_VECTOR_TYPES
+}
+
+/**
+ * Checks if this Kotlin file contains ImageVector properties.
+ *
+ * A file is considered to contain ImageVector if it has at least one top-level
+ * property with an ImageVector type annotation.
+ */
+fun KtFile.hasImageVectorProperties(): Boolean {
+    val properties = declarations.filterIsInstance<KtProperty>()
+    if (properties.isEmpty()) return false
+
+    return properties.any { it.isImageVector() }
+}
+
+/**
+ * Creates an icon representation from this Kotlin file containing an ImageVector.
+ *
+ * This function parses the ImageVector definition and renders it as a Swing Icon
+ * suitable for display in the IDE UI (project view, gutter, completion, etc.).
+ */
+fun KtFile.createImageVectorIcon(): Icon? {
+    val irImageVector = ImageVectorPsiParser.parseToIrImageVector(this) ?: return null
+    val vectorXml = irImageVector.toVectorXmlString()
+
+    return ImageVectorIcon(
+        vectorXml = vectorXml,
+        aspectRatio = irImageVector.aspectRatio,
+        dominantShade = irImageVector.dominantShadeColor,
+    )
+}

tools/idea-plugin/src/main/resources/META-INF/plugin.xml

@@ -75,6 +75,8 @@ Allows to create organized icon pack with an extension property of you pack obje
             language="kotlin"
             order="first, before KotlinCompletionContributor, before AndroidComposeCompletionContributor"
             implementationClass="io.github.composegears.valkyrie.completion.ImageVectorCompletionContributor"/>
+
+        <iconProvider implementation="io.github.composegears.valkyrie.icon.ImageVectorIconProvider" order="first"/>
     </extensions>
 
     <extensions defaultExtensionNs="org.jetbrains.kotlin">