feat: sql tab use plain language format

linyimin0812 · linyimin0812 · commit a70a3294b434 · 2022-02-13T03:14:42.000+08:00
diff --git a/src/main/java/io/github/linyimin/plugin/service/SqlParamGenerateService.java b/src/main/java/io/github/linyimin/plugin/service/SqlParamGenerateService.java
@@ -50,7 +50,7 @@ public String generateSql(Project project, String methodQualifiedName, String pa
             MybatisPojoCompile.compile(project);
         }
 
-        return MybatisSqlUtils.getSql(mybatisConfig, methodQualifiedName, params);
+        return MybatisSqlUtils.getSql(mybatisConfig, methodQualifiedName, params, false);
 
     }
 
@@ -60,7 +60,7 @@ private boolean checkNeedCompile(String mybatisConfig, String methodQualifiedNam
         }
 
         try {
-            MybatisSqlUtils.getSql(mybatisConfig, methodQualifiedName, params);
+            MybatisSqlUtils.getSql(mybatisConfig, methodQualifiedName, params, true);
         } catch (Exception | NoClassDefFoundError e) {
             return true;
         }
diff --git a/src/main/java/io/github/linyimin/plugin/utils/MybatisSqlUtils.java b/src/main/java/io/github/linyimin/plugin/utils/MybatisSqlUtils.java
@@ -3,7 +3,9 @@
 import com.alibaba.fastjson.JSON;
 import com.alibaba.fastjson.TypeReference;
 import com.github.vertical_blank.sqlformatter.SqlFormatter;
+import com.intellij.openapi.ui.Messages;
 import io.github.linyimin.plugin.compile.MybatisPojoCompile;
+import io.github.linyimin.plugin.dom.Constant;
 import org.apache.commons.collections.CollectionUtils;
 import org.apache.commons.io.IOUtils;
 import org.apache.ibatis.io.Resources;
