OpenIdUserService.java

package com.gmail.at.zhuikov.aleksandr.root;

import static com.gmail.at.zhuikov.aleksandr.root.domain.GrantedAuthority.USER;
import static org.springframework.util.StringUtils.hasText;

import java.util.List;
import java.util.Locale;

import javax.inject.Inject;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.security.core.userdetails.AuthenticationUserDetailsService;
import org.springframework.security.core.userdetails.UserDetails;
import org.springframework.security.core.userdetails.UsernameNotFoundException;
import org.springframework.security.openid.OpenIDAttribute;
import org.springframework.security.openid.OpenIDAuthenticationToken;
import org.springframework.stereotype.Component;

import com.gmail.at.zhuikov.aleksandr.root.domain.User;
import com.gmail.at.zhuikov.aleksandr.root.repository.UserRepository;

@Component("openIdUserService")
public class OpenIdUserService implements
		AuthenticationUserDetailsService<OpenIDAuthenticationToken> {

	private static final Logger LOG = LoggerFactory
			.getLogger(OpenIdUserService.class);

	@Inject
	private UserRepository userRepository;

	@Override
	public UserDetails loadUserDetails(OpenIDAuthenticationToken token)
			throws UsernameNotFoundException {

		String id = token.getIdentityUrl();

		User user = userRepository.findOne(id);

		if (user == null) {
			user = new User(id);
			user.getAuthorities().add(USER);
		}

		String country = null;
		String language = null;

		List<OpenIDAttribute> attributes = token.getAttributes();

		for (OpenIDAttribute attribute : attributes) {

			switch (attribute.getName()) {
			case "country":
				country = attribute.getValues().get(0);
				LOG.info("OpenId provided country [" + country + "]");
				break;
			case "language":
				language = attribute.getValues().get(0);
				LOG.info("OpenId provided language [" + language + "]");
				break;
			}
		}

		Locale providedLocale = createLocale(language, country);

		if (providedLocale != null) {
			user.setLocale(providedLocale);
		}

		userRepository.save(user);

		return user;
	}

	private Locale createLocale(String language, String country) {

		if (hasText(country) && hasText(language)) {
			return new Locale(language, country);
		} else if (hasText(language)) {
			return new Locale(language);
		}

		return null;
	}
}