feat(devti): add module information to SketchRunContext #350

- Add moduleInfo parameter to SketchRunContext
- Implement logic to detect and display module information for multi-module projects
- Update sketch.vm templates to include module information

Author: phodal

core/src/main/kotlin/cc/unitmesh/devti/sketch/SketchRunContext.kt

@@ -3,6 +3,7 @@ package cc.unitmesh.devti.sketch
 import cc.unitmesh.devti.agent.tool.search.RipgrepSearcher
 import cc.unitmesh.devti.gui.chat.message.ChatActionType
 import cc.unitmesh.devti.gui.chat.ui.relativePath
+import cc.unitmesh.devti.mcp.host.Response
 import cc.unitmesh.devti.provider.BuildSystemProvider
 import cc.unitmesh.devti.provider.context.ChatContextItem
 import cc.unitmesh.devti.provider.context.ChatContextProvider
@@ -17,6 +18,7 @@ import com.intellij.openapi.diagnostic.logger
 import com.intellij.openapi.editor.Editor
 import com.intellij.openapi.fileEditor.FileDocumentManager
 import com.intellij.openapi.fileEditor.FileEditorManager
+import com.intellij.openapi.module.ModuleManager
 import com.intellij.openapi.project.Project
 import com.intellij.openapi.project.ProjectManager
 import com.intellij.openapi.project.guessProjectDir
@@ -44,6 +46,7 @@ data class SketchRunContext(
     val buildTool: String = "",
     val searchTool: String = "localSearch",
     val rule: String = "",
+    val moduleInfo: String = "",
 ) : TemplateContext {
     companion object {
         suspend fun create(project: Project, myEditor: Editor?, input: String): SketchRunContext {
@@ -87,6 +90,8 @@ data class SketchRunContext(
                 ""
             }
 
+            val moduleInfo = moduleContext(project)
+
             return SketchRunContext(
                 currentFile = currentFile?.relativePath(project),
                 currentElement = currentElement,
@@ -101,8 +106,37 @@ data class SketchRunContext(
                 buildTool = buildTool,
                 searchTool = lookupSearchTool(),
                 rule = rule,
+                moduleInfo = moduleInfo,
             )
         }
+
+        private val JAVA_ONE_PROJECT_SIZE_IN_IDEA = 2
+        fun moduleContext(project: Project): String {
+            val moduleManager = ModuleManager.getInstance(project)
+            val allModules = moduleManager.modules.map { it.name }
+            if (allModules.size <= JAVA_ONE_PROJECT_SIZE_IN_IDEA) {
+                return ""
+            }
+
+            val rootProjectName = allModules.minByOrNull { it.length } ?: return ""
+            
+            val cleanedModules = allModules
+                .filter { it != rootProjectName && !it.endsWith(".test") }
+                .map { moduleName ->
+                    moduleName
+                        .removePrefix("$rootProjectName.")
+                        .removeSuffix(".main")
+                }
+                .distinct() // Remove duplicates after cleaning
+                .filter { it.isNotEmpty() } // Remove any empty strings that might result
+            
+            return if (cleanedModules.isNotEmpty()) {
+                val modules = cleanedModules.joinToString(", ") { it }
+                "- This project is a mono-repo projects, Please careful to create file in module. Here's modules: $modules"
+            } else {
+                ""
+            }
+        }
     }
 }
 
@@ -130,3 +164,4 @@ private fun workspace(myProject: Project? = null): String {
     val project = myProject ?: ProjectManager.getInstance().openProjects.firstOrNull()
     return project?.guessProjectDir()?.path ?: ""
 }
+

core/src/main/resources/genius/en/code/sketch.vm

@@ -18,6 +18,7 @@ Your main goal is to follow the USER's instructions at each message.
 - User's workspace context is: ${context.frameworkContext}
 - Current time is: ${context.time}
 ${context.rule}
+${context.moduleInfo}
 
 You have tools at your disposal to solve the coding task. We design a DSL call DevIn for you to call tools. If the USER's
 task is general or you already know the answer, just respond without calling tools.

core/src/main/resources/genius/zh/code/sketch.vm

@@ -15,6 +15,7 @@ Here is basic information about the current workspace:
 - User's workspace context is: ${context.frameworkContext}
 - Current time is: ${context.time}
 ${context.rule}
+${context.moduleInfo}
 
 在编码之前，请确保您拥有足够的上下文信息。请遵循 DevIn <devin /> 指令编写代码，以节省用户的时间。请不要直接基于用户的问题进行编码，而是先通过
 tool calls 获取上下文信息来了解用户的代码库。