gcube-sdi-suite/sdi-generic-client/src/main/java/org/gcube/spatial/data/clients/AbstractGenericRESTClient.java

package org.gcube.spatial.data.clients;


import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

import javax.ws.rs.client.Entity;
import javax.ws.rs.core.MediaType;

import org.gcube.common.clients.Call;
import org.gcube.common.gxrest.request.GXWebTargetAdapterRequest;
import org.gcube.common.gxrest.response.inbound.GXInboundResponse;
import org.gcube.spatial.data.clients.model.ConnectionDescriptor;
import org.gcube.spatial.data.sdi.model.credentials.AccessType;
import org.gcube.spatial.data.sdi.model.credentials.Credentials;
import org.gcube.spatial.data.sdi.model.faults.RemoteException;
import org.glassfish.jersey.client.authentication.HttpAuthenticationFeature;

import lombok.Getter;
import lombok.Setter;
import lombok.extern.slf4j.Slf4j;

@Slf4j
public abstract class AbstractGenericRESTClient implements GenericLoginClient{

	@Getter
	private ConnectionDescriptor conn;
	
	

	@Setter
	private String basePath=null;
	
	@Setter
	private MediaType defaultMediaType=MediaType.APPLICATION_JSON_TYPE;
	
	
	
	private Set<Class<?>> toRegisterClasses=new HashSet<Class<?>>();
	private Set<Object> toRegisterObjects=new HashSet<>();
	
	protected AbstractGenericRESTClient(ConnectionDescriptor conn) {
		this.conn=conn;
		
		// defaults
		toRegisterClasses.add(FollowRedirectFilter.class);
		
	}
	
	
	protected String getExistsPath() {
		return conn.getEndpoint();
	}
	
	@Override
	public boolean exist() {
		try {
			get(getExistsPath());
			return true;
		}catch(IOException e) {
			log.warn("Unable to check instance "+conn,e);
			return false;
		} catch (Exception e) {
			log.info("Server replied with error. Ignore this if it requires authentication. "+conn,e);
			return true;
		}
	}
	
	
	@Override
	public ConnectionDescriptor getInstance() {
		return conn;
	}
	
	public void authenticate(AccessType type) {
		log.info("Setting access type {} ",type);
		conn.getCredentials().forEach((Credentials c)->{
			if(c.getAccessType().equals(type))
				setHTTPBasicAuth(c.getUsername(), c.getPassword());
		});
	}
	
	public void authenticate() {
		log.info("Setting default access type between credentials {}",conn.getCredentials());
		if(conn.getCredentials().size()>0) {
			Credentials c =conn.getCredentials().get(0);
			setHTTPBasicAuth(c.getUsername(), c.getPassword());
		}else log.info("No credentials found in connection descriptor {} ",conn);
	};
	
	
	public void setHTTPBasicAuth(String user,String pwd) {
		log.info("Setting basic authentication, user : {} ",user);
		toRegisterObjects.add(HttpAuthenticationFeature.universal(user, pwd));
	}
	
	protected void register(Class<?> providerClass) {
		toRegisterClasses.add(providerClass);
	}
	
	protected void register(Object provider) {
		toRegisterObjects.add(provider);
	}
	
	
	
	protected GXWebTargetAdapterRequest resolve() throws UnsupportedEncodingException {
		GXWebTargetAdapterRequest toReturn =GXWebTargetAdapterRequest.newHTTPSRequest(conn.getEndpoint());
		
		toRegisterClasses.forEach((Class<?> clazz)->{toReturn.register(clazz);});
		toRegisterObjects.forEach((Object obj)->{toReturn.register(obj);});
		
		
		if(basePath!=null) return toReturn.path(basePath);
		
		else return toReturn;
	}
	
	
	protected <T> T makeCall(Call<GXWebTargetAdapterRequest, T> call) throws RemoteException, Exception{
		try{
			return call.call(resolve());
		}catch(RemoteException e) {
			throw e;
		}catch(Throwable t) {
			throw new Exception(t);
		}
	}
	
	
	
	
	//****** METHODS
	
	
	
