package com.monkeylessey.gen.handler;

import com.baomidou.mybatisplus.generator.config.po.TableInfo;
import com.monkeylessey.gen.domain.GenerateData;
import com.monkeylessey.gen.template.VueTemplate;
import com.monkeylessey.gen.utils.WordsUtils;
import com.monkeylessey.sys.domain.vo.TableColumnVO;

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

/**
 * @author xp
 * @data 2022/9/24
 */
public class VueHandler {

    private static final List<String> removeFields = Arrays.asList("gmt_create", "gmt_update", "deleted", "update_by", "create_by");

    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");
        // 设置VUE相关数据
        VueTemplate vueTemplate = new VueTemplate();
        vueTemplate.setName(entityName);
        // 设置view文件的名称
        vueTemplate.setViewName(entityName + "View");
        // 设置首字母小写的名字
        vueTemplate.setLowerName(WordsUtils.firstWordToLower(entityName));
        // 设置view路径
        vueTemplate.setViewFilePath(packageMap.get("Parent") + packageMap.get("ModuleName") + ".vue");

        List<TableColumnVO> fields = data.getColumns().stream()
                .filter(item -> !removeFields.contains(item.getColumnName()))
                .collect(Collectors.toList());

        // 设置vuex中表单的字段
        vueTemplate.setVuexFormFields(fields);
        List<TableColumnVO> useSelectFields = fields.stream()
                .filter(item -> "select".equals(item.getInputWay()) && Objects.nonNull(item.getDataSourceId()))
                .collect(Collectors.toList());
        vueTemplate.setUseSelectFields(useSelectFields);

        objectMap.put("vueInfo", vueTemplate);
        return objectMap;
    }


}