代码生成支持选择前端模板类型

RuoYi

2023-11-30 370a299d05029e085be9e259b103db419144cf87

 src/main/java/com/ruoyi/project/tool/gen/util/VelocityUtils.java

@@ -3,8 +3,10 @@
import java.util.ArrayList;

import java.util.HashSet;

import java.util.List;

import java.util.Set;

import org.apache.velocity.VelocityContext;

import com.alibaba.fastjson.JSONObject;

import com.alibaba.fastjson2.JSON;

import com.alibaba.fastjson2.JSONObject;

import com.ruoyi.common.constant.GenConstants;

import com.ruoyi.common.utils.DateUtils;

import com.ruoyi.common.utils.StringUtils;

@@ -29,7 +31,7 @@


    /**

     * 设置模板变量信息

     * 
     *

     * @return 模板列表

     */

    public static VelocityContext prepareContext(GenTable genTable)

@@ -54,14 +56,19 @@
        velocityContext.put("author", genTable.getFunctionAuthor());

        velocityContext.put("datetime", DateUtils.getDate());

        velocityContext.put("pkColumn", genTable.getPkColumn());

        velocityContext.put("importList", getImportList(genTable.getColumns()));

        velocityContext.put("importList", getImportList(genTable));

        velocityContext.put("permissionPrefix", getPermissionPrefix(moduleName, businessName));

        velocityContext.put("columns", genTable.getColumns());

        velocityContext.put("table", genTable);

        velocityContext.put("dicts", getDicts(genTable));

        setMenuVelocityContext(velocityContext, genTable);

        if (GenConstants.TPL_TREE.equals(tplCategory))

        {

            setTreeVelocityContext(velocityContext, genTable);

        }

        if (GenConstants.TPL_SUB.equals(tplCategory))

        {

            setSubVelocityContext(velocityContext, genTable);

        }

        return velocityContext;

    }

@@ -69,7 +76,7 @@
    public static void setMenuVelocityContext(VelocityContext context, GenTable genTable)

    {

        String options = genTable.getOptions();

        JSONObject paramsObj = JSONObject.parseObject(options);

        JSONObject paramsObj = JSON.parseObject(options);

        String parentMenuId = getParentMenuId(paramsObj);

        context.put("parentMenuId", parentMenuId);

    }

@@ -77,7 +84,7 @@
    public static void setTreeVelocityContext(VelocityContext context, GenTable genTable)

    {

        String options = genTable.getOptions();

        JSONObject paramsObj = JSONObject.parseObject(options);

        JSONObject paramsObj = JSON.parseObject(options);

        String treeCode = getTreecode(paramsObj);

        String treeParentCode = getTreeParentCode(paramsObj);

        String treeName = getTreeName(paramsObj);

@@ -96,13 +103,37 @@
        }

    }



    public static void setSubVelocityContext(VelocityContext context, GenTable genTable)

    {

        GenTable subTable = genTable.getSubTable();

        String subTableName = genTable.getSubTableName();

        String subTableFkName = genTable.getSubTableFkName();

        String subClassName = genTable.getSubTable().getClassName();

        String subTableFkClassName = StringUtils.convertToCamelCase(subTableFkName);



        context.put("subTable", subTable);

        context.put("subTableName", subTableName);

        context.put("subTableFkName", subTableFkName);

        context.put("subTableFkClassName", subTableFkClassName);

        context.put("subTableFkclassName", StringUtils.uncapitalize(subTableFkClassName));

        context.put("subClassName", subClassName);

        context.put("subclassName", StringUtils.uncapitalize(subClassName));

        context.put("subImportList", getImportList(genTable.getSubTable()));

    }



    /**

     * 获取模板信息

     * 
     * @param tplCategory 生成的模板

     * @param tplWebType 前端类型

     * @return 模板列表

     */

    public static List<String> getTemplateList(String tplCategory)

    public static List<String> getTemplateList(String tplCategory, String tplWebType)

    {

        String useWebType = "vm/vue";

        if ("element-plus".equals(tplWebType))

        {

            useWebType = "vm/vue/v3";

        }

        List<String> templates = new ArrayList<String>();

        templates.add("vm/java/domain.java.vm");

        templates.add("vm/java/mapper.java.vm");

@@ -114,11 +145,16 @@
        templates.add("vm/js/api.js.vm");

        if (GenConstants.TPL_CRUD.equals(tplCategory))

        {

            templates.add("vm/vue/index.vue.vm");

            templates.add(useWebType + "/index.vue.vm");

        }

        else if (GenConstants.TPL_TREE.equals(tplCategory))

        {

            templates.add("vm/vue/index-tree.vue.vm");

            templates.add(useWebType + "/index-tree.vue.vm");

        }

        else if (GenConstants.TPL_SUB.equals(tplCategory))

        {

            templates.add(useWebType + "/index.vue.vm");

            templates.add("vm/java/sub-domain.java.vm");

        }

        return templates;

    }

