Fix Test Hello

[aaf/authz.git] / auth / auth-hello / src / main / java / org / onap / aaf / auth / hello / API_Hello.java

diff --git a/auth/auth-hello/src/main/java/org/onap/aaf/auth/hello/API_Hello.java b/auth/auth-hello/src/main/java/org/onap/aaf/auth/hello/API_Hello.java
index e225223..97727ab 100644 (file)
--- a/auth/auth-hello/src/main/java/org/onap/aaf/auth/hello/API_Hello.java
+++ b/auth/auth-hello/src/main/java/org/onap/aaf/auth/hello/API_Hello.java
@@ -21,6 +21,9 @@
 
 package org.onap.aaf.auth.hello;
 
+import java.io.BufferedReader;
+import java.io.InputStreamReader;
+
 import javax.servlet.ServletOutputStream;
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
@@ -40,6 +43,9 @@ import org.onap.aaf.misc.env.TimeTaken;
 public class API_Hello {
 
 
+       private static final String APPLICATION_JSON = "application/json";
+       protected static final byte[] NOT_JSON = "Data does not look like JSON".getBytes();
+
        // Hide Public Constructor
        private API_Hello() {}
        
@@ -52,8 +58,9 @@ public class API_Hello {
         */
        public static void init(final AAF_Hello oauthHello) throws Exception {
                ////////
-               // Overall APIs
+               // Simple "GET" API
                ///////
+               
                oauthHello.route(HttpMethods.GET,"/hello/:perm*",API.TOKEN,new HttpCode<AuthzTrans, AAF_Hello>(oauthHello,"Hello OAuth"){
                        @Override
                        public void handle(AuthzTrans trans, HttpServletRequest req, HttpServletResponse resp) throws Exception {
@@ -84,5 +91,98 @@ public class API_Hello {
                        }
                }); 
 
+////////////////
+// REST APIs
+////////////////
+
+               ////////////////
+               // CREATE/POST
+               ////////////////
+               oauthHello.route(oauthHello.env,HttpMethods.POST,"/resthello/:id",new HttpCode<AuthzTrans, AAF_Hello>(oauthHello,"REST Hello Create") {
+                       @Override
+                       public void handle(AuthzTrans trans, HttpServletRequest req, HttpServletResponse resp) throws Exception {
+                               BufferedReader br = new BufferedReader(new InputStreamReader(req.getInputStream()));
+                               StringBuilder sb = new StringBuilder();
+                               while(br.ready()) {
+                                       sb.append(br.readLine());
+                               }
+                               String content = sb.toString();
+                               trans.info().printf("Content from %s: %s\n", pathParam(req, ":id"),content);
+                               if(content.startsWith("{") && content.endsWith("}")) {
+                                       resp.setStatus(201 /* OK */);
+                               } else {
+                                       resp.getOutputStream().write(NOT_JSON);
+                                       resp.setStatus(406);
+                               }
+                       }
+               },APPLICATION_JSON); 
+
+
+               ////////////////
+               // READ/GET
+               ////////////////
+               oauthHello.route(oauthHello.env,HttpMethods.GET,"/resthello/:id",new HttpCode<AuthzTrans, AAF_Hello>(oauthHello,"REST Hello Read") {
+                       @Override
+                       public void handle(AuthzTrans trans, HttpServletRequest req, HttpServletResponse resp) throws Exception {
+                               resp.setStatus(200 /* OK */);
+                               StringBuilder sb = new StringBuilder("{\"resp\": \"Hello REST AAF\",\"principal\": \"");
+                               sb.append(req.getUserPrincipal().getName());
+                               sb.append('"');
+                               String perm = pathParam(req, "perm");
+                               trans.info().printf("Read request from %s: %s\n", pathParam(req, ":id"),perm);
+                               if(perm!=null && perm.length()>0) {
+                                       TimeTaken tt = trans.start("Authorize perm", Env.REMOTE);
+                                       try {
+                                               sb.append(",\"validation\": { \"permission\" : \"");
+                                               sb.append(perm);
+                                               sb.append("\",\"has\" : \"");
+                                               sb.append(req.isUserInRole(perm));
+                                               sb.append("\"}");
+                                       } finally {
+                                               tt.done();
+                                       }
+                               }
+                               sb.append("}");
+                               ServletOutputStream os = resp.getOutputStream();
+                               os.println(sb.toString());
+                               trans.info().printf("Said 'RESTful Hello' to %s, Authentication type: %s",trans.getUserPrincipal().getName(),trans.getUserPrincipal().getClass().getSimpleName());
+                       }
+               },APPLICATION_JSON); 
+               
+               ////////////////
+               // UPDATE/PUT
+               ////////////////
+               oauthHello.route(oauthHello.env,HttpMethods.PUT,"/resthello/:id",new HttpCode<AuthzTrans, AAF_Hello>(oauthHello,"REST Hello Update") {
+                       @Override
+                       public void handle(AuthzTrans trans, HttpServletRequest req, HttpServletResponse resp) throws Exception {
+                               BufferedReader br = new BufferedReader(new InputStreamReader(req.getInputStream()));
+                               StringBuilder sb = new StringBuilder();
+                               while(br.ready()) {
+                                       sb.append(br.readLine());
+                               }
+                               String content = sb.toString();
+                               trans.info().printf("Content from %s: %s\n", pathParam(req, ":id"),content);
+                               if(content.startsWith("{") && content.endsWith("}")) {
+                                       resp.setStatus(200 /* OK */);
+                                       resp.getOutputStream().print(content);
+                               } else {
+                                       resp.getOutputStream().write(NOT_JSON);
+                                       resp.setStatus(406);
+                               }
+                       }
+               },APPLICATION_JSON); 
+
+
+               ////////////////
+               // DELETE
+               ////////////////
+               oauthHello.route(oauthHello.env,HttpMethods.DELETE,"/resthello/:id",new HttpCode<AuthzTrans, AAF_Hello>(oauthHello,"REST Hello Delete") {
+                       @Override
+                       public void handle(AuthzTrans trans, HttpServletRequest req, HttpServletResponse resp) throws Exception {
+                               trans.info().printf("Delete requested on %s\n", pathParam(req, ":id"));
+                               resp.setStatus(200 /* OK */);
+                       }
+               },APPLICATION_JSON); 
+
        }
 }