Android Studio 插件：按等号对齐进行格式化

背景

项目中有很多常量类，看上去不整齐，有时候想找一个常量眼特别花。于是写了一个小插件，用来把它们按照等号进行格式化。

效果

实现

public class FormatAction extends AnAction {

    @Override
    public void actionPerformed(AnActionEvent e) {
        // 获取编辑器model
        final Editor mEditor = e.getData(PlatformDataKeys.EDITOR);

        final Project project = e.getProject();
        if (null == mEditor || project == null) {
            return;
        }

        // 获取代码内容
        final Document document = mEditor.getDocument();
        // 获取光标model
        final CaretModel caretModel = mEditor.getCaretModel();
        // 获取选中区域model
        final SelectionModel selectionModel = mEditor.getSelectionModel();

        String s = selectionModel.getSelectedText();
        if (s == null || s.length() == 0) {
            System.out.println("no selection");
            return;
        }

        // 找到最远的等号的 index
        String[] lines = s.split("\n");
        int maxIndexOfEqual = -1;
        for (String line : lines) {
            int indexOfEqual = line.indexOf("=");
            if (indexOfEqual > maxIndexOfEqual) {
                maxIndexOfEqual = indexOfEqual;
            }
        }

        if (maxIndexOfEqual < 0) {
            System.out.println("no equal sign");
            return;
        }

        // 给每一行的等号前添加空格
        StringBuilder newDocumentBuilder = new StringBuilder();
        for (String line : lines) {
            String newLine = line;
            int indexOfEqual = line.indexOf("=");
            if (indexOfEqual > 0 && indexOfEqual < maxIndexOfEqual) {
                String lineHead = line.substring(0, indexOfEqual);
                String lineEnd = line.substring(indexOfEqual, line.length());

                int blankCount = maxIndexOfEqual - indexOfEqual;
                StringBuilder lineMidBuilder = new StringBuilder();
                for (int i = 0; i < blankCount; i++) {
                    lineMidBuilder.append(" ");
                }
                String lineMid = lineMidBuilder.toString();

                newLine = lineHead + lineMid + lineEnd;
            }
            newDocumentBuilder.append(newLine).append('\n');
        }

        // 替换旧代码
        int start = selectionModel.getSelectionStart();
        int end = selectionModel.getSelectionEnd();
        WriteCommandAction.runWriteCommandAction(project, () -> {
            document.replaceString(start, end, newDocumentBuilder.toString());
        });
    }
}