package com.monkeylessey.gen.engine;

import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
import com.baomidou.mybatisplus.generator.config.OutputFile;
import com.baomidou.mybatisplus.generator.config.po.TableInfo;
import com.baomidou.mybatisplus.generator.engine.VelocityTemplateEngine;
import com.monkeylessey.enums.general.MenuTypeEnum;
import com.monkeylessey.enums.general.StatusEnum;
import com.monkeylessey.framework.utils.SecurityUtil;
import com.monkeylessey.gen.domain.GenerateData;
import com.monkeylessey.gen.enums.GenFileTypeEnum;
import com.monkeylessey.gen.handler.FormHandler;
import com.monkeylessey.gen.template.VueTemplate;
import com.monkeylessey.gen.utils.WordsUtils;
import com.monkeylessey.sys.domain.entity.SysMenu;
import com.monkeylessey.sys.mapper.SysMenuMapper;

import java.io.File;
import java.text.SimpleDateFormat;
import java.util.*;

/**
 * @author xp
 * @version 1.0
 * @date 2022/5/17
 */
public class MyVelocityTemplateEngine extends VelocityTemplateEngine {

    private GenerateData data;
    private SysMenuMapper sysMenuMapper;

    public MyVelocityTemplateEngine(GenerateData data, SysMenuMapper sysMenuMapper) {
        this.data = data;
        this.sysMenuMapper = sysMenuMapper;
    }

    private Integer count = 0;

    @Override
    protected void outputCustomFile( Map<String, String> customFile,  TableInfo tableInfo,  Map<String, Object> objectMap) {
        String entityName = tableInfo.getEntityName();
        String forApiFileName = WordsUtils.firstWordToLower(entityName);
        String otherPath = getPathInfo(OutputFile.other);
        customFile.forEach((key, value) -> {
            String fileName = "";
            if (GenFileTypeEnum.FORM.getValue().equals(key)) {
                fileName = String.format((otherPath + File.separator + "form" + File.separator + entityName  + "%s" + suffixJavaOrKt()), key);
            }
            else if (GenFileTypeEnum.VO.getValue().equals(key)) {
                fileName = String.format((otherPath + File.separator + "vo" + File.separator + entityName  + "%s" + suffixJavaOrKt()), key);
            }
            else if (GenFileTypeEnum.QUERY.getValue().equals(key)) {
                fileName = String.format((otherPath + File.separator + "query" + File.separator + entityName  + "%s" + suffixJavaOrKt()), key);
            }
            else if (GenFileTypeEnum.VIEW.getValue().equals(key) ||
                    GenFileTypeEnum.DIALOG.getValue().equals(key) ||
                    GenFileTypeEnum.TABLE.getValue().equals(key))
            {
                fileName = String.format((otherPath + File.separator + "vue" + File.separator + entityName  + "%s" + ".vue"), key);
            }
            else if (GenFileTypeEnum.STORE.getValue().equals(key)) {
                fileName = String.format((otherPath + File.separator + "vue" + File.separator + "index.js"));
            } else if (GenFileTypeEnum.API.getValue().equals(key)) {
                fileName = String.format((otherPath + File.separator + "vue" + File.separator + forApiFileName  + ".js"));
            }
            outputFile(new File(fileName), objectMap, value, getConfigBuilder().getInjectionConfig().isFileOverride());
        });
    }

