初始化项目

Author: lk

build.gradle

@@ -45,10 +45,6 @@ repositories {
     maven { url "https://www.jetbrains.com/intellij-repository/releases" }
 }
 
-repositories {
-    mavenCentral()
-}
-
 dependencies {
     testCompile group: 'junit', name: 'junit', version: '4.12'
 }

builds/plugin.intellij.assistant.mybaitslog-1.0.0.jar

@@ -0,0 +1,31 @@
+package com.plugins.mybaitslog;
+
+import com.intellij.openapi.project.Project;
+import com.intellij.openapi.wm.ToolWindow;
+import com.intellij.openapi.wm.ToolWindowFactory;
+import com.intellij.ui.content.Content;
+import com.intellij.ui.content.ContentFactory;
+import com.plugins.mybaitslog.console.ConsolePanel;
+import com.plugins.mybaitslog.icons.Icons;
+import org.jetbrains.annotations.NotNull;
+
+import javax.swing.*;
+
+/**
+ * @author lk
+ * @version 1.0
+ * @date 2020/4/10 21:46
+ */
+public class MyBatisLogToolWindow implements ToolWindowFactory {
+
+
+    @Override
+    public void createToolWindowContent(@NotNull Project project, @NotNull ToolWindow toolWindow) {
+        ConsolePanel consolePanel = new ConsolePanel();
+        final JComponent jComponent = consolePanel.ConsolePanel(project);
+        ContentFactory contentFactory = ContentFactory.SERVICE.getInstance();
+        Content content = contentFactory.createContent(jComponent, "", false);
+        toolWindow.setIcon(Icons.MyBatisIcon);
+        toolWindow.getContentManager().addContent(content);
+    }
+}

image/2020-03-25_09-28-47.jpg

@@ -11,7 +11,6 @@
 import com.intellij.openapi.wm.ToolWindowManager;
 import com.intellij.ui.JBColor;
 import com.plugins.mybaitslog.icons.Icons;
-import com.plugins.mybaitslog.console.ConsoleRunExecutor;
 import com.plugins.mybaitslog.util.ConfigUtil;
 import com.plugins.mybaitslog.util.PrintUtil;
 import com.plugins.mybaitslog.util.RestoreSqlUtil;
