|
| 1 | +$output.java("${configuration.rootPackage}.config.locale", "AngularCookieLocaleResolver")## |
| 2 | + |
| 3 | +$output.require("org.springframework.context.i18n.LocaleContext")## |
| 4 | +$output.require("org.springframework.context.i18n.TimeZoneAwareLocaleContext")## |
| 5 | +$output.require("org.springframework.util.StringUtils")## |
| 6 | +$output.require("org.springframework.web.servlet.i18n.CookieLocaleResolver")## |
| 7 | +$output.require("org.springframework.web.util.WebUtils")## |
| 8 | + |
| 9 | +$output.require("java.util.Locale")## |
| 10 | +$output.require("java.util.TimeZone")## |
| 11 | + |
| 12 | +$output.require("javax.servlet.http.*")## |
| 13 | + |
| 14 | +/** |
| 15 | + * Angular cookie saved the locale with a double quote (%22en%22). |
| 16 | + * So the default CookieLocaleResolver#StringUtils.parseLocaleString(localePart) |
| 17 | + * is not able to parse the locale. |
| 18 | + * |
| 19 | + * This class will check if a double quote has been added, if so it will remove it. |
| 20 | + */ |
| 21 | +public class AngularCookieLocaleResolver extends CookieLocaleResolver { |
| 22 | + |
| 23 | + @Override |
| 24 | + public Locale resolveLocale(HttpServletRequest request) { |
| 25 | + parseLocaleCookieIfNecessary(request); |
| 26 | + return (Locale) request.getAttribute(LOCALE_REQUEST_ATTRIBUTE_NAME); |
| 27 | + } |
| 28 | + |
| 29 | + @Override |
| 30 | + public LocaleContext resolveLocaleContext(final HttpServletRequest request) { |
| 31 | + parseLocaleCookieIfNecessary(request); |
| 32 | + return new TimeZoneAwareLocaleContext() { |
| 33 | + @Override |
| 34 | + public Locale getLocale() { |
| 35 | + return (Locale) request.getAttribute(LOCALE_REQUEST_ATTRIBUTE_NAME); |
| 36 | + } |
| 37 | + |
| 38 | + @Override |
| 39 | + public TimeZone getTimeZone() { |
| 40 | + return (TimeZone) request.getAttribute(TIME_ZONE_REQUEST_ATTRIBUTE_NAME); |
| 41 | + } |
| 42 | + }; |
| 43 | + } |
| 44 | + |
| 45 | + @Override |
| 46 | + public void addCookie(HttpServletResponse response, String cookieValue) { |
| 47 | + // Mandatory cookie modification for angular to support the locale switching on the server side. |
| 48 | + cookieValue = "%22" + cookieValue + "%22"; |
| 49 | + super.addCookie(response, cookieValue); |
| 50 | + } |
| 51 | + |
| 52 | + private void parseLocaleCookieIfNecessary(HttpServletRequest request) { |
| 53 | + if (request.getAttribute(LOCALE_REQUEST_ATTRIBUTE_NAME) == null) { |
| 54 | + // Retrieve and parse cookie value. |
| 55 | + Cookie cookie = WebUtils.getCookie(request, getCookieName()); |
| 56 | + Locale locale = null; |
| 57 | + TimeZone timeZone = null; |
| 58 | + if (cookie != null) { |
| 59 | + String value = cookie.getValue(); |
| 60 | + |
| 61 | + // Remove the double quote |
| 62 | + value = StringUtils.replace(value, "%22", ""); |
| 63 | + |
| 64 | + String localePart = value; |
| 65 | + String timeZonePart = null; |
| 66 | + int spaceIndex = localePart.indexOf(' '); |
| 67 | + if (spaceIndex != -1) { |
| 68 | + localePart = value.substring(0, spaceIndex); |
| 69 | + timeZonePart = value.substring(spaceIndex + 1); |
| 70 | + } |
| 71 | + locale = (!"-".equals(localePart) ? StringUtils.parseLocaleString(localePart.replace('-', '_')) : null); |
| 72 | + if (timeZonePart != null) { |
| 73 | + timeZone = StringUtils.parseTimeZoneString(timeZonePart); |
| 74 | + } |
| 75 | + if (logger.isTraceEnabled()) { |
| 76 | + logger.trace("Parsed cookie value [" + cookie.getValue() + "] into locale '" + locale + |
| 77 | + "'" + (timeZone != null ? " and time zone '" + timeZone.getID() + "'" : "")); |
| 78 | + } |
| 79 | + } |
| 80 | + request.setAttribute(LOCALE_REQUEST_ATTRIBUTE_NAME, |
| 81 | + (locale != null ? locale: determineDefaultLocale(request))); |
| 82 | + |
| 83 | + request.setAttribute(TIME_ZONE_REQUEST_ATTRIBUTE_NAME, |
| 84 | + (timeZone != null ? timeZone : determineDefaultTimeZone(request))); |
| 85 | + } |
| 86 | + } |
| 87 | +} |
0 commit comments