Compare commits
5 Commits
adfcf34176
...
ccd1edea7e
| Author | SHA1 | Date |
|---|---|---|
|
|
ccd1edea7e | 2 years ago |
|
|
81971295c0 | 2 years ago |
|
|
06ce1ff07f | 2 years ago |
|
|
2b1fef5b0e | 2 years ago |
|
|
b3a52913a0 | 2 years ago |
@ -0,0 +1,44 @@
|
||||
package space.kklochko.spring_rest_example.interceptors;
|
||||
|
||||
import javax.servlet.http.HttpServletRequest;
|
||||
import javax.servlet.http.HttpServletResponse;
|
||||
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.http.HttpMethod;
|
||||
import org.springframework.web.servlet.HandlerInterceptor;
|
||||
import space.kklochko.spring_rest_example.security.access.AuthorizedValidator;
|
||||
|
||||
public class AuthorizedInterceptor implements HandlerInterceptor {
|
||||
@Autowired
|
||||
AuthorizedValidator authorizedValidator;
|
||||
|
||||
@Override
|
||||
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
|
||||
if (request.getMethod().equals(HttpMethod.GET.name())) {
|
||||
return true;
|
||||
}
|
||||
|
||||
String token = request.getHeader("Authorization");
|
||||
|
||||
Boolean accessStatus;
|
||||
|
||||
if (request.getMethod().equals(HttpMethod.DELETE.name())) {
|
||||
accessStatus = authorizedValidator.isAdmin(token);
|
||||
}else {
|
||||
accessStatus = authorizedValidator.isUser(token);
|
||||
}
|
||||
|
||||
if(accessStatus == null) {
|
||||
response.sendError(HttpServletResponse.SC_UNAUTHORIZED, authorizedValidator.noToken());
|
||||
return false;
|
||||
}
|
||||
|
||||
if(!accessStatus) {
|
||||
response.sendError(HttpServletResponse.SC_UNAUTHORIZED, authorizedValidator.accessDeniedNoPermission());
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
}
|
||||
|
||||
@ -0,0 +1,29 @@
|
||||
package space.kklochko.spring_rest_example.security.access;
|
||||
|
||||
public class AccessPower {
|
||||
int ADMIN = 2;
|
||||
int USER = 1;
|
||||
|
||||
public int accessPower(String role) {
|
||||
if(role.equals("ADMIN"))
|
||||
return ADMIN | USER;
|
||||
|
||||
if(role.equals("USER"))
|
||||
return USER;
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
public boolean isUser(String role) {
|
||||
int power = accessPower(role);
|
||||
|
||||
return (power & USER) != 0;
|
||||
}
|
||||
|
||||
public boolean isAdmin(String role) {
|
||||
int power = accessPower(role);
|
||||
|
||||
return (power & ADMIN) != 0;
|
||||
}
|
||||
}
|
||||
|
||||
@ -0,0 +1,56 @@
|
||||
package space.kklochko.spring_rest_example.security.access;
|
||||
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.stereotype.Component;
|
||||
import space.kklochko.spring_rest_example.db.repositories.TokenRepository;
|
||||
import space.kklochko.spring_rest_example.db.repositories.UserRepository;
|
||||
import space.kklochko.spring_rest_example.models.Token;
|
||||
import space.kklochko.spring_rest_example.models.User;
|
||||
|
||||
@Component
|
||||
public class AuthorizedValidator {
|
||||
@Autowired
|
||||
private UserRepository userRepository;
|
||||
|
||||
@Autowired
|
||||
private TokenRepository tokenRepository;
|
||||
|
||||
AccessPower accessPower = new AccessPower();
|
||||
|
||||
public Boolean isUser(String token){
|
||||
Token userToken = tokenRepository.read(token);
|
||||
|
||||
if(userToken == null)
|
||||
return null;
|
||||
|
||||
User user = userRepository.read(userToken.getUsername());
|
||||
|
||||
if(user == null)
|
||||
return null;
|
||||
|
||||
return accessPower.isUser(user.getRole());
|
||||
}
|
||||
|
||||
public Boolean isAdmin(String token){
|
||||
Token userToken = tokenRepository.read(token);
|
||||
|
||||
if(userToken == null)
|
||||
return null;
|
||||
|
||||
User user = userRepository.read(userToken.getUsername());
|
||||
|
||||
if(user == null)
|
||||
return null;
|
||||
|
||||
return accessPower.isAdmin(user.getRole());
|
||||
}
|
||||
|
||||
public String noToken(){
|
||||
return "Access denied, you have invalid token or no token!!!";
|
||||
}
|
||||
|
||||
public String accessDeniedNoPermission(){
|
||||
return "Access denied, you have no permission!!!";
|
||||
}
|
||||
}
|
||||
|
||||
Loading…
Reference in new issue