feat(project setup): K2适配，优化项目配置流程及模块选择

Author: EmptyIrony

gradle.properties

@@ -1,6 +1,6 @@
 kotlin.stdlib.default.dependency=false
 pluginSinceBuild=243
-version=1.31-SNAPSHOT
+version=1.4-SNAPSHOT
 kotlin.code.style=official
 
 # 启用 Kotlin 增量编译以提高构建性能

src/main/kotlin/org/tabooproject/development/ProjectBuilder.kt

@@ -59,8 +59,35 @@ class ProjectBuilder : AbstractNewProjectWizardBuilder() {
     }
 
     override fun cleanup() {
+        // 在清理前保存当前配置为默认设置
+        saveCurrentConfigurationAsDefaults()
+        
         super.cleanup()
         ConfigurationPropertiesStep.refreshTemporaryData()
         OptionalPropertiesStep.refreshTemporaryData()
     }
+
+    /**
+     * 保存当前配置为默认设置
+     */
+    private fun saveCurrentConfigurationAsDefaults() {
+        val settings = org.tabooproject.development.settings.TabooLibProjectSettings.getInstance()
+        val configProperty = ConfigurationPropertiesStep.property
+        val optionalProperty = OptionalPropertiesStep.property
+        
+        // 提取包名前缀
+        val packagePrefix = if (configProperty.mainClass.contains(".")) {
+            configProperty.mainClass.substringBeforeLast(".")
+                .substringBeforeLast(".") // 获取包名前缀，去掉最后两级
+        } else {
+            "org.example" // 默认值
+        }
+        
+        settings.saveAsDefaults(
+            packagePrefix = packagePrefix,
+            author = optionalProperty.authors.firstOrNull() ?: "",
+            selectedModules = configProperty.modules,
+            templateMirror = configProperty.mirrorIndex
+        )
+    }
 }

src/main/kotlin/org/tabooproject/development/component/AddDeleteStringListPanel.kt

