feat(pair): add LayeredArch and ProjectPackageTree classes

Add the `LayeredArch` and `ProjectPackageTree` classes to the `cc.unitmesh.devti.pair.arch` package.

Author: phodal

java/src/main/kotlin/cc/unitmesh/idea/provider/JavaLayeredArchProvider.kt

@@ -0,0 +1,33 @@
+package cc.unitmesh.idea.provider
+
+import cc.unitmesh.devti.pair.arch.ProjectPackageTree
+import cc.unitmesh.devti.provider.architecture.LayeredArchProvider
+import com.intellij.openapi.project.Project
+import com.intellij.openapi.projectRoots.JavaSdkType
+import com.intellij.openapi.roots.ProjectRootManager
+import com.intellij.psi.JavaPsiFacade
+
+class JavaLayeredArchProvider : LayeredArchProvider {
+    private val layeredArch = ProjectPackageTree()
+    override fun isApplicable(project: Project): Boolean {
+        val projectSdk = ProjectRootManager.getInstance(project).projectSdk ?: return false
+
+        return projectSdk.sdkType is JavaSdkType
+    }
+
+    override fun getLayeredArch(project: Project): ProjectPackageTree {
+        val psiFacade = JavaPsiFacade.getInstance(project)
+
+        val projectRootManager = ProjectRootManager.getInstance(project)
+        val contentRoots = projectRootManager.contentRoots
+        for (contentRoot in contentRoots) {
+            // todo implement this
+            val psiPackage = psiFacade.findPackage(contentRoot.url)
+            if (psiPackage != null) {
+                layeredArch.addPackage(psiPackage.qualifiedName)
+            }
+        }
+
+        return layeredArch
+    }
+}

java/src/main/resources/cc.unitmesh.idea.xml

@@ -52,6 +52,9 @@
                 language="JAVA"
                 implementationClass="cc.unitmesh.idea.provider.JavaCustomPromptProvider" />
 
+        <layeredArchProvider
+                implementation="cc.unitmesh.idea.provider.JavaLayeredArchProvider" />
+
         <autoDevIntention>
             <className>cc.unitmesh.idea.actions.AutoCrudAction</className>
             <bundleName>messages.AutoDevBundle</bundleName>

src/233/main/resources/META-INF/autodev-core.xml

@@ -50,13 +50,13 @@
                     icon="cc.unitmesh.devti.AutoDevIcons.AI_COPILOT"
                     factoryClass="cc.unitmesh.devti.gui.AutoDevToolWindowFactory"/>
 
-<!--        <toolWindow id="AutoDevPair"-->
-<!--                    doNotActivateOnStart="true"-->
-<!--                    anchor="right"-->
-<!--                    secondary="true"-->
-<!--                    canCloseContents="true"-->
-<!--                    icon="cc.unitmesh.devti.AutoDevIcons.AI_PAIR"-->
-<!--                    factoryClass="cc.unitmesh.devti.gui.AutoDevPairToolWindowFactory"/>-->
+        <toolWindow id="AutoDev Pair"
+                    doNotActivateOnStart="true"
+                    anchor="left"
+                    secondary="true"
+                    canCloseContents="false"
+                    icon="cc.unitmesh.devti.AutoDevIcons.AI_PAIR"
+                    factoryClass="cc.unitmesh.devti.gui.AutoDevPairToolWindowFactory"/>
 
         <notificationGroup id="AI notification group" displayType="STICKY_BALLOON" bundle="messages.AutoDevBundle"
                            key="name"/>
@@ -135,6 +135,10 @@
                         interface="cc.unitmesh.devti.provider.DevFlowProvider"
                         dynamic="true"/>
 
+        <extensionPoint qualifiedName="cc.unitmesh.layeredArchProvider"
+                        interface="cc.unitmesh.devti.provider.architecture.LayeredArchProvider"
+                        dynamic="true"  />
+
         <!-- TODO: find better way to share context -->
         <extensionPoint qualifiedName="cc.unitmesh.contextPrompter"
                         interface="cc.unitmesh.devti.provider.ContextPrompter"

src/main/kotlin/cc/unitmesh/devti/gui/AutoDevPairToolWindowFactory.kt

@@ -1,27 +1,44 @@
 package cc.unitmesh.devti.gui
 
+import cc.unitmesh.devti.provider.architecture.LayeredArchProvider
+import com.intellij.openapi.Disposable
 import com.intellij.openapi.project.Project
+import com.intellij.openapi.ui.NullableComponent
+import com.intellij.openapi.ui.SimpleToolWindowPanel
 import com.intellij.openapi.wm.ToolWindow
 import com.intellij.openapi.wm.ToolWindowFactory
 import com.intellij.ui.dsl.builder.panel
