package com.product.patch.service;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.google.common.collect.Lists;
import com.product.common.io.FileUtils;
import com.product.common.lang.StringUtils;
import com.product.core.config.Global;
import com.product.core.dao.BaseDao;
import com.product.core.entity.DataTableEntity;
import com.product.core.entity.FieldSetEntity;
import com.product.core.exception.BaseException;
import com.product.core.service.support.AbstractBaseService;
import com.product.patch.config.CmnConst;
import com.product.patch.config.ErrorCode;
import com.product.patch.service.idel.IPatchExtractService;
import com.product.patch.util.RSAUtil;
import com.product.util.BaseUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.io.File;
import java.text.SimpleDateFormat;
import java.util.Arrays;
import java.util.Date;
import java.util.List;
import java.util.Map;

/**
 * Copyright © 6c
 *
 * @Date: 2022-04-02 16:59
 * @Author: 6c
 * @Description: 补丁抽取
 */
@Service
public class PatchExtractService extends AbstractBaseService implements IPatchExtractService {
    @Autowired
    private BaseDao baseDao;

    public void testExtract() {
        FieldSetEntity fse = new FieldSetEntity();
        fse.setTableName("temp_table");
        fse.setValue(CmnConst.FIELD_CLIENT_UUID, "a34f8818-d36d-4d82-b1b0-dbc0d2e70a03");
        fse.setValue("deploy_date", new SimpleDateFormat("yyyy-MM-dd").format(new Date()));
        fse.setValue("operator", "admin");

        JSONObject paramObj = new JSONObject();

        // 表数据
        JSONObject tableReportObj = new JSONObject();
        tableReportObj.put("6c_test_contract", "id<5");
        tableReportObj.put("6c_test_renewal", "id<3");
        paramObj.put(CmnConst.PARAM_TABLE_REPORT, tableReportObj.toJSONString());

        // 表结构
        JSONArray tableStructureArray = new JSONArray();
        JSONObject tableStructureObj = new JSONObject();
        tableStructureObj.put("table_name", "6c_test_contract");
        tableStructureObj.put("field", "title,money");
        tableStructureObj.put("index", "id,uuid");
        tableStructureArray.add(tableStructureObj);
        paramObj.put(CmnConst.PARAM_TABLE_REPORT, tableStructureArray.toJSONString());

        // 功能建模
        JSONObject functionObj = new JSONObject();
        functionObj.put("c4abd684-0788-4a39-9422-3c24bc73d2bc", "1,2,3");
        functionObj.put("d0edf393-db19-44b1-87aa-95d28751b0d8", "1,2");
        tableStructureArray.add(functionObj);
        paramObj.put(CmnConst.PARAM_FUNCTION, functionObj.toJSONString());

        // 流程建模
        JSONObject flowObj = new JSONObject();
        flowObj.put("7c93182a-d091-4a07-b223-56b9bb9fcde8", "1,2,3");
        tableStructureArray.add(functionObj);
        paramObj.put(CmnConst.PARAM_FLOW, flowObj.toJSONString());

        fse.setValue(CmnConst.DATA, paramObj);
        extract(fse);
    }

