com.meltmedia.cadmium.servlets.jersey.AuthorizationCache Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of cadmium-servlets Show documentation
The newest version!
package com.meltmedia.cadmium.servlets.jersey;

import com.meltmedia.cadmium.core.AuthorizationApi;
import com.meltmedia.cadmium.core.Scheduled;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.inject.Inject;
import javax.inject.Singleton;
import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;
import java.util.TreeSet;
import java.util.concurrent.TimeUnit;

/**
 * Holds and manages cache of github authorizations.
 */
@Singleton
public class AuthorizationCache {
  protected Logger logger = LoggerFactory.getLogger(getClass());
  protected static class Cache {
    Map usernames = new HashMap();
    Map teamIds = new HashMap();
    Map authorizations = new HashMap();
  }
  protected Cache currentCache;

  @Inject
  protected ApiService apiService;

  public void checkToken(String token) throws Exception {
    Cache cache = currentCache;
    if(!cache.authorizations.containsKey(token) || !cache.authorizations.get(token)){
      String username = fetchUsernameFromGithub(token);
      if(username == null) {
        cache.authorizations.put(token, false);
        throw new Exception("Token has been revoked.");
      } else {
        cache.usernames.put(token, username);
        cache.authorizations.put(token, true);
      }
    }
  }

  public Integer[] getTeamIds(String token) throws Exception {
    checkToken(token);
    Cache cache = currentCache;
    Integer teamIdList[] = cache.teamIds.get(token);
    if(teamIdList == null) {
      teamIdList = fetchTeamIdsFromGithub(token);
      if(teamIdList != null) {
        cache.teamIds.put(token, teamIdList);
      }
    }
    return teamIdList;
  }

  public String getUserName(String token) throws Exception {
    checkToken(token);
    Cache cache = currentCache;
    return currentCache.usernames.get(token);
  }

  @Scheduled(interval = 1, unit = TimeUnit.HOURS)
  public void clearCache() {
    logger.debug("Resetting cache.");
    currentCache = new Cache();
  }

  private Integer[] fetchTeamIdsFromGithub(String token) {
    try {
      AuthorizationApi apiClient = apiService.getAuthorizationApi(token);
      String orgs[] = apiClient.getAuthorizedOrgs();
      Set teams = new TreeSet();
      if(orgs != null) {
        for(String org: orgs) {
          Integer teamsInOrg[] = apiClient.getAuthorizedTeamsInOrg(org);
          if(teamsInOrg != null) {
            teams.addAll(Arrays.asList(teamsInOrg));
          }
        }
      }
      return teams.toArray(new Integer[] {});
    } catch (Exception e) {
      logger.warn("Failed to fetch team ids.", e);
    }
    return null;
  }

  private String fetchUsernameFromGithub(String token) {
    try {
      AuthorizationApi apiClient = apiService.getAuthorizationApi(token);
      return apiClient.getUserName();
    } catch(Exception e) {
      logger.warn("Failed to get user name.", e);
    }
    return null;
  }
}