package com.dy.pmsWechat;

import com.alibaba.fastjson2.JSON;
import com.dy.common.webUtil.BaseResponse;
import com.dy.common.webUtil.BaseResponseUtils;
import com.dy.pmsWechat.util.JwtUtil;
import jakarta.servlet.*;
import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse;
import lombok.extern.slf4j.Slf4j;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.boot.web.servlet.FilterRegistrationBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.ComponentScan;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.EnableAspectJAutoProxy;

import java.io.IOException;
import java.io.PrintWriter;

@SpringBootApplication
@EnableAspectJAutoProxy
@ComponentScan(basePackages = {"com.dy.common", "com.dy.pmsGlobal", "com.dy.pmsWechat"})
@MapperScan(basePackages={"com.dy.pmsGlobal.dao*"})
public class PmsWechatApplication {
    public static void main(String[] args) {
        SpringApplication.run(PmsWechatApplication.class, args);
    }

    @Configuration
    public static class WechatModuleConfig {

        @Bean
        public FilterRegistrationBean<LoginCheckFilter> wechatFilter() {
            FilterRegistrationBean<LoginCheckFilter> registrationBean = new FilterRegistrationBean<>();
            registrationBean.setFilter(new LoginCheckFilter());
            registrationBean.addUrlPatterns("/wechat/*");
            registrationBean.setOrder(1);
            return registrationBean;
        }
    }
    @Slf4j
    public static class LoginCheckFilter implements Filter {
        @Value("${wechat.jwt.secret-key}")
        private String secretKey;
        @Override
        public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain)
                throws IOException, ServletException {
            HttpServletRequest req = (HttpServletRequest) request;
            HttpServletResponse resp = (HttpServletResponse) response;
            String url = req.getRequestURI().toString();
            // 在这里编写过滤器逻辑
            if (url.contains("login")) {
                chain.doFilter(request, response); // "登录操作，放行" 放行请求，继续执行后续过滤器链或目标资源
                return; // 方法结束，后续逻辑不再执行
            }
            // 获取请求头中的 "token" 值
            String jwt = req.getHeader("token");
            // 如果请求头中没有 "token"，返回未登录错误信息
            if (jwt == null || jwt.trim().isEmpty()) { // 使用 Spring 的 StringUtils 检查 jwt 是否为空
                PrintWriter pw = null ;
                try {
                    BaseResponse<?> res = BaseResponseUtils.buildToLogin();
                    String jsonString = JSON.toJSONString(res);
                    response.setCharacterEncoding("UTF-8");
                    response.setContentType("application/json; charset=utf-8");
                    pw = response.getWriter() ;
                    pw.write(jsonString);
                    pw.flush();
                } catch (IOException e) {
                    e.printStackTrace();
                }finally {
                    if(pw != null){
                        pw.close();
                    }
                }
                return; // 方法结束，后续逻辑不再执行
            }

            // 解析令牌
            try {
                JwtUtil.parseJWT(secretKey,jwt); // 尝试解析令牌，验证其合法性
            } catch (Exception e) {
                PrintWriter pw = null ;
                try {
                    BaseResponse<?> res = BaseResponseUtils.buildToLogin();
                    String jsonString = JSON.toJSONString(res);
                    response.setCharacterEncoding("UTF-8");
                    response.setContentType("application/json; charset=utf-8");
                    pw = response.getWriter() ;
                    pw.write(jsonString);
                    pw.flush();
                } catch (IOException ex) {
                    ex.printStackTrace();
                }finally {
                    if(pw != null){
                        pw.close();
                    }
                }
                return; // 方法结束，后续逻辑不再执行
            }
            chain.doFilter(request, response);
        }
    }

}