EdenPlusServer/src/main/java/com/miekir/shibei/tool/RequestTool.java

package com.miekir.shibei.tool;

import com.alibaba.fastjson.JSON;
import com.miekir.shibei.bean.User;
import com.miekir.shibei.constants.ServerConstants;
import com.miekir.shibei.repository.UserRepository;
import org.springframework.http.HttpHeaders;

import java.util.List;

public class RequestTool {
    private RequestTool() {}

    public static boolean isRequestValid(HttpHeaders header, UserRepository userRepository) {
        String token = header.getFirst("token");
        String email = header.getFirst("email");

        if (TextUtils.isEmpty(token) || TextUtils.isEmpty(email)) {
            return false;
        }

        // 根据email查找用户，查询用户的token是否相同
        User dbUserBean;
        try {
            dbUserBean = userRepository.findUserByEmail(email);
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }

        // 用户token不一致
        if (dbUserBean == null || !TextUtils.equals(token, dbUserBean.getToken())) {
            return false;
        }
        return true;
    }

    public static boolean isRequestAdminValid(HttpHeaders header, UserRepository userRepository) {
        String email = header.getFirst("email");
        if (!TextUtils.equals(email, ServerConstants.EMAIL_ADMIN)) {
            return false;
        }
        return isRequestValid(header, userRepository);
    }

    public static User getUserByToken(UserRepository userRepository, String token) {
        // 根据email查找用户，查询用户的token是否相同
        List<User> dbUserBeanList;
        try {
            dbUserBeanList = userRepository.findUserByToken(token);
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }

        if (dbUserBeanList == null || dbUserBeanList.size() != 1) {
            return null;
        }

        return dbUserBeanList.get(0);
    }
}