package com.ruoyi.business.appuser.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.ruoyi.business.appuser.service.DefaultLoginService;
import com.ruoyi.common.constant.CacheConstants;
import com.ruoyi.common.core.domain.AjaxResult;
import com.ruoyi.common.core.domain.entity.SysUser;
import com.ruoyi.common.core.domain.model.LoginBody;
import com.ruoyi.common.core.redis.RedisCache;
import com.ruoyi.common.utils.SecurityUtils;
import com.ruoyi.framework.web.service.SysLoginService;
import com.ruoyi.framework.web.service.SysPermissionService;
import com.ruoyi.framework.web.service.TokenService;
import com.ruoyi.system.mapper.SysUserRoleMapper;
import com.ruoyi.system.service.ISysUserService;
import lombok.RequiredArgsConstructor;
import org.apache.commons.lang3.RandomStringUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.List;
import java.util.Set;
import java.util.concurrent.TimeUnit;

/**
 * 默认登录
 *
 * @author Tellsea
 * @date 2022/9/2
 */
@Service
@RequiredArgsConstructor(onConstructor_ = @Autowired)
public class DefaultLoginServiceImpl implements DefaultLoginService {

    private final SysLoginService loginService;
    private final SysPermissionService permissionService;
    private final RedisCache redisCache;
    private final ISysUserService sysUserService;

    @Override
    public AjaxResult login(LoginBody entity) {
        return AjaxResult.success("登录成功", loginService.login(entity.getUsername(), entity.getPassword()));
    }

    @Override
    public AjaxResult getInfo() {
        SysUser user = SecurityUtils.getLoginUser().getUser();
        // 角色集合
        Set<String> roles = permissionService.getRolePermission(user);
        // 权限集合
        Set<String> permissions = permissionService.getMenuPermission(user);
        AjaxResult ajax = AjaxResult.success();
        ajax.put("user", user);
        ajax.put("roles", roles);
        ajax.put("permissions", permissions);
        return ajax;
    }

    @Override
    public AjaxResult getCode(String username) {
        String key = CacheConstants.PHONE_CODE + username;
        String code = RandomStringUtils.randomNumeric(4);
        if (redisCache.hasKey(key)) {
            return AjaxResult.success("无需重复发送验证码");
        }
        redisCache.setCacheObject(key, code, 5, TimeUnit.SECONDS);
        // 调用短信平台发送验证码
        return AjaxResult.success("验证码发送成功");
    }

    @Override
    public AjaxResult loginByCode(LoginBody entity) {
        String key = CacheConstants.PHONE_CODE + entity.getUsername();
        if (!redisCache.hasKey(key)) {
            return AjaxResult.error("请先获取短信验证码");
        }
        String cacheCode = redisCache.getCacheObject(key);
        if (StringUtils.isEmpty(cacheCode)) {
            return AjaxResult.error("验证码已失效，请重新获取");
        }
        if (!StringUtils.equals(cacheCode, entity.getCode())) {
            return AjaxResult.error("短信验证码错误");
        }
        entity.setPassword("123456");
        return AjaxResult.success("登录成功", loginService.login(entity.getUsername(), entity.getPassword()));
    }

    @Override
    public AjaxResult register(LoginBody entity) {
        String key = CacheConstants.PHONE_CODE + entity.getUsername();
        if (!redisCache.hasKey(key)) {
            return AjaxResult.error("请先获取短信验证码");
        }
        String cacheCode = redisCache.getCacheObject(key);
        if (StringUtils.isEmpty(cacheCode)) {
            return AjaxResult.error("验证码已失效，请重新获取");
        }
        if (!StringUtils.equals(cacheCode, entity.getCode())) {
            return AjaxResult.error("短信验证码错误");
        }
        List<SysUser> list = sysUserService.list(new LambdaQueryWrapper<SysUser>()
                .eq(SysUser::getUserName, entity.getUsername()));
        if (list.size() > 1) {
            return AjaxResult.error("账号异常，存在多个");
        }
        String token = loginService.getToken(list, entity.getUsername(), entity.getUsername());
        return AjaxResult.success("登录成功", token);
    }
}