    @Override
    public void writer(Map<String, Object> objectMap, String templatePath, File outputFile) throws Exception {
        // 获取table
        TableInfo tableInfo = (TableInfo) objectMap.get("table");
        // 取出Service接口的名字，进行首字母小写处理
        String serviceNameFirstWord = tableInfo.getServiceName().substring(0,1).toLowerCase();
        String serviceNameFirstWordToLower = serviceNameFirstWord + tableInfo.getServiceName().substring(1);
        objectMap.put("serviceNameFirstWordToLower", serviceNameFirstWordToLower);

        // 取出mapper接口的名字，进行首字母小写处理
        String mapperNameFirstWord = tableInfo.getMapperName().substring(0, 1).toLowerCase();
        String mapperNameFirstWordToLower = mapperNameFirstWord + tableInfo.getMapperName().substring(1);
        objectMap.put("mapperNameFirstWordToLower", mapperNameFirstWordToLower);

        // 对Form、VO、query的处理
        objectMap = new FormHandler().handler(objectMap, data);
        super.writer(objectMap, templatePath, outputFile);

        count += 1;
        if (count == 1) {

            String menuPath = "/" + objectMap.get("controllerMappingHyphen");
            String routerName = ((VueTemplate) objectMap.get("vueInfo")).getName();
            String routerComponent = ((VueTemplate) objectMap.get("vueInfo")).getViewName();
            Date now = new Date();
            String dateStr = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(now);

            SysMenu hasMenu = new LambdaQueryChainWrapper<>(sysMenuMapper)
                    .eq(SysMenu::getMenuName, ((TableInfo) objectMap.get("table")).getComment())
                    .eq(SysMenu::getMenuPath, menuPath)
                    .one();
            if (Objects.nonNull(hasMenu)) {
                return;
            }
            SysMenu menu = new SysMenu();
            menu.setMenuName(((TableInfo) objectMap.get("table")).getComment());
            menu.setMenuType(MenuTypeEnum.MENU);
            menu.setStatus(StatusEnum.ACTIVE);
            menu.setMenuIcon(null);
            menu.setMenuPath(menuPath);
            menu.setRouterName(routerName);
            menu.setRouterComponent(routerComponent);
            menu.setPermission(null);
            menu.setOrderNum(100);
            menu.setParentId("0");
            menu.setGmtCreate(now);
            menu.setCreateBy(SecurityUtil.getCurrentUserName());
            menu.setGmtUpdate(now);
            menu.setDeleted(0);
            sysMenuMapper.insert(menu);

            // 然后添加6个操作按钮
            String lowerName = ((VueTemplate) objectMap.get("vueInfo")).getLowerName();
            List<String> permissons = Arrays.asList(lowerName + ":add", lowerName + ":edit", lowerName + ":del",
                    lowerName + ":del:batch", lowerName + ":page", lowerName + ":list", lowerName + "detail");
            for (int i = 0; i < permissons.size(); i++) {
                String per = permissons.get(i);
                String menuName = "";
                if (per.equals(lowerName + ":add")) {
                    menuName = "添加";
                } else if (per.equals(lowerName + ":edit")) {
                    menuName = "修改";
                } else if (per.equals(lowerName + ":del")) {
                    menuName = "删除";
                } else if (per.equals(lowerName + ":del:batch")) {
                    menuName = "批量删除";
                } else if (per.equals(lowerName + ":page")) {
                    menuName = "分页查询";
                } else if (per.equals(lowerName + ":list")) {
                    menuName = "列表查询";
                } else if (per.equals(lowerName + "detail")) {
                    menuName = "查看详情";
                }
                SysMenu button = new SysMenu();
                button.setMenuName(menuName);
                button.setMenuType(MenuTypeEnum.BUTTON);
                button.setStatus(StatusEnum.ACTIVE);
                button.setPermission(per);
                button.setOrderNum(i);
                button.setParentId(menu.getId());
                button.setGmtCreate(now);
                button.setCreateBy(SecurityUtil.getCurrentUserName());
                button.setGmtUpdate(now);
                button.setDeleted(0);
                sysMenuMapper.insert(button);
            }
//            System.out.println("以下为添加菜单的SQL：");
//            System.out.println(String.format("INSERT INTO sys_menu (menu_name, menu_type, STATUS, menu_icon, menu_path, router_name, router_component, permission, order_num, parent_id, gmt_create, create_by, gmt_update, deleted) VALUES" + "(%s, '1', '0', NULL, %s, %s, %s, NULL, 1, 0, %s, 'admin', %s, 0)",
//                    ((TableInfo) objectMap.get("table")).getComment(), menuPath, routerName, routerComponent, dateStr, dateStr));

        }
    }
}