EDIT

/**

* Copyright (C) 2021 EDIT

* European Distributed Institute of Taxonomy

* http://www.e-taxonomy.eu

*

* The contents of this file are subject to the Mozilla Public License Version 1.1

* See LICENSE.TXT at the top of this package for the full license terms.

*/

package eu.etaxonomy.cdm.api.service.security;

import java.util.HashMap;

import java.util.Map;

import java.util.Optional;

import javax.mail.internet.AddressException;

import javax.mail.internet.InternetAddress;

import org.apache.logging.log4j.LogManager;

import org.apache.logging.log4j.Logger;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.beans.factory.annotation.Qualifier;

import org.springframework.dao.DataAccessException;

import org.springframework.mail.MailException;

import org.springframework.mail.MailPreparationException;

import org.springframework.scheduling.annotation.Async;

import org.springframework.scheduling.annotation.AsyncResult;

import org.springframework.security.core.userdetails.UserDetails;

import org.springframework.security.core.userdetails.UsernameNotFoundException;

import org.springframework.stereotype.Service;

import org.springframework.transaction.annotation.Transactional;

import org.springframework.util.Assert;

import org.springframework.util.concurrent.ListenableFuture;

import eu.etaxonomy.cdm.api.security.AbstractRequestToken;

import eu.etaxonomy.cdm.api.security.IAbstractRequestTokenStore;

import eu.etaxonomy.cdm.api.security.PasswordResetRequest;

import eu.etaxonomy.cdm.model.permission.User;

/**

 * @author a.kohlbecker

 * @since Oct 26, 2021

 */

@Service

@Transactional(readOnly = false)

public class PasswordResetService extends AccountSelfManagementService implements IPasswordResetService {

    private static Logger logger = LogManager.getLogger();

    @Autowired

    @Qualifier("passwordResetTokenStore")

    private IAbstractRequestTokenStore<PasswordResetRequest, User> passwordResetTokenStore;

    @Override

    @Async

    public ListenableFuture<Boolean> emailResetToken(String userNameOrEmail, String passwordRequestFormUrlTemplate) throws MailException, EmailAddressNotFoundException {

        try {

            // give calling methods a bit time to register the Listeners to the Future

            Thread.sleep(10);

        } catch (InterruptedException e1) {

            // IGNORE

        }

        if(logger.isTraceEnabled()) {

            logger.trace("emailResetToken trying to aquire from rate limiter [rate: " + emailResetToken_rateLimiter.getRate() + ", timeout: " + getRateLimiterTimeout().toMillis() + "ms]");

        }

        if (emailResetToken_rateLimiter.tryAcquire(getRateLimiterTimeout())) {

            logger.trace("emailResetToken allowed by rate limiter");

            try {

                User user = findUser(userNameOrEmail);

                String emailAddress = user.getEmailAddress();

                if(emailAddress == null) {

                    throw new MailPreparationException("no email address found for " + userNameOrEmail);

                }

                AbstractRequestToken resetRequest = passwordResetTokenStore.create(emailAddress, user);

                String passwordRequestFormUrl = String.format(passwordRequestFormUrlTemplate, resetRequest.getToken());

                Map<String, String> additionalValues = new HashMap<>();

                additionalValues.put("linkUrl", passwordRequestFormUrl);

                sendEmail(emailAddress, user.getUsername(),

                        UserAccountEmailTemplates.RESET_REQUEST_EMAIL_SUBJECT_TEMPLATE,

                        UserAccountEmailTemplates.REGISTRATION_REQUEST_EMAIL_BODY_TEMPLATE, additionalValues);

                logger.info("A password reset request for  " + user.getUsername() + " has been send to "

                        + emailAddress);

            } catch (EmailAddressNotFoundException e) {

                throw e;

            } catch (UsernameNotFoundException e) {

                logger.warn("Password reset request for unknown user, cause: " + e.getMessage());

            } catch (MailException e) {

                throw e;

            }

            return new AsyncResult<Boolean>(true);

        } else {

            logger.trace("blocked by rate limiter");

            return new AsyncResult<Boolean>(false);

        }

    }

   @Override

   @Async

   public ListenableFuture<Boolean> resetPassword(String token, String newPassword) throws AccountSelfManagementException, MailException {

       if (resetPassword_rateLimiter.tryAcquire(getRateLimiterTimeout())) {

           Optional<PasswordResetRequest> resetRequest = passwordResetTokenStore.findRequest(token);

           if (resetRequest.isPresent()) {

               try {

                   UserDetails user = userService.loadUserByUsername(resetRequest.get().getUserName());

                   Assert.isAssignable(user.getClass(), User.class);

                   userService.encodeUserPassword((User)user, newPassword);

                   userDao.saveOrUpdate((User)user);

                   passwordResetTokenStore.remove(token);

                   sendEmail(resetRequest.get().getUserEmail(), resetRequest.get().getUserName(),

                           UserAccountEmailTemplates.RESET_SUCCESS_EMAIL_SUBJECT_TEMPLATE,

                           UserAccountEmailTemplates.RESET_SUCCESS_EMAIL_BODY_TEMPLATE, null);

                   return new AsyncResult<Boolean>(true);

               } catch (DataAccessException | IllegalArgumentException | UsernameNotFoundException e) {

                   logger.error("Failed to change password of User " + resetRequest.get().getUserName(), e);

                   sendEmail(resetRequest.get().getUserEmail(), resetRequest.get().getUserName(),

                           UserAccountEmailTemplates.RESET_FAILED_EMAIL_SUBJECT_TEMPLATE,

                           UserAccountEmailTemplates.RESET_FAILED_EMAIL_BODY_TEMPLATE, null);

               }

           } else {

               throw new AccountSelfManagementException("Invalid password reset token");

           }

       }

       return new AsyncResult<Boolean>(false);

   }

    protected User findUser(String userNameOrEmail) throws UsernameNotFoundException, EmailAddressNotFoundException {

        User user;

        try {

            InternetAddress emailAddr = new InternetAddress(userNameOrEmail);

            emailAddr.validate();

            user = userDao.findByEmailAddress(userNameOrEmail);

            if (user == null) {

                throw new EmailAddressNotFoundException(

                        "No user with the email address'" + userNameOrEmail + "' found.");

            }

        } catch (AddressException ex) {

            user = userDao.findUserByUsername(userNameOrEmail);

            if (user == null) {

                throw new UsernameNotFoundException("No user with the user name: '" + userNameOrEmail + "' found.");

            }

        }

        return user;

    }

}