```

refactor: Improve component group processing and add source dependencies

This commit refactors the component group processing logic in `ComponentProcessor` and `SampleCodeProcessor`, and adds source dependencies to the generated files.

- **Component Group Processing:**
    - Simplified the logic for organizing component functions into groups.
    - Use `getOrPut` to manage component functions.
    - If component package not mapped, then the default group is "/".
- **Dependencies:**
    - Added `componentPackageProperty` and `componentScreenFile` to store component files.
    - Added source dependencies to the generated files using `Dependencies`.
- **Sample Code Processing:**
    - Use `getOrPut` to manage sample code functions.
```

Author: Sanlorng

gallery-processor/src/jvmMain/kotlin/io/github/composefluent/gallery/processor/ComponentProcessor.kt

@@ -5,6 +5,7 @@ import com.google.devtools.ksp.processing.Dependencies
 import com.google.devtools.ksp.processing.KSPLogger
 import com.google.devtools.ksp.impl.symbol.kotlin.KSPropertyDeclarationImpl
 import com.google.devtools.ksp.symbol.KSAnnotation
+import com.google.devtools.ksp.symbol.KSFile
 import com.google.devtools.ksp.symbol.KSFunctionDeclaration
 import com.google.devtools.ksp.symbol.KSPropertyDeclaration
 import com.google.devtools.ksp.symbol.KSValueArgument
@@ -33,6 +34,8 @@ class ComponentProcessor(private val logger: KSPLogger, private val codeGenerato
     private val componentFunctions = mutableMapOf<String, MutableList<Pair<KSAnnotation, KSFunctionDeclaration>>>()
     private val componentGroups = mutableMapOf<String, Pair<KSAnnotation, KSPropertyDeclarationImpl>>()
     private val componentPackageMap = mutableMapOf<String, String>()
+    private val componentPackageProperty = mutableSetOf<KSFile>()
+    private val componentScreenFile = mutableSetOf<KSFile>()
 
     private val componentPackage = "io.github.jpy.wangposefluent.gallery.component"
     private val componentItemClass = ClassName(componentPackage, "ComponentItem")
@@ -52,19 +55,12 @@ class ComponentProcessor(private val logger: KSPLogger, private val codeGenerato
     private fun arrangeComponentGroup() {
         val mapPackage = componentFunctions.remove("/_Auto") ?: emptyList()
         mapPackage.forEach { pair ->
-            val group = componentPackageMap[pair.second.packageName.asString()]
-            if (!group.isNullOrEmpty()) {
-                val list =
-                    componentFunctions[group] ?: mutableListOf<Pair<KSAnnotation, KSFunctionDeclaration>>().apply {
-                        componentFunctions[group] = this
-                    }
-                list.add(pair)
-            } else {
-                val list = componentFunctions["/"] ?: mutableListOf<Pair<KSAnnotation, KSFunctionDeclaration>>().apply {
-                    componentFunctions["/"] = this
-                }
-                list.add(pair)
+            val group = componentPackageMap[pair.second.packageName.asString()].let {
+                if (it.isNullOrEmpty()) "/" else it
             }
+            componentFunctions
+                .getOrPut(group) { mutableListOf() }
+                .add(pair)
         }
     }
 
@@ -79,11 +75,10 @@ class ComponentProcessor(private val logger: KSPLogger, private val codeGenerato
                     }
                 }
                 val group = (groupArg?.value as? String)?.prefixIfNot("/") ?: return@forEach
-                val list =
-                    componentFunctions[group] ?: mutableListOf<Pair<KSAnnotation, KSFunctionDeclaration>>().apply {
-                        componentFunctions[group] = this
-                    }
-                list.add(annotation to function)
+                componentFunctions
+                    .getOrPut(group) { mutableListOf() }
+                    .add(annotation to function)
+                componentScreenFile.add(function.containingFile ?: return)
                 return
             }
         }
@@ -99,6 +94,7 @@ class ComponentProcessor(private val logger: KSPLogger, private val codeGenerato
             if (psi is KtProperty) {
                 val groupName = psi.initializer?.text?.removePrefix("\"")?.removeSuffix("\"")?.prefixIfNot("/")
                     ?: return
+                property.containingFile?.let { componentPackageProperty.add(it) }
                 componentGroups[groupName] = annotation to property
                 val packageNameValue =
                     annotation.arguments.firstOrNull { it.name?.asString() == "packageMap" }?.value as? String
@@ -200,15 +196,15 @@ class ComponentProcessor(private val logger: KSPLogger, private val codeGenerato
                 .build()
         )
         val file = codeGenerator.createNewFile(
-            Dependencies(true),
+            Dependencies(aggregating = true, sources = componentPackageProperty.toTypedArray()),
             fileSpecBuilder.packageName,
             fileSpecBuilder.name
         )
         val pathFileSpec = FileSpec.builder(
             componentPackage, "ComponentPagePath"
         ).addType(componentPagePathType.build()).build()
         val pathFile = codeGenerator.createNewFile(
-            Dependencies(true),
+            Dependencies(true, sources = componentScreenFile.toTypedArray()),
             componentPackage,
             pathFileSpec.name
         )

gallery-processor/src/jvmMain/kotlin/io/github/composefluent/gallery/processor/SampleCodeProcessor.kt

@@ -24,11 +24,9 @@ class SampleCodeProcessor(private val logger: KSPLogger, private val codeGenerat
         super.onFunctionVisit(function)
         function.annotations.forEach {
             if (it.isTargetAnnotation(sampleAnnotation)) {
-                val list = sampleCodeFunctions[function.packageName.asString()]
-                    ?: mutableListOf<KSFunctionDeclaration>().apply {
-                        sampleCodeFunctions[function.packageName.asString()] = this
-                    }
-                list.add(function)
+                sampleCodeFunctions
+                    .getOrPut(function.packageName.asString()) { mutableListOf() }
+                    .add(function)
             }
             return
         }