@@ -11,22 +11,35 @@ import com.intellij.ui.AddDeleteListPanel
  */
 class AddDeleteStringListPanel(
     title: String,
-    initial: List<String>,
+    private val initialData: MutableList<String>,
     defaultHeight: Int = 175,
-) : AddDeleteListPanel<String>(title, initial), Disposable {
+) : AddDeleteListPanel<String>(title, initialData.toList()), Disposable {
 
     init {
         preferredSize = preferredSize.apply {
             width += 300
             height += defaultHeight
         }
+        
+        // 初始化时加载已有数据
+        refreshData()
     }
 
     override fun findItemToAdd(): String? {
         // 简单实现，可以扩展为输入对话框
         return null
     }
 
+    /**
+     * 刷新面板数据以反映initialData的变化
+     */
+    fun refreshData() {
+        myListModel.clear()
+        initialData.forEach { item ->
+            myListModel.addElement(item)
+        }
+    }
+
     fun export(): List<String> = listItems.map { it as String }
 
     /**

src/main/kotlin/org/tabooproject/development/component/CheckModuleList.kt

@@ -21,7 +21,7 @@ class CheckModuleList : JScrollPane(), Disposable {
         isFocusable = false
     }
 
-    private val treeNode = TreeModel(root)
+    private val treeNode: TreeModel? = TreeModel(root)
     var onModuleSelectionChanged: ((List<Module>) -> Unit)? = null
 
     private val checkBoxList = CheckboxTreeBase(object : CheckboxTreeCellRenderer() {
@@ -82,22 +82,89 @@ class CheckModuleList : JScrollPane(), Disposable {
         setFocusable(false)
         autoscrolls = true
         setViewportView(checkBoxList)
+
+        updateUI()
     }
 
     /**
      * 设置模块数据
      */
     fun setModules(modules: Map<String, List<Module>>) {
         root.removeAllChildren()
-        modules.forEach { (category, moduleList) ->
-            val categoryNode = CheckedTreeNode(category)
-            moduleList.forEach { module ->
-                val moduleNode = CheckedTreeNode(module)
-                categoryNode.add(moduleNode)
+        modules.map {
+            DefaultMutableTreeNode(it.key).apply {
+                it.value.forEach {
+                    add(CheckedTreeNode(it).apply {
+                        isChecked = false
+                        isFocusable = false
+                    })
+                }
+                isFocusable = false
             }
-            root.add(categoryNode)
+        }.forEach {
+            root.add(it)
+        }
+
+        updateUI()
+    }
+
+    /**
+     * 设置指定模块的选中状态
+     */
+    fun setModuleSelected(moduleId: String, selected: Boolean) {
+        // 遍历树节点，找到对应的模块并设置选中状态
+        findAndSetModuleSelection(root, moduleId, selected)
+        treeNode?.reload()
+    }
+
+    /**
+     * 批量设置模块选中状态
+     */
+    fun setSelectedModules(moduleIds: List<String>) {
+        // 先清除所有选择
+        clearAllSelections(root)
+        
+        // 然后设置指定的模块为选中
+        moduleIds.forEach { moduleId ->
+            findAndSetModuleSelection(root, moduleId, true)
+        }
+        treeNode?.reload()
+        
+        // 通知选择变更
+        onModuleSelectionChanged?.invoke(export())
+    }
+
+    /**
+     * 递归查找并设置模块选中状态
+     */
+    private fun findAndSetModuleSelection(node: CheckedTreeNode, moduleId: String, selected: Boolean) {
+        // 检查当前节点
+        if (node.userObject is Module && (node.userObject as Module).id == moduleId) {
+            node.isChecked = selected
+            return
+        }
+        
+        // 递归检查子节点
+        for (i in 0 until node.childCount) {
+            val child = node.getChildAt(i) as? DefaultMutableTreeNode ?: continue
+            for (j in 0 until child.childCount) {
+                val treeNode = child.getChildAt(j) as? CheckedTreeNode ?: continue
+                if ((treeNode.userObject as? Module)?.id == moduleId) {
+                    treeNode.isChecked = selected
+                }
+            }
+        }
+    }
+
+    /**
+     * 清除所有选择
+     */
+    private fun clearAllSelections(node: CheckedTreeNode) {
+        node.isChecked = false
+        for (i in 0 until node.childCount) {
+            val child = node.getChildAt(i) as? CheckedTreeNode ?: continue
+            clearAllSelections(child)
         }
-        treeNode.reload()
     }
 
     /**
@@ -114,6 +181,7 @@ class CheckModuleList : JScrollPane(), Disposable {
      */
     override fun updateUI() {
         super.updateUI()
+        treeNode?.reload()
     }
 
     override fun dispose() {

src/main/kotlin/org/tabooproject/development/component/CheckModulePanel.kt

@@ -1,6 +1,7 @@
 package org.tabooproject.development.component
 
 import com.intellij.openapi.Disposable
+import com.intellij.openapi.observable.util.whenSizeChanged
 import com.intellij.openapi.util.Disposer
 import com.intellij.ui.components.JBList
 import com.intellij.ui.components.JBPanel
@@ -19,12 +20,11 @@ import javax.swing.JPanel
  * @since 1.31
  */
 class CheckModulePanel(
-    displayModuleList: DisplayModuleList = DisplayModuleList()
+    private val displayModuleList: DisplayModuleList = DisplayModuleList()
 ) : JBPanel<CheckModulePanel>(), Disposable {
 
     private val checkModuleList = CheckModuleList()
-    private val displayModuleList = displayModuleList
-    
+
     private val checkModuleScrollPane = JBScrollPane(checkModuleList)
     private val displayModuleScrollPane = JBScrollPane(displayModuleList)
 
@@ -68,6 +68,15 @@ class CheckModulePanel(
      */
     fun setModules(modules: Map<String, List<Module>>) {
         checkModuleList.setModules(modules)
+
+        checkModuleScrollPane.size = checkModuleScrollPane.preferredSize
+    }
+
+    /**
+     * 设置选中的模块
+     */
+    fun setSelectedModules(moduleIds: List<String>) {
+        checkModuleList.setSelectedModules(moduleIds)
     }
 
     /**