IDEA 插件开发实战

 关注
发布于: 2020 年 05 月 26 日
一. 简介IntelliJ IDEA是一款开发工具，提供很多插件功能，比如阿里规范插件（Alibaba Java Coding Guidelines），但是随着日常业务展开，很多工作重复性编码，浪费很多时间，需要自定义抽象出来一些插件，自动化的方式解决问题，这也是工程师文化的体现。
二.原理2.1 背景IntelliJ平台是开源的，基于Apache许可协议，提供很多丰富的工具，提供组件驱动，基于跨平台JVM，可以在创建菜单栏、列表、弹出菜单、对话框等等。可以适用于多种语言，提供相关解析器和PSI模型，解析文件，构建语义模型。
2.2 基本原理组件模型
负责生命周期管理以及连接组件之间的相互依赖关系。
Application level components，在IDEA启动的时候创建和初始化，可以使用 getComponent(Class) 获取它们。
Project level components，在IDEA中每个Project实例创建的，甚至可以为未打开的项目创建组件，可以使用 getComponent(Class)方法从Project实例中获取它们。
Module level components，它们是为IDEA中加载的每个项目中每个模块创建，使用getComponent(Class)方法可以从Module实例获取模块级别组件。
 
生命周期：
创建，调用构造函数
初始化，initComponent调用该方法（如果组件实现ApplicationComponent接口）
配置，保存和加载每个组件的状态。（PersistentStateComponent和JDOMExternalizable，实例化配置）。
注册，对于模块组件，将调用接口的moduleAdded方法ModuleComponent将模块添加到项目中，对于项目组件，调用接口的projectOpened方法ProjectComponent加载项目。
保存配置，JDOMExternalizable，PersistentStateComponent的调用。
输出，disposeComponent调用输出。
 
线程模型
平台相关数据结构由读/写锁覆盖，适用于PSI，VFS 和项目模型。允许从任何线程读取数据。从UI线程读取数据不需要任何特殊的工作。但是，从任何其他线程执行的读取操作都需要使用ApplicationManager.getApplication().runReadAction()或ReadAction.run/compute。
仅允许从UI线程写入数据，并且写入操作始终需要用ApplicationManager.getApplication().runWriteAction()或WriteAction.run()/compute()。
 
后台流程管理
后台进度由ProgressManager类管理，该类有很多方法可以使用模式（对话框），非模式（在状态栏中可见）或不可见进度来执行给定代码。在所有情况下，代码都是在与ProgressIndicator对象关联的后台线程上执行的。
 
讯息传递
平台中可用的消息传递基础结构，基于 Observer设计模式扩展实现的，通过该模式能够更好的梳理的一对多关系，实现提供了附加功能，例如在层次结构上进行广播和特殊的嵌套事件处理（此处的嵌套事件是指从另一个事件的回调中（直接或间接）触发新事件的情况）。
 
2.3 小结具体相关原理研究，可查看官网（http://www.jetbrains.org/intellij/sdk/docs/welcome.html）。
三.api3.1 框架结构.IntelliJIDEA/
└── plugins
    └── code_plugin
        └── lib
            ├── lib_foo.jar
            ├── lib_bar.jar
            │   ...
            │   ...
            └── sample.jar
                ├── com/foo/...
                │   ...
                │   ...
                └── META-INF
                    ├── plugin.xml
                    ├── pluginIcon.svg
                    └── pluginIcon_dark.svg
└──src
├──com.code
 
基本的框架结构，如果要导入依赖放到lib文件夹中，还有另一种建立框架的方式，那个是基于Gradle管理。
META-INF，配置文件件，管理注册的类。
3.2 常用API介绍VFS 
提供一个处理文件的通用API，而不关心文件的具体位置（无论文件位于磁盘上、归档文件中还是HTTP服务器上）。
追踪文件变化，并且在检测到文件内容发生更改时能提供新旧两个版本的文件。
建立文件在VFS和持久化存储之间的关联。
从本地IO文件中获取
File ioFile = new File("./io.java")
VritualFile virtualFile = LocalFileSystem.getInstance().refreshAndFindFileByIoFile(ioFile)
virtualFile.refresh(false, true)
对VirtualFile进行读写操作：
和Android一样，Intellij Platform不允许直接在主线程进行实时的文件写入，需要通过一个异步任务进行。
WriteCommandAction.runWriteCommandAction(project, new Runnable() {
     @Override
    public void run() {
    //   virtualFile.getInputStream() / virtualFile.getOutputStream()         
    }
 });