-import javax.swing.JComponent
 
 class AutoDevPairToolWindowFactory : ToolWindowFactory {
     override fun createToolWindowContent(project: Project, toolWindow: ToolWindow) {
-        val panel = AutoDevPairToolWindow(project)
+        val disposable = toolWindow.disposable
+        val panel = AutoDevPairToolWindow(project, disposable)
+
         val contentManager = toolWindow.contentManager
+
         val content = contentManager.factory.createContent(panel, "", false)
         contentManager.addContent(content)
     }
 }
 
-class AutoDevPairToolWindow(val project: Project) : JComponent() {
+class AutoDevPairToolWindow(val project: Project, val disposable: Disposable) : SimpleToolWindowPanel(true, true),
+    NullableComponent {
     init {
-        // A table like todo list
-        panel {
+        val layeredArch = LayeredArchProvider.find(project)?.getLayeredArch(project)
+        val panel = panel {
             row {
                 label("Hello World")
             }
+            row {
+                // show a tree in a table
+                label(layeredArch.toString())
+            }
         }
+
+        setContent(panel)
+    }
+
+    override fun isNull(): Boolean {
+        return false
     }
 }

src/main/kotlin/cc/unitmesh/devti/pair/arch/LayeredArch.kt

@@ -0,0 +1,13 @@
+package cc.unitmesh.devti.pair.arch
+
+data class LayeredArch(val layers: List<Layer>) {
+    fun getLayerByName(name: String): Layer? {
+        return layers.find { it.name == name }
+    }
+}
+
+data class Layer(val name: String, val packages: List<String>) {
+    fun getPackageByName(name: String): String? {
+        return packages.find { it == name }
+    }
+}

src/main/kotlin/cc/unitmesh/devti/pair/arch/ProjectPackageTree.kt

@@ -0,0 +1,37 @@
+package cc.unitmesh.devti.pair.arch
+
+class ProjectPackageTree {
+    private var tree: TreeNode? = null
+
+    private fun getTree(): TreeNode {
+        if (tree == null) {
+            tree = TreeNode("root")
+        }
+        return tree!!
+    }
+
+    fun setTree(tree: TreeNode) {
+        this.tree = tree
+    }
+
+    fun addPackage(packageName: String) {
+        val tree = getTree()
+        val packages = packageName.split(".")
+        var node = tree
+        for (pkg in packages) {
+            val child = node.children.find { it.name == pkg }
+            if (child != null) {
+                node = child
+            } else {
+                val newNode = TreeNode(pkg)
+                node.addChild(newNode)
+                node = newNode
+            }
+        }
+    }
+
+    fun print(): String {
+        return getTree().print()
+    }
+}
+

src/main/kotlin/cc/unitmesh/devti/pair/arch/TreeNode.kt

@@ -0,0 +1,17 @@
+package cc.unitmesh.devti.pair.arch
+
+class TreeNode(val name: String, val children: MutableList<TreeNode> = mutableListOf()) {
+    fun addChild(child: TreeNode) {
+        children.add(child)
+    }
+
+    fun print(): String {
+        val sb = StringBuilder()
+        sb.append(name)
+        sb.append("\n")
+        for (child in children) {
+            sb.append(child.print())
+        }
+        return sb.toString()
+    }
+}

src/main/kotlin/cc/unitmesh/devti/provider/architecture/LayeredArchProvider.kt

@@ -0,0 +1,24 @@
+package cc.unitmesh.devti.provider.architecture
+
+import cc.unitmesh.devti.pair.arch.ProjectPackageTree
+import com.intellij.openapi.extensions.ExtensionPointName
+import com.intellij.openapi.project.Project
+
+interface LayeredArchProvider {
+    fun isApplicable(project: Project): Boolean
+    fun getLayeredArch(project: Project): ProjectPackageTree
+
+    companion object {
+
+        private val EP_NAME = ExtensionPointName<LayeredArchProvider>("cc.unitmesh.layeredArchProvider")
+
+        fun find(project: Project): LayeredArchProvider? {
+            val providers = EP_NAME.extensionList.filter {
+                it.isApplicable(project)
+            }
+
+            return providers.firstOrNull()
+        }
+    }
+}
+

src/main/kotlin/cc/unitmesh/devti/provider/builtin/LanguageContextProvider.kt

@@ -15,11 +15,7 @@ class LanguageContextProvider : ChatContextProvider {
     override suspend fun collect(project: Project, creationContext: ChatCreationContext): List<ChatContextItem> {
         val language = AutoDevSettingsState.getInstance().language
 
-        return listOf(
-            ChatContextItem(
-                LanguageContextProvider::class,
-                "You MUST Use $language to reply me!"
-            )
-        )
+        val text = "You MUST Use $language to reply me!"
+        return listOf(ChatContextItem(LanguageContextProvider::class, text))
     }
 }