package com.tievd.cube.modules.system.service.impl;

import cn.dev33.satoken.stp.SaTokenInfo;
import cn.dev33.satoken.stp.StpUtil;
import cn.hutool.core.math.Calculator;
import cn.hutool.core.util.RandomUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.crypto.SecureUtil;
import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.tievd.cube.application.config.properties.CaptchaConfigProperties;
import com.tievd.cube.common.system.api.ISysBaseAPI;
import com.tievd.cube.commons.base.Result;
import com.tievd.cube.commons.constant.CacheConst;
import com.tievd.cube.commons.exception.CubeAppException;
import com.tievd.cube.commons.intf.ISmsSender;
import com.tievd.cube.commons.utils.CommonUtil;
import com.tievd.cube.commons.utils.SystemContextUtil;
import com.tievd.cube.commons.utils.captcha.CaptchaGenerator;
import com.tievd.cube.commons.utils.crypto.PasswordTransportUtil;
import com.tievd.cube.commons.utils.crypto.PasswordUtil;
import com.tievd.cube.commons.utils.spring.RedisUtil;
import com.tievd.cube.modules.system.entity.SysDepart;
import com.tievd.cube.modules.system.entity.SysUser;
import com.tievd.cube.modules.system.mapper.SysDepartMapper;
import com.tievd.cube.modules.system.mapper.SysLogMapper;
import com.tievd.cube.modules.system.mapper.SysUserMapper;
import com.tievd.cube.modules.system.model.SmsType;
import com.tievd.cube.modules.system.model.api.response.LogInfoResponse;
import com.tievd.cube.modules.system.model.api.response.LoginResponse;
import com.tievd.cube.modules.system.model.api.response.VisitsInfo;
import com.tievd.cube.modules.system.service.ILoginService;
import com.tievd.cube.modules.system.service.ISysDictService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.Calendar;
import java.util.Date;
import java.util.GregorianCalendar;
import java.util.List;

@Slf4j
@Service
public class LoginServiceImpl implements ILoginService {

    @Autowired
    private SysDepartMapper sysDepartMapper;
    @Autowired
    private SysUserMapper sysUserMapper;
    @Autowired
    private SysLogMapper sysLogMapper;
    @Autowired
    private RedisUtil redisUtil;
    @Autowired
    private ISysBaseAPI sysBaseAPI;
    @Autowired
    private ISysDictService sysDictService;
    @Autowired
    private CaptchaConfigProperties captchaConfigProperties;
    @Autowired
    private CaptchaGenerator captchaGenerator;
    @Autowired(required = false)
    private ISmsSender smsSender;

    @Override
    public LoginResponse getUserInfo(SysUser sysUser) {
        String username = sysUser.getUsername();
        // 获取用户部门信息
        LoginResponse loginResponse = new LoginResponse();
        List<SysDepart> departs = sysDepartMapper.queryUserDeparts(sysUser.getId());
        loginResponse.setDeparts(departs);
        if (departs.size() == 1) {
            SysDepart sysDepart = departs.get(0);
            sysUser.setOrgCode(sysDepart.getOrgCode());
            sysUser.setOrgCodeTxt(sysDepart.getDepartName());
            sysUserMapper.updateUserDepart(username, sysDepart.getOrgCode());
        }
        // 生成token
        StpUtil.login(username);
        SaTokenInfo saTokenInfo = StpUtil.getTokenInfo();
        loginResponse.setToken(saTokenInfo.getTokenValue());
        loginResponse.setTokenTimeout(saTokenInfo.getTokenTimeout());
        loginResponse.setUserInfo(sysUser);
        loginResponse.setUserRoles(sysBaseAPI.queryUserRoles(sysUser.getUsername()));
        loginResponse.setSysAllDictItems(sysDictService.queryAllDictItems());
        // 登录日志
        SystemContextUtil.log("登录成功，登录凭证：" + saTokenInfo.getTokenValue());
        // 删除原有用户信息缓存
        redisUtil.del(String.format("%s::%s", CacheConst.SYS_USERS_CACHE, username));
        return loginResponse;
    }

    @Override
    public LoginResponse loginWithPassword(String username, String password, String captcha, String checkKey) {
        //AES CBC 解密
        password = PasswordTransportUtil.getPlaintext(password);
        if (captchaConfigProperties.isEnable()) {
            if (StrUtil.isEmpty(captcha)) {
                throw new CubeAppException("请填写验证码！");
            }
            String lowerCaseCaptcha = captcha.toLowerCase();
            String realKey = SecureUtil.md5(lowerCaseCaptcha + checkKey);
            Object checkCode = redisUtil.get(realKey);
            //当进入登录页时，有一定几率出现验证码错误 #1714
            if (checkCode == null || !String.valueOf(checkCode).equals(lowerCaseCaptcha)) {
                throw new CubeAppException("验证码错误！");
            }
        }
        //1. 校验用户是否有效
        LambdaQueryWrapper<SysUser> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.eq(SysUser::getUsername, username);
        SysUser sysUser = sysUserMapper.selectOne(queryWrapper);
        Result<?> userState = sysBaseAPI.checkUserState(sysUser);
        if (!userState.isSuccess()) {
            throw new CubeAppException(userState.getMessage());
        }
        //2. 校验用户名或密码是否正确
        String userInputPassword = PasswordUtil.encrypt(username, password, sysUser.getSalt());
        String userPassword = sysUser.getPassword();
        if (!userPassword.equals(userInputPassword)) {
            throw new CubeAppException("用户名或密码错误！");
        }
        return this.getUserInfo(sysUser);
    }