@@ -146,6 +182,10 @@
        if (template.contains("domain.java.vm"))

        {

            fileName = StringUtils.format("{}/domain/{}.java", javaPath, className);

        }

        if (template.contains("sub-domain.java.vm") && StringUtils.equals(GenConstants.TPL_SUB, genTable.getTplCategory()))

        {

            fileName = StringUtils.format("{}/domain/{}.java", javaPath, genTable.getSubTable().getClassName());

        }

        else if (template.contains("mapper.java.vm"))

        {

@@ -188,26 +228,31 @@


    /**

     * 获取包前缀

     * 
     *

     * @param packageName 包名称

     * @return 包前缀名称

     */

    public static String getPackagePrefix(String packageName)

    {

        int lastIndex = packageName.lastIndexOf(".");

        String basePackage = StringUtils.substring(packageName, 0, lastIndex);

        return basePackage;

        return StringUtils.substring(packageName, 0, lastIndex);

    }



    /**

     * 根据列类型获取导入包

     * 

     * @param column 列集合

     * @param genTable 业务表对象

     * @return 返回需要导入的包列表

     */

    public static HashSet<String> getImportList(List<GenTableColumn> columns)

    public static HashSet<String> getImportList(GenTable genTable)

    {

        List<GenTableColumn> columns = genTable.getColumns();

        GenTable subGenTable = genTable.getSubTable();

        HashSet<String> importList = new HashSet<String>();

        if (StringUtils.isNotNull(subGenTable))

        {

            importList.add("java.util.List");

        }

        for (GenTableColumn column : columns)

        {

            if (!column.isSuperColumn() && GenConstants.TYPE_DATE.equals(column.getJavaType()))

@@ -224,8 +269,46 @@
    }



    /**

     * 获取权限前缀

     * 根据列类型获取字典组

     * 

     * @param genTable 业务表对象

     * @return 返回字典组

     */

    public static String getDicts(GenTable genTable)

    {

        List<GenTableColumn> columns = genTable.getColumns();

        Set<String> dicts = new HashSet<String>();

        addDicts(dicts, columns);

        if (StringUtils.isNotNull(genTable.getSubTable()))

        {

            List<GenTableColumn> subColumns = genTable.getSubTable().getColumns();

            addDicts(dicts, subColumns);

        }

        return StringUtils.join(dicts, ", ");

    }



    /**

     * 添加字典列表

     * 
     * @param dicts 字典列表

     * @param columns 列集合

     */

    public static void addDicts(Set<String> dicts, List<GenTableColumn> columns)

    {

        for (GenTableColumn column : columns)

        {

            if (!column.isSuperColumn() && StringUtils.isNotEmpty(column.getDictType()) && StringUtils.equalsAny(

                    column.getHtmlType(),

                    new String[] { GenConstants.HTML_SELECT, GenConstants.HTML_RADIO, GenConstants.HTML_CHECKBOX }))

            {

                dicts.add("'" + column.getDictType() + "'");

            }

        }

    }



    /**

     * 获取权限前缀

     *

     * @param moduleName 模块名称

     * @param businessName 业务名称

     * @return 返回权限前缀

@@ -237,13 +320,14 @@


    /**

     * 获取上级菜单ID字段

     * 
     * @param options 生成其他选项

     *

     * @param paramsObj 生成其他选项

     * @return 上级菜单ID字段

     */

    public static String getParentMenuId(JSONObject paramsObj)

    {

        if (StringUtils.isNotEmpty(paramsObj) && paramsObj.containsKey(GenConstants.PARENT_MENU_ID))

        if (StringUtils.isNotEmpty(paramsObj) && paramsObj.containsKey(GenConstants.PARENT_MENU_ID)

                && StringUtils.isNotEmpty(paramsObj.getString(GenConstants.PARENT_MENU_ID)))

        {

            return paramsObj.getString(GenConstants.PARENT_MENU_ID);

        }

@@ -252,8 +336,8 @@


    /**

     * 获取树编码

     * 
     * @param options 生成其他选项

     *

     * @param paramsObj 生成其他选项

     * @return 树编码

     */

    public static String getTreecode(JSONObject paramsObj)

@@ -267,8 +351,8 @@


    /**

     * 获取树父编码

     * 
     * @param options 生成其他选项

     *

     * @param paramsObj 生成其他选项

     * @return 树父编码

     */

    public static String getTreeParentCode(JSONObject paramsObj)

@@ -282,8 +366,8 @@


    /**

     * 获取树名称

     * 
     * @param options 生成其他选项

     *

     * @param paramsObj 生成其他选项

     * @return 树名称

     */

    public static String getTreeName(JSONObject paramsObj)

@@ -297,14 +381,14 @@


    /**

     * 获取需要在哪一列上面显示展开按钮

     * 
     *

     * @param genTable 业务表对象

     * @return 展开按钮列序号

     */

    public static int getExpandColumn(GenTable genTable)

    {

        String options = genTable.getOptions();

        JSONObject paramsObj = JSONObject.parseObject(options);

        JSONObject paramsObj = JSON.parseObject(options);

        String treeName = paramsObj.getString(GenConstants.TREE_NAME);

        int num = 0;

        for (GenTableColumn column : genTable.getColumns())

@@ -321,4 +405,4 @@
        }

        return num;

    }

}
}