package com.wgcloud.controller;

import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;
import com.github.pagehelper.PageInfo;
import com.wgcloud.config.CommonConfig;
import com.wgcloud.dto.MessageDto;
import com.wgcloud.entity.AccountInfo;
import com.wgcloud.entity.DbInfo;
import com.wgcloud.service.DbInfoService;
import com.wgcloud.service.DbTableService;
import com.wgcloud.service.LogInfoService;
import com.wgcloud.util.*;
import com.wgcloud.util.jdbc.ConnectionUtil;
import com.wgcloud.util.jdbc.RDSConnection;
import com.wgcloud.util.staticvar.StaticKeys;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.servlet.mvc.support.RedirectAttributes;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * @version v3.3
 * @ClassName:DbInfoController.java
 * @author: http://www.wgstart.com
 * @date: 2021年1月16日
 * @Description: 数据源配置
 * @Copyright: 2019-2021 wgcloud. All rights reserved.
 */
@Controller
@RequestMapping("/dbInfo")
public class DbInfoController {


    private static final Logger logger = LoggerFactory.getLogger(DbInfoController.class);

    @Resource
    private DbInfoService dbInfoService;
    @Resource
    private DbTableService dbTableService;
    @Resource
    private LogInfoService logInfoService;
    @Resource
    private ConnectionUtil connectionUtil;
    @Resource
    private RedisUtil redisUtil;
    @Resource
    private MongoDbUtil mongoDbUtil;
    @Autowired
    private CommonConfig commonConfig;
    @Autowired
    private TokenUtils tokenUtils;


    /**
     * agent查询监控数据源列表
     *
     * @param paramBean
     * @return
     */
    @ResponseBody
    @RequestMapping(value = "agentList")
    public String agentList(@RequestBody String paramBean) {
        JSONObject agentJsonObject = (JSONObject) JSONUtil.parse(paramBean);
        if (!tokenUtils.checkAgentToken(agentJsonObject)) {
            logger.error(StaticKeys.TOKEN_ERROR);
            return ResDataUtils.resetErrorJson(StaticKeys.TOKEN_ERROR);
        }
        Map<String, Object> params = new HashMap<String, Object>();
        try {
            if (null != agentJsonObject.get("dbInfoNames") && !StringUtils.isEmpty(agentJsonObject.get("dbInfoNames").toString())) {
                params.put("dbInfoNames", agentJsonObject.get("dbInfoNames").toString().split(","));
            }
            List<DbInfo> dbInfoList = dbInfoService.selectAllByParams(params);
            //数据源信息加密处理 begin
            for (DbInfo dbInfo : dbInfoList) {
                dbInfo.setUserName(DESUtil.encryption(dbInfo.getUserName()));
                dbInfo.setDbUrl(DESUtil.encryption(dbInfo.getDbUrl()));
                dbInfo.setPasswd(DESUtil.encryption(dbInfo.getPasswd()));
            }
            //数据源信息加密处理 end

            //将server-backup节点处理的数据源，标记起来，server节点就不监控这些数据源了
            ServerBackupUtil.cacheSaveDbInfoId(dbInfoList);
            return ResDataUtils.resetSuccessJson(dbInfoList);
        } catch (Exception e) {
            logger.error("agent获取监控数据源信息错误", e);
            logInfoService.save("agent获取监控数据源信息错误", e.toString(), StaticKeys.LOG_XTCZ);
            return ResDataUtils.resetErrorJson(e.toString());
        }
    }

    /**
     * 测试数据库连接
     *
     * @param model
     * @param request
     * @return
     */
    @ResponseBody
    @RequestMapping(value = "validate")
    public String valdateDbInfo(DbInfo DbInfo, Model model, HttpServletRequest request) {
        MessageDto messageDto = new MessageDto();
        try {
            if (RDSConnection.KEY_REDIS.equals(DbInfo.getDbType())) {
                String pong = redisUtil.connectRedis(DbInfo);
                if (StringUtils.isEmpty(pong)) {
                    messageDto.setCode("1");
                    messageDto.setMsg("连接Redis错误，请检查参数是否正确。请在系统信息里查看日志");
                } else {
                    messageDto.setCode("0");
                    messageDto.setMsg("连接Redis成功");
                }
            } else if (RDSConnection.KEY_MONGODB.equals(DbInfo.getDbType())) {
                mongoDbUtil.connectMongoDb(DbInfo);
                if (StaticKeys.DOWN_STATE.equals(DbInfo.getDbState())) {
                    messageDto.setCode("1");
                    messageDto.setMsg("连接MongoDb错误，请检查参数是否正确。请在系统信息里查看日志");
                } else {
                    messageDto.setCode("0");
                    messageDto.setMsg("连接MongoDb成功");
                }
            } else {
                JdbcTemplate JdbcTemplate = connectionUtil.getJdbcTemplate(DbInfo);
                if (JdbcTemplate == null) {
                    messageDto.setCode("1");
                    messageDto.setMsg("连接数据库错误，请检查参数是否正确。请在系统信息里查看日志");
                } else {
                    messageDto.setCode("0");
                    messageDto.setMsg("连接数据库成功");
                }
            }
        } catch (Exception e) {
            logger.error("测试数据源信息错误", e);
            logInfoService.save("测试数据源信息错误", e.toString(), StaticKeys.LOG_XTCZ);

        }
        return JSONUtil.toJsonStr(messageDto);
    }


