OpenmrsCookieLocaleResolver.java

/**
 * The contents of this file are subject to the OpenMRS Public License
 * Version 1.0 (the "License"); you may not use this file except in
 * compliance with the License. You may obtain a copy of the License at
 * http://license.openmrs.org
 *
 * Software distributed under the License is distributed on an "AS IS"
 * basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the
 * License for the specific language governing rights and limitations
 * under the License.
 *
 * Copyright (C) OpenMRS, LLC.  All Rights Reserved.
 */
package org.openmrs.web;

import java.util.Locale;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;

import org.openmrs.api.context.Context;
import org.springframework.web.servlet.i18n.CookieLocaleResolver;

public class OpenmrsCookieLocaleResolver extends CookieLocaleResolver {
	
	public Locale resolveLocale(HttpServletRequest request) {
		
		Locale locale;
		
		locale = Context.getLocale();
		if (locale != null)
			return locale;
		
		//fall back to cookie that was set
		return super.resolveLocale(request);
	}
	
	public void setLocale(HttpServletRequest request, HttpServletResponse response, Locale locale) {
		
		HttpSession session = (HttpSession) request.getSession();
		
		// if a user clicks on the locale change links 
		// AND their current default locale is different (so the msg isn't repeated)
		if (request.getParameter("lang") != null) {
			if (Context.isAuthenticated() && !Context.getLocale().equals(locale)) {
				session.setAttribute(WebConstants.OPENMRS_MSG_ARGS, request.getContextPath());
				session.setAttribute(WebConstants.OPENMRS_MSG_ATTR, "User.localeChangeHint");
			}
		}
		
		Context.setLocale(locale);
		
		response.setLocale(locale);
		
		//still set the cookie for later possible use.
		super.setLocale(request, response, locale);
		
	}
	
	public String getCookieName() {
		return WebConstants.OPENMRS_LANGUAGE_COOKIE_NAME;
	}
	
}