package com.wms_admin.sys.service.impl;

import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.wms_admin.clients.UserLoginUtils;
import com.wms_admin.shiro.realm.ShiroRealm;
import com.wms_admin.sys.entity.User;
import com.wms_admin.sys.mapper.UserMapper;
import com.wms_admin.sys.service.UserService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.wms_admin.utils.*;
import io.swagger.models.auth.In;
import org.apache.shiro.SecurityUtils;
import org.apache.shiro.authc.IncorrectCredentialsException;
import org.apache.shiro.authc.LockedAccountException;
import org.apache.shiro.authc.UnknownAccountException;
import org.apache.shiro.authc.UsernamePasswordToken;
import org.apache.shiro.crypto.hash.Md5Hash;
import org.apache.shiro.subject.Subject;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
import org.springframework.util.FileSystemUtils;
import org.springframework.util.ObjectUtils;

import javax.annotation.Resource;
import java.io.File;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Objects;

/**
 * <p>
 *  服务实现类
 * </p>
 *
 * @author 江苏鵷雏网络科技有限公司
 * @since 2023-05-24
 */
@Service
public class UserServiceImpl extends ServiceImpl<UserMapper, User> implements UserService {

    @Resource
    private UserMapper mapper;

    @Value("${file.path}")
    private String FILE_PATH;

    @Value("${avatar}")
    private String avatar;

    @Override
    public User accordingUsernameSelectAll(String account) {
        LambdaQueryWrapper<User> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(User::getAccount,account);
        return mapper.selectOne(wrapper);
    }

    @Override
    public Boolean AddUser(User user) {
        LambdaQueryWrapper<User> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(User::getAccount, user.getAccount());
        User userIsNull = mapper.selectOne(wrapper);
        if (!ObjectUtils.isEmpty(userIsNull)){ // 如果查询数据返回false
            return false;
        } else { // 没有查询到则进行添加操作
            //1.获取随机盐
            String salt = MyUtils.getSalt(10);
            user.setSalt(salt);
            //3.明文密码进行md5 + salt + hash散列
            Md5Hash MD5 = new Md5Hash(user.getPassword(),salt,1024);
            user.setPassword(MD5.toString());
            user.setAvatar(avatar);
            mapper.insert(user);
            return true;
        }
    }

    @Override
    public Result<Map<String, Object>> login(String account, String password) {
        boolean loginSuccess = false;
        Subject subject = SecurityUtils.getSubject();
        if (!subject.isAuthenticated()) {
            UsernamePasswordToken usernamePasswordToken = new UsernamePasswordToken(account, password);
//            usernamePasswordToken.setRememberMe(true);
            try {
                subject.login(usernamePasswordToken);
                loginSuccess = true;
            } catch (UnknownAccountException e) {
                return Result.fail(202, "账户不存在");
            } catch (LockedAccountException e) {
                return Result.fail(202, "账户被锁定");
            } catch (IncorrectCredentialsException e) {
                return Result.fail(202, "密码不匹配");
            }
        }
        if (loginSuccess) {
            // 获取shiroRealm中的数据
            ShiroRealm bean = SpringUtil.getBean(ShiroRealm.class);
            User user = accordingUsernameSelectAll(account);
            user.setPassword(null);
            user.setSalt(null);
            UserLoginUtils bean1 = SpringUtil.getBean(UserLoginUtils.class);
            return bean1.LoginExamine(user);
        }else {
            return Result.fail("登录失败");
        }
    }

    @Override
    public Map<String, Object> getUserInfo(String token) {
        //根据token获取用户信息
        Object obj = RedisUtil.get(token);
        if (obj != null){
            User loginUser = JSON.parseObject(JSON.toJSONString(obj),User.class);
//            Map<String, Object> map = loginUser.getMap();
//            Object residualTime = map.get("residualTime");
//            int i = Integer.parseInt(residualTime.toString());
//            String remind = null;
//            if (i <= 30) {
//                remind = "您所使用的系统即将过期！请联系管理员！";
//            }
            Map<String, Object> data = new HashMap<>();
            data.put("name",loginUser.getUsername());
            data.put("avatar",loginUser.getAvatar());
            data.put("authority", loginUser.getAuthority());
            data.put("account", loginUser.getAccount());
//            data.put("remind", remind);
            return data;
        }
        return null;
    }

