package com.jsl.oa.utils;

import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.jsl.oa.dao.PermissionDAO;
import com.jsl.oa.dao.RoleDAO;
import com.jsl.oa.dao.UserDAO;
import com.jsl.oa.exception.ClassCopyException;
import com.jsl.oa.model.dodata.*;
import com.jsl.oa.model.vodata.PermissionContentVo;
import com.jsl.oa.model.vodata.ProjectSimpleVO;
import com.jsl.oa.model.vodata.UserCurrentBackVO;
import org.jetbrains.annotations.Contract;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import org.springframework.validation.BindingResult;
import org.springframework.validation.ObjectError;

import javax.servlet.http.HttpServletRequest;
import java.lang.reflect.Field;
import java.sql.Timestamp;
import java.util.*;

/**
 * <h1>自定义快捷工具类</h1>
 * <hr/>
 *
 * @author 筱锋xiao_lfeng
 * @version v1.0.0
 * @since v1.0.0
 */
public class Processing {

    /**
     * <h1>获取参数校验错误信息</h1>
     * <hr/>
     * 用于获取参数校验错误信息
     *
     * @param bindingResult 参数校验结果
     * @return {@link ArrayList<String>}
     * @since v1.0.0
     */
    public static @NotNull ArrayList<String> getValidatedErrorList(BindingResult bindingResult) {
        ArrayList<String> arrayList = new ArrayList<>();
        for (ObjectError objectError : bindingResult.getAllErrors()) {
            arrayList.add(objectError.getDefaultMessage());
        }
        return arrayList;
    }

    /**
     * <h1>生成工号</h1>
     * <hr/>
     * 用于生成工号，默认长度为10
     *
     * @param type 0:学生 1:教师 2:其他
     * @return {@link String}
     * @since v1.0.0
     */
    public static @NotNull String createJobNumber(Short type) {
        return createJobNumber(type, (short) 10);
    }

    /**
     * <h1>生成工号</h1>
     * <hr/>
     * 用于生成工号
     *
     * @param type 0:学生 1:教师 2:其他
     * @param size 工号长度
     * @return {@link String}
     * @since v1.0.0
     */
    public static @NotNull String createJobNumber(Short type, Short size) {
        StringBuilder stringBuilder = new StringBuilder();
        if (type == 0) {
            stringBuilder.append("STU");
        } else if (type == 1) {
            stringBuilder.append("TCH");
        } else {
            stringBuilder.append("OTH");
        }
        // 生成工号
        Random random = new Random();
        for (int i = 0; i < size - 3; i++) {
            stringBuilder.append(random.nextInt(10));
        }
        return stringBuilder.toString();
    }

    /**
     * <h1>生成验证码</h1>
     * <hr/>
     * 用于生成验证码，默认长度为6
     *
     * @return {@link Integer}
     */
    public static @NotNull Integer createCode(Integer size) {
        if (size == null) {
            size = 6;
        }
        StringBuilder stringBuilder = new StringBuilder();
        // 生成验证码
        Random random = new Random();
        for (int i = 0; i < size; i++) {
            stringBuilder.append(random.nextInt(10));
        }
        return Integer.valueOf(stringBuilder.toString());
    }

    /**
     * <h1>生成256位字符串</h1>
     * <hr/>
     * 用于生成256位字符串（用于加密）
     *
     * @param input 输入
     * @return 返回加密后的字符串
     */
    public static String generateKey(Long input) {
        String inputString = String.valueOf(input);
        // 使用简单的伪随机算法生成256位字符串
        StringBuilder result = new StringBuilder();
        for (int i = 0; i < 256; i++) {
            int charIndex;
            int math;
            if (i == 0) {
                math = Math.abs(inputString.hashCode());
            } else {
                math = Math.abs(inputString.hashCode() % i);
            }
            if (i < 62) {
                charIndex = (math % 62);
            } else {
                charIndex = (math / 6 % 62);
            }
            char nextChar = getCharFromIndex(charIndex);
            result.append(nextChar);
        }

        return result.toString();
    }

    /**
     * <h2>获取Authorization Header</h2>
     * <hr/>
     * 用于获取Authorization Header
     *
     * @param request 请求
     */
    public static @Nullable Long getAuthHeaderToUserId(@NotNull HttpServletRequest request) {
        String token = request.getHeader("Authorization");
        if (token == null || token.isEmpty()) {
            return null;
        } else {
            // 解析Bearer后面的令牌
            token = token.replace("Bearer ", "");
            return JwtUtil.getUserId(token);
        }
    }

