package com.ruoyi.common.filter;

import lombok.extern.slf4j.Slf4j;
import org.springframework.context.i18n.LocaleContextHolder;
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.util.Enumeration;
import java.util.Locale;

@Slf4j
public class MyI18nInterceptor implements HandlerInterceptor {

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler){
        try {
            final String key = "language";
            //String language = "en-US";
            //String language = "zh-CN";
            Enumeration<String> headers = request.getHeaderNames();
            String language = request.getHeader(key);
            if (language == null) {
                //String language = "en-US";
                language = "zh-CN";
            }
            // 前端传递的language必须是zh-CN格式的，中间的-必须要完整，不能只传递zh或en
            log.debug("当前语言={}", language);
            Locale locale = new Locale(language.split("-")[0], language.split("-")[1]);
            // 这样赋值以后，MessageUtils.message方法就不用修改了
            LocaleContextHolder.setLocale(locale);
        }
        catch (Exception ex)
        {
            log.error(ex.getMessage());
            return false;
        }
        return true;
    }

    /**
     * 请求处理之后进行调用，但是在视图被渲染之前（Controller方法调用之后）
     */
    @Override
    public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) {
    }

    /**
     * 在整个请求结束之后被调用，也就是在DispatcherServlet 渲染了对应的视图之后执行（主要是用于进行资源清理工作）
     */
    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) {
    }

}