基于IntelliJ IDEA Ultimate版开发的一个代码生成插件，主要通过自定义模板来生成各种你想要的代码

package com.sjhy.plugin.ui; import com.intellij.openapi.fileChooser.FileChooser; import com.intellij.openapi.fileChooser.FileChooserDescriptorFactory; import com.intellij.openapi.module.Module; import com.intellij.openapi.module.ModuleManager; import com.intellij.openapi.project.Project; import com.intellij.openapi.ui.DialogWrapper; import com.intellij.openapi.ui.Messages; import com.intellij.openapi.vfs.VirtualFile; import com.intellij.util.ExceptionUtil; import com.sjhy.plugin.constants.StrState; import com.sjhy.plugin.dict.GlobalDict; import com.sjhy.plugin.dto.GenerateOptions; import com.sjhy.plugin.dto.SettingsStorageDTO; import com.sjhy.plugin.entity.TableInfo; import com.sjhy.plugin.entity.Template; import com.sjhy.plugin.service.CodeGenerateService; import com.sjhy.plugin.service.SettingsStorageService; import com.sjhy.plugin.service.TableInfoSettingsService; import com.sjhy.plugin.tool.CacheDataUtils; import com.sjhy.plugin.tool.ModuleUtils; import com.sjhy.plugin.tool.ProjectUtils; import com.sjhy.plugin.tool.StringUtils; import com.sjhy.plugin.ui.component.TemplateSelectComponent; import org.jetbrains.annotations.Nullable; import javax.swing.*; import java.awt.*; import java.awt.event.FocusAdapter; import java.awt.event.FocusEvent; import java.lang.reflect.Constructor; import java.lang.reflect.InvocationTargetException; import java.lang.reflect.Method; import java.util.LinkedList; import java.util.List; /** * 选择保存路径 * * @author makejava * @version 1.0.0 * @since 2018/07/17 13:10 */ public class SelectSavePath extends DialogWrapper { /** * 主面板 */ private JPanel contentPane; /** * 模型下拉框 */ private JComboBox<String> moduleComboBox; /** * 包字段 */ private JTextField packageField; /** * 路径字段 */ private JTextField pathField; /** * 前缀字段 */ private JTextField preField; /** * 包选择按钮 */ private JButton packageChooseButton; /** * 路径选择按钮 */ private JButton pathChooseButton; /** * 模板面板 */ private JPanel templatePanel; /** * 统一配置复选框 */ private JCheckBox unifiedConfigCheckBox; /** * 弹框选是复选框 */ private JCheckBox titleSureCheckBox; /** * 格式化代码复选框 */ private JCheckBox reFormatCheckBox; /** * 弹框全否复选框 */ private JCheckBox titleRefuseCheckBox; /** * 数据缓存工具类 */ private CacheDataUtils cacheDataUtils = CacheDataUtils.getInstance(); /** * 表信息服务 */ private TableInfoSettingsService tableInfoService; /** * 项目对象 */ private Project project; /** * 代码生成服务 */ private CodeGenerateService codeGenerateService; /** * 当前项目中的module */ private List<Module> moduleList; /** * 实体模式生成代码 */ private boolean entityMode; /** * 模板选择组件 */ private TemplateSelectComponent templateSelectComponent; /** * 构造方法 */ public SelectSavePath(Project project) { this(project, false); } @Override protected @Nullable JComponent createCenterPanel() { return this.contentPane; } /** * 构造方法 */ public SelectSavePath(Project project, boolean entityMode) { super(project); this.entityMode = entityMode; this.project = project; this.tableInfoService = TableInfoSettingsService.getInstance(); this.codeGenerateService = CodeGenerateService.getInstance(project); // 初始化module，存在资源路径的排前面 this.moduleList = new LinkedList<>(); for (Module module : ModuleManager.getInstance(project).getModules()) { // 存在源代码文件夹放前面，否则放后面 if (ModuleUtils.existsSourcePath(module)) { this.moduleList.add(0, module); } else { this.moduleList.add(module); } } this.initPanel(); this.refreshData(); this.initEvent(); init(); setTitle(GlobalDict.TITLE_INFO); //初始化路径 refreshPath(); } private void initEvent() { //监听module选择事件 moduleComboBox.addActionListener(e -> { // 刷新路径 refreshPath(); }); try { Class<?> cls = Class.forName("com.intellij.ide.util.PackageChooserDialog"); //添加包选择事件 packageChooseButton.addActionListener(e -> { try { Constructor<?> constructor = cls.getConstructor(String.class, Project.class); Object dialog = constructor.newInstance("Package Chooser", project); // 显示窗口 Method showMethod = cls.getMethod("show"); showMethod.invoke(dialog); // 获取选中的包名 Method getSelectedPackageMethod = cls.getMethod("getSelectedPackage"); Object psiPackage = getSelectedPackageMethod.invoke(dialog); if (psiPackage != null) { Method getQualifiedNameMethod = psiPackage.getClass().getMethod("getQualifiedName"); String packageName = (String) getQualifiedNameMethod.invoke(psiPackage); packageField.setText(packageName); // 刷新路径 refreshPath(); } } catch (NoSuchMethodException | IllegalAccessException | InstantiationException | InvocationTargetException e1) { ExceptionUtil.rethrow(e1); } }); // 添加包编辑框失去焦点事件 packageField.addFocusListener(new FocusAdapter() { @Override public void focusLost(FocusEvent e) { // 刷新路径 refreshPath(); } }); } catch (ClassNotFoundException e) { // 没有PackageChooserDialog，并非支持Java的IDE，禁用相关UI组件 packageField.setEnabled(false); packageChooseButton.setEnabled(false); } //选择路径 pathChooseButton.addActionListener(e -> { //将当前选中的model设置为基础路径 VirtualFile path = ProjectUtils.getBaseDir(project); Module module = getSelectModule(); if (module != null) { path = ModuleUtils.getSourcePath(module); } VirtualFile virtualFile = FileChooser.chooseFile(FileChooserDescriptorFactory.createSingleFolderDescriptor(), project, path); if (virtualFile != null) { pathField.setText(virtualFile.getPath()); } }); } private void refreshData() { // 获取选中的表信息（鼠标右键的那张表），并提示未知类型 TableInfo tableInfo; if(entityMode) { tableInfo = tableInfoService.getTableInfo(cacheDataUtils.getSelectPsiClass()); } else { tableInfo = tableInfoService.getTableInfo(cacheDataUtils.getSelectDbTable()); } // 设置默认配置信息 if (!StringUtils.isEmpty(tableInfo.getSaveModelName())) { moduleComboBox.setSelectedItem(tableInfo.getSaveModelName()); } if (!StringUtils.isEmpty(tableInfo.getSavePackageName())) { packageField.setText(tableInfo.getSavePackageName()); } if (!StringUtils.isEmpty(tableInfo.getPreName())) { preField.setText(tableInfo.getPreName()); } SettingsStorageDTO settings = S