全栈开发工具-java类型转TypeScript

全栈开发工具-java类型转TypeScript

package com.cai.orderingfood.common.utils;

import cn.hutool.core.io.FileUtil;
import io.swagger.annotations.ApiModelProperty;
import org.apache.commons.lang3.reflect.FieldUtils;

import java.io.File;
import java.lang.reflect.Field;
import java.net.URL;
import java.nio.charset.Charset;
import java.nio.file.Paths;
import java.util.*;
import java.util.stream.Collectors;

public class JavaToTypescript {

    //class加载器
    final static ClassLoader CLASS_LOADER = JavaToTypescript.class.getClassLoader();
    //class根路径
    final static String base_PATH = CLASS_LOADER.getResource("").getPath().substring(1);
    //类型映射
    final static Map TYPE_MAP = new HashMap<>();

    //super class
    static Class superClass;
    //super class 字段名集合
    static List superClassFieldNameList = new ArrayList();

    static {
        TYPE_MAP.put("Integer", "number");
        TYPE_MAP.put("String", "string");
        TYPE_MAP.put("Date", "string");
        TYPE_MAP.put("DateTime", "string");
        TYPE_MAP.put("BigDecimal", "number");

        try {
            //基类
            superClass = CLASS_LOADER.loadClass("com.cai.orderingfood.common.base.baseEntity");
        } catch (ClassNotFoundException e) {
            e.printStackTrace();
        }
    }

    public static void main(String[] args) throws ClassNotFoundException {
        String superClassTypeStr = loadSuperClass();
        String packageName = "com.cai.orderingfood.common.model.entity";
        URL url = CLASS_LOADER.getResource(packageName.replaceAll("\.", "/"));
        //扫描包中的文件
        List allJavaFile = getAllJavaFile(new File(url.getPath()), new ArrayList());
        //把包中的class加载到内存中
        List> allClass = getAllClass(allJavaFile);
        //转换类型
        String result = allClass.stream().map(clazz -> getTypescript(clazz)).collect(Collectors.joining("n"));
        System.out.println(superClassTypeStr);
        System.out.println(result);

        //写入文件
        File file = new File(base_PATH + "type.d.ts");
        if(file.exists()){
            file.delete();
        }
        FileUtil.appendString(superClassTypeStr,file, Charset.defaultCharset());
        FileUtil.appendString(result,file, Charset.defaultCharset());
    }

    
    public static String loadSuperClass() throws ClassNotFoundException {
        try {
            List fieldNameList = FieldUtils
                    .getAllFieldsList(superClass)
                    .stream()
                    .map(field -> field.getName())
                    .collect(Collectors.toList());
            superClassFieldNameList.addAll(fieldNameList);
            return getTypescript(superClass, true);
        } catch (Exception e) {
            e.printStackTrace();
            return "";
        }
    }

    //转换类型
    public static String getTypescript(Class clazz) {
        return getTypescript(clazz, false);
    }

    //转换类型
    public static String getTypescript(Class clazz, boolean isSupperClass) {
        String template = "interface =className= {n=typeBody=n}n";
        if (Objects.nonNull(superClass) && !isSupperClass) {
            template = "interface =className= extends =superClassName= {n=typeBody=n}n"
//                    .replaceAll("=superClassName=", "baseDto");
                    .replaceAll("=superClassName=", superClass.getSimpleName());
        }
        String typeTemplate = "     // =remark=n     =fieldName=?: =typeName=";
        //获取class全部的字段
        String typeBody = FieldUtils
                .getAllFieldsList(clazz)
                .stream()
                //只保留有set get方法的属性
                .filter((field) -> {
                    String fieldName = field.getName();
                    String setMethodName = "set" + fieldName.substring(0, 1).toUpperCase() + fieldName.substring(1);
                    String getMethodName = "get" + fieldName.substring(0, 1).toUpperCase() + fieldName.substring(1);
                    try {
                        clazz.getMethod(setMethodName, field.getType());
                        clazz.getMethod(getMethodName);
                        if (isSupperClass) {
                            return true;
                        }
                        //继续判断字段是否存在于父类基类
                        return !superClassFieldNameList.contains(field.getName());
                    } catch (NoSuchMethodException e) {
                        return false;
                    }
                })
                .map(field -> typeTemplate
                        .replaceAll("=fieldName=", field.getName())
                        .replaceAll("=typeName=", TYPE_MAP.get(field.getType().getSimpleName()))
                        .replaceAll("=remark=", getFieldRemark(field))
                )
                .collect(Collectors.joining("n"));
        return template
                .replaceAll("=className=", clazz.getSimpleName())
                .replaceAll("=typeBody=", typeBody);
    }

    //获取字段注释
    public static String getFieldRemark(Field field) {
        String remark = "";
        try {
            ApiModelProperty apiModelProperty = field.getAnnotation(ApiModelProperty.class);
            if (Objects.nonNull(apiModelProperty)){
                return apiModelProperty.value();
            }
        } catch (Throwable e) {
            e.printStackTrace();
        }
        return remark;
    }

    //获取加载的所有class,并且加载到内存
    public static List> getAllClass(List files) {
        String basePath = Paths.get(base_PATH).toString();
        return files.stream().map(filePath -> {
            filePath = Paths.get(filePath).toString();
            String packageName = filePath
                    .replace(basePath, "")
                    .substring(1)
                    .replaceAll("\.class", "")
                    .replaceAll(File.separator + File.separator, ".");
            try {
                return CLASS_LOADER.loadClass(packageName);
            } catch (ClassNotFoundException e) {
                e.printStackTrace();
                return null;
            }
        }).filter(item -> Objects.nonNull(item)).collect(Collectors.toList());
    }

    //加载class文件
    public static List getAllJavaFile(File file, List files) {
        if (!file.isDirectory()) {
            return files;
        }
        file.listFiles((file1) -> {
            if (file1.isDirectory()) {
                getAllJavaFile(file1, files);
            } else {
                String name = file1.getName();
                if (name.toLowerCase().endsWith(".class")) {
                    files.add(file1.getAbsolutePath());
                }
            }
            return false;
        });
        return files;
    }

}

支持基类
取ApiModelProperty value作为注释
生成效果