org.wallride.service.AuthorizedUserDetailsService.java Source code

Introduction

Here is the source code for org.wallride.service.AuthorizedUserDetailsService.java

Source

/*
 * Copyright 2014 Tagbangers, Inc.
 *
 * 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.wallride.service;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.dao.DataAccessException;
import org.springframework.security.core.Authentication;
import org.springframework.security.core.userdetails.UserDetails;
import org.springframework.security.core.userdetails.UserDetailsService;
import org.springframework.security.core.userdetails.UsernameNotFoundException;
import org.springframework.security.web.authentication.SavedRequestAwareAuthenticationSuccessHandler;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.StringUtils;
import org.wallride.domain.User;
import org.wallride.repository.UserRepository;
import org.wallride.support.AuthorizedUser;

import javax.annotation.Resource;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.Date;

@Transactional(rollbackFor = Exception.class)
public class AuthorizedUserDetailsService extends SavedRequestAwareAuthenticationSuccessHandler
        implements UserDetailsService {

    @Resource
    private UserRepository userRepository;

    private static Logger logger = LoggerFactory.getLogger(AuthorizedUserDetailsService.class);

    @Override
    public UserDetails loadUserByUsername(final String username)
            throws UsernameNotFoundException, DataAccessException {
        if (!StringUtils.hasText(username)) {
            throw new UsernameNotFoundException("Username is empty");
        }

        User user;
        if (!username.contains("@")) {
            user = userRepository.findOneByLoginId(username);
        } else {
            user = userRepository.findOneByEmail(username);
        }

        if (user == null) {
            throw new UsernameNotFoundException("User ID not existing. [" + username + "]");
        }

        return new AuthorizedUser(user);
    }

    @Transactional(readOnly = false)
    @Override
    public void onAuthenticationSuccess(HttpServletRequest request, HttpServletResponse response,
            Authentication authentication) throws IOException, ServletException {
        AuthorizedUser authorizedUser = (AuthorizedUser) authentication.getPrincipal();
        userRepository.updateLastLoginTime(authorizedUser.getId(), new Date());

        logger.info("\"{}\" logged in. IP: [{}]", authorizedUser.toString(), request.getRemoteAddr());

        super.onAuthenticationSuccess(request, response, authentication);
    }
}