    /**
     * <h2>检查用户是否是管理员</h2>
     * <hr/>
     * 该方法用于检查用户是否是管理员，类型封装后字节返回结果
     *
     * @param request    请求
     * @return 如果为 true 是管理员，false 不是管理员
     */
    public static @NotNull Boolean checkUserIsAdmin(HttpServletRequest request, @NotNull RoleDAO roleDAO) {
        RoleUserDO roleUserDO = roleDAO.getRoleUserByUid(Processing.getAuthHeaderToUserId(request));
        if (roleUserDO != null) {
            RoleDO roleDO = roleDAO.getRoleByRoleName("admin");
            return roleUserDO.getRid().equals(roleDO.getId());
        } else {
            return false;
        }
    }

    /**
     * 检查用户是否是老师
     *
     * @param request    请求
     * @return 如果为 true 是老师，false 不是老师
     */
    public static @NotNull Boolean checkUserIsTeacher(HttpServletRequest request, @NotNull RoleDAO roleDAO) {
        RoleUserDO roleUserDO = roleDAO.getRoleUserByUid(Processing.getAuthHeaderToUserId(request));
        if (roleUserDO != null) {
            RoleDO roleDO = roleDAO.getRoleByRoleName("teacher");
            return roleUserDO.getRid().equals(roleDO.getId());
        } else {
            return false;
        }
    }


    private static char getCharFromIndex(int index) {
        // 生成字符集合，可以根据需要自定义
        String charset = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789";
        return charset.charAt(index);
    }


    /**
     * 将属性从源对象复制到目标对象。
     *
     * @param <T>    目标对象的类型。
     * @param <S>    源对象的类型。
     * @param source 从中复制属性的源对象。
     * @param target 属性将复制到的目标对象。
     * @throws ClassCopyException 如果在复制过程中出现错误。
     */
    @Contract(pure = true)
    public static <T, S> void copyProperties(@NotNull S source, @NotNull T target) throws ClassCopyException {
        Class<?> sourceClass = source.getClass();
        Class<?> targetClass = target.getClass();

        try {
            Field[] sourceFields = sourceClass.getDeclaredFields();
            for (Field sourceField : sourceFields) {
                String fieldName = sourceField.getName();
                Field targetField;
                try {
                    targetField = targetClass.getDeclaredField(fieldName);
                } catch (NoSuchFieldException e) {
                    // 目标对象不存在该属性，忽略
                    continue;
                }

                sourceField.setAccessible(true);
                targetField.setAccessible(true);

                Object value = sourceField.get(source);

                if (value == null) {
                    continue;
                }

                //如果获取的值不为数字且等于“”，则跳过
                if ("".equals(value)) {
                    continue;
                }

                if (!sourceField.getType().equals(targetField.getType())) {
                    continue;
                }

                targetField.set(target, value);
            }
        } catch (IllegalAccessException ignored) {
            throw new ClassCopyException();
        }
    }

    /**
     * <h2>获取性别</h2>
     * <hr/>
     * 用于获取性别
     *
     * @param sex 性别ID
     * @return 返回中文性别
     */
    @Contract(pure = true)
    public static @NotNull String getSex(short sex) {
        switch (sex) {
            case 1: return "男";
            case 2: return "女";
            default: return "保密";
        }
    }

    /**
     * <h2>封装返回内容</h2>
     * <hr/>
     * 封装返回内容
     *
     * @param userDO 用户信息
     * @return {@link BaseResponse}
     */
    public static @NotNull UserCurrentBackVO.UserCurrent returnUserInfo(
            @NotNull UserDO userDO,
            RoleDAO roleDAO,
            PermissionDAO permissionDAO
    ) {
        UserCurrentBackVO.UserCurrent userCurrent = new UserCurrentBackVO.UserCurrent();
        // 获取用户角色
        RoleUserDO getUserRole = roleDAO.getRoleUserByUid(userDO.getId());
        if (getUserRole == null) {
            getUserRole = new RoleUserDO();
            getUserRole.setRid(0L).setCreatedAt(new Timestamp(System.currentTimeMillis()));
        } else {
            getUserRole.setUid(null);
        }
        // 获取用户权限
        RoleUserDO roleUserDO = roleDAO.getRoleUserByUid(userDO.getId());
        List<String> getPermissionForString;
        if (roleUserDO != null) {
            // 获取全部根权限
            getPermissionForString = permissionDAO.getAllPermissionBuildString();
        } else {
            // 获取权限列表信息
            getPermissionForString = permissionDAO.getPermission(userDO.getId());
        }
        userCurrent.setUser(new UserCurrentBackVO.ReturnUser()
                        .setId(userDO.getId())
                        .setJobId(userDO.getJobId())
                        .setUsername(userDO.getUsername())
                        .setAddress(userDO.getAddress())
                        .setPhone(userDO.getPhone())
                        .setEmail(userDO.getEmail())
                        .setAge(userDO.getAge())
                        .setSignature(userDO.getSignature())
                        .setAvatar(userDO.getAvatar())
                        .setNickname(userDO.getNickname())
                        .setSex(userDO.getSex())
                        .setEnabled(userDO.getEnabled())
                        .setAccountNoExpired(userDO.getAccountNoExpired())
                        .setCredentialsNoExpired(userDO.getCredentialsNoExpired())
                        .setRecommend(userDO.getRecommend())
                        .setAccountNoLocked(userDO.getAccountNoLocked())
                        .setDescription(userDO.getDescription())
                        .setCreatedAt(userDO.getCreatedAt())
                        .setUpdatedAt(userDO.getUpdatedAt())
                        .setIsDelete(userDO.getIsDelete()))
                .setRole(new UserCurrentBackVO.ReturnUserRole().setRid(getUserRole.getRid()))
                .setPermission(getPermissionForString);
        return userCurrent;
    }

