org.summerb.webappboilerplate.users.i18n.LocaleResolverUserBasedImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of summerb-webboilerplate Show documentation
A very opinionated view on building blocks of simple web applications
The newest version!
/*******************************************************************************
 * Copyright 2015-2024 Sergey Karpushin
 *
 * Licensed under the Apache License, Version 2.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://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  See the
 * License for the specific language governing permissions and limitations under
 * the License.
 ******************************************************************************/
package org.summerb.webappboilerplate.users.i18n;

import java.util.Locale;

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

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.util.StringUtils;
import org.springframework.web.servlet.i18n.CookieLocaleResolver;
import org.summerb.spring.security.SecurityConstants;
import org.summerb.spring.security.api.SecurityContextResolver;
import org.summerb.users.api.UserService;
import org.summerb.users.api.dto.User;
import org.summerb.users.api.exceptions.UserNotFoundException;
import org.summerb.validation.ValidationException;

public class LocaleResolverUserBasedImpl extends CookieLocaleResolver {
  protected Logger log = LoggerFactory.getLogger(getClass());

  @Autowired protected SecurityContextResolver securityContextResolver;
  @Autowired protected UserService userService;

  public Locale resolveLocaleFromCookie(HttpServletRequest request) {
    return super.resolveLocale(request);
  }

  @Override
  public Locale resolveLocale(HttpServletRequest request) {
    try {
      boolean isRegisteredUser = securityContextResolver.hasRole(SecurityConstants.ROLE_USER);
      if (isRegisteredUser) {
        User user = securityContextResolver.getUser();
        Locale userLocale = StringUtils.parseLocaleString(user.getLocale());
        return userLocale;
      }
    } catch (Throwable t) {
      log.warn("Failed to resolve locale based on user data", t);
    }

    return super.resolveLocale(request);
  }

  @Override
  public void setLocale(HttpServletRequest request, HttpServletResponse response, Locale locale) {
    try {
      boolean isRegisteredUser = securityContextResolver.hasRole(SecurityConstants.ROLE_USER);
      if (isRegisteredUser) {
        updateUserProfileWithNewLocale(locale.toString());
      }
    } catch (Throwable t) {
      log.warn("Failed to resolve locale based on user data", t);
    }

    super.setLocale(request, response, locale);
  }

  protected void updateUserProfileWithNewLocale(String newLocale)
      throws UserNotFoundException, ValidationException {
    User user = securityContextResolver.getUser();
    if (user.getLocale().equalsIgnoreCase(newLocale)) {
      return;
    }

    user.setLocale(newLocale);
    userService.updateUser(user);
  }
}