    /**
     * 获取表、视图、索引信息
     * @return
     */
    public JSONObject listTableInfo() {
        String dataBaseName = baseDao.getDataBaseName();
        JSONObject resultObj = new JSONObject();

        StringBuilder sql = new StringBuilder(512);
        sql.append("\nselect table_name");
        sql.append("\nfrom information_schema.`TABLES`");
        sql.append("\nwhere table_schema=?");
        sql.append("\nand table_type='BASE TABLE'");
        sql.append("\norder by table_name");
        DataTableEntity dte = baseDao.listTable(sql.toString(), new Object[]{dataBaseName});
        resultObj.put(CmnConst.PARAM_TABLE_REPORT, BaseUtil.dataTableEntityToJson(dte));

        sql.setLength(0);
        sql.append("\nselect *");
        sql.append("\nfrom (");
        sql.append("\n    select table_name,1 table_type,1 operate_type,column_name name");
        sql.append("\n    from information_schema.`COLUMNS`");
        sql.append("\n    where table_schema=?");
        sql.append("\n    and table_name in (");
        sql.append("\n        select table_name");
        sql.append("\n        from information_schema.`TABLES`");
        sql.append("\n        where table_schema=?");
        sql.append("\n        and table_type='BASE TABLE'");
        sql.append("\n    )");
        sql.append("\n    union all");
        sql.append("\n    select table_name,1 table_type,2 operate_type,column_name name");
        sql.append("\n    from information_schema.`STATISTICS`");
        sql.append("\n    where table_schema=?");
        sql.append("\n    and table_name in (");
        sql.append("\n        select table_name");
        sql.append("\n        from information_schema.`TABLES`");
        sql.append("\n        where table_schema=?");
        sql.append("\n        and table_type='BASE TABLE'");
        sql.append("\n    )");
        sql.append("\n    union all");
        sql.append("\n    select table_name,2 table_type,null,null");
        sql.append("\n    from information_schema.`VIEWS`");
        sql.append("\n    where table_schema=?");
        sql.append("\n    order by table_name");
        sql.append("\n) t");
        sql.append("\norder by table_type,table_name,operate_type,name");
        dte = baseDao.listTable(sql.toString(), new Object[]{dataBaseName, dataBaseName, dataBaseName, dataBaseName, dataBaseName});
        resultObj.put(CmnConst.PARAM_TABLE_STRUCTURE, BaseUtil.dataTableEntityToJson(dte));

        sql.setLength(0);
        sql.append("\nselect uuid,tricode,function_name,t.*");
        sql.append("\nfrom product_sys_functions f,(");
        sql.append("\n    select \"按钮\" type union all select \"页面\" union all select \"连线\"");
        sql.append("\n) t");
        sql.append("\nwhere data_type=1");
        sql.append("\norder by tricode");
        dte = baseDao.listTable(sql.toString(), new Object[]{});
        resultObj.put(CmnConst.PARAM_FUNCTION, BaseUtil.dataTableEntityToJson(dte));

        sql.setLength(0);
        sql.append("\nselect uuid,title,t.*");
        sql.append("\nfrom product_sys_flow_model f,(");
        sql.append("\n    select \"节点\" type union all select \"处理器\" union all select \"连线\"");
        sql.append("\n) t");
        sql.append("\nwhere length(type_code)>0");
        sql.append("\norder by title");
        dte = baseDao.listTable(sql.toString(), new Object[]{});
        resultObj.put(CmnConst.PARAM_FLOW, BaseUtil.dataTableEntityToJson(dte));

        return resultObj;
    }

    /**
     * 补丁抽取
     * @param fse
     */
    public void extract(FieldSetEntity fse) {
        String aimClientUUID = fse.getString(CmnConst.FIELD_CLIENT_UUID);
        if (StringUtils.isEmpty(aimClientUUID)) {
            throw new BaseException(ErrorCode.EXTRACT_NO_AIM_CLIENT_UUID);
        }
        JSONObject paramObj = JSON.parseObject(fse.getString(CmnConst.DATA));
        JSONArray resultArray = new JSONArray();
        // 表数据
        resultArray.addAll(extractTableRecord(paramObj.getJSONObject(CmnConst.PARAM_TABLE_REPORT)));
        // 表结构
        resultArray.addAll(extractTableStructure(paramObj.getJSONArray(CmnConst.PARAM_TABLE_STRUCTURE)));
        // 功能建模
        resultArray.addAll(extractFunction(paramObj.getJSONArray(CmnConst.PARAM_FUNCTION)));
        // 流程建模
        resultArray.addAll(extractTableStructure(paramObj.getJSONArray(CmnConst.PARAM_FLOW)));

        Date deployDate = fse.getDate(CmnConst.PARAM_DEPLOY_DATE);
        String operator = fse.getString(CmnConst.PARAM_OPERATOR);
        writeToFile(resultArray.toString(), deployDate, operator, aimClientUUID);
    }

