pro.fessional.wings.faceless.project.ProjectAuthGenerator Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of faceless-codegen Show documentation
Code Generator of Faceless
The newest version!
package pro.fessional.wings.faceless.project;

import lombok.Getter;
import lombok.Setter;
import org.springframework.jdbc.core.RowMapper;
import pro.fessional.wings.faceless.codegen.ConstantNaviGenerator;
import pro.fessional.wings.faceless.codegen.ConstantNaviGenerator.Entry;
import pro.fessional.wings.faceless.codegen.JdbcDataLoadHelper;

import java.util.List;

/**
 * In IDEA, the execution of `main` is different from spring, workdir is different.
 *
 * @author trydofor
 * @since 2021-02-20
 */
@Setter
@Getter
public class ProjectAuthGenerator {

    protected String targetDir;
    protected String targetPkg;

    public void genPerm(String jdbc, String user, String pass) {
        genPerm(JdbcDataLoadHelper.use(jdbc, user, pass));
    }

    public void genPerm(JdbcDataLoadHelper helper) {
        genPerm(helper, "PermConstant", "");
    }

    public void genPerm(JdbcDataLoadHelper helper, String javaName, String prefixCode) {
        final List perms = helper.load(permSelect, permMapper);

        ConstantNaviGenerator generator = new ConstantNaviGenerator();
        generator.setPackageName(targetPkg);
        generator.setTargetDir(targetDir);

        generator.generate(javaName, prefixCode, perms);
    }

    public void genRole(String jdbc, String user, String pass) {
        genRole(JdbcDataLoadHelper.use(jdbc, user, pass), "RoleConstant", "ROLE_");
    }

    public void genRole(JdbcDataLoadHelper helper) {
        genRole(helper, "RoleConstant", "ROLE_");
    }

    public void genRole(JdbcDataLoadHelper helper, String javaName, String prefixCode) {
        final List roles = helper.load(roleSelect, roleMapper);

        ConstantNaviGenerator generator = new ConstantNaviGenerator();
        generator.setPackageName(targetPkg);
        generator.setTargetDir(targetDir);

        generator.generate(javaName, prefixCode, roles);
    }

    ///
    public static final String permSelect = "select id, scopes, action, remark from win_perm_entry where delete_dt = '1000-01-01'";
    public static final RowMapper permMapper = (rs, rowNum) -> {
        Entry en = new Entry();
        en.setId(rs.getLong("id"));
        en.setName(rs.getString("scopes") + "." + rs.getString("action"));
        en.setRemark(rs.getString("remark"));
        return en;
    };

    public static final String roleSelect = "select id, name, remark from win_role_entry where delete_dt = '1000-01-01'";
    public static final RowMapper roleMapper = (rs, rowNum) -> {
        Entry en = new Entry();
        en.setId(rs.getLong("id"));
        en.setName(rs.getString("name"));
        en.setRemark(rs.getString("remark"));
        return en;
    };
}