package com.ruoyi.system.filter;

import com.alibaba.fastjson.JSON;
import com.ruoyi.account.api.feignClient.AppUserClient;
import com.ruoyi.account.api.model.TAppUser;
import com.ruoyi.common.core.constant.TokenConstants;
import com.ruoyi.common.core.domain.R;
import com.ruoyi.common.core.utils.StringUtils;
import com.ruoyi.system.api.domain.SysUser;
import com.ruoyi.system.api.feignClient.SysUserClient;
import com.ruoyi.system.service.ISysUserService;
import org.apache.logging.log4j.core.config.Order;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Lazy;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.stereotype.Component;

import javax.annotation.Resource;
import javax.servlet.*;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.PrintWriter;

/**
 * @author zhibing.pu
 * @Date 2024/8/23 11:22
 */
@Order(-200)
@Component
public class AuthFilter implements Filter {
	private static final Logger log = LoggerFactory.getLogger(AuthFilter.class);
	
	@Lazy
	@Resource
	private AppUserClient appUserClient;
	
	@Lazy
	@Resource
	private ISysUserService sysUserService;
	
	
	@Override
	public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain filterChain) throws IOException, ServletException {
		HttpServletRequest request = (HttpServletRequest) servletRequest;
		HttpServletResponse response = (HttpServletResponse) servletResponse;
		String userid = request.getHeader("user_id");
		if(StringUtils.isEmpty(userid)){
			filterChain.doFilter(request, response);
			return;
		}
		String userType = request.getHeader("user_type");
		//管理后台用户
		if ("system".equals(userType)) {
			SysUser sysUser = sysUserService.getById(userid);
			if(null == sysUser || "2".equals(sysUser.getDelFlag())){
				log.error("[账户异常处理]请求账户id:{}", userid);
				unauthorizedResponse(response,"无效的账户");
				return;
			}
			if("1".equals(sysUser.getStatus())){
				log.error("[账户异常处理]请求账户id:{}", userid);
				unauthorizedResponse(response,"账户已被停用，请联系系统管理员！");
				return;
			}
		}
		//小程序用户
		if ("applet".equals(userType)) {
			TAppUser appUser = appUserClient.getUserById(Long.valueOf(userid)).getData();
			if(null == appUser || appUser.getDelFlag() || 3 == appUser.getStatus()){
				log.error("[账户异常处理]请求账户id:{}", userid);
				unauthorizedResponse(response,"无效的账户");
				return;
			}
			if(2 == appUser.getStatus()){
				log.error("[账户异常处理]请求账户id:{}", userid);
				unauthorizedResponse(response,"账户已被冻结，请联系系统管理员！");
				return;
			}
		}
		filterChain.doFilter(servletRequest, servletResponse);
	}
	
	
	
	private void unauthorizedResponse(HttpServletResponse response, String msg) {
		response.setStatus(HttpStatus.OK.value());
		response.setHeader(HttpHeaders.CONTENT_TYPE, MediaType.APPLICATION_JSON_UTF8_VALUE);
		PrintWriter writer = null;
		try {
			writer = response.getWriter();
		} catch (IOException e) {
			throw new RuntimeException(e);
		}
		writer.println(JSON.toJSONString(R.fail(msg)));
		writer.flush();
		writer.close();
	}
	
	
	
	/**
	 * 获取请求token
	 */
	private String getToken(HttpServletRequest request) {
		String token = request.getHeader(TokenConstants.AUTHENTICATION);
		// 如果前端设置了令牌前缀，则裁剪掉前缀
		if (StringUtils.isNotEmpty(token) && token.startsWith(TokenConstants.PREFIX)) {
			token = token.replaceFirst(TokenConstants.PREFIX, StringUtils.EMPTY);
		}
		return token;
	}
	
	
}