    /**
     * 表数据
     * @param paramObj
     */
    private JSONArray extractTableRecord(JSONObject paramObj) {
        DataTableEntity dte;
        JSONArray resultArray = new JSONArray();
        for (Map.Entry<String, Object> entry : paramObj.entrySet()) {
            dte = baseDao.listTable(entry.getKey(), entry.getValue() == null ? "" : entry.getValue().toString(), new Object[]{});
            resultArray.addAll(dte.toJson());
        }
        return resultArray;
    }

    /**
     * 表结构
     * @param paramArr
     */
    private JSONArray extractTableStructure(JSONArray paramArr) {
        JSONArray resultArray = new JSONArray();
        JSONObject resultObj;
        JSONObject tempObj;
        for (Object obj : paramArr) {
            tempObj = (JSONObject) obj;
            if ("1".equals(tempObj.getString(CmnConst.TABLE_TYPE))) {
                // 表
                if ("1".equals(tempObj.getString(CmnConst.OPERATE_TYPE))) {
                    // 字段
                    resultObj = getTable(tempObj);
                } else if ("2".equals(tempObj.getString(CmnConst.OPERATE_TYPE))) {
                    // 索引
                    resultObj = getIndex(tempObj);
                } else {
                    throw new BaseException(ErrorCode.EXTRACT_INVALID_OPERATE_TYPE);
                }
            } else if ("2".equals(tempObj.getString(CmnConst.TABLE_TYPE))) {
                // 视图
                resultObj = getView(tempObj);
            } else {
                throw new BaseException(ErrorCode.EXTRACT_INVALID_TABLE_TYPE);
            }
            if (resultObj != null && !resultObj.isEmpty()) {
                resultArray.add(resultObj);
            }
        }
        return resultArray;
    }

    /**
     * 表结构-字段
     * @param jsonObject
     * @return
     */
    private JSONObject getTable(JSONObject jsonObject) {
        String name = jsonObject.getString(CmnConst.NAME);
        if (StringUtils.isEmpty(name)) {
            return new JSONObject();
        }
        String[] nameArr = name.split(",");
        String dataBaseName = baseDao.getDataBaseName();
        List<String> paramList = Lists.newArrayList();
        paramList.add(dataBaseName);
        paramList.add(jsonObject.getString(CmnConst.TABLE_NAME));
        paramList.addAll(Arrays.asList(nameArr));
        String filter = String.format("%s=? and %s='%s' and %s=? and %s", CmnConst.DATABASE_COLUMN, CmnConst.TABLE_TYPE_COLUMN, CmnConst.TABLE_TYPE_TABLE, CmnConst.TABLE_NAME, BaseUtil.buildQuestionMarkFilter(CmnConst.COLUMNS_NAME_COLUMN, nameArr.length, true));
        String tableName = String.format("%s.`%s`", CmnConst.DATABASE, CmnConst.TABLE);
        FieldSetEntity fse = baseDao.getFieldSetByFilter(tableName, filter, paramList.toArray(), false);
        return BaseUtil.fieldSetEntityToJson(fse);
    }

    /**
     * 表结构-索引
     * @param jsonObject
     * @return
     */
    private JSONObject getIndex(JSONObject jsonObject) {
        String name = jsonObject.getString(CmnConst.NAME);
        if (StringUtils.isEmpty(name)) {
            return new JSONObject();
        }
        String[] nameArr = name.split(",");
        String dataBaseName = baseDao.getDataBaseName();
        List<String> paramList = Lists.newArrayList();
        paramList.add(dataBaseName);
        paramList.add(jsonObject.getString(CmnConst.TABLE_NAME));
        paramList.addAll(Arrays.asList(nameArr));
        String filter = String.format("%s=? and %s=? and %s=?", CmnConst.DATABASE_COLUMN, CmnConst.TABLE_NAME, BaseUtil.buildQuestionMarkFilter(CmnConst.INDEX_NAME_COLUMN, nameArr.length, true));
        String tableName = String.format("%s.`%s`", CmnConst.DATABASE, CmnConst.INDEX);
        FieldSetEntity fse = baseDao.getFieldSetByFilter(tableName, filter, paramList.toArray(), false);
        return BaseUtil.fieldSetEntityToJson(fse);
    }