    /**
     * 根据条件查询列表
     *
     * @param model
     * @param request
     * @return
     */
    @RequestMapping(value = "list")
    public String dbInfoList(DbInfo DbInfo, Model model, HttpServletRequest request) {
        Map<String, Object> params = new HashMap<String, Object>();
        try {

            StringBuffer url = new StringBuffer();
            if (!StringUtils.isEmpty(DbInfo.getAccount())) {
                params.put("account", DbInfo.getAccount());
                url.append("&account=").append(DbInfo.getAccount());
            }

            //校验是否需要添加过滤用户查询条件
            HostUtil.addAccountquery(request, params);

            PageInfo<DbInfo> pageInfo = dbInfoService.selectByParams(params, DbInfo.getPage(), DbInfo.getPageSize());
            //设置累积告警次数 begin
            if (StaticKeys.TRUE_VAL.equals(commonConfig.getShowWarnCount())) {
                Map<String, Object> paramsLogInfo = new HashMap<String, Object>();
                for (DbInfo dbInfo : pageInfo.getList()) {
                    String warnQueryWd = "数据源连接失败告警：" + dbInfo.getAliasName();
                    paramsLogInfo.clear();
                    paramsLogInfo.put("hostname", warnQueryWd);
                    dbInfo.setWarnCount(logInfoService.countByParams(paramsLogInfo));
                    dbInfo.setWarnQueryWd(warnQueryWd);
                }
            }
            //设置累积告警次数 end

            //设置数据库logo
            dbInfoService.dbAddLogo(pageInfo.getList());

            PageUtil.initPageNumber(pageInfo, model);

            //设置用户列表
            HostUtil.addAccountListModel(model);

            model.addAttribute("pageUrl", "/dbInfo/list?1=1" + url.toString());
            model.addAttribute("page", pageInfo);
        } catch (Exception e) {
            logger.error("查询数据源信息错误", e);
            logInfoService.save("查询数据源信息错误", e.toString(), StaticKeys.LOG_XTCZ);

        }
        return "mysql/dbList";
    }


    /**
     * 添加
     *
     * @param model
     * @param request
     * @return
     */
    @RequestMapping(value = "edit")
    public String edit(Model model, HttpServletRequest request) {
        String errorMsg = "添加数据源错误";
        String id = request.getParameter("id");
        DbInfo dbInfo = new DbInfo();
        try {

            if (StringUtils.isEmpty(id)) {
                model.addAttribute("dbInfo", dbInfo);
                return "mysql/addDb";
            }
            dbInfo = dbInfoService.selectById(id);
            model.addAttribute("dbInfo", dbInfo);
        } catch (Exception e) {
            logger.error(errorMsg, e);
            logInfoService.save(errorMsg, e.toString(), StaticKeys.LOG_XTCZ);
        }
        return "mysql/addDb";
    }


    /**
     * 保存数据源信息
     *
     * @param DbInfo
     * @param model
     * @param request
     * @return
     */
    @RequestMapping(value = "save")
    public String saveDbInfo(DbInfo DbInfo, Model model, HttpServletRequest request) {
        try {
            if (StringUtils.isEmpty(DbInfo.getId())) {
                AccountInfo accountInfo = HostUtil.getAccountByRequest(request);
                if (null != accountInfo) {
                    if (!StaticKeys.ROLE_ADMIN.equals(accountInfo.getRole())) {
                        DbInfo.setAccount(accountInfo.getAccount());
                    }
                }
                dbInfoService.save(DbInfo);
                dbInfoService.saveLog(request, StaticKeys.LOG_ADD, DbInfo);
            } else {
                dbInfoService.updateById(DbInfo);
                dbInfoService.saveLog(request, StaticKeys.LOG_UPDATE, DbInfo);
            }
        } catch (Exception e) {
            logger.error("保存数据源错误", e);
            logInfoService.save("保存数据源错误", e.toString(), StaticKeys.LOG_XTCZ);
        }
        return "redirect:/dbInfo/list?msg=save";
    }


    /**
     * 删除数据源
     *
     * @param model
     * @param request
     * @param redirectAttributes
     * @return
     */
    @RequestMapping(value = "del")
    public String delete(Model model, HttpServletRequest request, RedirectAttributes redirectAttributes) {
        String errorMsg = "删除数据源信息错误";
        DbInfo DbInfo = new DbInfo();
        try {
            if (!StringUtils.isEmpty(request.getParameter("id"))) {
                String[] ids = request.getParameter("id").split(",");
                for (String id : ids) {
                    DbInfo = dbInfoService.selectById(id);
                    dbInfoService.saveLog(request, StaticKeys.LOG_DEL, DbInfo);
                    dbTableService.deleteByDbInfoId(DbInfo.getId());
                }
                dbInfoService.deleteById(ids);
            }
        } catch (Exception e) {
            logger.error(errorMsg, e);
            logInfoService.save(errorMsg, e.toString(), StaticKeys.LOG_XTCZ);
        }
        return "redirect:/dbInfo/list?msg=del";
    }

}