package com.monkeylessey.sys.service.impl;

import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.monkeylessey.gen.domain.ReturnColumn;
import com.monkeylessey.gen.utils.WordsUtils;
import com.monkeylessey.sys.domain.entity.SysGenSet;
import com.monkeylessey.sys.domain.entity.SysTable;
import com.monkeylessey.gen.domain.GenerateData;
import com.monkeylessey.gen.utils.GenerateCodeUtil;
import com.monkeylessey.sys.domain.vo.TableColumnVO;
import com.monkeylessey.sys.mapper.SysGenSetMapper;
import com.monkeylessey.sys.mapper.SysTableMapper;
import com.monkeylessey.response.Result;
import com.monkeylessey.sys.service.SysTableService;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;
import org.springframework.util.StringUtils;

import java.math.BigDecimal;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;

/**
 * @author 29443
 * @date 2022/5/22
 */
@Service
@RequiredArgsConstructor
public class SysTableServiceImpl implements SysTableService {

    private final List<String> excludeTable = Arrays.asList("QRTZ_BLOB_TRIGGERS",
            "QRTZ_CALENDARS",
            "QRTZ_CRON_TRIGGERS",
            "QRTZ_FIRED_TRIGGERS",
            "QRTZ_JOB_DETAILS",
            "QRTZ_LOCKS",
            "QRTZ_PAUSED_TRIGGER_GRPS",
            "QRTZ_SCHEDULER_STATE",
            "QRTZ_SIMPLE_TRIGGERS",
            "QRTZ_SIMPROP_TRIGGERS",
            "QRTZ_TRIGGERS",
            "sys_file",
            "sys_op_log");

    private final List<String> excludeColumn = Arrays.asList("deleted");

    private final SysTableMapper tableMapper;
    private final SysGenSetMapper genSetMapper;
    private final GenerateCodeUtil generateCodeUtil;


    @Override
    public Result getTableNames(String dataBaseName) {
        List<SysTable> tables = tableMapper.getTableNames(dataBaseName);
        List<SysTable> exclude = new ArrayList<>();
        for (SysTable table : tables) {
            if (excludeTable.contains(table.getTableName())) {
                exclude.add(table);
            }
        }
        tables.removeAll(exclude);
        return Result.ok().data(tables);
    }

    @Override
    public Result getTableColumns(String tableName, String dataBaseName) throws JsonProcessingException {
        if (! StringUtils.hasText(dataBaseName) || ! StringUtils.hasText(tableName)) {
            throw new IllegalArgumentException("参数错误");
        }
        List<TableColumnVO> tableColumns = tableMapper.getTableColumns(tableName, dataBaseName);
        SysGenSet set = new LambdaQueryChainWrapper<>(genSetMapper)
                .eq(SysGenSet::getTableName, tableName)
                .one();
        if (Objects.nonNull(set)) {
            ObjectMapper json = new ObjectMapper();
            return Result.ok().data(json.readValue(set.getContent(), GenerateData.class).getColumns());
        }
        List<TableColumnVO> data = tableColumns.stream()
                .filter(item -> !excludeColumn.contains(item.getColumnName()))
                .collect(Collectors.toList());
        data.stream().forEach(item -> {
            if (item.getColumnName().equals("id")) {
                item.setAddRequired(Boolean.FALSE);
                item.setEditRequired(Boolean.TRUE);
            } else {
                item.setAddRequired(Boolean.TRUE);
                item.setEditRequired(Boolean.TRUE);
            }
            item.setInputWay("input");
        });
        return Result.ok().data(data);
    }

    @Override
    public Result getTableColumnsSelect(String tableName, String dataBaseName, String keyword) {
        if (! StringUtils.hasText(dataBaseName) || ! StringUtils.hasText(tableName)) {
            throw new IllegalArgumentException("参数错误");
        }
        List<String> tableColumns = tableMapper.getTableColumnsSelect(tableName, dataBaseName, keyword);
        return Result.ok().data(tableColumns);
    }

    @Override
    public Result generateCode(GenerateData data) throws JsonProcessingException {
        data.getColumns().stream().forEach(item -> {
            // 处理一下，方便模板使用
            if (!CollectionUtils.isEmpty(item.getReturnColumnList())) {
                item.getReturnColumnList().stream().forEach(returnC -> {
                    returnC.setProperty(WordsUtils.convertToCamelCase(returnC.getColumn()));
                });
            }
            // 转换类型
            if ("int".equals(item.getColumnType())) {
                item.setColumnType("Integer");
            }
            else if ("bigint".equals(item.getColumnType())) {
                item.setColumnType("Long");
            }
            else if ("varchar".equals(item.getColumnType()) || "text".equals(item.getColumnType())) {
                item.setColumnType("String");
            }
            else if ("char".equals(item.getColumnType())) {
                item.setColumnType("String");
            }
            else if ("datetime".equals(item.getColumnType())) {
                item.setColumnType("LocalDateTime");
            }
            if ("decimal".equals(item.getColumnType())) {
                item.setColumnType("BigDecimal");
            }
            // 字段名转换
            item.setPropertyName(WordsUtils.convertToCamelCase(item.getColumnName()));
            if (StringUtils.hasText(item.getRefTableName())) {
                item.setRefTableVoName(WordsUtils.firstWordToUp(WordsUtils.convertToCamelCase(item.getRefTableName())) + "VO");
                item.setRefTableProperty(WordsUtils.firstWordToUp(WordsUtils.convertToCamelCase(item.getRefTableName())) + "VO");
            }
        });
        SysGenSet set = new LambdaQueryChainWrapper<>(genSetMapper)
                .eq(SysGenSet::getTableName, data.getTableName())
                .one();
        ObjectMapper json = new ObjectMapper();
        String jstr = json.writeValueAsString(data);
        if (Objects.isNull(set)) {
            set = new SysGenSet();
            set.setTableName(data.getTableName());
            set.setContent(jstr);
            genSetMapper.insert(set);
        } else {
            set.setContent(jstr);
            genSetMapper.updateById(set);
        }
        generateCodeUtil.generateCode(data);
        return Result.ok("代码生成成功");
    }
}