    /**
     * 表结构-视图
     * @param jsonObject
     * @return
     */
    private JSONObject getView(JSONObject jsonObject) {
        String name = jsonObject.getString(CmnConst.NAME);
        if (StringUtils.isEmpty(name)) {
            return new JSONObject();
        }
        String[] nameArr = name.split(",");
        String dataBaseName = baseDao.getDataBaseName();
        List<String> paramList = Lists.newArrayList();
        paramList.add(dataBaseName);
        paramList.add(jsonObject.getString(CmnConst.TABLE_NAME));
        paramList.addAll(Arrays.asList(nameArr));
        String filter = String.format("%s=? and %s=? and %s=?", CmnConst.DATABASE_COLUMN, CmnConst.TABLE_NAME, BaseUtil.buildQuestionMarkFilter(CmnConst.INDEX_NAME_COLUMN, nameArr.length, true));
        String tableName = String.format("%s.`%s`", CmnConst.DATABASE, CmnConst.VIEWS);
        FieldSetEntity fse = baseDao.getFieldSetByFilter(tableName, filter, paramList.toArray(), false);
        return BaseUtil.fieldSetEntityToJson(fse);
    }

    /**
     * 功能建模
     * @param paramArr
     * @return
     */
    private JSONArray extractFunction(JSONArray paramArr) {
        JSONArray resultArray = new JSONArray();
        JSONObject paramObj;
        for (int i = 0;i < paramArr.size();i++) {
            paramObj = paramArr.getJSONObject(i);
            // todo
//            resultArray.add();
        }
        return resultArray;
    }

    /**
     * 补丁输出到文件
     * @param patchContent
     * @param deployDate
     * @param operator
     * @param clientUUID
     */
    private void writeToFile(String patchContent, Date deployDate, String operator, String clientUUID) {
        deployDate = deployDate == null ? new Date() : deployDate;
        operator = StringUtils.isEmpty(operator) ? "admin" : operator;
        String patchDir = Global.getSystemConfig("patch.extract.dir","");
        File dir = new File(patchDir);
        dir.mkdirs();
        String deployDateStr = new SimpleDateFormat(CmnConst.DATE_FORMAT_1).format(deployDate);
        String fileName = patchDir + File.separator + deployDateStr + "-" + operator + ".patch";
        File file = new File(fileName);
        boolean appendFlag = false;
        if (file.exists()) {
            appendFlag = true;
        } else {
            try {
                file.createNewFile();
            } catch (Exception e) {
                throw new BaseException(ErrorCode.EXTRACT_CREATE_FILE_FAIL);
            }
        }
        StringBuilder content = new StringBuilder(1024);
        content.append(deployDateStr).append("-").append(operator).append(";\n")
                .append(patchContent);
        String encodedData = content.toString();

        // 加密
        FieldSetEntity mainFse = baseDao.getFieldSetEntityByFilter(CmnConst.TABLE_PRODUCT_SYS_ORG_LEVELS, "org_level_id=1", new Object[]{}, false);
        FieldSetEntity mainCryptFse = baseDao.getFieldSetEntityByFilter(CmnConst.TABLE_PRODUCT_SYS_PATCH_KEY, "org_level_uuid=?", new Object[]{mainFse.getUUID()}, false);
        String  mainPrivateKey = mainCryptFse.getString(CmnConst.FIELD_PIRVATE_KEY);
        FieldSetEntity clientCryptFse = baseDao.getFieldSetEntityByFilter(CmnConst.TABLE_PRODUCT_SYS_PATCH_KEY, "org_level_uuid=?", new Object[]{clientUUID}, false);
        String clientPublicKey = clientCryptFse.getString(CmnConst.FIELD_PUBLIC_KEY);
        try {
            encodedData = RSAUtil.privateEncrypt(encodedData, RSAUtil.getPrivateKey(mainPrivateKey));
            encodedData = RSAUtil.publicEncrypt(encodedData, RSAUtil.getPublicKey(clientPublicKey));
        } catch (Exception e) {
            e.printStackTrace();
            throw new BaseException(ErrorCode.EXTRACT_ENCODE_FAIL);
        }

        FileUtils.writeToFile(fileName, encodedData, "utf-8", appendFlag);
    }
}