UserEntityContextResolverImpl.java

package fi.otavanopisto.muikku.security.impl;

import fi.otavanopisto.muikku.model.users.UserEmailEntity;
import fi.otavanopisto.muikku.model.users.UserEntity;
import fi.otavanopisto.muikku.security.UserContextResolver;
import fi.otavanopisto.security.ContextReference;

public class UserEntityContextResolverImpl implements UserContextResolver {

  @Override
  public boolean handlesContextReference(ContextReference contextReference) {
    return 
        UserEntity.class.isInstance(contextReference) ||
        UserEmailEntity.class.isInstance(contextReference);
  }

  @Override
  public UserEntity resolveUser(ContextReference contextReference) {
    if (UserEntity.class.isInstance(contextReference))
      return (UserEntity) contextReference;
    
    if (UserEmailEntity.class.isInstance(contextReference))
      return ((UserEmailEntity) contextReference).getUserSchoolDataIdentifier().getUserEntity();
    
    return null;
  }
  
}