@@ -28,17 +30,24 @@
  **/
 public class MybatisSqlUtils {
 
-    public static String getSql(String mybatisConfiguration, String qualifiedMethod, String params) {
+    public static String getSql(String mybatisConfiguration, String qualifiedMethod, String params, boolean isCheck) {
 
-        InputStream in = IOUtils.toInputStream(mybatisConfiguration, Charset.defaultCharset());
-        Resources.setDefaultClassLoader(MybatisPojoCompile.classLoader);
+        try {
+            InputStream in = IOUtils.toInputStream(mybatisConfiguration, Charset.defaultCharset());
+            Resources.setDefaultClassLoader(MybatisPojoCompile.classLoader);
 
-        SqlSessionFactory sqlSessionFactory = new SqlSessionFactoryBuilder().build(in);
-        Configuration configuration = sqlSessionFactory.getConfiguration();
+            SqlSessionFactory sqlSessionFactory = new SqlSessionFactoryBuilder().build(in);
+            Configuration configuration = sqlSessionFactory.getConfiguration();
 
-        BoundSql sql = getBoundSql(configuration, qualifiedMethod, params);
+            BoundSql sql = getBoundSql(configuration, qualifiedMethod, params);
 
-        return formatSql(configuration, sql);
+            return formatSql(configuration, sql);
+        } catch (Exception e) {
+            if (!isCheck) {
+                Messages.showInfoMessage(e.getMessage(), Constant.APPLICATION_NAME);
+            }
+            throw e;
+        }
     }
 
     private static String formatSql(Configuration configuration, BoundSql boundSql) {
diff --git a/src/main/java/io/github/linyimin/plugin/view/JsonTextField.java b/src/main/java/io/github/linyimin/plugin/view/JsonTextField.java
@@ -8,7 +8,6 @@
 import com.intellij.openapi.editor.EditorSettings;
 import com.intellij.openapi.editor.ex.EditorEx;
 import com.intellij.openapi.project.Project;
-import com.intellij.openapi.ui.Messages;
 import com.intellij.psi.PsiDocumentManager;
 import com.intellij.psi.PsiFile;
 import com.intellij.psi.codeStyle.CodeStyleManager;
diff --git a/src/main/java/io/github/linyimin/plugin/view/MyTextField.java b/src/main/java/io/github/linyimin/plugin/view/MyTextField.java
@@ -0,0 +1,87 @@
+package io.github.linyimin.plugin.view;
+
+import com.intellij.lang.Language;
+import com.intellij.openapi.application.ReadAction;
+import com.intellij.openapi.command.WriteCommandAction;
+import com.intellij.openapi.editor.Document;
+import com.intellij.openapi.editor.EditorSettings;
+import com.intellij.openapi.editor.ex.EditorEx;
+import com.intellij.openapi.fileTypes.LanguageFileType;
+import com.intellij.openapi.project.Project;
+import com.intellij.psi.PsiDocumentManager;
+import com.intellij.psi.PsiFile;
+import com.intellij.psi.codeStyle.CodeStyleManager;
+import com.intellij.ui.ErrorStripeEditorCustomization;
+import com.intellij.ui.LanguageTextField;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+
+/**
+ * @author yiminlin
+ * @date 2022/02/13 2:11 上午
+ * @description TODO:
+ **/
+public class MyTextField extends LanguageTextField {
+    private final Project myProject;
+    private final Language language;
+    private final LanguageFileType fileType;
+
+    public MyTextField(Project myProject, Language language, LanguageFileType fileType) {
+        super(language, myProject, "", false);
+        this.myProject = myProject;
+        this.language = language;
+        this.fileType = fileType;
+    }
+
+    @Override
+    protected @NotNull EditorEx createEditor() {
+        EditorEx editor = super.createEditor();
+        setUpEditor(editor);
+        return editor;
+    }
+
+    @Override
+    public void setText(@Nullable String text) {
+        super.setFileType(fileType);
+        ReadAction.nonBlocking(() -> {
+            Document document = createDocument(text, language, myProject, new SimpleDocumentCreator());
+            setDocument(document);
+            PsiFile psiFile = PsiDocumentManager.getInstance(myProject).getPsiFile(document);
+            if (psiFile != null) {
+                WriteCommandAction.runWriteCommandAction(
+                        myProject,
+                        () -> {
+                            CodeStyleManager.getInstance(getProject()).reformat(psiFile);
+                        }
+                );
+            }
+        }).executeSynchronously();
+    }
+
+    private void setUpEditor(EditorEx editor) {
+        editor.offsetToVisualPosition(100);
+        editor.offsetToLogicalPosition(100);
+        editor.setHorizontalScrollbarVisible(true);
+        editor.setShowPlaceholderWhenFocused(true);
+        editor.setVerticalScrollbarVisible(true);
+        editor.setCaretEnabled(true);
+        editor.setEmbeddedIntoDialogWrapper(true);
+        EditorSettings settings = editor.getSettings();
+        settings.setLeadingWhitespaceShown(true);
+        settings.setTrailingWhitespaceShown(true);
+        settings.setGutterIconsShown(true);
+        settings.setSmartHome(true);
+        settings.setLineNumbersShown(true);
+        settings.setIndentGuidesShown(true);
+        settings.setUseSoftWraps(true);
+        settings.setAutoCodeFoldingEnabled(true);
+        settings.setFoldingOutlineShown(true);
+        settings.setAllowSingleLogicalLineFolding(true);
+        settings.setRightMarginShown(true);
+        settings.setCaretRowShown(true);
+        settings.setLineMarkerAreaShown(true);
+        settings.setDndEnabled(true);
+        //开启右侧的错误条纹
+        ErrorStripeEditorCustomization.ENABLED.customize(editor);
+    }
+}
diff --git a/src/main/java/io/github/linyimin/plugin/view/MybatisSqlViewerToolWindow.form b/src/main/java/io/github/linyimin/plugin/view/MybatisSqlViewerToolWindow.form
@@ -40,12 +40,13 @@
             <properties/>
             <border type="none"/>
             <children>
-              <component id="28fa2" class="javax.swing.JTextArea" binding="sql">
+              <grid id="d9725" binding="sql" custom-create="true" layout-manager="GridLayoutManager" row-count="1" column-count="1" same-size-horizontally="false" same-size-vertically="false" hgap="-1" vgap="-1">
+                <margin top="0" left="0" bottom="0" right="0"/>
                 <constraints/>
-                <properties>
-                  <editable value="false"/>
-                </properties>
-              </component>
+                <properties/>
+                <border type="none"/>
+                <children/>
+              </grid>
             </children>
           </scrollpane>
           <scrollpane id="1709">
diff --git a/src/main/java/io/github/linyimin/plugin/view/MybatisSqlViewerToolWindow.java b/src/main/java/io/github/linyimin/plugin/view/MybatisSqlViewerToolWindow.java
@@ -1,5 +1,9 @@
 package io.github.linyimin.plugin.view;
 
+import com.intellij.json.JsonFileType;
+import com.intellij.json.JsonLanguage;
+import com.intellij.openapi.fileTypes.PlainTextFileType;
+import com.intellij.openapi.fileTypes.PlainTextLanguage;
 import com.intellij.openapi.project.Project;
 import com.intellij.openapi.ui.SimpleToolWindowPanel;
 import com.intellij.openapi.wm.ToolWindow;
@@ -34,7 +38,7 @@ public class MybatisSqlViewerToolWindow extends SimpleToolWindowPanel {
     private JTextArea connectionInfoTextArea;
     private JPanel params;
     private JTextArea result;
-    private JTextArea sql;
+    private JPanel sql;
 
     private final Project myProject;
 
@@ -68,28 +72,29 @@ public void refresh(Project project) {
 
         methodName.setText(config.getMethod());
         result.setText(config.getResult());
-        ((JsonTextField) params).setText(config.getParams());
-        sql.setText(config.getSql());
+        ((MyTextField) params).setText(config.getParams());
+        ((MyTextField) sql).setText(config.getSql());
+
         // 默认每次打开，都展示第一个tab
         tabbedPane.setSelectedIndex(0);
     }
 
     private void createUIComponents() {
-        params = new JsonTextField(this.myProject);
-
+        params = new MyTextField(this.myProject, JsonLanguage.INSTANCE, JsonFileType.INSTANCE);
+        sql = new MyTextField(this.myProject, PlainTextLanguage.INSTANCE, PlainTextFileType.INSTANCE);
     }
 
     private void addComponentListener() {
         host.getDocument().addDocumentListener(new DatasourceChangeListener());
         port.getDocument().addDocumentListener(new DatasourceChangeListener());
         database.getDocument().addDocumentListener(new DatasourceChangeListener());
 
-        ((JsonTextField) params).addDocumentListener(new com.intellij.openapi.editor.event.DocumentListener() {
+        ((MyTextField) params).addDocumentListener(new com.intellij.openapi.editor.event.DocumentListener() {
             @Override
             public void documentChanged(com.intellij.openapi.editor.event.@NotNull DocumentEvent event) {
                 MybatisSqlConfiguration config = myProject.getService(MybatisSqlStateComponent.class).getState();
                 assert config != null;
-                config.setParams(((JsonTextField) params).getText());
+                config.setParams(((MyTextField) params).getText());
             }
         });
 
@@ -115,7 +120,7 @@ public void documentChanged(com.intellij.openapi.editor.event.@NotNull DocumentE
 
             // 点击sql tab时生成sql
             if (selectedIndex == TabbedComponentType.sql.index) {
-                sql.setText("Loading...");
+                ((MyTextField) sql).setText("Loading...");
                 generateSql();
             }
 
@@ -137,7 +142,7 @@ private void generateSql() {
         String sqlStr = generateService.generateSql(myProject, sqlConfig.getMethod(), sqlConfig.getParams());
         sqlConfig.setSql(sqlStr);
 
-        sql.setText(sqlStr);
+        ((MyTextField) sql).setText(sqlStr);
     }
 
     private void executeSql() {
@@ -146,7 +151,7 @@ private void executeSql() {
         String passwordText = String.valueOf(password.getPassword());
         String resultText;
         try {
-            resultText = MybatisSqlUtils.executeSql(urlText, user.getText(), passwordText, sql.getText());
+            resultText = MybatisSqlUtils.executeSql(urlText, user.getText(), passwordText, ((MyTextField) sql).getText());
         } catch (SQLException e) {
             resultText = "Execute Sql Failed. err: " + e.getMessage();
         }

Original file line number	Diff line number	Diff line change
`@@ -50,7 +50,7 @@ public String generateSql(Project project, String methodQualifiedName, String pa`
`50`	`50`	`MybatisPojoCompile.compile(project);`
`51`	`51`	`}`
`52`	`52`
`53`		`- return MybatisSqlUtils.getSql(mybatisConfig, methodQualifiedName, params);`
	`53`	`+ return MybatisSqlUtils.getSql(mybatisConfig, methodQualifiedName, params, false);`
`54`	`54`
`55`	`55`	`}`
`56`	`56`
`@@ -60,7 +60,7 @@ private boolean checkNeedCompile(String mybatisConfig, String methodQualifiedNam`
`60`	`60`	`}`
`61`	`61`
`62`	`62`	`try {`
`63`		`- MybatisSqlUtils.getSql(mybatisConfig, methodQualifiedName, params);`
	`63`	`+ MybatisSqlUtils.getSql(mybatisConfig, methodQualifiedName, params, true);`
`64`	`64`	`} catch (Exception \| NoClassDefFoundError e) {`
`65`	`65`	`return true;`
`66`	`66`	`}`