package com.monkeylessey.gen.handler;

import com.baomidou.mybatisplus.generator.config.po.TableInfo;
import com.monkeylessey.gen.FormTemplate;
import com.monkeylessey.gen.domain.GenerateData;
import com.monkeylessey.sys.domain.vo.TableColumnVO;

import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * @author xp
 * @version 1.0
 * @date 2022/5/18
 */

public class FormHandler {

    public Map handler(Map objectMap, GenerateData data) {
        // 获取table
        TableInfo tableInfo = (TableInfo) objectMap.get("table");
        // 获取包信息
        Map<String, String> packageMap = (Map<String, String>) objectMap.get("package");
        // 实体类名称
        String entityName = (String) objectMap.get("entity");
        // 设置Form类相关
        FormTemplate formTemplate = new FormTemplate();
        // 设置Form名称
        formTemplate.setFormName(entityName + "Form");
        // 设置Form路径
        formTemplate.setFormPath(packageMap.get("Parent") + packageMap.get("ModuleName") + ".sys.domain.form");
        // 设置form的字段，排除entity的一些字段即可

        List<TableColumnVO> formColumns = data.getColumns().stream()
                .filter(item -> !FormTemplate.removeFields.contains(item.getPropertyName()))
                .collect(Collectors.toList());
        formTemplate.setFields(formColumns);
        // 设置form的父类名
        formTemplate.setBaseFormName("AbsForm");
        formTemplate.setBaseFormPath("com.monkeylessey.sys.domain.base");
        objectMap.put("formInfo", formTemplate);
        return new VoHandler().handler(objectMap, data);
    }
}