Merge pull request eugenp#149 from Doha2012/master

fix submit reddit

Author: Eugen

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

@@ -17,6 +17,8 @@
 import org.springframework.security.oauth2.client.resource.UserRedirectRequiredException;
 import org.springframework.stereotype.Controller;
 import org.springframework.ui.Model;
+import org.springframework.util.LinkedMultiValueMap;
+import org.springframework.util.MultiValueMap;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RequestParam;
 
@@ -43,48 +45,48 @@ public String getInfo(Model model) {
             throw e;
         } catch (Exception e) {
             LOGGER.error("Error occurred", e);
-            model.addAttribute("error", e.getLocalizedMessage());
+            model.addAttribute("error", e.getMessage());
         }
         return "reddit";
     }
 
     @RequestMapping("/submit")
     public String submit(Model model, @RequestParam Map<String, String> formParams) {
         try {
-            System.out.println(formParams.keySet());
-            HttpHeaders headers = new HttpHeaders();
-            headers.setContentType(MediaType.APPLICATION_JSON);
-            HttpEntity req = new HttpEntity(headers);
-
-            Map<String, String> param = new HashMap<String, String>();
-            param.put("api_type", "json");
-            param.put("kind", "link");
-            param.putAll(formParams);
-
-            System.out.println(param.keySet());
-            System.out.println(param.entrySet());
-            ResponseEntity<String> result = redditRestTemplate.postForEntity("https://oauth.reddit.com/api/submit", req, String.class, param);
-            model.addAttribute("error", result.getBody());
+            MultiValueMap<String, String> param = new LinkedMultiValueMap<String, String>();
+            param.add("api_type", "json");
+            param.add("kind", "link");
+            param.add("resubmit", "true");
+            param.add("sendreplies", "false");
+            param.add("then", "comments");
+
+            for (Map.Entry<String, String> entry : formParams.entrySet()) {
+                param.add(entry.getKey(), entry.getValue());
+            }
+
+            LOGGER.info("User submitting Link with these parameters: " + formParams.entrySet());
+            String result = redditRestTemplate.postForObject("https://oauth.reddit.com/api/submit", param, String.class);
+            LOGGER.info("Full Reddit Response: " + result);
+            String responseMsg = parseResponse(result);
+            model.addAttribute("msg", responseMsg);
         } catch (UserApprovalRequiredException e) {
             throw e;
         } catch (UserRedirectRequiredException e) {
             throw e;
         } catch (Exception e) {
             LOGGER.error("Error occurred", e);
-            model.addAttribute("error", e.getLocalizedMessage());
+            model.addAttribute("msg", e.getLocalizedMessage());
         }
-        return "reddit";
+        return "submissionResponse";
     }
 
     @RequestMapping("/post")
     public String showSubmissionForm(Model model) throws JsonProcessingException, IOException {
         try {
             String needsCaptchaResult = needsCaptcha();
             if (needsCaptchaResult.equalsIgnoreCase("true")) {
-                String newCaptchaResult = getNewCaptcha();
-                String[] split = newCaptchaResult.split("\"");
-                String iden = split[split.length - 2];
-                model.addAttribute("iden", iden.trim());
+                String iden = getNewCaptcha();
+                model.addAttribute("iden", iden);
             }
         } catch (UserApprovalRequiredException e) {
             throw e;
@@ -98,7 +100,7 @@ public String showSubmissionForm(Model model) throws JsonProcessingException, IO
         return "submissionForm";
     }
 
-    //
+    // === private
 
     private List<String> getSubreddit() throws JsonProcessingException, IOException {
         String result = redditRestTemplate.getForObject("https://oauth.reddit.com/subreddits/popular?limit=50", String.class);
@@ -125,11 +127,29 @@ private String getNewCaptcha() {
         param.put("api_type", "json");
 
         ResponseEntity<String> result = redditRestTemplate.postForEntity("https://oauth.reddit.com/api/new_captcha", req, String.class, param);
-        return result.getBody();
+        String[] split = result.getBody().split("\"");
+        return split[split.length - 2];
+    }
+
+    private String parseResponse(String responseBody) throws JsonProcessingException, IOException {
+        String result = "";
+        JsonNode node = new ObjectMapper().readTree(responseBody);
+        JsonNode errorNode = node.get("json").get("errors").get(0);
+        if (errorNode != null) {
+            for (JsonNode child : errorNode) {
+                result = result + child.toString().replaceAll("\"|null", "") + "<br>";
+            }
+            return result;
+        } else {
+            if (node.get("json").get("data") != null && node.get("json").get("data").get("url") != null)
+                return "Post submitted successfully <a href=\"" + node.get("json").get("data").get("url").asText() + "\"> check it out </a>";
+            else
+                return "Error Occurred";
+        }
     }
 
     public void setRedditRestTemplate(OAuth2RestTemplate redditRestTemplate) {
         this.redditRestTemplate = redditRestTemplate;
     }
 
-}
+}

spring-security-oauth/src/main/webapp/WEB-INF/jsp/reddit.jsp

@@ -3,20 +3,24 @@
 <head>
 
 <title>Spring Security OAuth</title>
+<link rel="stylesheet" href="https://maxcdn.bootstrapcdn.com/bootstrap/3.3.2/css/bootstrap.min.css">
+
 </head>
 <body>
+<div class="container">
 <c:choose>
     <c:when test="${info != null}">
         <h1>Your Reddit Info</h1>
         <b>Your reddit username is </b>${info}
         <br><br><br>
-        <a href="post">Submit to Reddit</a>
+        <a href="post" class="btn btn-primary">Submit to Reddit</a>
     </c:when>
     <c:otherwise> 
         <b>Sorry, error occurred</b> 
         <br><br>
         <div>${error}</div>
     </c:otherwise>
 </c:choose>
+</div>
 </body>
 </html>

spring-security-oauth/src/main/webapp/WEB-INF/jsp/submissionResponse.jsp

@@ -0,0 +1,15 @@
+<%@ taglib prefix="c" uri="http://java.sun.com/jsp/jstl/core"%>
+<html xmlns="http://www.w3.org/1999/xhtml">
+<head>
+
+<title>Spring Security OAuth</title>
+<link rel="stylesheet" href="https://maxcdn.bootstrapcdn.com/bootstrap/3.3.2/css/bootstrap.min.css">
+
+</head>
+<body>
+<div class="container">
+<h1>${msg}</h1>
+<a href="post" class="btn btn-primary">Submit another link to Reddit</a>
+</div>
+</body>
+</html>

spring-security-oauth/src/main/webapp/index.jsp

@@ -5,11 +5,12 @@
 <head>
 <meta http-equiv="Content-Type" content="text/html; charset=utf-8" />
 <title>Spring Security OAuth</title>
+<link rel="stylesheet" href="https://maxcdn.bootstrapcdn.com/bootstrap/3.3.2/css/bootstrap.min.css">
 </head>
 <body>
-	
+<div class="container">
 <h1>Welcome to Spring Security OAuth</h1>
-<a href="info">Login with Reddit</a>
+<a href="info" class="btn btn-primary">Login with Reddit</a>
 
 </body>
 </html>