生成Dialog，通过输入的内容生成模板代码

一、代码

  1. package action;
  3. import com.intellij.openapi.actionSystem.AnAction;
  4. import com.intellij.openapi.actionSystem.AnActionEvent;
  5. import com.intellij.openapi.actionSystem.PlatformDataKeys;
  6. import com.intellij.openapi.diagnostic.Logger;
  7. import com.intellij.openapi.project.Project;
  8. import org.codehaus.groovy.macro.transform.MacroTransformation;
  9. import org.jetbrains.annotations.NotNull;
  10. import util.AppUtils;
  11. import util.FileUtil;
  12. import util.StringUtil;
  13. import view.MvpDialog;
  15. import java.util.Map;
  18. public class SdgjMvpActivityPluginAction extends AnAction {
  19.     private Project project;
  20.     private String basePath; //选中的路径,如项目/login/src/main
  21.     private String packageName = ""; //包名
  22.     private String mIntro = "";
  23.     private String mAuthor = "";//作者
  24.     private String dirPackagename = ""; //点击目录,如 com.sdjg.mvp.page.login.pwd_login
  25.     private String mModuleName; //模块名称，如 pwd_login
  26.     private String mModuleDirName = ""; //模块目录名，如Login
  27.     private int mType; //0activity 1fragment
  28.     private static  final Logger logger= Logger.getInstance(SdgjMvpActivityPluginAction.class);
  30.     private enum CodeType {
  31.         Activity,Fragment,Layout,AndroidManifest
  32.     }
  34.     @Override
  35.     public void actionPerformed(AnActionEvent e) {
  36.         project=e.getData(PlatformDataKeys.PROJECT);
  37.         init();
  38.     }
  40.     private void init() {
  41.         MvpDialog mvpDialog = new MvpDialog(new MvpDialog.DialogCallback() {
  42.             @Override
  43.             public void ok(String moduleName, String author, String intro, int type) {
  44.                 mModuleName = moduleName;
  45.                 mAuthor = author;
  46.                 mIntro = intro;
  47.                 mType=type;
  48.                 createClassFile();
  49.             }
  50.         });
  51.         mvpDialog.setVisible(true);
  52.     }
  54.     /**
  55.      * 生成类文件
  56.      */
  57.     private void createClassFile() {
  58.         if(mType==0){
  59.             createClassFile(CodeType.Activity);
  60.             createClassFile(CodeType.AndroidManifest);
  61.         }
  62.         else{
  63.             createClassFile(CodeType.Fragment);
  64.         }
  65.         createClassFile(CodeType.Layout);
  66.         createClassFile(CodeType.ViewModel);
  67.     }
  69.     private void createClassFile(CodeType codeType) {
  70.         String content="";
  71.         switch (codeType) {
  72.             case Activity:
  73.             case Fragment:
  74.                 String type="";
  75.                 if(mType==0){
  76.                     type="Activity";
  77.                 }
  78.                 else{
  79.                     type="Fragment";
  80.                 }
  81.                 content= readTemplateFile("Template"+type+".txt");
  82.                 content = dealActivityTemplateContent(content);
  84.                 FileUtil.writeToFile(content, basePath + "/" + mModuleName, StringUtil.pathToMethodName(mModuleName) + type+ ".kt");
  85.                 break;
  86.             case Layout:
  87.                 content= readTemplateFile("TemplateLayout.txt");
  88.                 String layoutPath=StringUtil.pathExtractDirPath(basePath,"main");
  89.                 if(!layoutPath.isEmpty()){
  90.                     layoutPath+="res/layout";
  91.                 }
  93.                 String layoutContent="";
  94.                 if(mType==0) {
  95.                     layoutContent= readTemplateFile("TemplateLayoutTitle.txt");
  96.                     content=content.replace("$content$",layoutContent);
  97.                     FileUtil.writeToFile(content, layoutPath, "activity_" + mModuleName + ".xml");
  98.                 }
  99.                 else{
  100.                     content=content.replace("$content$","");
  101.                     FileUtil.writeToFile(content, layoutPath, "fragment_" + mModuleName + ".xml");
  102.                 }
  103.                 break;
  105.             case AndroidManifest:
  106.                 content= readTemplateFile("TemplateAndroidManifest.txt");
  107.                 content=dealAndroidMainifestTemplateContent(content);
  108.                 writeManifest(content,"AndroidManifest.xml");
  109.                 break;
  110.         }
  111.     }
  114.     /**
  115.      * 往androidManifest中写入新增的Android
  116.      */
  117.     private void writeManifest(String content,String path){
  118.         String mainPath=StringUtil.pathExtractDirPath(basePath,"main");
  119.         if(!mainPath.isEmpty()){
  120.             mainPath+=path;
  121.         }
  122.         if(FileUtil.isFileExists(mainPath)) {
  123.             String mainifestContent = FileUtil.readProjectFile(mainPath);
  124.             mainifestContent = mainifestContent.replace("</application>", content + "\n    </application>");
  125.             FileUtil.writeToFile(mainifestContent, mainPath, "");
  126.         }
  127.     }
  129.     /**
  130.      * 读取文件
  131.      * @param fileName
  132.      * @return
  133.      */
  134.     private String readTemplateFile(String fileName){
  135.         String content = FileUtil.readTemplateFile(this, "/Template/mvp/" + fileName);
  136.         return content;
  137.     }
  139.     private String readFunctionTemplateFile(String fileName){
  140.         String content = FileUtil.readTemplateFile(this, "/Template/mvp/functionTemplate/" + fileName);
  141.         return content;
  142.     }
  144.     //替换Activity模板中字符
  145.     private String dealActivityTemplateContent(String content){
  146.         content = content.replace("$basePackageName$", packageName);
  147.         content = content.replace("$name$", mModuleName);
  148.         String[] packageNameList = packageName.split("\\.");
  149.         String arouterName = packageNameList[packageNameList.length - 1];
  150.         content = content.replace("$Arouter$", arouterName.toUpperCase());
  151.         content=dealTemplateContent(content);
  152.         return content;
  153.     }
  155.     //替换AndroidManifest模板中字符
  156.     private String dealAndroidMainifestTemplateContent(String content){
  157.         content = content.replace("$fileName$", StringUtil.pathToMethodName(mModuleName));
  158.         String manPath=StringUtil.pathExtractDirPathLater(basePath,"page");
  159.         manPath="."+ manPath.replace("/",".");
  160.         content=content.replace("$package$",manPath);
  161.         return content;
  162.     }
  164.     //默认替换模板中字符
  165.     private String dealTemplateContent(String content) {
  166.         content = content.replace("$packagename$", dirPackagename);
  167.         content = content.replace("$intro$", mIntro);
  168.         content = content.replace("$Author$", mAuthor);
  169.         content = content.replace("$fileName$", StringUtil.pathToMethodName(mModuleName));
  170.         return content;
  171.     }
  174.     /**
  175.      * update用于判断该插件是否能使用
  176.      * 在actionPerformed前执行，这里判断必须是main路径才能使用
  177.      *
  178.      * @param e
  179.      */
  180.     @Override
  181.     public void update(@NotNull AnActionEvent e) {
  182.         basePath = AppUtils.getSelectPath(e);
  183.         String moduleStr = AppUtils.getModulePath(basePath);
  184.         if (moduleStr.isEmpty()) {
  185.             e.getPresentation().setEnabled(false);
  186.         } else if (!AppUtils.isDirExist(basePath,"page")) {
  187.             e.getPresentation().setEnabled(false);
  188.         } else {
  189.             packageName = AppUtils.getPackageName(moduleStr);
  190.             dirPackagename = AppUtils.packageSuffix(basePath, packageName);
  191.             String[] packs = packageName.split("\\.");
  192.             mModuleDirName = packs[packs.length - 1];
  193.             mModuleDirName = StringUtil.firstToUpperCase(mModuleDirName);
  194.         }
  195.         super.update(e);
  196.     }
  197. }

其实就是获取路径，通过文件读写的方式写进去，如果已经存在，则使用$标记$的方式来把它替换掉，即可实现自动生成代码的方式。
这里只是简单的代码生成，还可以自动生成Retrofit+okhtt的接口代码，然后界面生成ViewModel并自动引入，自动调用的方式减少代码工作量。

二、配套的工具类

FileUtil.java
StringUtil.java
SystemUtil.java
AppUtils.java