    @Override
    public Result<?> storeFile(String token, byte[] bytes, String originalFilename) {
        // 根据token获取Redis数据
        User user = (User)RedisUtil.get(token);
        FileSaveUtils bean = SpringUtil.getBean(FileSaveUtils.class);
        // 保存文件并返回文件名
        String fileName = bean.StoreFile(bytes, originalFilename);
        // 更新数据库头像路径
        LambdaUpdateWrapper<User> wrapper = new LambdaUpdateWrapper<>();
        wrapper.eq(User::getAccount, user.getAccount());
        wrapper.set(User::getAvatar, "/" + fileName);
        int update = mapper.update(null, wrapper);
        if (update == 1){
            // 删除原来图片在set
            if (!user.getAvatar().equals(avatar)){
                FileSystemUtils.deleteRecursively(new File(FILE_PATH + user.getAvatar()));
            }
            user.setAvatar("/" + fileName);
            RedisUtil.set(token, user, 0);
            return Result.success("更新头像成功");
        } else {
            // 删除原来图片在set
            FileSystemUtils.deleteRecursively(new File(FILE_PATH + "/"+ fileName));
            return Result.fail("更新头像失败");
        }
    }

    @Override
    public void logout(String token) {
        RedisUtil.del(token);
    }

    @Override
    public List<User> getUserList() {
        LambdaQueryWrapper<User> wrapper = new LambdaQueryWrapper<>();
        wrapper.select(User::getId,User::getUsername,User::getAvatar, User::getAuthority, User::getAccount);
        return mapper.selectList(wrapper);
    }

    @Override
    public Integer UpdatePassword(String token, String oldPassword, String newPassword) {
        LambdaQueryWrapper<User> wrapper = new LambdaQueryWrapper<>();
        User redisUser = (User)RedisUtil.get(token);
        Integer id = redisUser.getId();
        wrapper.eq(User::getId, id);
        wrapper.select(User::getPassword, User::getSalt);
        User user = mapper.selectOne(wrapper);
        if (!ObjectUtils.isEmpty(user)){ // 非空
            String mysqlPassword = user.getPassword();
            String salt = user.getSalt();
            String oldMD5 = new Md5Hash(oldPassword,salt,1024).toString();
            if (Objects.equals(oldMD5, mysqlPassword)){
                String newMD5 = new Md5Hash(newPassword,salt,1024).toString();
                if (Objects.equals(mysqlPassword, newMD5)){
                    return 2;
                }
                LambdaUpdateWrapper<User> updateWrapper = new LambdaUpdateWrapper<>();
                updateWrapper.eq(User::getId, id);
                updateWrapper.set(User::getPassword, newMD5);
                mapper.update(null, updateWrapper);
                RedisUtil.del(token);
                return 1;
            }
        }
        return 0;
    }

    @Override
    public Integer DeleteUserId(String id) {
        LambdaQueryWrapper<User> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(User::getId, id);
        return mapper.delete(wrapper);
    }

    @Override
    public Integer UpdateUser(String token, String username, String account) {
        User user = (User)RedisUtil.get(token);
        Integer id = user.getId();
        LambdaUpdateWrapper<User> wrapper = new LambdaUpdateWrapper<>();
        wrapper.eq(User::getId, id);
        wrapper.set(User::getUsername, username);
        wrapper.set(User::getAccount, account);
        user.setUsername(username);
        user.setAccount(account);
        RedisUtil.set(token, user, 0);
        return mapper.update(null, wrapper);
    }
}