chore: refactor

MC-XiaoHei · Jun 2, 2024 · 2e149ca · 2e149ca
1 parent 226db97
commit 2e149ca
Show file tree

Hide file tree

Showing 3 changed files with 36 additions and 23 deletions.
diff --git a/src/main/kotlin/cn/xor7/xiaohei/leavesknife/activities/ProjectStartupActivity.kt b/src/main/kotlin/cn/xor7/xiaohei/leavesknife/activities/ProjectStartupActivity.kt
@@ -1,13 +1,7 @@
 package cn.xor7.xiaohei.leavesknife.activities
 
-import cn.xor7.xiaohei.leavesknife.CommonBundle
 import cn.xor7.xiaohei.leavesknife.services.LEAVESKNIFE_CONFIG_FILE
 import cn.xor7.xiaohei.leavesknife.services.leavesknifeConfigService
-import com.intellij.notification.Notification
-import com.intellij.notification.NotificationAction
-import com.intellij.notification.NotificationGroupManager
-import com.intellij.notification.NotificationType
-import com.intellij.openapi.actionSystem.AnActionEvent
 import com.intellij.openapi.project.Project
 import com.intellij.openapi.project.guessProjectDir
 import com.intellij.openapi.startup.ProjectActivity
@@ -28,29 +22,15 @@ class ProjectStartupActivity : ProjectActivity {
                 )
             ) {
                 project.leavesknifeConfigService.enablePlugin = true
+                // TODO 检查配置文件合法性
                 return@let
             }
             GradleConnector.newConnector()
                 .forProjectDirectory(File(it.path))
                 .connect().use { connection ->
                     val ideaProject: IdeaProject = connection.getModel(IdeaProject::class.java)
                     if (ideaProject.modules.any { it.name == "paper-api-generator" }) {
-                        @Suppress("DialogTitleCapitalization")
-                        NotificationGroupManager.getInstance()
-                            .getNotificationGroup("LeavesKnife")
-                            .createNotification(
-                                CommonBundle.message("notification.configure.title"),
-                                NotificationType.INFORMATION
-                            )
-                            .addAction(object : NotificationAction(
-                                CommonBundle.message("notification.configure.action")
-                            ) {
-                                override fun actionPerformed(e: AnActionEvent, notification: Notification) {
-                                    project.leavesknifeConfigService.enablePlugin = true
-                                    notification.hideBalloon()
-                                }
-                            })
-                            .notify(project)
+                        project.leavesknifeConfigService.needConfigure = true
                         return@let
                     }
                 }

diff --git a/src/main/kotlin/cn/xor7/xiaohei/leavesknife/dialogs/PluginConfigurationDialog.kt b/src/main/kotlin/cn/xor7/xiaohei/leavesknife/dialogs/PluginConfigurationDialog.kt
@@ -0,0 +1,4 @@
+package cn.xor7.xiaohei.leavesknife.dialogs
+
+class PluginConfigurationDialog {
+}
diff --git a/src/main/kotlin/cn/xor7/xiaohei/leavesknife/services/ProjectConfigService.kt b/src/main/kotlin/cn/xor7/xiaohei/leavesknife/services/ProjectConfigService.kt
@@ -1,5 +1,11 @@
 package cn.xor7.xiaohei.leavesknife.services
 
+import cn.xor7.xiaohei.leavesknife.CommonBundle
+import com.intellij.notification.Notification
+import com.intellij.notification.NotificationAction
+import com.intellij.notification.NotificationGroupManager
+import com.intellij.notification.NotificationType
+import com.intellij.openapi.actionSystem.AnActionEvent
 import com.intellij.openapi.application.runInEdt
 import com.intellij.openapi.components.Service
 import com.intellij.openapi.project.Project
@@ -19,7 +25,30 @@ class ProjectConfigService(private val project: Project) {
                     ?.isAvailable = value
             }
         }
+    var needConfigure = false
+        set(value) {
+            if (field == value) return
+            field = value
+            if (value) {
+                @Suppress("DialogTitleCapitalization")
+                NotificationGroupManager.getInstance()
+                    .getNotificationGroup("LeavesKnife")
+                    .createNotification(
+                        CommonBundle.message("notification.configure.title"),
+                        NotificationType.INFORMATION
+                    )
+                    .addAction(object : NotificationAction(
+                        CommonBundle.message("notification.configure.action")
+                    ) {
+                        override fun actionPerformed(e: AnActionEvent, notification: Notification) {
+                            project.leavesknifeConfigService.enablePlugin = true
+                            notification.hideBalloon()
+                        }
+                    })
+                    .notify(project)
+            }
+        }
 }
 
 val Project.leavesknifeConfigService: ProjectConfigService
-    get() = this.getService(ProjectConfigService::class.java)
+    get() = this.getService(ProjectConfigService::class.java)