@@ -44,15 +43,14 @@ public void actionPerformed(AnActionEvent e) {
         CaretModel caretModel = e.getData(LangDataKeys.EDITOR).getCaretModel();
         Caret currentCaret = caretModel.getCurrentCaret();
         String sqlText = currentCaret.getSelectedText();
-        ToolWindow toolWindow = ToolWindowManager.getInstance(project).getToolWindow(ConsoleRunExecutor.TOOLWINDOWS_ID);
-        if (!ConfigUtil.active || !toolWindow.isAvailable()) {
-            new ShowLogInConsoleAction(project).showLogInConsole(project);
+        ToolWindow toolWindow = ToolWindowManager.getInstance(project).getToolWindow("MyBatis Log");
+        if (!toolWindow.isAvailable()) {
+            //激活Restore Sql tab
+            toolWindow.activate(null);
         }
-        //激活Restore Sql tab
-        toolWindow.activate(null);
         final String preparing = ConfigUtil.getPreparing(project);
         final String parameters = ConfigUtil.getParameters(project);
-        if(StringUtils.isNotEmpty(sqlText)){
+        if (StringUtils.isNotEmpty(sqlText)) {
             String[] sqlArr = sqlText.split("\n");
             if (isSelectedText(project, sqlText, sqlArr, preparing, parameters)) {
                 setSelectedTextFormat(project, sqlArr, preparing, parameters);

src/main/java/com/plugins/mybaitslog/MyBatisLogStartupActivity.java

@@ -0,0 +1,38 @@
+package com.plugins.mybaitslog.console;
+
+import com.intellij.icons.AllIcons;
+import com.intellij.openapi.actionSystem.AnAction;
+import com.intellij.openapi.actionSystem.AnActionEvent;
+import com.intellij.openapi.project.DumbAware;
+
+/**
+ * @author lk
+ * @version 1.0
+ * @date 2020/4/10 22:00
+ */
+public class ConsoleActionGroup {
+
+    private static Runnable myFilterAction;
+
+    public static void withFilter(Runnable filter) {
+        ConsoleActionGroup.myFilterAction = filter;
+    }
+
+    public static Runnable getMyFilterAction() {
+        return ConsoleActionGroup.myFilterAction;
+    }
+
+    /**
+     * 过滤按钮
+     */
+    public static class FilterAction extends AnAction implements DumbAware {
+        public FilterAction() {
+            super("Filter", "Filter", AllIcons.General.Filter);
+        }
+
+        @Override
+        public void actionPerformed(AnActionEvent e) {
+            ConsoleActionGroup.myFilterAction.run();
+        }
+    }
+}

src/main/java/com/plugins/mybaitslog/MyBatisLogToolWindow.java

@@ -0,0 +1,98 @@
+package com.plugins.mybaitslog.console;
+
+import com.intellij.execution.filters.Filter;
+import com.intellij.execution.filters.TextConsoleBuilder;
+import com.intellij.execution.filters.TextConsoleBuilderFactory;
+import com.intellij.execution.ui.ConsoleView;
+import com.intellij.openapi.actionSystem.ActionManager;
+import com.intellij.openapi.actionSystem.ActionToolbar;
+import com.intellij.openapi.actionSystem.AnAction;
+import com.intellij.openapi.actionSystem.DefaultActionGroup;
+import com.intellij.openapi.project.Project;
+import com.intellij.openapi.ui.SimpleToolWindowPanel;
+import com.plugins.mybaitslog.action.gui.FilterSetting;
+import com.plugins.mybaitslog.util.PrintUtil;
+import org.jetbrains.annotations.NotNull;
+
+import javax.swing.*;
+import java.awt.*;
+import java.util.ArrayList;
+import java.util.List;
+
+/**
+ * console 描述
+ */
+public class ConsolePanel {
+
+    private final List<Filter> myFilterList = new ArrayList<>();
+
+    /**
+     * 创建 输出面板
+     *
+     * @param project
+     * @return
+     */
+    private ConsoleView createConsole(@NotNull Project project) {
+        TextConsoleBuilder consoleBuilder = TextConsoleBuilderFactory.getInstance().createBuilder(project);
+        consoleBuilder.filters(myFilterList);
+        ConsoleView console = consoleBuilder.getConsole();
+        PrintUtil.setConsoleView(console);
+        return console;
+    }
+
+
+    private static JComponent createConsolePanel(ConsoleView view) {
+        JPanel panel = new JPanel();
+        panel.setLayout(new BorderLayout());
+        panel.add(view.getComponent(), BorderLayout.CENTER);
+        return panel;
+    }
+
+
+    /**
+     * 构建面板
+     */
+    public JComponent ConsolePanel(final Project myProject) {
+        final ConsoleView consoleView = createConsole(myProject);
+        //左边显示的控件
+        final JComponent consolePanel = createConsolePanel(consoleView);
+        final ActionToolbar actionToolbar = createActionToolbar(myProject, consolePanel, consoleView);
+        actionToolbar.setTargetComponent(consolePanel);
+
+        SimpleToolWindowPanel panel = new SimpleToolWindowPanel(false, true);
+        panel.setContent(consolePanel);
+        panel.setToolbar(actionToolbar.getComponent());
+
+        DefaultActionGroup actions = new DefaultActionGroup();
+        for (AnAction action : consoleView.createConsoleActions()) {
+            actions.add(action);
+        }
+        final JComponent component = panel.getComponent();
+        return component;
+    }
+
+    /**
+     * 创建工具栏
+     *
+     * @param consolePanel
+     * @param consoleView
+     * @return
+     */
+    @NotNull
+    private ActionToolbar createActionToolbar(final Project project, JComponent consolePanel, ConsoleView consoleView) {
+        final DefaultActionGroup actionGroup = new DefaultActionGroup();
+        ConsoleActionGroup.withFilter(() -> {
+            //启动filter配置
+            FilterSetting dialog = new FilterSetting(project);
+            dialog.pack();
+            dialog.setSize(600, 400);//配置大小
+            dialog.setLocationRelativeTo(null);//位置居中显示
+            dialog.setVisible(true);
+        });
+        actionGroup.add(new ConsoleActionGroup.FilterAction());
+        actionGroup.add(consoleView.createConsoleActions()[2]);
+        actionGroup.add(consoleView.createConsoleActions()[3]);
+        actionGroup.add(consoleView.createConsoleActions()[5]);
+        return ActionManager.getInstance().createActionToolbar("EventLog", actionGroup, false);
+    }
+}