package com.monkeylessey.gen.handler;

import com.baomidou.mybatisplus.generator.config.po.TableInfo;
import com.monkeylessey.gen.QueryTemplate;
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
 * @date 2022/11/30
 */
public class QueryHandler {

    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");
        // 设置query类相关
        QueryTemplate queryTemplate = new QueryTemplate();
        // 设置query类名称
        queryTemplate.setQueryName(entityName + "Query");
        // 设置query类路径
        queryTemplate.setQueryPath(packageMap.get("Parent") + packageMap.get("ModuleName") + ".sys.domain.query");
        // 设置query类的字段
        List<TableColumnVO> columnS = data.getColumns().stream()
                .filter(item -> Boolean.TRUE == item.getSearch())
                .collect(Collectors.toList());
        queryTemplate.setFields(columnS);
        // 设置query的父类名
        queryTemplate.setBaseQueryName("AbsQuery");
        queryTemplate.setBaseQueryPath("com.monkeylessey.base");
        objectMap.put("queryInfo", queryTemplate);
        return new VueHandler().handler(objectMap, data);
    }

}