prerna.auth.AppTokens Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of semoss Show documentation
SEMOSS
The newest version!
package prerna.auth;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Base64;
import java.util.Hashtable;
import java.util.List;

import org.apache.http.NameValuePair;
import org.apache.http.client.entity.UrlEncodedFormEntity;
import org.apache.http.client.methods.CloseableHttpResponse;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.message.BasicNameValuePair;

import prerna.om.AbstractValueObject;
import prerna.security.HttpHelperUtility;
import prerna.util.SocialPropertiesUtil;
import prerna.util.git.GitRepoUtils;
import prerna.util.Constants;

import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;

public class AppTokens extends AbstractValueObject{
	protected static final Logger logger = LogManager.getLogger(AppTokens.class);

	// name of this user in the SEMOSS system if there is one
	
	private static AppTokens app = null;
	
	private static SocialPropertiesUtil socialData = null;
	private static AccessToken twitToken = null;
	private static AccessToken googAppToken = null; 
	
	// need to have an access token store
	private Hashtable accessTokens = new Hashtable();
	
	private AppTokens() {
		
	}
	
	public static AppTokens getInstance() {
		if(app == null) {
			app = new AppTokens();
			loginGoogleApp();
			loginTwitterApp();
			
			if(googAppToken != null) {
				app.setAccessToken(googAppToken);
			}
			if(twitToken != null) {
				app.setAccessToken(twitToken);
			}
			
			socialData = SocialPropertiesUtil.getInstance();
		}
		return app;
	}
	
	public void setAccessToken(AccessToken value) {
		AuthProvider name = value.getProvider();
		accessTokens.put(name, value);
	}
	
	public AccessToken getAccessToken(AuthProvider name) {
		AccessToken token = accessTokens.get(name);
		if(token == null) {
			if(name == AuthProvider.TWITTER) {
				loginTwitterApp();
				if(twitToken != null) {
					app.setAccessToken(twitToken);
				}
			} else if(name == AuthProvider.GOOGLE_MAP) {
				loginGoogleApp();
				if(googAppToken != null) {
					app.setAccessToken(googAppToken);
				}
			}
			// try again...
			token = accessTokens.get(name);
		}
		return token;
	}
	
	public void dropAccessToken(AuthProvider name) {
		accessTokens.remove(name);
	}
	
	private static void loginTwitterApp() {
		// getting the bearer token on twitter for app authentication is a lot simpler
		// need to just combine the id and secret
		// base 64 and send as authorization
		GitRepoUtils.addCertForDomain("https://twitter.com");
		
		InputStream is = null;
		InputStreamReader isr = null;
		BufferedReader rd = null;
		CloseableHttpClient httpclient = null;
		if(twitToken == null) {
			try {
				String prefix = "twitter_";
				String clientId = "***REMOVED***";
				String clientSecret = "***REMOVED***";
				if(socialData != null && socialData.containsKey(prefix+"client_id")) {
					clientId = socialData.getProperty(prefix+"client_id");
				}
				if(socialData != null && socialData.containsKey(prefix+"secret_key")) {
					clientSecret = socialData.getProperty(prefix+"secret_key");
				}
				
				// make a joint string
				String jointString = clientId + ":" + clientSecret;

				// encde this base 64
				String encodedJointString = new String(Base64.getEncoder().encode(jointString.getBytes()));
				httpclient = HttpClients.createDefault();
				HttpPost httppost = new HttpPost("https://api.twitter.com/oauth2/token");
				httppost.addHeader("Authorization", "Basic " + encodedJointString);

				List paramList = new ArrayList();
				paramList.add(new BasicNameValuePair("grant_type", "client_credentials"));
				httppost.setEntity(new UrlEncodedFormEntity(paramList));

				CloseableHttpResponse authResp = httpclient.execute(httppost);

				System.out.println("Response Code " + authResp.getStatusLine().getStatusCode());

				is = authResp.getEntity().getContent();
				isr = new InputStreamReader(is);
				rd = new BufferedReader(isr);
				StringBuffer result = new StringBuffer();
				String line = "";
				while ((line = rd.readLine()) != null) {
					result.append(line);
				}

				twitToken = HttpHelperUtility.getJAccessToken(result.toString());
				twitToken.setProvider(AuthProvider.TWITTER);
			} catch(Exception ex) {
				logger.error(Constants.STACKTRACE, ex);
			} finally {
				if(is != null) {
					try {
						is.close();
					} catch(IOException e) {
						// ignore
					}
				}
				if(isr != null) {
					try {
						isr.close();
					} catch(IOException e) {
						// ignore
					}
				}
				if(rd != null) {
					try {
						rd.close();
					} catch(IOException e) {
						// ignore
					}
				}
				if(httpclient != null) {
					try {
						httpclient.close();
					} catch(IOException e) {
						// ignore
					}
				}
			}
		}
	}
	
	private static void loginGoogleApp() {
		// nothing big here
		// set the name on accesstoken
		if(socialData != null && googAppToken == null) {
			googAppToken = new AccessToken();
			googAppToken.setAccess_token(socialData.getProperty("google_maps_api"));
			googAppToken.setProvider(AuthProvider.GOOGLE_MAP);
		}
	}

}