package com.xindazhou.framework.security.core.rpc;

import com.xindazhou.framework.common.util.json.JsonUtils;
import com.xindazhou.framework.security.core.LoginUser;
import com.xindazhou.framework.security.core.util.SecurityFrameworkUtils;
import feign.RequestInterceptor;
import feign.RequestTemplate;
import lombok.extern.slf4j.Slf4j;

import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;

/**
 * LoginUser 的 RequestInterceptor 实现类：Feign 请求时，将 {@link LoginUser} 设置到 header 中，继续透传给被调用的服务
 *
 * @author hmy
 */
@Slf4j
public class LoginUserRequestInterceptor implements RequestInterceptor {

    @Override
    public void apply(RequestTemplate requestTemplate) {
        LoginUser user = SecurityFrameworkUtils.getLoginUser();
        if (user == null) {
            // Token 验证等场景下，LoginUser 可能为空，这是正常情况，使用 DEBUG 级别
            log.debug("[Feign请求拦截器] LoginUser 为空，无法传递用户信息到目标服务: {}", requestTemplate.url());
            return;
        }
        try {
            String userStr = JsonUtils.toJsonString(user);
            userStr = URLEncoder.encode(userStr, StandardCharsets.UTF_8); // 编码，避免中文乱码
            requestTemplate.header(SecurityFrameworkUtils.LOGIN_USER_HEADER, userStr);
            log.info("[Feign请求拦截器] 成功传递 LoginUser 到目标服务: {}, userId: {}", requestTemplate.url(), user.getId());
        } catch (Exception ex) {
            log.error("[apply][序列化 LoginUser({}) 发生异常]", user, ex);
            throw ex;
        }
    }

}