    public static List<UserDO> orderUser(List<UserDO> userDOS, String order, String orderBy) {

        Comparator<UserDO> comparator = null;

        if (order.equals("asc")) {
            if (orderBy.equals("userName")) {
                comparator = Comparator.comparing(UserDO::getUsername);
            } else if (orderBy.equals("userId")) {
                comparator = Comparator.comparingLong(UserDO::getId);
            }
        } else if (order.equals("desc")) {
            if (orderBy.equals("userName")) {
                comparator = Comparator.comparing(UserDO::getUsername).reversed();
            } else if (orderBy.equals("userId")) {
                comparator = Comparator.comparingLong(UserDO::getId).reversed();
            }
        }

        userDOS.sort(comparator);
        return userDOS;
    }

    public static void projectTosimply(
            ProjectSimpleVO projectSimpleVO,
            ProjectDO projectDO,
            UserDAO userDAO,
            ObjectMapper objectMapper
    ) {
        projectSimpleVO.setId(projectDO.getId());
        projectSimpleVO.setName(projectDO.getName());
        projectSimpleVO.setTags(projectDO.getTags());
        projectSimpleVO.setCycle(projectDO.getCycle());
        projectSimpleVO.setIsFinish(projectDO.getIsFinish());
        projectSimpleVO.setWorkLoad(projectDO.getWorkLoad());
        projectSimpleVO.setPrincipalUser(userDAO.getUserById(projectDO.getPrincipalId()).getUsername());
        // 解析JSON字符串
        JsonNode rootNode = null;
        try {
            rootNode = objectMapper.readTree(projectDO.getDescription());
            // 访问特定的key
            JsonNode targetNode = rootNode.get("description");
            if (targetNode != null && !rootNode.isNull()) {
                projectSimpleVO.setDescription(targetNode.asText());
            } else {
                projectSimpleVO.setDescription("null");
            }
        } catch (JsonProcessingException ignored) {

        }

        //return ProjectSimpleVO;
    }

    /**
     * @Description: 将Permission归纳为父子关系的json形式
     * @Date: 2024/1/20
     * @Param permissions: 权限实体类
     **/
    public static List<PermissionContentVo> convertToVoList(List<PermissionDO> permissions) {
        List<PermissionContentVo> vos = new ArrayList<>();
        Map<Long, List<PermissionDO>> childrenMap = new HashMap<>();

        for (PermissionDO permission : permissions) {
            if (permission.getPid() != null) {
                List<PermissionDO> children = childrenMap.getOrDefault(permission.getPid(), new ArrayList<>());
                children.add(permission);
                childrenMap.put(permission.getPid(), children);
            }
        }

        for (PermissionDO permission : permissions) {
            if (permission.getPid() == null) {
                PermissionContentVo vo = convertToVo(permission, childrenMap);
                vos.add(vo);
            }
        }

        return vos;
    }

    /**
     * @Description: 封装PermissionContentVo的子类，被convertToVoList方法调用
     * @Date: 2024/1/20
     * @Param permission: 权限实体类
     * @Param childrenMap: 要封装的子类
     **/
    public static PermissionContentVo convertToVo(PermissionDO permission, Map<Long, List<PermissionDO>> childrenMap) {
        PermissionContentVo vo = new PermissionContentVo();
        copyProperties(permission, vo);

        List<PermissionDO> children = childrenMap.get(permission.getId());
        if (children != null) {
            List<PermissionContentVo> childVos = new ArrayList<>();
            for (PermissionDO child : children) {
                PermissionContentVo childVo = convertToVo(child, childrenMap);
                childVos.add(childVo);
            }
            vo.setChildren(childVos);
        }

        return vo;
    }
}