    @Override
    public LoginResponse loginWithCode(String phone, String code) {
        //校验用户有效性
        SysUser sysUser = sysUserMapper.getUserByPhone(phone);
        Result<?> userState = sysBaseAPI.checkUserState(sysUser);
        if (!userState.isSuccess()) {
            throw new CubeAppException(userState.getMessage());
        }
        Object verificationCode = redisUtil.get(phone);
        if (!code.equals(verificationCode)) {
            throw new CubeAppException("手机验证码错误！");
        }
        return this.getUserInfo(sysUser);
    }

    @Override
    public void logout() {
        // 删除用户的缓存信息（包括部门信息），例如sys:cache:user::<username>
        redisUtil.del(String.format("%s::%s", CacheConst.SYS_USERS_CACHE, StpUtil.getLoginIdAsString()));
        StpUtil.logout();
        SystemContextUtil.log("登出成功，登录凭证：" + StpUtil.getTokenInfo().getTokenValue());
    }

    @Override
    public LogInfoResponse visitsToday() {
        // 获取一天的开始和结束时间
        Calendar calendar = new GregorianCalendar();
        calendar.set(Calendar.HOUR_OF_DAY, 0);
        calendar.set(Calendar.MINUTE, 0);
        calendar.set(Calendar.SECOND, 0);
        calendar.set(Calendar.MILLISECOND, 0);
        Date dayStart = calendar.getTime();
        calendar.add(Calendar.DATE, 1);
        Date dayEnd = calendar.getTime();
        // 获取系统访问记录
        Long totalVisitCount = sysLogMapper.findTotalVisitCount();
        Long todayVisitCount = sysLogMapper.findTodayVisitCount(dayStart, dayEnd);
        Long todayIp = sysLogMapper.findTodayIp(dayStart, dayEnd);
        return new LogInfoResponse(totalVisitCount, todayVisitCount, todayIp);
    }

    @Override
    public List<VisitsInfo> visitsSevenDay() {
        Calendar calendar = new GregorianCalendar();
        calendar.set(Calendar.HOUR_OF_DAY, 0);
        calendar.set(Calendar.MINUTE, 0);
        calendar.set(Calendar.SECOND, 0);
        calendar.set(Calendar.MILLISECOND, 0);
        calendar.add(Calendar.DAY_OF_MONTH, 1);
        Date dayEnd = calendar.getTime();
        calendar.add(Calendar.DAY_OF_MONTH, -7);
        Date dayStart = calendar.getTime();
        String dbType = CommonUtil.getDatabaseType();
        return sysLogMapper.findVisitCount(dayStart, dayEnd, dbType);
    }

    @Override
    public SysUser selectDepart(String orgCode) {
        String username = StpUtil.getLoginIdAsString();
        // 重新选择部门时删除原有缓存
        redisUtil.del(String.format("%s::%s", CacheConst.SYS_USERS_CACHE, username));
        if (StrUtil.isNotEmpty(orgCode)) {
            sysUserMapper.updateUserDepart(username, orgCode);
        }
        SysUser sysUser = sysUserMapper.getUserByName(username);
        String departName = sysBaseAPI.translateDictFromTable("sys_depart", "depart_name", "org_code", orgCode);
        sysUser.setOrgCodeTxt(departName);
        return sysUser;
    }

    @Override
    public void sendSms(String phone, String mode) {
        String captcha = RandomUtil.randomNumbers(6);
        SmsType smsType = SmsType.SECURITY_TEMPLATE;
        switch (mode) {
            case "0":
                smsType = SmsType.LOGIN_TEMPLATE;
                break;
            case "1":
                SysUser sysUser = sysUserMapper.getUserByPhone(phone);
                if (sysUser != null) {
                    throw new CubeAppException("手机号已经注册，请直接登录！");
                }
                break;
            case "2":
                smsType = SmsType.FORGET_PASSWORD_TEMPLATE;
                break;
        }
        redisUtil.set(phone, captcha, 600);
        if (smsSender != null) {
            JSONObject params = JSONUtil.createObj();
            params.set("code", captcha);
            smsSender.send(phone, smsType, params);
        }
        log.info("Send SMS template={} code={} phone={} !", smsType.name(), captcha, phone);
    }

    @Override
    public String randomImage(String key) {
        String base64 = captchaGenerator.generate();
        String lowerCaseCode = captchaGenerator.getCode().toLowerCase();
        if (captchaConfigProperties.isMath()) {
            int calculateResult = (int) Calculator.conversion(lowerCaseCode);
            lowerCaseCode = String.valueOf(calculateResult);
        }
        String realKey = SecureUtil.md5(lowerCaseCode + key);
        redisUtil.set(realKey, lowerCaseCode, 60 * 2);
        return base64;
    }

    @Override
    public void checkCaptcha(String captcha, String checkKey) {
        String lowerCaseCaptcha = captcha.toLowerCase();
        String realKey = SecureUtil.md5(lowerCaseCaptcha + checkKey);
        Object checkCode = redisUtil.get(realKey);
        if (checkCode == null || !checkCode.equals(lowerCaseCaptcha)) {
            throw new CubeAppException("验证码错误！");
        }
    }
}