Merge pull request #475 from joshzcold/master

Update changes to rest test serverlet. merge most recent pwm changes

Author: jrivard

rest-test-service/src/main/java/password/pwm/resttest/ExternalMacroServlet.java

@@ -24,6 +24,8 @@
 package password.pwm.resttest;
 
 import com.google.gson.Gson;
+import com.google.gson.JsonObject;
+import com.google.gson.JsonParser;
 import org.apache.commons.io.IOUtils;
 
 import javax.servlet.ServletException;
@@ -34,23 +36,26 @@
 import java.io.IOException;
 import java.io.InputStream;
 import java.io.PrintWriter;
-import java.time.Instant;
 
 @WebServlet(
         name = "NewUserServlet",
-        urlPatterns = { "/sms", "/macro" }
+        urlPatterns = { "/sms", "/macro", "/external-token-destination", "/external-password-check" }
 )
 
 public class ExternalMacroServlet extends HttpServlet
 {
     private static final String USERNAME_PARAMETER = "username";
     private static final String SUCCESSFUL = "true";
     private static final String UNSUCCESSFUL = "false";
+    private static final String SMS_URL = "/sms";
+    private static final String MACRO_URL = "/macro";
+    private static final String EXTERNAL_TOKEN_DESTINATION_URL = "/external-token-destination";
+    private static final String EXTERNAL_PASSWORD_CHECK_URL = "/external-password-check";
 
     @Override
     protected void doPost( final HttpServletRequest req, final HttpServletResponse resp ) throws ServletException, IOException
     {
-        if ( req.getServletPath().equals( "/sms" ) )
+        if ( req.getServletPath().equals( SMS_URL ) )
         {
             final SmsResponse instance = SmsResponse.getInstance();
             final InputStream inputStream = req.getInputStream();
@@ -59,8 +64,7 @@ protected void doPost( final HttpServletRequest req, final HttpServletResponse r
             final String[] messageContent = body.split( "=" );
             final String message = messageContent[messageContent.length - 1];
             final String username = message.split( "\\+" )[0];
-            final Instant currentDate = Instant.now();
-            final SmsPostResponseBody messageBody = new SmsPostResponseBody( message, currentDate );
+            final SmsPostResponseBody messageBody = new SmsPostResponseBody( message );
 
             instance.addToMap( username, messageBody );
 
@@ -72,12 +76,41 @@ protected void doPost( final HttpServletRequest req, final HttpServletResponse r
             writer.write(  "{\"output\":\"Message Received\"}" );
             writer.close();
         }
+        else if ( req.getServletPath().equals( EXTERNAL_TOKEN_DESTINATION_URL ) )
+        {
+            System.out.println( "External Token Destination" );
+            final InputStream inputStream = req.getInputStream();
+            final String body = IOUtils.toString( inputStream );
+            final JsonObject jsonObject = new JsonParser().parse( body ).getAsJsonObject();
+            final String email = jsonObject.getAsJsonObject( "tokenDestination" ).get( "email" ).getAsString();
+            final String sms = jsonObject.getAsJsonObject( "tokenDestination" ).get( "sms" ).getAsString();
+            final String displayValue = "YourTokenDestination";
+
+            resp.setHeader( "Content-Type", "application/json" );
+
+            final PrintWriter writer = resp.getWriter();
+            final String response = "{\"email\":\"" + email + "\",\"sms\":\"" + sms + "\",\"displayValue\":\"" + displayValue + "\"}";
+            writer.write( response );
+            writer.close();
+        }
+        else if ( req.getServletPath().equals( EXTERNAL_PASSWORD_CHECK_URL ) )
+        {
+            System.out.println( "External Password Check" );
+            final boolean error = false;
+            final String errorMessage = "No error.";
+            resp.setHeader( "Content-Type", "application/json" );
+
+            final PrintWriter writer = resp.getWriter();
+            final String response = "{\"error\":\"" + error + "\",\"errorMessage\":\"" + errorMessage + "\"}";
+            writer.write( response );
+            writer.close();
+        }
     }
 
     @Override
     protected void doGet( final HttpServletRequest req, final HttpServletResponse resp ) throws IOException
     {
-        if ( req.getServletPath().equals( "/sms" ) )
+        if ( req.getServletPath().equals( SMS_URL ) )
         {
             //Check request
             final SmsResponse instance = SmsResponse.getInstance();

rest-test-service/src/main/java/password/pwm/resttest/SmsPostResponseBody.java

@@ -27,49 +27,37 @@
 public class SmsPostResponseBody
 {
     private String messageContent;
-    private Instant date;
-
-    public SmsPostResponseBody( final String message, final Instant date )
-    {
-        final String[] strings = message.split( "&" );
-        this.messageContent = strings[strings.length - 1];
-        this.date = date;
-    }
+//    private Instant date;
 
     public SmsPostResponseBody( final String message )
     {
         final String[] strings = message.split( "&" );
         this.messageContent = strings[strings.length - 1];
     }
 
-    public SmsPostResponseBody( final Instant date )
-    {
-        this.date = date;
-        this.messageContent = "";
-    }
-
     public SmsPostResponseBody()
     {
 
     }
 
+
     public String getMessageContent()
     {
         return messageContent;
     }
 
-    public void setMessageContent( final String messageContent )
-    {
-        this.messageContent = messageContent;
-    }
+//    public void setMessageContent( final String messageContent )
+//    {
+//        this.messageContent = messageContent;
+//    }
 
     public Instant getDate()
     {
-        return date;
+        return Instant.now();
     }
 
-    public void setDate( final Instant date )
-    {
-        this.date = date;
-    }
+//    public void setDate( final Instant date )
+//    {
+//        this.date = Instant.now();
+//    }
 }