package com.iplatform.generator.controller;

import com.fasterxml.jackson.databind.JsonNode;
import com.iplatform.base.SystemController;
import com.iplatform.generator.GenConstants;
import com.iplatform.generator.JdbcGeneratorEngine;
import com.iplatform.generator.config.JdbcGeneratorProperties;
import com.iplatform.generator.pojo.TableParam;
import com.iplatform.generator.service.CodeGenServiceImpl;
import com.iplatform.generator.util.CodeGenUtils;
import com.iplatform.model.po.S_gen_column;
import com.iplatform.model.po.S_gen_table;
import com.walker.db.TableInfo;
import com.walker.db.page.GenericPager;
import com.walker.db.page.ListPageContext;
import com.walker.db.page.PageSearch;
import com.walker.dbmeta.FieldInfo;
import com.walker.infrastructure.utils.JsonUtils;
import com.walker.infrastructure.utils.StringUtils;
import com.walker.web.ResponseValue;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.io.IOException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 代码生成界面管理接口。
 * @author 时克英
 * @date 2022-11-22
 */
@RestController
@RequestMapping("/tool/gen")
public class CodeGenController extends SystemController {

    public static final String PACKAGE_NAME = "com.iplatform.model.po";

    private CodeGenServiceImpl codeGenService;
    private JdbcGeneratorEngine jdbcGeneratorEngine;
    private JdbcGeneratorProperties jdbcGeneratorProperties;

    @Autowired
    public CodeGenController(CodeGenServiceImpl codeGenService
            , JdbcGeneratorEngine jdbcGeneratorEngine, JdbcGeneratorProperties jdbcGeneratorProperties){
        this.codeGenService = codeGenService;
        this.jdbcGeneratorEngine = jdbcGeneratorEngine;
        this.jdbcGeneratorProperties = jdbcGeneratorProperties;
    }

    @RequestMapping("/list")
    public ResponseValue listGenTable(S_gen_table genTable){
//        this.preparePageSearch();
        GenericPager<S_gen_table> pager = this.codeGenService.queryPageGenTable(genTable.getTable_name(), genTable.getTable_comment());
//        return this.acquireTablePage(pager.getDatas(), pager.getTotalRows());
        return ResponseValue.success(pager);
    }

    @RequestMapping("/db/list")
    public ResponseValue listDbTable(TableParam tableParam){
//        PageSearch pageSearch = this.preparePageSearch();
        PageSearch pageSearch = ListPageContext.getPageSearch();
        ListPageContext.setCurrentPageIndex(pageSearch.getPageIndex());
        ListPageContext.setCurrentPageSize(pageSearch.getPageSize());
        GenericPager<TableInfo> pager = this.jdbcGeneratorEngine.queryPageTableNamesByLike(tableParam.getTableName());
        logger.debug("tableInfo list = " + pager);
//        return this.acquireTablePage(pager.getDatas(), pager.getTotalRows());
        return ResponseValue.success(pager);
    }

    @RequestMapping("/import")
    public ResponseValue saveImportTable(String tables){
        if(StringUtils.isEmpty(tables)){
            return ResponseValue.error("请提供表名称");
        }
        String[] tableArray = StringUtils.toArray(tables);
        if(tableArray.length > 1){
            return ResponseValue.error("暂不支持多表同时导出，请选择一个");
        }
        logger.info("当前保存导入(生成)表 = " + tableArray[0]);
        TableInfo tableInfo = this.jdbcGeneratorEngine.queryOneTableInfo(tableArray[0]);
        if(tableInfo == null){
            return ResponseValue.error("选择的表不存在:" + tables);
        }

        String userName = this.getCurrentUser().getUser_name();
        List<FieldInfo> fieldList = this.jdbcGeneratorEngine.getFieldsObject(tableInfo.getName());
        logger.debug("fieldList = " + fieldList);

        String packageName = this.jdbcGeneratorProperties.getPackageName();
        if(StringUtils.isEmpty(packageName)){
            packageName = PACKAGE_NAME;
            logger.error("packageName 未找到配置项，使用默认值: " + PACKAGE_NAME);
        }
        S_gen_table s_gen_table = CodeGenUtils.createGenTable(tableInfo, userName, packageName);
        List<S_gen_column> columnList = new ArrayList<>();
        if(fieldList != null){
            for(FieldInfo fieldInfo : fieldList){
                S_gen_column s_gen_column = CodeGenUtils.createGenColumn(fieldInfo, userName, s_gen_table.getTable_id());
                columnList.add(s_gen_column);
            }
        }
        this.codeGenService.execInsertOneTableAndColumnList(s_gen_table, columnList);
        return ResponseValue.success();
    }

    /**
     * 重用 /tool/gen/db/** 的权限点，删除已生成的表记录。
     * @param tableIds
     * @return
     * @date 2022-11-26
     */
    @RequestMapping("/db/remove/{tableIds}")
    public ResponseValue removeGenTable(@PathVariable Long[] tableIds){
        if(tableIds == null || tableIds.length == 0){
            return ResponseValue.error("请提供表ID");
        }
        if(tableIds.length > 1){
            return ResponseValue.error("暂不支持多表删除，请选择一个");
        }
        this.codeGenService.execDeleteTableAndColumnList(tableIds[0]);
        return ResponseValue.success();
    }

    /**
     * 显示一个表结构信息，修改界面使用。
     * @param tableId
     * @return
     */
    @GetMapping(value = "/db/{tableId}")
    public ResponseValue showTableInfo(@PathVariable Long tableId){
        if(tableId == null | tableId.longValue() <= 0){
            return ResponseValue.error("缺少表参数");
        }
        S_gen_table s_gen_table = this.codeGenService.get(new S_gen_table(tableId));
        this.setTableFromOptions(s_gen_table);
        List<S_gen_column> columnList = this.codeGenService.queryGenColumnList(tableId);
        Map<String, Object> map = new HashMap<String, Object>(4);
        map.put("info", s_gen_table);
        map.put("rows", columnList);
        return ResponseValue.success(map);
    }

    @GetMapping("/db/gen_po/{tableName}")
    public void downloadGenPoZip(@PathVariable("tableName") String tableName){
        logger.debug(".......... tableName = " + tableName);
        if(StringUtils.isEmpty(tableName)){
            throw new IllegalArgumentException("未提供表明，无法下载生成PO");
        }
        byte[] data = this.jdbcGeneratorEngine.generateOnePoZipFile(tableName);
        try {
            this.downloadSimpleFile(data, tableName + ".zip");
        } catch (IOException e) {
            throw new RuntimeException("下载PO生成异常: " + tableName,e);
        }
    }

    private void setTableFromOptions(S_gen_table genTable){
        String options = genTable.getOptions();
        if(StringUtils.isNotEmpty(options)){
            try {
                JsonNode json = JsonUtils.jsonStringToObject(options, JsonNode.class);
                String treeCode = json.get(GenConstants.TREE_CODE).asText();
                String treeParentCode = json.get(GenConstants.TREE_PARENT_CODE).asText();
                String treeName = json.get(GenConstants.TREE_NAME).asText();
                String parentMenuId = json.get(GenConstants.PARENT_MENU_ID).asText();
                String parentMenuName = json.get(GenConstants.PARENT_MENU_NAME).asText();

                genTable.setTreeCode(treeCode);
                genTable.setTreeParentCode(treeParentCode);
                genTable.setTreeName(treeName);
                genTable.setParentMenuId(parentMenuId);
                genTable.setParentMenuName(parentMenuName);
            } catch (Exception e) {
                throw new RuntimeException(e);
            }
        }
    }
}