feat: compatible with other plugins

Author: linyimin0812

build.gradle

@@ -4,7 +4,7 @@ plugins {
 }
 
 group 'org.linyimin'
-version '1.0.0'
+version '1.0.1'
 
 repositories {
     maven { url 'https://maven.aliyun.com/repository/public/' }
@@ -28,14 +28,22 @@ intellij {
 }
 patchPluginXml {
     changeNotes = """
-      <ul>
-        <li>navigate from the code to mapper and from the mapper back to code</li>
-        <li>The xml statement/method in the mapper file convert to sql statement, which supports parameters mock, data source configuration, and sql execution</li>
-    </ul>
-    <ul>
-        <li>快速从代码跳转到mapper及从mapper返回代码</li>
-        <li>mapper文件中的xml语句/方法转换成sql语句，支持参数mock、数据源配置、sql执行</li>
-    </ul>
+        <h4>1.0.1</h4>
+        <ul>
+            <li>Compatible with other plugins</li>
+        </ul>
+        <ul>
+            <li>兼容其他插件</li>
+        </ul>
+        <h4>1.0.0</h4>
+        <ul>
+            <li>navigate from the code to mapper and from the mapper back to code</li>
+            <li>The xml statement/method in the mapper file convert to sql statement, which supports parameters mock, data source configuration, and sql execution</li>
+        </ul>
+        <ul>
+            <li>快速从代码跳转到mapper及从mapper返回代码</li>
+            <li>mapper文件中的xml语句/方法转换成sql语句，支持参数mock、数据源配置、sql执行</li>
+        </ul>
     """
 }
 

src/main/java/io/github/linyimin/plugin/cache/MybatisXmlContentCache.java

@@ -0,0 +1,183 @@
+package io.github.linyimin.plugin.cache;
+
+import com.google.common.collect.Lists;
+import com.intellij.openapi.project.Project;
+import com.intellij.openapi.roots.ProjectRootManager;
+import com.intellij.openapi.vfs.VirtualFile;
+import com.intellij.psi.PsiFile;
+import com.intellij.psi.PsiFileSystemItem;
+import com.intellij.psi.util.PsiUtil;
+import com.intellij.psi.xml.XmlAttribute;
+import com.intellij.psi.xml.XmlFile;
+import com.intellij.psi.xml.XmlTag;
+import io.github.linyimin.plugin.utils.MapperDomUtils;
+
+import java.util.*;
+
+/**
+ * @author banzhe
+ * @date 2022/08/10 23:28
+ **/
+public class MybatisXmlContentCache {
+
+    private static final List<String> SUB_TAGS = Lists.newArrayList("insert", "update", "delete", "select");
+
+    private static final Map<Project, Map<String /* path */, String /* configuration */>> projectMybatisConfigurationMap = new HashMap<>();
+
+    private static final Map<Project, Map<String /* namespace */, List<String> /* method name list */>> projectMybatisMapperMap = new HashMap<>();
+
+    private static final Map<Project, Map<String /* namespace */, List<XmlTag>>> projectMapperNamespaceMap = new HashMap<>();
+
+    private static final Map<Project, Map<String /* method qualified name */, List<XmlTag>>> projectMapperMethodMap = new HashMap<>();
+
+
+    public static List<String> acquireConfigurations(Project project) {
+
+        addXmlCache(project);
+
+        Map<String, String> cacheMap = projectMybatisConfigurationMap.getOrDefault(project, Collections.emptyMap());
+
+        return new ArrayList<>(cacheMap.values());
+    }
+
+    public static List<String> acquireByNamespace(Project project) {
+
+        addXmlCache(project);
+
+        Set<String> namespaces = projectMybatisMapperMap.getOrDefault(project, new HashMap<>()).keySet();
+        return new ArrayList<>(namespaces);
+    }
+
+    public static List<XmlTag> acquireByNamespace(Project project, String namespace) {
+
+        addXmlCache(project);
+
+        Map<String /* namespace */, List<XmlTag>> cache = projectMapperNamespaceMap.getOrDefault(project, new HashMap<>());
+
+        return cache.getOrDefault(namespace, new ArrayList<>());
+    }
+
+    public static List<XmlTag> acquireByMethodName(Project project, String methodQualifiedName) {
+
+        addXmlCache(project);
+
+        Map<String /* namespace */, List<XmlTag>> cache = projectMapperMethodMap.getOrDefault(project, new HashMap<>());
+
+        return cache.getOrDefault(methodQualifiedName, new ArrayList<>());
+    }
+
+    private static void addXmlCache(Project project) {
+
+        ProjectRootManager.getInstance(project).getFileIndex().iterateContent(fileOrDir -> {
+
+            PsiFileSystemItem item = PsiUtil.findFileSystemItem(project, fileOrDir);
+
+            if (item == null) {
+                return true;
+            }
+
+            PsiFile psiFile = item.getContainingFile();
+
+            if (MapperDomUtils.isMybatisConfigurationFile(psiFile)) {
+                addConfigurationCache(project, fileOrDir, psiFile);
+            }
+
+            if (MapperDomUtils.isMybatisMapperFile(psiFile)) {
+                addMapperCache(project, psiFile);
+            }
+
+            return true;
+        });
+    }
+
+    private static void addMapperCache(Project project, PsiFile psiFile) {
+
+        XmlTag rootTag = ((XmlFile) psiFile).getRootTag();
+
+        if (rootTag == null || rootTag.getAttribute("namespace") == null) {
+            return;
+        }
+
+
+        XmlAttribute namespaceAttribute = rootTag.getAttribute("namespace");
+
+        if (namespaceAttribute == null) {
+            return;
+        }
+
+        String namespace = namespaceAttribute.getValue();
+
+        addNamespaceXmlTagCache(project, namespace, rootTag);
+
+        XmlTag[] subTags = rootTag.getSubTags();
+
+        for (XmlTag subTag : subTags) {
+            if (!SUB_TAGS.contains(subTag.getName())) {
+                continue;
+            }
+
+            XmlAttribute subAttribute = subTag.getAttribute("id");
+
+            if (subAttribute == null) {
+                continue;
+            }
+
+            String id = subAttribute.getValue();
+
+            addMethodXmlTagCache(project, namespace, id, subTag);
+
+            addNamespaceCache(project, namespace, id);
+
+        }
+
+    }
+
+    private static void addNamespaceXmlTagCache(Project project, String namespace, XmlTag xmlTag) {
+
+        Map<String, List<XmlTag>> namespaceCacheMap = projectMapperNamespaceMap.getOrDefault(project, new HashMap<>());
+
+        List<XmlTag> tags = namespaceCacheMap.getOrDefault(namespace, new ArrayList<>());
+        tags.add(xmlTag);
+
+        namespaceCacheMap.put(namespace, tags);
+
+        projectMapperNamespaceMap.put(project, namespaceCacheMap);
+
+    }
+
+    private static void addMethodXmlTagCache(Project project, String namespace, String id, XmlTag xmlTag) {
+
+        Map<String, List<XmlTag>> methodCacheMap = projectMapperMethodMap.getOrDefault(project, new HashMap<>());
+
+        String methodQualifiedName = namespace + "." + id;
+
+        List<XmlTag> tags = methodCacheMap.getOrDefault(methodQualifiedName, new ArrayList<>());
+        tags.add(xmlTag);
+
+        methodCacheMap.put(methodQualifiedName, tags);
+
+        projectMapperMethodMap.put(project, methodCacheMap);
+    }
+
+    private static void addNamespaceCache(Project project, String namespace, String id) {
+
+        Map<String, List<String>> cacheMap = projectMybatisMapperMap.getOrDefault(project, new HashMap<>());
+
+        List<String> ids = cacheMap.getOrDefault(namespace, new ArrayList<>());
+        ids.add(id);
+
+        cacheMap.put(namespace, ids);
+
+        projectMybatisMapperMap.put(project, cacheMap);
+    }
+
+    private static void addConfigurationCache(Project project, VirtualFile fileOrDir, PsiFile psiFile) {
+        Map<String, String> cacheMap = projectMybatisConfigurationMap.getOrDefault(project, new HashMap<>());
+
+        String path = fileOrDir.getPath();
+        String configuration = psiFile.getText();
+
+        cacheMap.put(path, configuration);
+        projectMybatisConfigurationMap.put(project, cacheMap);
+    }
+}