在异步任务结束后，切回UI线程进行UI更新：
ApplicationManager.getApplication().invokeLater(new Runnable(){ 
 ...
})
 
PSI
PSI（Program Structure Interface）是Intellij Platform中一个非常重要的概念，在IDE所管理的Project中，每个目录，Package，源代码和资源文件都会被抽象成相应的PSI对象。
常用子类：PsiDirectory、PsiJavaFile和XmlFile。
创建目录和文件：
//创建目录
PsiDirectory baseDir =PsiDirectoryFactory.getInstance(project).createDirectory(project.getBaseDir());
//创建Java文件
PsiJavaFile psiFile = (PsiJavaFile) PsiFileFactory.getInstance(project).createFileFromText("",StdFileTypes.JAVA, "");
//创建Xml文件
XmlFile psiFile = (XmlFile) PsiFileFactory.getInstance(project).createFileFromText("",StdFileTypes.XML, "");
 
读写文件：和写入VirtualFile一样，读写操作都需要在WriteCommandAction异步线程中进行。
创建Class文件类：
PsiClass clazz =JavaDirectoryService.getInstance().createClass(subDir, className)
//还有通过freemarker模板建立class类。
 
psiClass类中添加接口：
PsiClass view = myFactory.createInterface("View");
psiClass.add(view);
设置包名：
PsiJavaFile javaFile = (PsiJavaFile) psiClass.getContainingFile();
PsiPackage psiPackage = myDirectoryService.getPackage(directory);
javaFile.setPackageName(psiPackage.getQualifiedName());
设置类权限：
psiClass.getModifierList().setModifierProperty(PsiModifier.PUBLIC,true);
四.实例架构平时开发过程中，代码结构会分层，类似MVC思想，这里面有很多可以抽象出来的公共类，比如JavaBean，DTO，Service等等，我这个实例结合类似场景，实现自动化插件。
﻿
﻿
架构
五.准备工作创建插件项目:
﻿
﻿
还可以用Gradle方式创建项目，我用的idea版本2019.2.4，上述内容中提到框架结构，现在可以在src目录中编码。
六.编码总共有几个部分组成。
BaseAnAction
AnActionEvent一些基本信息。
public abstract class BaseAnAction extends AnAction {
    private AnActionEvent anActionEvent;
    private DataContext dataContext;
    private Presentation presentation;
    private Module module;
    private IdeView view;
    private ModuleType moduleType;
    private Project project;
    private PsiDirectory psiDirectory;
    private DialogBuilder builder;
    private PsiFile file;
    private JavaDirectoryService javaDirectoryService;
    private MysqlJdbc mysqlJdbc = MysqlJdbc.getMysqlJdbc();
    private PropertiesUtil properties = PropertiesUtil.getConfigProperties();
﻿
﻿
    public void init(AnActionEvent anActionEvent) {
        this.javaDirectoryService = new JavaDirectoryServiceImpl();
        this.anActionEvent = anActionEvent;
        IdeView ideView = (IdeView)anActionEvent.getRequiredData(LangDataKeys.IDE_VIEW);
        this.psiDirectory = ideView.getOrChooseDirectory();
        this.project = this.psiDirectory.getProject();
    }
﻿
﻿
    public PropertiesUtil getProperties() {
        return this.properties;
    }
﻿
﻿
    public MysqlJdbc getMysqlJdbc() {
        return this.mysqlJdbc;
    }
﻿
﻿
    public PsiDirectory getPsiDirectory() {
        return this.psiDirectory;
    }
﻿
﻿
    public JavaDirectoryService getJavaDirectoryService() {
        return this.javaDirectoryService;
    }
﻿
﻿
    public AnActionEvent getAnActionEvent() {
        return this.anActionEvent;
    }
﻿
﻿
    public void setAnActionEvent(AnActionEvent anActionEvent) {
        this.anActionEvent = anActionEvent;
    }
﻿
﻿
    public DataContext getDataContext() {
        return this.dataContext;
    }
﻿
﻿
    public void setDataContext(DataContext dataContext) {
        this.dataContext = dataContext;
    }
﻿
﻿
    public Module getModule() {
        return this.module;
    }
﻿
﻿
    public void setModule(Module module) {
        this.module = module;
    }
﻿
﻿
    public IdeView getView() {
        return this.view;
    }
﻿
﻿
    public void setView(IdeView view) {
        this.view = view;
    }
﻿
﻿
    public ModuleType getModuleType() {
        return this.moduleType;
    }
﻿
﻿
    public void setModuleType(ModuleType moduleType) {
        this.moduleType = moduleType;
    }
﻿
﻿
    public Project getProject() {
        return this.project;
    }
﻿
﻿
    public void setProject(Project project) {
        this.project = project;
    }
﻿
﻿
    public DialogBuilder getBuilder() {
        return this.builder;
    }
﻿
﻿
    public void setBuilder(DialogBuilder builder) {
        this.builder = builder;
    }
﻿
﻿
    public PsiFile getFile() {
        return this.file;
    }
﻿
﻿
    public void setFile(PsiFile file) {
        this.file = file;
    }
﻿
﻿
    public Presentation getPresentation() {
        return this.presentation;
    }
﻿
﻿
    public void setPresentation(Presentation presentation) {
        this.presentation = presentation;
    }
﻿
﻿
    @Override
    public void update(AnActionEvent e) {
        try {
            this.presentation = e.getPresentation();
            this.onMenuUpade(e, (PsiFile)e.getData(DataKeys.PSI_FILE), ((IdeView)LangDataKeys.IDE_VIEW.getData(e.getDataContext())).getOrChooseDirectory());
        }
        catch (Exception exception) {
            // empty catch block
        }
    }
﻿
﻿
    public void show() {
        this.presentation.setEnabled(true);
        this.presentation.setVisible(true);
    }
﻿
﻿
    public void hide() {
        this.presentation.setEnabled(false);
        this.presentation.setVisible(false);
    }
﻿
﻿
    public void onMenuUpade(AnActionEvent e, PsiFile file, P
CodeComponent
应用管理。
public class CodeComponent implements ApplicationComponent {
    @Override
    public void initComponent() {
﻿
﻿
        // TODO: insert component initialization logic here
﻿
﻿
    }
﻿
﻿
﻿
﻿
﻿
﻿
    @Override
    public void disposeComponent() {
﻿
﻿
        // TODO: insert component disposal logic here
﻿
﻿
    }
﻿
﻿
﻿
﻿
﻿
﻿
    @Override
    @NotNull
    public String getComponentName() {
        if ("CreateMicroServiceProjectComponent" == null) {
            CodeComponent.reportNull(0);
        }
        return "CreateMicroServiceProjectComponent";
    }
﻿
﻿
    private static  void reportNull(int n) {
        throw new IllegalStateException(String.format("@NotNull method %s.%s must not return null", "com/code/action/CodeComponent", "getComponentName"));
还有一些工具类，比如操作MySQL数据库，操作字符串等等。一些freemarker模板，Action动作。
MMS_DO.java.ft
#if (${PACKAGE_NAME} && ${PACKAGE_NAME} != "")package ${PACKAGE_NAME};#end
import lombok.Data;
import java.util.*;
import java.math.BigDecimal;
﻿
﻿
/**
 * @author ${USER} E-mail:${E_MAIL}
 * @version 创建时间：${DATE} ${TIME}
 *     ${doCalssName}DO对象
 */
@Data
public class ${doCalssName}DO {
}
CreateServiceAction
创建Service接口和实现类。
public class CreateServiceAction extends BaseAnAction {
    @Override
    public void actionPerformed(@NotNull AnActionEvent anActionEvent) {
        this.init(anActionEvent);
        String serviceName = Messages.showInputDialog((String)"Service name", (String)"Create service", (Icon)Messages.getInformationIcon());
        Map<String, String> param = new HashMap<String, String>();
        param.put("doCalssName", BaseUtils.firstLetterUpperCase(BaseUtils.markToHump(serviceName, "_", null)));
        param.put("tableName", serviceName);
        PsiDirectory implDir = this.getPsiDirectory().findSubdirectory("impl");
        if (implDir == null) {
            implDir = this.getPsiDirectory().createSubdirectory("impl");
        }
        PsiClass servicePsiClass = this.getJavaDirectoryService().createClass(this.getPsiDirectory(), "", "MMS_Service", false, param);
        String packagePath = servicePsiClass.getQualifiedName();
        String implT = "impl";
        param.put(implT, packagePath + ";");
        param.put("serviceName", serviceName);
        this.getJavaDirectoryService().createClass(implDir, "", "MMS_ServiceImpl", false, param);
    }
}
GitHub项目
项目内容放到GitHub中，地址：https://github.com/77954309/code_plugin.git
有些依赖得自行下载，
﻿
﻿
七.部署在code_plugin项目鼠标右击，或者build 点击Prepare Plugin Module '插件名称（codeplugin）' For Deployment 生成插件包（zip/jar）。
在IDEA 文件夹，File->Settings->Plugins->Install Plugin from Disk，安装打出插件，查看目录，重启。
导入插件
﻿
﻿
效果展示
插件位置
项目，鼠标右击，新建New，有CreateDO、CreateDTO、CreateService三个功能窗口。
﻿
﻿
创建DO
这个实体是跟MySQL业务表像映射的，窗口填的是数据库表名称。
﻿
﻿
﻿
﻿
创建DTO
DTO是跟DO相映射的，符合阿里的编程规范，用于处理Service层业务处理，这个代码中写上包名称，DO得在特定包名下，DTO才能映射。
public class CreateDTOAction extends BaseAnAction  {
    @Override
    public void actionPerformed(@NotNull AnActionEvent anActionEvent) {
        this.init(anActionEvent);
        String doName = Messages.showInputDialog((String)"DO name", (String)"Create DTO", (Icon)Messages.getInformationIcon());
        final PsiElementFactory factory = JavaPsiFacade.getInstance((Project)this.getProject()).getElementFactory();
        HashMap<String, String> param = new HashMap<String, String>();
        String doClassName = BaseUtils.firstLetterUpperCase(BaseUtils.markToHump(doName.substring(0, doName.length() - 2), "_", null));
        param.put("doCalssName", doClassName);
        GlobalSearchScope searchScope = GlobalSearchScope.allScope((Project)this.getProject());
        PsiPackage psiPackage = JavaPsiFacade.getInstance((Project)this.getProject()).findPackage("com.lm.model");
        PsiClass[] doPsiClasss = psiPackage.findClassByShortName(doName, searchScope);
        PsiClass doPsiClass = doPsiClasss[0];
        PsiClass dtoPsiClass = JavaPsiFacade.getInstance((Project)this.getProject()).findClass(doPsiClass.getQualifiedName(), searchScope);
        final PsiField[] psiFields = dtoPsiClass.getFields();
        final PsiClass psiClass = this.getJavaDirectoryService().createClass(this.getPsiDirectory(), "", "MMS_DTO", false, param);
        WriteCommandAction.runWriteCommandAction((Project)this.getProject(), (Runnable)new Runnable(){
﻿
﻿
            @Override
            public void run() {
                for (PsiField psiField : psiFields) {
                    String comment = psiField.getDocComment().getText().replaceAll("\\*", "").replaceAll("/", "").replaceAll(" ", "").replaceAll("\n", "");
                    StringBuffer fieldStrBuf = new StringBuffer(psiField.getDocComment().getText()).append("\nprivate ").append(psiField.getType().getPresentableText()).append(" ").append(psiField.getName()).append(";");
                    psiClass.add((PsiElement)factory.createFieldFromText(fieldStrBuf.toString(), (PsiElement)psiClass));
                }
            }
        });
    }
}
﻿
八.总结总体的IDEA插件开发介绍完毕，这个可以基于模板快速拓展，有兴趣的朋友可以尝试下，毕竟授人以鱼不如授人以渔，自动化是工程师文化的一个重要体现。
还可以把插件发布到仓库，支持Plugin中搜索安装，参考：
http://www.jetbrains.org/intellij/sdk/docs/basics/getting_started/publishing_plugin.html
就是注册账号，提交jar，填写信息，等着审核就可以了。
文中项目地址：https://github.com/77954309/code_plugin
九.作者介绍李孟，目前就职于知因智慧数据科技有限公司，负责数据中台数据引擎基础架构设计和中间件开发，专注云计算大数据方向。
博客：https://blog.csdn.net/qq_19968255
﻿