Update REST Samples for required behavior

Issue-ID: AAF-361
Change-Id: Ia6dabfa3eeec2253c3f017ddc2d70afe4afbd54b
Signed-off-by: Instrumental <jonathan.gathman@att.com>

2 files changed, 189 insertions, 0 deletions

diff --git a/cadi/oauth-enduser/src/main/java/org/onap/aaf/cadi/enduser/ClientFactory.java b/cadi/oauth-enduser/src/main/java/org/onap/aaf/cadi/enduser/ClientFactory.java
new file mode 100644
index 00000000..50eaa759
--- /dev/null
+++ b/cadi/oauth-enduser/src/main/java/org/onap/aaf/cadi/enduser/ClientFactory.java
@@ -0,0 +1,56 @@
+/**
+ * ============LICENSE_START====================================================
+ * org.onap.aaf
+ * ===========================================================================
+ * Copyright (c) 2018 AT&T Intellectual Property. All rights reserved.
+ * ===========================================================================
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ * 
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ * 
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ============LICENSE_END====================================================
+ *
+ */
+package org.onap.aaf.cadi.enduser;
+
+import java.io.IOException;
+import java.net.URISyntaxException;
+import java.security.GeneralSecurityException;
+
+import org.onap.aaf.cadi.Access;
+import org.onap.aaf.cadi.CadiException;
+import org.onap.aaf.cadi.LocatorException;
+import org.onap.aaf.cadi.PropAccess;
+import org.onap.aaf.cadi.config.Config;
+import org.onap.aaf.cadi.oauth.TokenClientFactory;
+import org.onap.aaf.misc.env.APIException;
+
+public class ClientFactory {
+	private final TokenClientFactory tcf;
+	public ClientFactory(final PropAccess access) throws APIException, CadiException {
+		try {
+			tcf = TokenClientFactory.instance(access);
+		} catch (GeneralSecurityException | IOException e) {
+			throw new CadiException(e);
+		}
+	}
+	
+	public ClientFactory(String[] args) throws APIException, CadiException {
+		this(new PropAccess(args));
+	}
+
+	public SimpleRESTClient simpleRESTClient(final String endpoint, final String ... scopes) throws URISyntaxException, LocatorException, CadiException, APIException {
+		return new SimpleRESTClient(tcf, Config.AAF_OAUTH2_TOKEN_URL, endpoint, scopes);
+	}
+
+	public Access getAccess() {
+		return tcf.access;
+	}
+}
diff --git a/cadi/oauth-enduser/src/main/java/org/onap/aaf/cadi/enduser/SimpleRESTClient.java b/cadi/oauth-enduser/src/main/java/org/onap/aaf/cadi/enduser/SimpleRESTClient.java
new file mode 100644
index 00000000..9535ad64
--- /dev/null
+++ b/cadi/oauth-enduser/src/main/java/org/onap/aaf/cadi/enduser/SimpleRESTClient.java
@@ -0,0 +1,133 @@
+/**
+ * ============LICENSE_START====================================================
+ * org.onap.aaf
+ * ===========================================================================
+ * Copyright (c) 2018 AT&T Intellectual Property. All rights reserved.
+ * ===========================================================================
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ * 
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ * 
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ============LICENSE_END====================================================
+ *
+ */
+package org.onap.aaf.cadi.enduser;
+
+import java.io.IOException;
+import java.net.ConnectException;
+import java.security.Principal;
+
+import org.onap.aaf.cadi.CadiException;
+import org.onap.aaf.cadi.LocatorException;
+import org.onap.aaf.cadi.client.Future;
+import org.onap.aaf.cadi.client.Rcli;
+import org.onap.aaf.cadi.client.Result;
+import org.onap.aaf.cadi.client.Retryable;
+import org.onap.aaf.cadi.config.Config;
+import org.onap.aaf.cadi.oauth.TimedToken;
+import org.onap.aaf.cadi.oauth.TokenClient;
+import org.onap.aaf.cadi.oauth.TokenClientFactory;
+import org.onap.aaf.cadi.oauth.TzClient;
+import org.onap.aaf.cadi.principal.TaggedPrincipal;
+import org.onap.aaf.misc.env.APIException;
+
+public class SimpleRESTClient {
+	private static final String[] EMPTY = new String[0];
+	private final TokenClient tokenClient;
+	private final TzClient restClient;
+	private int callTimeout;
+	private String client_id;
+	private String app;
+	private String chain;
+	private Headers headers = new Headers() {
+		@Override
+		public String[] headers() {
+			return EMPTY;
+		}};
+	
+	public SimpleRESTClient(final TokenClientFactory tcf, final String tokenURL, final String endpoint, final String[] scope) throws CadiException, LocatorException, APIException {
+		callTimeout = Integer.parseInt(tcf.access.getProperty(Config.AAF_CALL_TIMEOUT,Config.AAF_CALL_TIMEOUT_DEF));
+		tokenClient = tcf.newClient(tokenURL);
+		Result<TimedToken> rtt = tokenClient.getToken(scope);
+		if(rtt.isOK()) {
+			restClient = tcf.newTzClient(endpoint);
+			
+			if((client_id = tcf.access.getProperty(Config.AAF_APPID, null))==null) {
+				if((client_id = tcf.access.getProperty(Config.CADI_ALIAS, null))==null) {
+					throw new CadiException(Config.AAF_APPID + " or " + Config.CADI_ALIAS + " needs to be defined");
+				}				
+			}
+			try {
+				restClient.setToken(client_id,rtt.value);
+			} catch (IOException e) {
+				throw new CadiException(e);
+			}
+		} else {
+			throw new CadiException(rtt.error);
+		}
+	}
+	
+	public SimpleRESTClient timeout(int newTimeout) {
+		callTimeout = newTimeout;
+		return this;
+	}
+
+	//Format:<ID>:<APP>:<protocol>[:AS][,<ID>:<APP>:<protocol>]*
+	public SimpleRESTClient as(Principal principal) {
+		if(principal==null) {
+			chain = null;
+		} else {
+			if(principal instanceof TaggedPrincipal) {
+				TaggedPrincipal tp = (TaggedPrincipal)principal;
+				chain = tp.getName() + ':' + (app==null?"":app) + ':' + tp.tag() + ":AS";
+			} else {
+				chain = principal.getName() + (app==null?"":':'+app);
+			}
+		}
+		return this;
+	}
+	
+	public String get(final String path) throws CadiException, LocatorException, APIException  {
+		return get(path,"application/json");
+	}
+
+	public String get(final String path, final String accepts) throws CadiException, LocatorException, APIException  {
+		return restClient.best(new Retryable<String>() {
+			@Override
+			public String code(Rcli<?> client) throws CadiException, ConnectException, APIException {
+				Future<String> future = client.read(path,accepts, headers());
+				if(future.get(callTimeout)) {
+					return future.value;
+				} else {
+					throw new APIException(future.code()  + future.body());
+				}					
+			}
+		});
+	}
+	
+	public interface Headers {
+		String[] headers();
+	}
+	
+	public String[] headers() {
+		if(chain==null) {
+			return headers.headers();
+		} else {
+			String[] strs = headers.headers();
+			String[] rv = new String[strs.length+2];
+			rv[0]=Config.CADI_USER_CHAIN;
+			rv[1]=chain;
+			for(int i = 0;i<strs.length;++i) {
+				rv[i+2]=strs[i];
+			}
+			return rv;
+		}
+	}
+}