de.steilerdev.myVerein.server.security.RememberMeTokenDataProvider.java Source code

Introduction

Here is the source code for de.steilerdev.myVerein.server.security.RememberMeTokenDataProvider.java

Source

/**
 * Copyright (C) 2015 Frank Steiler <frank@steilerdev.de>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package de.steilerdev.myVerein.server.security;

import de.steilerdev.myVerein.server.model.RememberMeToken;
import de.steilerdev.myVerein.server.model.RememberMeTokenRepository;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.web.authentication.rememberme.PersistentRememberMeToken;
import org.springframework.security.web.authentication.rememberme.PersistentTokenRepository;

import java.util.Date;
import java.util.List;

/**
 * This data provider is implementing the necessary functions to enable a persistent token repository. This provider is needed to user the remember me functionality.
 */
public class RememberMeTokenDataProvider implements PersistentTokenRepository {
    @Autowired
    RememberMeTokenRepository rememberMeTokenRepository;

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

    @Override
    public void createNewToken(PersistentRememberMeToken persistentRememberMeToken) {
        logger.debug("Creating new remember me token for user " + persistentRememberMeToken.getUsername());
        rememberMeTokenRepository.save(new RememberMeToken(persistentRememberMeToken));
    }

    @Override
    public void updateToken(String series, String tokenValue, Date lastUsed) {
        RememberMeToken token = rememberMeTokenRepository.findRememberMeTokenBySeries(series);
        if (token != null) {
            logger.debug("Updating remember me token of series " + series);
            token = new RememberMeToken(token.getUsername(), series, tokenValue, lastUsed);
            rememberMeTokenRepository.save(token);
        } else {
            logger.warn("Unable to update remember me token of series " + series);
        }
    }

    @Override
    public PersistentRememberMeToken getTokenForSeries(String series) {
        logger.debug("Getting token for series " + series);
        return rememberMeTokenRepository.findRememberMeTokenBySeries(series).toPersistentRememberMeToken();
    }

    @Override
    public void removeUserTokens(String username) {
        List<RememberMeToken> tokens = rememberMeTokenRepository.findRememberMeTokenByUsername(username);
        if (tokens != null) {
            logger.debug("Deleting remember me token for user " + username);
            try {
                rememberMeTokenRepository.delete(tokens);
            } catch (IllegalArgumentException e) {
                logger.error("Unable to delete remember me token for user " + username + ": " + e.getMessage());
            }
        } else {
            logger.warn("Unable to delete remember me token for user " + username);
        }
    }
}