iSharkFly-Docs
/
java-tutorials
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

modify need captcha

This commit is contained in:
DOHA 2015-03-10 20:57:01 +02:00
parent fee0f90b8e
commit 07af9a97b7
2 changed files with 27 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -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 class User {
this.tokenExpiration = tokenExpiration;
}
public boolean isCaptchaNeeded() {
return needCaptcha;
}
public void setNeedCaptcha(boolean needCaptcha) {
this.needCaptcha = needCaptcha;
}
public List<Post> getPosts() {
return posts;
}

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

@ -71,8 +71,8 @@ public class RedditController {
@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 class RedditController {
@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 class RedditController {
@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 class RedditController {
@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 class RedditController {
// === 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 @@ public class RedditController {
user.setAccessToken(token.getValue());
user.setRefreshToken(token.getRefreshToken().getValue());
user.setTokenExpiration(token.getExpiration());
}
final String needsCaptchaResult = needsCaptcha();
if (needsCaptchaResult.equalsIgnoreCase("true")) {
user.setNeedCaptcha(true);
} else {
user.setNeedCaptcha(false);
}
userReopsitory.save(user);
}
}
}