IDEA插件开发（三）

public class GenerateDialog extends DialogWrapper {
    public GenerateDialog() {
        super(true);
        setTitle("Where语句生成");
        init();
    }
    @Override
    protected JComponent createCenterPanel() {
        return new WhereGenerateWindow().getRootPanel();
    }
}

public class GenerateWhereAction extends AnAction {
    @Override
    public void actionPerformed(AnActionEvent e) {
        this.show(e);
    }
    private void show(AnActionEvent e){
        new GenerateDialog().show();
    }
}

public class GenerateWhereAction extends AnAction {
    @Override
    public void actionPerformed(AnActionEvent e) {
        this.show(e);
    }
    private void show(AnActionEvent e){
        WhereGenerateWindow window = new WhereGenerateWindow();
        DialogBuilder dialogBuilder = new DialogBuilder(e.getProject());
        dialogBuilder.setCenterPanel(window.getRootPanel());
        dialogBuilder.setTitle("Where语句生成");
        dialogBuilder.show();
    }
}

public class GenerateWhereAction extends AnAction {
    @Override
    public void actionPerformed(AnActionEvent e) {
        // 获取编辑器
        Editor editor = e.getRequiredData(CommonDataKeys.EDITOR);
        // 当前打开的文件
        VirtualFile virtualFile = e.getRequiredData(CommonDataKeys.PSI_FILE).getViewProvider().getVirtualFile();
        // 获取选择的数据模型
        SelectionModel selectionModel = editor.getSelectionModel();
        // 获取当前选择的文本
        String selectedText = selectionModel.getSelectedText();
        if(StringUtils.isBlank(selectedText)){
            // 获取当前打开的文件
            Document document = editor.getDocument();
            show(e, document.getText());
        }
        else {
            show(e, selectedText);
        }
    }
    private void show(AnActionEvent e, String text){
        WhereGenerateWindow window = new WhereGenerateWindow(text);
        DialogBuilder dialogBuilder = new DialogBuilder(e.getProject());
        dialogBuilder.setCenterPanel(window.getRootPanel());
        dialogBuilder.setTitle("Where语句生成");
        dialogBuilder.show();
    }
}

public class WhereGenerateWindow {
    private JPanel rootPanel;
    private JTextField prefix;
    private JButton generate;
    private JTextArea whereStatement;
    public WhereGenerateWindow(String text){
        this.init(text);
    }
    private void init(String text){
        whereStatement.setText(text);
        generate.addActionListener(e -> {
            whereStatement.setText(prefix.getText() + "." + whereStatement.getText());
        });
    }
    public JPanel getRootPanel() {
        return rootPanel;
    }
}

dependencies{
    implementation("cn.hutool:hutool-all:5.6.2")
}