From c6e8c774877ca8233f286fcdf1144675e48973f6 Mon Sep 17 00:00:00 2001 From: makejava <1353036300@qq.com> Date: Tue, 17 Aug 2021 17:05:47 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E6=B8=85=E7=90=86=E9=85=8D?= =?UTF-8?q?=E7=BD=AE=E5=8A=9F=E8=83=BD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../com/sjhy/plugin/actions/MainActionGroup.java | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) diff --git a/src/main/java/com/sjhy/plugin/actions/MainActionGroup.java b/src/main/java/com/sjhy/plugin/actions/MainActionGroup.java index 0484204..42128fe 100644 --- a/src/main/java/com/sjhy/plugin/actions/MainActionGroup.java +++ b/src/main/java/com/sjhy/plugin/actions/MainActionGroup.java @@ -3,7 +3,10 @@ package com.sjhy.plugin.actions; import com.intellij.database.psi.DbTable; import com.intellij.openapi.actionSystem.*; import com.intellij.openapi.project.Project; +import com.intellij.openapi.ui.Messages; import com.intellij.psi.PsiElement; +import com.sjhy.plugin.dict.GlobalDict; +import com.sjhy.plugin.service.TableInfoSettingsService; import com.sjhy.plugin.tool.CacheDataUtils; import org.jetbrains.annotations.NotNull; import org.jetbrains.annotations.Nullable; @@ -109,8 +112,19 @@ public class MainActionGroup extends ActionGroup { configAction = new ConfigAction("Config Table"); actionManager.registerAction(configActionId, configAction); } + AnAction clearConfigAction = new AnAction("Clear Config") { + @Override + public void actionPerformed(@NotNull AnActionEvent e) { + DbTable dbTable = CacheDataUtils.getInstance().getSelectDbTable(); + if (dbTable == null) { + return; + } + TableInfoSettingsService.getInstance().resetTableInfo(dbTable); + Messages.showInfoMessage(dbTable.getName() + "表配置信息已重置成功", GlobalDict.TITLE_INFO); + } + }; // 返回所有菜单 - return new AnAction[]{mainAction, configAction}; + return new AnAction[]{mainAction, configAction, clearConfigAction}; }