modify need captcha

Author: Doha2012

spring-security-oauth/src/main/java/org/baeldung/persistence/model/User.java

@@ -24,6 +24,8 @@ public class User {
 
     private Date tokenExpiration;
 
+    private boolean needCaptcha;
+
     @OneToMany(mappedBy = "user")
     private List<Post> posts;
 
@@ -71,6 +73,14 @@ public void setTokenExpiration(Date tokenExpiration) {
         this.tokenExpiration = tokenExpiration;
     }
 
+    public boolean isCaptchaNeeded() {
+        return needCaptcha;
+    }
+
+    public void setNeedCaptcha(boolean needCaptcha) {
+        this.needCaptcha = needCaptcha;
+    }
+
     public List<Post> getPosts() {
         return posts;
     }

spring-security-oauth/src/main/java/org/baeldung/web/RedditController.java

@@ -71,8 +71,8 @@ public final String submit(final Model model, @RequestParam final Map<String, St
 
     @RequestMapping("/post")
     public final String showSubmissionForm(final Model model) {
-        final String needsCaptchaResult = needsCaptcha();
-        if (needsCaptchaResult.equalsIgnoreCase("true")) {
+        final boolean isCaptchaNeeded = getCurrentUser().isCaptchaNeeded();
+        if (isCaptchaNeeded) {
             final String iden = getNewCaptcha();
             model.addAttribute("iden", iden);
         }
@@ -81,8 +81,8 @@ public final String showSubmissionForm(final Model model) {
 
     @RequestMapping("/postSchedule")
     public final String showSchedulePostForm(final Model model) {
-        final String needsCaptchaResult = needsCaptcha();
-        if (needsCaptchaResult.equalsIgnoreCase("true")) {
+        final boolean isCaptchaNeeded = getCurrentUser().isCaptchaNeeded();
+        if (isCaptchaNeeded) {
             model.addAttribute("msg", "Sorry, You do not have enought karma");
             return "submissionResponse";
         }
@@ -92,7 +92,7 @@ public final String showSchedulePostForm(final Model model) {
     @RequestMapping(value = "/schedule", method = RequestMethod.POST)
     public final String schedule(final Model model, @RequestParam final Map<String, String> formParams) throws ParseException {
         logger.info("User scheduling Post with these parameters: " + formParams.entrySet());
-        final User user = userReopsitory.findByAccessToken(redditRestTemplate.getAccessToken().getValue());
+        final User user = getCurrentUser();
         final Post post = new Post();
         post.setUser(user);
         post.setSent(false);
@@ -116,7 +116,7 @@ public final String schedule(final Model model, @RequestParam final Map<String,
 
     @RequestMapping("/posts")
     public final String getScheduledPosts(final Model model) {
-        final User user = userReopsitory.findByAccessToken(redditRestTemplate.getAccessToken().getValue());
+        final User user = getCurrentUser();
         final List<Post> posts = postReopsitory.findByUser(user);
         model.addAttribute("posts", posts);
         return "postListView";
@@ -160,6 +160,10 @@ public String updatePost(Model model, @PathVariable("id") final Long id, @Reques
 
     // === private
 
+    private User getCurrentUser() {
+        return userReopsitory.findByAccessToken(redditRestTemplate.getAccessToken().getValue());
+    }
+
     private final MultiValueMap<String, String> constructParams(final Map<String, String> formParams) {
         final MultiValueMap<String, String> param = new LinkedMultiValueMap<String, String>();
         param.add(RedditApiConstants.API_TYPE, "json");
@@ -211,9 +215,15 @@ private final void addUser(final String name, final OAuth2AccessToken token) {
             user.setAccessToken(token.getValue());
             user.setRefreshToken(token.getRefreshToken().getValue());
             user.setTokenExpiration(token.getExpiration());
-            userReopsitory.save(user);
         }
 
+        final String needsCaptchaResult = needsCaptcha();
+        if (needsCaptchaResult.equalsIgnoreCase("true")) {
+            user.setNeedCaptcha(true);
+        } else {
+            user.setNeedCaptcha(false);
+        }
+        userReopsitory.save(user);
     }
 
 }