package cn.stylefeng.guns.generator.core.generator.guns.sqls;

import cn.hutool.core.util.StrUtil;
import cn.stylefeng.guns.generator.core.generator.base.AbstractCustomGenerator;
import cn.stylefeng.guns.generator.core.util.TemplateUtil;
import com.baomidou.mybatisplus.core.toolkit.IdWorker;
import com.baomidou.mybatisplus.generator.config.po.TableInfo;
import org.beetl.core.Template;

import java.io.File;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

/**
 * Guns菜单相关的sql生成器
 *
 * @author fengshuonan
 * @date 2018-12-13-2:20 PM
 */
public class GunsMenuSqlGenerator extends AbstractCustomGenerator {

    public GunsMenuSqlGenerator(Map<String, Object> tableContext) {
        super(tableContext);
    }

    @Override
    public void bindingOthers(Template template) {
        TableInfo tableInfo = (TableInfo) tableContext.get("table");
        template.binding("menus", getMenus(tableInfo));
    }

    @Override
    public String getTemplateResourcePath() {
        return "/gunsTemplates/menu_sql.sql.btl";
    }

    @Override
    public String getGenerateFilePath() {
        String lowerEntity = (String) this.tableContext.get("lowerEntity");
        File file = new File(contextParam.getOutputPath() + "/sqls/" + lowerEntity + "_menus.sql");
        return file.getAbsolutePath();
    }

    private List<MenuModel> getMenus(TableInfo tableInfo) {
        ArrayList<MenuModel> menuModels = new ArrayList<>();

        MenuModel parentMenu = new MenuModel();
        parentMenu.setMenuId(IdWorker.getId());
        parentMenu.setCode(StrUtil.toUnderlineCase(tableInfo.getEntityName()).toUpperCase());
        parentMenu.setPcode("0");
        parentMenu.setPcodes("[0],");
        if (StrUtil.isBlank(TemplateUtil.cleanWhite(tableInfo.getComment()))) {
            parentMenu.setName("空表名");
        } else {
            parentMenu.setName(TemplateUtil.cleanWhite(tableInfo.getComment()));
        }
        parentMenu.setIcon("fa-star");
        parentMenu.setUrl("/" + TemplateUtil.lowerFirst((String) tableContext.get("entity")));
        parentMenu.setSort(999);
        parentMenu.setLevels(1);
        parentMenu.setMenuFlag("Y");
        menuModels.add(parentMenu);

        MenuModel addMenu = new MenuModel();
        addMenu.setMenuId(IdWorker.getId());
        addMenu.setCode(StrUtil.toUnderlineCase(tableInfo.getEntityName()).toUpperCase() + "_ADD");
        addMenu.setPcode(parentMenu.getCode());
        addMenu.setPcodes("[0],[" + parentMenu.getCode() + "],");
        addMenu.setName(parentMenu.getName() + "添加");
        addMenu.setIcon("fa-star");
        addMenu.setSort(999);
        addMenu.setLevels(2);
        addMenu.setMenuFlag("N");
        menuModels.add(addMenu);

        MenuModel editMenu = new MenuModel();
        editMenu.setMenuId(IdWorker.getId());
        editMenu.setCode(StrUtil.toUnderlineCase(tableInfo.getEntityName()).toUpperCase() + "_EDIT");
        editMenu.setPcode(parentMenu.getCode());
        editMenu.setPcodes("[0],[" + parentMenu.getCode() + "],");
        editMenu.setName(parentMenu.getName() + "修改");
        editMenu.setIcon("fa-star");
        editMenu.setSort(999);
        editMenu.setLevels(2);
        editMenu.setMenuFlag("N");
        menuModels.add(editMenu);

        MenuModel deleteMenu = new MenuModel();
        deleteMenu.setMenuId(IdWorker.getId());
        deleteMenu.setCode(StrUtil.toUnderlineCase(tableInfo.getEntityName()).toUpperCase() + "_DELETE");
        deleteMenu.setPcode(parentMenu.getCode());
        deleteMenu.setPcodes("[0],[" + parentMenu.getCode() + "],");
        deleteMenu.setName(parentMenu.getName() + "删除");
        deleteMenu.setIcon("fa-star");
        deleteMenu.setSort(999);
        deleteMenu.setLevels(2);
        deleteMenu.setMenuFlag("N");
        menuModels.add(deleteMenu);

        return menuModels;

    }
}
