/*
* eGov suite of products aim to improve the internal efficiency,transparency,
* accountability and the service delivery of the government organizations.
*
* Copyright (C) <2015> eGovernments Foundation
*
* The updated version of eGov suite of products as by eGovernments Foundation
* is available at http://www.egovernments.org
*
* 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 3 of the License, or
* 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/ or
* http://www.gnu.org/licenses/gpl.html .
*
* In addition to the terms of the GPL license to be adhered to in using this
* program, the following additional terms are to be complied with:
*
* 1) All versions of this program, verbatim or modified must carry this
* Legal Notice.
*
* 2) Any misrepresentation of the origin of the material is prohibited. It
* is required that all modified versions of this material be marked in
* reasonable ways as different from the original version.
*
* 3) This license does not grant any rights to any user of the program
* with regards to rights under trademark law for use of the trade names
* or trademarks of eGovernments Foundation.
*
* In case of any queries, you can reach eGovernments Foundation at contact@egovernments.org.
*/
package org.egov.infra.security.audit.service;
import org.egov.infra.admin.master.entity.User;
import org.egov.infra.admin.master.service.UserService;
import org.egov.infra.security.audit.entity.LoginAttempt;
import org.egov.infra.security.audit.repository.LoginAttemptRepository;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import java.util.Date;
import java.util.Optional;
import static org.egov.infra.security.utils.SecurityConstants.MAX_LOGIN_ATTEMPT_ALLOWED;
@Transactional
@Service
public class LoginAttemptService {
private final LoginAttemptRepository loginAttemptRepository;
@Autowired
private UserService userService;
@Autowired
public LoginAttemptService(LoginAttemptRepository loginAttemptRepository) {
this.loginAttemptRepository = loginAttemptRepository;
}
public Optional<LoginAttempt> updateFailedAttempt(String username) {
User user = userService.getUserByUsername(username);
if (user != null) {
LoginAttempt loginAttempt = getLoginAttempt(username);
if (loginAttempt == null) {
loginAttempt = new LoginAttempt();
loginAttempt.setUsername(username);
loginAttempt.setFailedAttempts(1);
} else {
loginAttempt.setFailedAttempts(loginAttempt.getFailedAttempts() + 1);
}
loginAttempt.setLastModifiedOn(new Date());
if (loginAttempt.getFailedAttempts() >= MAX_LOGIN_ATTEMPT_ALLOWED) {
user.setAccountLocked(true);
userService.updateUser(user);
}
return Optional.of(loginAttemptRepository.save(loginAttempt));
}
return Optional.empty();
}
public void resetFailedAttempt(String username) {
LoginAttempt loginAttempt = getLoginAttempt(username);
if (loginAttempt != null) {
loginAttempt.setFailedAttempts(0);
loginAttempt.setLastModifiedOn(new Date());
loginAttemptRepository.save(loginAttempt);
User user = userService.getUserByUsername(username);
user.setAccountLocked(false);
userService.updateUser(user);
}
}
@Transactional(readOnly = true)
public LoginAttempt getLoginAttempt(String username) {
return loginAttemptRepository.findByUsername(username);
}
}