IDEA插件：快速删除Java代码中的注释

路人甲Java 2021-05-02

展开全文

背景

有时，我们需要删除Java源代码中的注释。目前有不少方法，比如：

实现状态机。该方式较为通用，适用于多种语言（取决于状态机支持的注释符号）。
正则匹配。该方式容易误判，尤其是容易误删字符串。
利用第三方库。该方式局限性较强，比如不同语言可能有不同的第三方库。

本文针对Java语言，介绍一种利用第三方库的方式，可以方便快速地移除代码中的注释。

原理

这个第三方库叫做JavaParser。它可以分析Java源码，并生成语法分析树（AST），其中注释也属于AST中的节点。

因此核心思路即为：

JavaParser解析源码并得到AST。
识别出注释类型的节点并将其删掉。
将AST中剩余的节点按一定规则打印出来。

在实践之前，我们先要了解Java中的几种注释类型：

LineComment 单行注释。
BlockComent 块注释。
JavadocComment Java文档注释。

下面举个简单例子，说明三种注释的区别：

import java.util.ArrayList;import java.util.stream.Collectors;/**
 * @author xiaoxi666
 * @date 2021-02-15 17:13
 * 我是 Javadoc 注释 */public class Input {/** * 我是 Javadoc 注释
     *
     * @param param1
     * @param param2     */public static void someMethod(String param1, // 我是单行注释                                  String param2// 我是单行注释 String param3,/* 我是块注释 String param4,
                                  String param5,
                                  String param6 *//* 我是块注释 String param4 */)
 {// 我是单行注释int a = 1;/* 我是块注释，注意我和Javadoc注释的区别，我只有一个星号 */int b = 2;/* * 我是块注释         */int c = 3;
        String s1 = "// 我是字符串中的内容，不是注释";
        String s2 = "/* 我是字符串中的内容，不是注释 */";
        String s3 = "/** 我是字符串中的内容，不是注释 */";
    }
}

下面我们实践一下，看看怎么移除源码中的注释。
我这里使用maven管理项目，首先引入JavaParser依赖：

<dependencies><dependency> <groupId>com.github.javaparser</groupId> <artifactId>javaparser-symbol-solver-core</artifactId> <version>3.18.0</version></dependency></dependencies>

然后编写核心代码：

   =<CompilationUnit> result =<CompilationUnit> optionalCompilationUnit = (! ""=  <Comment> comments =<Comment> unwantedComments = =     comment  LineComment || comment

在上面的代码中，我们首先创建JavaParser，再解析源码，然后移除单行注释和块注释，最后再用LexicalPreservingPrinter将处理后的源码打印出来，这个打印器可以保留源代码所有词法，比如空格、换行之类的元素。上述代码已有注释，因此不再详述。

封装为IDEA插件

考虑到我们平时可能会大量使用该功能，因此将其封装为了IDEA插件，名为remove.comments。下面简要介绍该插件的工作原理及使用方式。PS：本文不会详细介绍如何编写IDEA插件。

IDEA插件的原理基本都是事件驱动，如下图所示，我们创建了一个事件监听器，当检测到编辑器中点击右键后，即可弹出菜单，我们的插件在菜单中的第一个位置。

接下来，实现事件处理器：

其中包含两段核心代码：

删除源码注释。首先读取当前文件内容也即源码，然后交给前面已经介绍过的CommentsRemover.doAction处理，就拿到了删除注释后的源码。
格式化代码。删除注释后，可能会引入多余的空格，因此我们自动格式化，这样用户就不用再手动格式化一次了。

/**
 * 移除代码中的注释
 *
 * @param editor
 * @return true if remove comments successfully */private boolean removeComments(Editor editor) {
    String src = editor.getDocument().getText();if (Strings.isNullOrEmpty(src)) {return false;
    }
    String dst = CommentsRemover.doAction(checkEndLineAndModifyIfNeed(src));if (Strings.isNullOrEmpty(dst)) {return false;
    }
    editor.getDocument().setText(dst);return true;
}/**
 * 由于我们保留了源码格式，移除注释之后会引入不必要的空格，因此需要再格式化一下
 *
 * @param editor
 * @param project */private void reformat(Editor editor, Project project) {
    PsiDocumentManager.getInstance(project).commitAllDocuments();
    PsiFile file = PsiDocumentManager.getInstance(project).getPsiFile(editor.getDocument());if (file == null) {return;
    }
    LastRunReformatCodeOptionsProvider provider = new LastRunReformatCodeOptionsProvider(PropertiesComponent.getInstance());
    ReformatCodeRunOptions currentRunOptions = provider.getLastRunOptions(file);
    TextRangeType processingScope = TextRangeType.WHOLE_FILE;
    currentRunOptions.setProcessingScope(processingScope);
    (new FileInEditorProcessor(file, editor, currentRunOptions)).processCode();
}

然后打包插件：