opt: 优化源代码文件路径获取

Author: CiyLei

Plugin/Plugin.zip

@@ -16,6 +16,7 @@ import com.intellij.openapi.application.ApplicationManager
 import com.intellij.openapi.module.Module
 import com.intellij.openapi.project.Project
 import com.intellij.openapi.project.rootManager
+import com.intellij.openapi.roots.ModuleRootManager
 import com.intellij.openapi.vfs.newvfs.impl.VirtualDirectoryImpl
 import java.io.File
 
@@ -69,8 +70,7 @@ class ShowInputDialogAction : AnAction() {
     fun analysisApiList(module: Module, packName: String, apiList: List<ApiBean>) {
         AnalysisApiListProgressDialog(apiList, AnalysisApiListProgressDialog.AnalysisApiListProgressDialogListener {
             // 找到模块源代码存放的根目录
-            val javaSrc: VirtualDirectoryImpl? =
-                module.rootManager.getSourceRoots(false).find { it2 -> it2.name == "java" } as? VirtualDirectoryImpl
+            val javaSrc: VirtualDirectoryImpl? = ModuleRootManager.getInstance(module).sourceRoots.find { it2 -> it2.path.endsWith("src/main/java") } as? VirtualDirectoryImpl
             if (javaSrc != null) {
                 val rootDir = File(javaSrc.path)
                 if (!rootDir.exists()) {

Plugin/src/com/ciy/plugin/ShowInputDialogAction.kt

@@ -10,6 +10,7 @@
 import com.intellij.openapi.module.Module;
 import com.intellij.openapi.module.ModuleManager;
 import com.intellij.openapi.project.Project;
+import com.intellij.openapi.roots.ModuleRootManager;
 import com.intellij.openapi.vfs.VirtualFile;
 import okhttp3.*;
 import org.jetbrains.annotations.NotNull;
@@ -192,43 +193,48 @@ public void onResponse(@NotNull Call call, @NotNull Response response) throws IO
     private void readSelectApi() {
         Module selectModule = moduleList.get(cbModule.getSelectedIndex());
         if (selectModule != null) {
-            String pack = tfPack.getText().replace(".", "/");
-            VirtualFile apiServiceFile = selectModule.getModuleFile().getParent().findFileByRelativePath("src/main/java/" +
-                    pack + "/" + ShowInputDialogAction.Companion.getUrlConstantClassName() + ".kt");
-            if (apiServiceFile != null && apiServiceFile.exists()) {
-                // URLConstant 存在读取里面的url
-                try {
-                    BufferedReader br = new BufferedReader(new InputStreamReader(apiServiceFile.getInputStream()));
-                    StringBuilder sb = new StringBuilder();
-                    String len = "";
-                    while ((len = br.readLine()) != null) {
-                        sb.append(len);
-                    }
-                    String sourceText = sb.toString();
-                    if (!sourceText.isEmpty()) {
-                        Pattern pattern = Pattern.compile("\\$" + URLConstantGenerate.INSTANCE.getPrefixPropertyName() +
-                                "(.+?)\\$" + URLConstantGenerate.INSTANCE.getSuffixPropertyName());
-                        Matcher matcher = pattern.matcher(sourceText);
-                        List<String> urlList = new ArrayList<>();
-                        while (matcher.find()) {
-                            if (matcher.group(1) != null) {
-                                urlList.add(matcher.group(1));
+            VirtualFile sourcePath = getSourcePath(selectModule);
+            if (sourcePath == null) {
+                lb.setText("无法找到源代码文件");
+            } else {
+                String pack = tfPack.getText().replace(".", "/");
+                VirtualFile apiServiceFile = sourcePath.findFileByRelativePath(pack + "/" +
+                        ShowInputDialogAction.Companion.getUrlConstantClassName() + ".kt");
+                if (apiServiceFile != null && apiServiceFile.exists()) {
+                    // URLConstant 存在读取里面的url
+                    try {
+                        BufferedReader br = new BufferedReader(new InputStreamReader(apiServiceFile.getInputStream()));
+                        StringBuilder sb = new StringBuilder();
+                        String len = "";
+                        while ((len = br.readLine()) != null) {
+                            sb.append(len);
+                        }
+                        String sourceText = sb.toString();
+                        if (!sourceText.isEmpty()) {
+                            Pattern pattern = Pattern.compile("\\$" + URLConstantGenerate.INSTANCE.getPrefixPropertyName() +
+                                    "(.+?)\\$" + URLConstantGenerate.INSTANCE.getSuffixPropertyName());
+                            Matcher matcher = pattern.matcher(sourceText);
+                            List<String> urlList = new ArrayList<>();
+                            while (matcher.find()) {
+                                if (matcher.group(1) != null) {
+                                    urlList.add(matcher.group(1));
+                                }
                             }
+                            apiList.stream().forEach(it -> {
+                                it.setSelect(urlList.contains(it.getPath()));
+                            });
                         }
-                        apiList.stream().forEach(it -> {
-                            it.setSelect(urlList.contains(it.getPath()));
-                        });
+                    } catch (IOException e) {
+                        e.printStackTrace();
                     }
-                } catch (IOException e) {
-                    e.printStackTrace();
+                } else {
+                    // URLConstant 不存在则全选
+                    apiList.forEach(it -> {
+                        it.setSelect(false);
+                    });
                 }
-            } else {
-                // URLConstant 不存在则全选
-                apiList.forEach(it -> {
-                    it.setSelect(false);
-                });
+                refreshListData();
             }
-            refreshListData();
         }
     }
 
@@ -277,15 +283,38 @@ private void loadModule() {
     private void onOK() {
         if (cbModule.getSelectedIndex() < moduleList.size() && !tfPack.getText().isEmpty()) {
             Module selectModule = moduleList.get(cbModule.getSelectedIndex());
-            if (listener != null) {
-                this.listener.onOk(selectModule, tfPack.getText(), apiList.stream().filter(ApiBean::getSelect).collect(Collectors.toList()));
+            VirtualFile sourcePath = getSourcePath(selectModule);
+            if (sourcePath == null) {
+                lb.setText("无法找到源代码文件");
+            } else {
+                if (listener != null) {
+                    this.listener.onOk(selectModule, tfPack.getText(), apiList.stream().filter(ApiBean::getSelect).collect(Collectors.toList()));
+                }
+                dispose();
             }
-            dispose();
         } else {
             lb.setText("模块或者包名格式错误");
         }
     }
 
+    /**
+     * 获取模块源代码文件
+     *
+     * @param module 模块
+     * @return 源代码文件
+     */
+    private VirtualFile getSourcePath(Module module) {
+        VirtualFile[] sourceRoots = ModuleRootManager.getInstance(module).getSourceRoots();
+        if (sourceRoots != null && sourceRoots.length > 0) {
+            for (VirtualFile sourceRoot : sourceRoots) {
+                if (sourceRoot.getPath().endsWith("src/main/java")) {
+                    return sourceRoot;
+                }
+            }
+        }
+        return null;
+    }
+
     private void onCancel() {
         // add your code here if necessary
         dispose();

Plugin/src/com/ciy/plugin/ui/SelectApiDialog.java

@@ -1,6 +1,6 @@
 plugins {
     id 'java'
-    id 'org.jetbrains.kotlin.jvm' version '1.3.11'
+    id 'org.jetbrains.kotlin.jvm' version '1.3.50'
 }
 
 /*