added formatter plugin

This commit is contained in:
Rohan Sircar 2020-08-11 17:30:48 +05:30
parent ced84a05a6
commit af4181185c
44 changed files with 600 additions and 528 deletions

1
.gitignore vendored
View File

@ -29,6 +29,7 @@ build/
### VS Code ### ### VS Code ###
.vscode/ .vscode/
.cache/
node_modules node_modules
bundle.js bundle.js

11
pom.xml
View File

@ -223,6 +223,17 @@
</excludeProperties> </excludeProperties>
</configuration> </configuration>
</plugin> </plugin>
<plugin>
<groupId>net.revelc.code.formatter</groupId>
<artifactId>formatter-maven-plugin</artifactId>
<version>2.12.0</version>
<configuration>
<configFile>${project.basedir}/eclipse-formatter.xml</configFile>
<compilerSource>11</compilerSource>
<compilerCompliance>11</compilerCompliance>
<compilerTargetPlatform>11</compilerTargetPlatform>
</configuration>
</plugin>
</plugins> </plugins>
</build> </build>
</project> </project>

View File

@ -28,8 +28,10 @@ public class V3__add_default_admin extends BaseJavaMigration {
/** /**
* Method will generate random string based on the parameters * Method will generate random string based on the parameters
* *
* @param len the length of the random string * @param len
* @param dic the dictionary used to generate the password * the length of the random string
* @param dic
* the dictionary used to generate the password
* @return the random password * @return the random password
*/ */
public String generatePassword(int len, String dic) { public String generatePassword(int len, String dic) {
@ -44,8 +46,10 @@ public class V3__add_default_admin extends BaseJavaMigration {
@Override @Override
public void migrate(final Context context) throws Exception { public void migrate(final Context context) throws Exception {
try (final PreparedStatement ps = context.getConnection() try (final PreparedStatement ps = context.getConnection()
.prepareStatement("insert into users (user_id, name, password) values (0,?,?)")) { .prepareStatement(
final String generatedPassword = generatePassword(60, ALPHA_CAPS + ALPHA + SPECIAL_CHARS); "insert into users (user_id, name, password) values (0,?,?)")) {
final String generatedPassword = generatePassword(60,
ALPHA_CAPS + ALPHA + SPECIAL_CHARS);
final BufferedWriter bw = new BufferedWriter( final BufferedWriter bw = new BufferedWriter(
new FileWriter("gen-password.txt")); new FileWriter("gen-password.txt"));
@ -53,9 +57,8 @@ public class V3__add_default_admin extends BaseJavaMigration {
bw.write("\nPlease delete this file"); bw.write("\nPlease delete this file");
bw.close(); bw.close();
final var perms = Files.getPosixFilePermissions(Paths.get( final var perms = Files
"gen-password.txt")); .getPosixFilePermissions(Paths.get("gen-password.txt"));
// add owners permission // add owners permission
perms.add(PosixFilePermission.OWNER_READ); perms.add(PosixFilePermission.OWNER_READ);
@ -71,14 +74,14 @@ public class V3__add_default_admin extends BaseJavaMigration {
perms.remove(PosixFilePermission.OTHERS_EXECUTE); perms.remove(PosixFilePermission.OTHERS_EXECUTE);
Files.setPosixFilePermissions(Paths.get("gen-password.txt"), perms); Files.setPosixFilePermissions(Paths.get("gen-password.txt"), perms);
ps.setString(1, "admin"); ps.setString(1, "admin");
ps.setString(2, passwordEncoder.encode(generatedPassword)); ps.setString(2, passwordEncoder.encode(generatedPassword));
ps.execute(); ps.execute();
} }
try (final PreparedStatement ps = context.getConnection() try (final PreparedStatement ps = context.getConnection()
.prepareStatement("insert into users_roles (user_id, role_id) values (1,0)")) { .prepareStatement(
"insert into users_roles (user_id, role_id) values (1,0)")) {
ps.execute(); ps.execute();
} }
} }

View File

@ -12,10 +12,12 @@ import org.springframework.security.web.authentication.www.BasicAuthenticationEn
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
@Component @Component
public final class RESTAuthenticationEntryPoint extends BasicAuthenticationEntryPoint { public final class RESTAuthenticationEntryPoint
extends BasicAuthenticationEntryPoint {
@Override @Override
public void commence(HttpServletRequest request, HttpServletResponse response, AuthenticationException authEx) public void commence(HttpServletRequest request,
HttpServletResponse response, AuthenticationException authEx)
throws IOException, ServletException { throws IOException, ServletException {
response.setStatus(HttpServletResponse.SC_UNAUTHORIZED); response.setStatus(HttpServletResponse.SC_UNAUTHORIZED);
PrintWriter writer = response.getWriter(); PrintWriter writer = response.getWriter();

View File

@ -6,7 +6,8 @@ import org.springframework.boot.web.servlet.support.SpringBootServletInitializer
public class ServletInitializer extends SpringBootServletInitializer { public class ServletInitializer extends SpringBootServletInitializer {
@Override @Override
protected SpringApplicationBuilder configure(SpringApplicationBuilder application) { protected SpringApplicationBuilder configure(
SpringApplicationBuilder application) {
return application.sources(ChattoApplication.class); return application.sources(ChattoApplication.class);
} }

View File

@ -10,10 +10,8 @@ public class WebConfig implements WebMvcConfigurer {
@Override @Override
public void addCorsMappings(CorsRegistry registry) { public void addCorsMappings(CorsRegistry registry) {
registry.addMapping("/api/**") registry.addMapping("/api/**").allowedOrigins("*")
.allowedOrigins("*") .allowedMethods("POST", "GET", "OPTIONS").allowedHeaders("*")
.allowedMethods("POST","GET","OPTIONS")
.allowedHeaders("*")
.allowCredentials(false).maxAge(3600); .allowCredentials(false).maxAge(3600);
} }

View File

@ -4,7 +4,10 @@ import java.awt.image.BufferedImage;
interface CaptchaBehaviour { interface CaptchaBehaviour {
public BufferedImage generateCaptcha(); public BufferedImage generateCaptcha();
public BufferedImage generateCaptcha(String captchaText); public BufferedImage generateCaptcha(String captchaText);
public String getRandomChars(int size); public String getRandomChars(int size);
public String getRandomChars(); public String getRandomChars();
} }

View File

@ -9,6 +9,7 @@ import lombok.Builder;
public class ManualCaptchaBehaviour implements CaptchaBehaviour { public class ManualCaptchaBehaviour implements CaptchaBehaviour {
private final int length; private final int length;
private final String style; private final String style;
@Override @Override
public BufferedImage generateCaptcha() { public BufferedImage generateCaptcha() {
// TODO Auto-generated method stub // TODO Auto-generated method stub
@ -33,5 +34,4 @@ public class ManualCaptchaBehaviour implements CaptchaBehaviour{
return null; return null;
} }
} }

View File

@ -10,10 +10,8 @@ import java.io.IOException;
import java.util.Random; import java.util.Random;
/** /**
* This class represents a simple captcha consisting * This class represents a simple captcha consisting of an image {@code png} and
* of an image {@code png} and its text value. * its text value. Comic Neue Bold Font. Capital english letters {@code ONLY}.
* Comic Neue Bold Font.
* Capital english letters {@code ONLY}.
* *
* @since 1.3 * @since 1.3
* @author Gennadiy Golovin * @author Gennadiy Golovin
@ -24,8 +22,8 @@ public final class SimpleCaptcha {
private char[] text; private char[] text;
/** /**
* Initializes a newly created default object * Initializes a newly created default object consisting of 8 capital
* consisting of 8 capital english letters. * english letters.
*/ */
public SimpleCaptcha() { public SimpleCaptcha() {
this.text = getRandomChars(); this.text = getRandomChars();
@ -38,14 +36,14 @@ public final class SimpleCaptcha {
} }
/** /**
* Initializes a newly created object, which length * Initializes a newly created object, which length depends on the passed
* depends on the passed {@code int} parameter, * {@code int} parameter, which {@code MUST} be greater than 0. If the
* which {@code MUST} be greater than 0. * condition is not met, initializes a newly created default object
* If the condition is not met, initializes a newly * consisting of 8 symbols.
* created default object consisting of 8 symbols.
* *
* @param length the quantity of symbols, that the * @param length
* captcha consists of, greater than 0. * the quantity of symbols, that the captcha consists of, greater
* than 0.
*/ */
public SimpleCaptcha(int length) { public SimpleCaptcha(int length) {
if (length < 1) { if (length < 1) {
@ -62,13 +60,14 @@ public final class SimpleCaptcha {
} }
/** /**
* Initializes a newly created object based on the passed * Initializes a newly created object based on the passed {@link String}
* {@link String} parameter, consisting of capital english * parameter, consisting of capital english letters. If the condition is not
* letters. If the condition is not met, initializes a newly * met, initializes a newly created default object consisting of 8 capital
* created default object consisting of 8 capital english letters. * english letters.
* *
* @param text the text string with the value of the captcha, * @param text
* length greater than 0. * the text string with the value of the captcha, length greater
* than 0.
*/ */
public SimpleCaptcha(String text) { public SimpleCaptcha(String text) {
if (text == null || text.equals("")) { if (text == null || text.equals("")) {
@ -134,7 +133,8 @@ public final class SimpleCaptcha {
BufferedImage[] images = new BufferedImage[charsQuantity]; BufferedImage[] images = new BufferedImage[charsQuantity];
for (int i = 0; i < charsQuantity; i++) { for (int i = 0; i < charsQuantity; i++) {
images[i] = ImageIO.read(SimpleCaptcha.class.getResourceAsStream("/pictures/" + this.text[i] + ".png")); images[i] = ImageIO.read(SimpleCaptcha.class
.getResourceAsStream("/pictures/" + this.text[i] + ".png"));
if (i % 2 == 0) { if (i % 2 == 0) {
images[i] = rotateImage(images[i], 25); images[i] = rotateImage(images[i], 25);
} else { } else {
@ -145,12 +145,17 @@ public final class SimpleCaptcha {
int imageSize = 30; int imageSize = 30;
int rotatedImageSize = (int) Math.sqrt(imageSize * imageSize * 2); int rotatedImageSize = (int) Math.sqrt(imageSize * imageSize * 2);
BufferedImage captchaImg = new BufferedImage(rotatedImageSize * (charsQuantity - 1) / 10 * 6 + rotatedImageSize, rotatedImageSize, BufferedImage.TYPE_INT_ARGB); BufferedImage captchaImg = new BufferedImage(
rotatedImageSize * (charsQuantity - 1) / 10 * 6
+ rotatedImageSize,
rotatedImageSize, BufferedImage.TYPE_INT_ARGB);
Graphics2D graphics2d = captchaImg.createGraphics(); Graphics2D graphics2d = captchaImg.createGraphics();
graphics2d.setBackground(Color.WHITE); graphics2d.setBackground(Color.WHITE);
graphics2d.clearRect(0, 0, captchaImg.getWidth(), captchaImg.getHeight()); graphics2d.clearRect(0, 0, captchaImg.getWidth(),
captchaImg.getHeight());
for (int i = 0; i < charsQuantity; i++) { for (int i = 0; i < charsQuantity; i++) {
captchaImg.getGraphics().drawImage(images[i], rotatedImageSize * i / 10 * 6, 0, null); captchaImg.getGraphics().drawImage(images[i],
rotatedImageSize * i / 10 * 6, 0, null);
} }
graphics2d.dispose(); graphics2d.dispose();
this.imagePng = captchaImg; this.imagePng = captchaImg;
@ -165,14 +170,18 @@ public final class SimpleCaptcha {
int width = buffImage.getWidth(); int width = buffImage.getWidth();
int height = buffImage.getHeight(); int height = buffImage.getHeight();
int nWidth = (int) Math.floor((double) width * cos + (double) height * sin); int nWidth = (int) Math
int nHeight = (int) Math.floor((double) height * cos + (double) width * sin); .floor((double) width * cos + (double) height * sin);
int nHeight = (int) Math
.floor((double) height * cos + (double) width * sin);
BufferedImage rotatedImage = new BufferedImage(nWidth, nHeight, BufferedImage.TYPE_INT_ARGB); BufferedImage rotatedImage = new BufferedImage(nWidth, nHeight,
BufferedImage.TYPE_INT_ARGB);
Graphics2D graphics = rotatedImage.createGraphics(); Graphics2D graphics = rotatedImage.createGraphics();
graphics.setRenderingHint(RenderingHints.KEY_INTERPOLATION, RenderingHints.VALUE_INTERPOLATION_BICUBIC); graphics.setRenderingHint(RenderingHints.KEY_INTERPOLATION,
RenderingHints.VALUE_INTERPOLATION_BICUBIC);
graphics.translate((nWidth - width) / 2, (nHeight - height) / 2); graphics.translate((nWidth - width) / 2, (nHeight - height) / 2);
graphics.rotate(radian, (double) (width / 2), (double) (height / 2)); graphics.rotate(radian, (double) (width / 2), (double) (height / 2));
graphics.drawImage(buffImage, 0, 0, null); graphics.drawImage(buffImage, 0, 0, null);

View File

@ -9,6 +9,7 @@ public class SimpleCaptchaBehavior implements CaptchaBehaviour {
SimpleCaptcha simpleCaptcha = new SimpleCaptcha(); SimpleCaptcha simpleCaptcha = new SimpleCaptcha();
return simpleCaptcha.getImagePng(); return simpleCaptcha.getImagePng();
} }
@Override @Override
public BufferedImage generateCaptcha(String captchaText) { public BufferedImage generateCaptcha(String captchaText) {
SimpleCaptcha simpleCaptcha = new SimpleCaptcha(captchaText); SimpleCaptcha simpleCaptcha = new SimpleCaptcha(captchaText);
@ -19,8 +20,7 @@ public class SimpleCaptchaBehavior implements CaptchaBehaviour {
return getRandomChars(8); return getRandomChars(8);
} }
public String getRandomChars(int quantity) public String getRandomChars(int quantity) {
{
char[] randomString = new char[quantity]; char[] randomString = new char[quantity];
Random random = new Random(); Random random = new Random();

View File

@ -25,7 +25,8 @@ public class WebCaptcha {
} }
} }
// WebCaptcha webCaptcha = WebCaptcha.builder().captchaBehaviour(new SimpleCaptchaBehavior()).build(); // WebCaptcha webCaptcha = WebCaptcha.builder().captchaBehaviour(new
// SimpleCaptchaBehavior()).build();
// webCaptcha.generateCaptcha(); // webCaptcha.generateCaptcha();
// //
// // @formatter:off // // @formatter:off

View File

@ -6,11 +6,14 @@ import org.ehcache.event.CacheEventListener;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
@Slf4j @Slf4j
public class CustomCacheEventLogger implements CacheEventListener<Object, Object> { public class CustomCacheEventLogger
implements CacheEventListener<Object, Object> {
@Override @Override
public void onEvent(CacheEvent<? extends Object, ? extends Object> cacheEvent) { public void onEvent(
log.debug("custom Caching event {} key = {} old {} new {} ", cacheEvent.getType(), cacheEvent.getKey(), CacheEvent<? extends Object, ? extends Object> cacheEvent) {
log.debug("custom Caching event {} key = {} old {} new {} ",
cacheEvent.getType(), cacheEvent.getKey(),
cacheEvent.getOldValue(), cacheEvent.getNewValue()); cacheEvent.getOldValue(), cacheEvent.getNewValue());
} }
} }

View File

@ -1,6 +1,5 @@
package org.ros.chatto.config; package org.ros.chatto.config;
import org.springframework.cache.annotation.EnableCaching; import org.springframework.cache.annotation.EnableCaching;
import org.springframework.context.annotation.Configuration; import org.springframework.context.annotation.Configuration;

View File

@ -89,8 +89,8 @@ public class ChatMessageController {
@PathVariable final String userName, @PathVariable final String userName,
@PathVariable final Instant lastMessageTime, @PathVariable final Instant lastMessageTime,
final Principal principal) { final Principal principal) {
final List<ChatMessageDTO> chatMessageDTOs = chatService.getNewMessages( final List<ChatMessageDTO> chatMessageDTOs = chatService
principal.getName(), userName, lastMessageTime); .getNewMessages(principal.getName(), userName, lastMessageTime);
return chatMessageDTOs; return chatMessageDTOs;
} }

View File

@ -0,0 +1,14 @@
package org.ros.chatto.controller;
import org.springframework.security.web.csrf.CsrfToken;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
@RestController
public class CsrfController {
@RequestMapping("/csrf")
public CsrfToken csrf(CsrfToken token) {
return token;
}
}

View File

@ -1,6 +1,5 @@
package org.ros.chatto.dto; package org.ros.chatto.dto;
import javax.validation.constraints.NotBlank; import javax.validation.constraints.NotBlank;
import javax.validation.constraints.Pattern; import javax.validation.constraints.Pattern;
import javax.validation.constraints.Size; import javax.validation.constraints.Size;

View File

@ -12,7 +12,9 @@ import lombok.Data;
@Data @Data
public class MessageCipherDTO { public class MessageCipherDTO {
@Pattern(regexp = "^(?:[A-Za-z0-9+/]{4})*(?:[A-Za-z0-9+/]{2}==|[A-Za-z0-9+/]{3}=)?$") // regex for base64 @Pattern(regexp = "^(?:[A-Za-z0-9+/]{4})*(?:[A-Za-z0-9+/]{2}==|[A-Za-z0-9+/]{3}=)?$") // regex
// for
// base64
@NotBlank @NotBlank
private String iv; private String iv;
@Max(1) @Max(1)

View File

@ -16,7 +16,9 @@ public class UserRegistrationDTO {
@Transient @Transient
@Size(min = 4, max = 75, message = "Password must be between 4 and 75 characters") @Size(min = 4, max = 75, message = "Password must be between 4 and 75 characters")
@NotBlank(message = "Password should not be blank") @NotBlank(message = "Password should not be blank")
// @Pattern(regexp = "^.*(?=.{6,})(?=.*d)(?=.*[A-Z])(?=.*[a-z])(?=.*[!@#$%^&*? ]).*$", message = "Invalid password format") // @Pattern(regexp =
// "^.*(?=.{6,})(?=.*d)(?=.*[A-Z])(?=.*[a-z])(?=.*[!@#$%^&*? ]).*$", message
// = "Invalid password format")
private String password; private String password;
private Long captchaID; private Long captchaID;

View File

@ -17,5 +17,4 @@ public class ErrorModel{
@JsonProperty("error_message") @JsonProperty("error_message")
private String messageError; private String messageError;
} }

View File

@ -4,5 +4,6 @@ import org.springframework.cache.annotation.CacheEvict;
public class TokenCacheUtil { public class TokenCacheUtil {
@CacheEvict(value = "userTokenCache", key = "#cacheKey") @CacheEvict(value = "userTokenCache", key = "#cacheKey")
public static void evictSingleTokenValue(String cacheKey) {} public static void evictSingleTokenValue(String cacheKey) {
}
} }

View File

@ -26,7 +26,8 @@ public class UserLoggingSessionListener implements HttpSessionBindingListener {
@Override @Override
public void valueBound(HttpSessionBindingEvent event) { public void valueBound(HttpSessionBindingEvent event) {
UserLoggingSessionListener user = (UserLoggingSessionListener) event.getValue(); UserLoggingSessionListener user = (UserLoggingSessionListener) event
.getValue();
log.debug("Incrementing session count for user {}", user.getUsername()); log.debug("Incrementing session count for user {}", user.getUsername());
@ -38,7 +39,8 @@ public class UserLoggingSessionListener implements HttpSessionBindingListener {
@Override @Override
public void valueUnbound(HttpSessionBindingEvent event) { public void valueUnbound(HttpSessionBindingEvent event) {
UserLoggingSessionListener user = (UserLoggingSessionListener) event.getValue(); UserLoggingSessionListener user = (UserLoggingSessionListener) event
.getValue();
log.debug("Decrementing session count for user {}", user.getUsername()); log.debug("Decrementing session count for user {}", user.getUsername());

View File

@ -11,7 +11,8 @@ import org.springframework.security.web.authentication.AuthenticationSuccessHand
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
@Component("myAuthenticationSuccessHandler") @Component("myAuthenticationSuccessHandler")
public class UserSessionLoggingLoginSuccessHandler implements AuthenticationSuccessHandler { public class UserSessionLoggingLoginSuccessHandler
implements AuthenticationSuccessHandler {
@Override @Override
public void onAuthenticationSuccess(HttpServletRequest request, public void onAuthenticationSuccess(HttpServletRequest request,
@ -19,7 +20,8 @@ public class UserSessionLoggingLoginSuccessHandler implements AuthenticationSucc
throws IOException { throws IOException {
HttpSession session = request.getSession(false); HttpSession session = request.getSession(false);
if (session != null) { if (session != null) {
UserLoggingSessionListener user = new UserLoggingSessionListener(authentication.getName()); UserLoggingSessionListener user = new UserLoggingSessionListener(
authentication.getName());
session.setAttribute("user", user); session.setAttribute("user", user);
} }
response.sendRedirect("/chat"); response.sendRedirect("/chat");

View File

@ -12,7 +12,8 @@ import org.springframework.security.web.authentication.logout.LogoutSuccessHandl
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
@Component("myLogoutSuccessHandler") @Component("myLogoutSuccessHandler")
public class UserSessionLoggingLogoutSuccessHandler implements LogoutSuccessHandler{ public class UserSessionLoggingLogoutSuccessHandler
implements LogoutSuccessHandler {
@Override @Override
public void onLogoutSuccess(HttpServletRequest request, public void onLogoutSuccess(HttpServletRequest request,
HttpServletResponse response, Authentication authentication) HttpServletResponse response, Authentication authentication)

View File

@ -16,7 +16,6 @@ import lombok.Data;
import lombok.EqualsAndHashCode; import lombok.EqualsAndHashCode;
import lombok.ToString; import lombok.ToString;
@Entity @Entity
@Table(name = "roles") @Table(name = "roles")
@Data @Data
@ -30,8 +29,8 @@ public class Role {
private String description; private String description;
@OneToMany(mappedBy = "role", cascade = { CascadeType.PERSIST, CascadeType.MERGE, CascadeType.DETACH, @OneToMany(mappedBy = "role", cascade = { CascadeType.PERSIST,
CascadeType.REFRESH }) CascadeType.MERGE, CascadeType.DETACH, CascadeType.REFRESH })
@JsonBackReference @JsonBackReference
@ToString.Exclude @ToString.Exclude
@EqualsAndHashCode.Exclude @EqualsAndHashCode.Exclude

View File

@ -4,6 +4,7 @@
// import org.springframework.stereotype.Repository; // import org.springframework.stereotype.Repository;
// //
// @Repository // @Repository
//public interface DBInitializerRepostory extends JpaRepository<Integer, Integer>{ // public interface DBInitializerRepostory extends JpaRepository<Integer,
// Integer>{
// //
// } // }

View File

@ -5,6 +5,7 @@ import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.stereotype.Repository; import org.springframework.stereotype.Repository;
@Repository @Repository
public interface MessageCipherRepository extends JpaRepository<MessageCipher, Long>{ public interface MessageCipherRepository
extends JpaRepository<MessageCipher, Long> {
} }

View File

@ -23,13 +23,17 @@ class UserRepositoryCustomImpl implements UserRepositoryCustom{
List<String> userNamesList = null; List<String> userNamesList = null;
// Session session = null; // Session session = null;
try { try {
CriteriaBuilder criteriaBuilder = entityManager.getCriteriaBuilder(); CriteriaBuilder criteriaBuilder = entityManager
CriteriaQuery<String> criteriaQuery = criteriaBuilder.createQuery(String.class); .getCriteriaBuilder();
CriteriaQuery<String> criteriaQuery = criteriaBuilder
.createQuery(String.class);
Root<ChatUser> root = criteriaQuery.from(ChatUser.class); Root<ChatUser> root = criteriaQuery.from(ChatUser.class);
criteriaQuery.select(root.get("userName")); criteriaQuery.select(root.get("userName"));
criteriaQuery.where(criteriaBuilder.notEqual(root.get("userName"), userName)); criteriaQuery.where(
criteriaBuilder.notEqual(root.get("userName"), userName));
userNamesList = entityManager.createQuery(criteriaQuery).getResultList(); userNamesList = entityManager.createQuery(criteriaQuery)
.getResultList();
// for(String un: userNamesList) // for(String un: userNamesList)
// { // {
// System.out.println(un); // System.out.println(un);

View File

@ -16,20 +16,29 @@ import org.springframework.security.web.authentication.AuthenticationSuccessHand
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
@Component @Component
public class AuthenticationSuccessHandlerImpl implements AuthenticationSuccessHandler { public class AuthenticationSuccessHandlerImpl
implements AuthenticationSuccessHandler {
private static final SimpleGrantedAuthority SUPER_USER_AUTHORITY = new SimpleGrantedAuthority("ROLE_SUPER_USER"); private static final SimpleGrantedAuthority SUPER_USER_AUTHORITY = new SimpleGrantedAuthority(
private static final SimpleGrantedAuthority ADMIN_AUTHORITY = new SimpleGrantedAuthority("ROLE_ADMIN"); "ROLE_SUPER_USER");
private static final SimpleGrantedAuthority ADMIN_AUTHORITY = new SimpleGrantedAuthority(
"ROLE_ADMIN");
private RedirectStrategy redirectStrategy = new DefaultRedirectStrategy(); private RedirectStrategy redirectStrategy = new DefaultRedirectStrategy();
@Override @Override
public void onAuthenticationSuccess(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse, public void onAuthenticationSuccess(HttpServletRequest httpServletRequest,
Authentication authentication) throws IOException, ServletException { HttpServletResponse httpServletResponse,
Collection<? extends GrantedAuthority> authorities = authentication.getAuthorities(); Authentication authentication)
if (authorities.contains(ADMIN_AUTHORITY) || authorities.contains(SUPER_USER_AUTHORITY)) { throws IOException, ServletException {
redirectStrategy.sendRedirect(httpServletRequest, httpServletResponse, "/admin"); Collection<? extends GrantedAuthority> authorities = authentication
.getAuthorities();
if (authorities.contains(ADMIN_AUTHORITY)
|| authorities.contains(SUPER_USER_AUTHORITY)) {
redirectStrategy.sendRedirect(httpServletRequest,
httpServletResponse, "/admin");
} else { } else {
redirectStrategy.sendRedirect(httpServletRequest, httpServletResponse, "/user"); redirectStrategy.sendRedirect(httpServletRequest,
httpServletResponse, "/user");
} }
} }
} }

View File

@ -11,16 +11,15 @@ public class CaptchaService {
private final WebCaptcha webCaptcha; private final WebCaptcha webCaptcha;
public CaptchaService() { public CaptchaService() {
webCaptcha = WebCaptcha.builder().captchaBehaviour(new SimpleCaptchaBehavior()).build(); webCaptcha = WebCaptcha.builder()
.captchaBehaviour(new SimpleCaptchaBehavior()).build();
} }
public BufferedImage createCaptchaImage(final String captchaText) public BufferedImage createCaptchaImage(final String captchaText) {
{
return webCaptcha.generateCaptcha(captchaText); return webCaptcha.generateCaptcha(captchaText);
} }
public String getRandomText() public String getRandomText() {
{
return webCaptcha.getRandomChars(); return webCaptcha.getRandomChars();
} }
} }

View File

@ -46,13 +46,15 @@ public class DBInitializerService {
}); });
} }
private void resetAllUserSessions(final Connection connection) throws SQLException { private void resetAllUserSessions(final Connection connection)
throws SQLException {
final PreparedStatement preparedStatement = connection final PreparedStatement preparedStatement = connection
.prepareStatement(dbInitializerConfig.getResetSessionsQuery()); .prepareStatement(dbInitializerConfig.getResetSessionsQuery());
preparedStatement.executeUpdate(); preparedStatement.executeUpdate();
} }
private void clearAllTokens(final Connection connection) throws SQLException { private void clearAllTokens(final Connection connection)
throws SQLException {
final PreparedStatement preparedStatement = connection final PreparedStatement preparedStatement = connection
.prepareStatement(dbInitializerConfig.getClearTokensQuery()); .prepareStatement(dbInitializerConfig.getClearTokensQuery());
preparedStatement.executeUpdate(); preparedStatement.executeUpdate();

View File

@ -28,7 +28,8 @@
// @Autowired // @Autowired
// private UserRoleRepository userRoleRepository; // private UserRoleRepository userRoleRepository;
// private final Logger logger = LoggerFactory.getLogger(ChattoApplicationTests.class); // private final Logger logger =
// LoggerFactory.getLogger(ChattoApplicationTests.class);
// // // //
// @Test // @Test
// public void contextLoads() { // public void contextLoads() {
@ -47,7 +48,8 @@
// @Test // @Test
// public void findAllOtherUsers() { // public void findAllOtherUsers() {
// logger.info("Usernames = {}",userRepository.findAllOtherUserNames("hmm").toString()); // logger.info("Usernames =
// {}",userRepository.findAllOtherUserNames("hmm").toString());
// } // }
// // @Test // // @Test
@ -65,7 +67,8 @@
// // } // // }
// /* // /*
// * @Test public void testSave() { ChatUser fromUser = new ChatUser(); fromUser = // * @Test public void testSave() { ChatUser fromUser = new ChatUser(); fromUser
// =
// * userRepository.findByUserName("hmm"); ChatUser toUser = new ChatUser(); // * userRepository.findByUserName("hmm"); ChatUser toUser = new ChatUser();
// * toUser = userRepository.findByUserName("user2"); ChatMessage chatMessage = // * toUser = userRepository.findByUserName("user2"); ChatMessage chatMessage =
// * new ChatMessage(); chatMessage.setMessage("Hello!"); // * new ChatMessage(); chatMessage.setMessage("Hello!");

View File

@ -55,7 +55,8 @@
// @Mock // @Mock
// private UserRepository userRepository; // private UserRepository userRepository;
// private UserService userService = new UserServiceImpl(userRepository, userRoleRepository, passwordEncoder, roleRepository, // private UserService userService = new UserServiceImpl(userRepository,
// userRoleRepository, passwordEncoder, roleRepository,
// userSessionRepository); // userSessionRepository);
// // private ChatUser chatUser; // // private ChatUser chatUser;