package com.xkl.authorization.resolvers;

import com.xkl.authorization.annotation.CurrentAdmin;
import com.xkl.config.Constants;
import com.xkl.controller.uploadsoft.UpSoftAccountController;
import com.xkl.controller.uspih.AdminAccountController;
import com.xkl.domain.XklAdminEntity;
import com.xkl.repository.AdminRepository;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.MethodParameter;
import org.springframework.stereotype.Component;
import org.springframework.web.bind.support.WebDataBinderFactory;
import org.springframework.web.context.request.NativeWebRequest;
import org.springframework.web.context.request.RequestAttributes;
import org.springframework.web.context.request.ServletWebRequest;
import org.springframework.web.method.support.HandlerMethodArgumentResolver;
import org.springframework.web.method.support.ModelAndViewContainer;
import org.springframework.web.multipart.support.MissingServletRequestPartException;

import javax.servlet.http.HttpServletRequest;

/**
 * 增加方法注入，将含有CurrentAdmin注解的方法参数注入当前登录用户
 *
 * @see CurrentAdmin
 */
@Component
public class CurrentAdminMethodArgumentResolver implements HandlerMethodArgumentResolver {

    @Autowired
    private AdminRepository adminRepository;

    @Override
    public boolean supportsParameter(MethodParameter parameter) {
        //如果参数类型是Admin并且有CurrentAdmin注解则支持
        if (parameter.getParameterType().isAssignableFrom(XklAdminEntity.class) &&
                parameter.hasParameterAnnotation(CurrentAdmin.class)) {
            return true;
        }
        return false;
    }

    @Override
    public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer mavContainer, NativeWebRequest webRequest, WebDataBinderFactory binderFactory) throws Exception {
        //取出鉴权时存入的登录用户Id
        String currentAdminId = ((String) webRequest.getAttribute(Constants.CURRENT_USER_ID, RequestAttributes.SCOPE_REQUEST)).
                replace(UpSoftAccountController.UPSOFT_TOKEN_PREFIX, "").replace(AdminAccountController.USPIH_TOKEN_PREFIX, "").replace(Constants.ADMIN_TOKEN_PREFIX, "");
        if (currentAdminId != null) {
            //从数据库中查询并返回
            XklAdminEntity admin = adminRepository.findByIdAndStatus(Long.parseLong(currentAdminId), Constants.STATUS_OK);
            HttpServletRequest request = ((ServletWebRequest) webRequest).getRequest();
            String ip = null;
            if (request.getHeader("x-forwarded-for") == null) {
                ip = request.getRemoteAddr();
            } else {
                ip = request.getHeader("x-forwarded-for");
            }
            admin.setIp(ip);
            return admin;
        }
        throw new MissingServletRequestPartException(Constants.CURRENT_USER_ID);
    }
}