	protected void delete(String path) throws RemoteException,Exception {
		makeCall(new Call<GXWebTargetAdapterRequest, Object>() {
			
			public Object call(GXWebTargetAdapterRequest endpoint) throws Exception {
				return check(resolve().path(path).delete(),null);
			};
			
		});
		
	}
	
	protected void post(String path,Object obj) throws RemoteException,Exception {
		post(path,Entity.entity(obj, MediaType.APPLICATION_JSON_TYPE));
	}
	
	
	
	protected <T> T  post (String path,Entity<?> entity,Class<T> returnClazz,Map<String,?> queryParams) throws RemoteException,Exception{
		return makeCall(new Call<GXWebTargetAdapterRequest, T>() {
			
			public T call(GXWebTargetAdapterRequest endpoint) throws Exception {
				Map<String,Object[]> actualMap=new HashMap<String, Object[]>();
				queryParams.forEach((String k, Object v)->{
					Object[] theArray=null;
					if(v instanceof Collection) {}
					else theArray=new Object[] {v};
					
					actualMap.put(k, theArray);
				});
				
//				for(java.util.Map.Entry<String,?> e:queryParams.entrySet()) {
//					String k=e.getKey();
//					if(!actualMap.containsKey(k))actualMap.
//				}
				return  check(endpoint.path(path).queryParams(actualMap).post(entity),returnClazz);
			};
			
		});
	}
	
	protected void post(String path,Entity<?> entity)throws RemoteException,Exception{
		post(path,entity,null,Collections.emptyMap());
	}
	
	protected <T> T post(String path,Entity<?> entity,Class<T> returnClazz)throws RemoteException,Exception{
		return post(path,entity,returnClazz,Collections.emptyMap());
	}
	
	protected void put(String path,Object obj) throws Exception {
		post(path,Entity.entity(obj, MediaType.APPLICATION_JSON_TYPE));
	}
	
	protected <T> T put(String path,Entity<?> entity,Class<T> returnClazz) throws Exception {
		return makeCall(new Call<GXWebTargetAdapterRequest, T>() {
			
			public T call(GXWebTargetAdapterRequest endpoint) throws Exception {
				return  check(endpoint.path(path).post(entity),returnClazz);
			};
			
		});
	}
	
	protected String get(String path) throws Exception {
		return get(path,String.class);
	}
	
	protected <T> T get(String path, Class<T> clazz) throws Exception {
		return get(path,clazz,defaultMediaType);
	}
	
	protected <T> T get(String path, Class<T> clazz, String mediaType) throws Exception{
		return makeCall(new Call<GXWebTargetAdapterRequest, T>() {
			
			public T call(GXWebTargetAdapterRequest endpoint) throws Exception {
				return  check(endpoint.path(path).header("accept",mediaType).get(),clazz);
			};
			
		});
	}
	
	protected <T> T get(String path, Class<T> clazz, MediaType type) throws Exception{
		return get(path,clazz,type+"");
	}
	
	
	
	
	protected static <T> T check(GXInboundResponse resp, Class<T> clazz) throws RemoteException{
		try {
		log.debug("Checking Response [Status : {}, Msg : {}]",resp.getHTTPCode(),resp.getMessage());
		if(!resp.isSuccessResponse()) {		
			RemoteException e=new RemoteException("Error received from server ");
			e.setRemoteMessage(resp.getMessage());
			e.setResponseHTTPCode(resp.getHTTPCode());
			e.setContent(resp.getStreamedContentAsString());
			throw e;
		}else {
			if(clazz==null) return null;			
			
			if(clazz==String.class) return (T) resp.getStreamedContentAsString();
			
			return resp.tryConvertStreamedContentFromJson(clazz);
		}
		}catch(RemoteException e) {
			throw e;
		}catch(Exception e1) {
			log.error("Reporting Remote Exception for ",e1);
			RemoteException e = new RemoteException("Unable to read  response from server.", e1);
			e.setRemoteMessage(resp.getMessage());
			e.setResponseHTTPCode(resp.getHTTPCode());
			try{
				e.setContent(resp.getStreamedContentAsString());
			}catch(IOException e2) {
				log.error("Error while reading resposne content",e2);
			}
			throw e;
		}
	}
	
	
	
}