package compbio.controllers;

import java.util.Collection;

import org.springframework.security.core.GrantedAuthority;
import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.security.core.userdetails.UserDetails;

public class BasicController {

	protected String getPrincipalName() {
		Object principal = SecurityContextHolder.getContext().getAuthentication().getPrincipal();
		if (principal instanceof UserDetails) {
			UserDetails details = (UserDetails) principal;
			String rolefix = "";
			String role = details.getUsername();
			Collection<? extends GrantedAuthority> au = details.getAuthorities();
			for (GrantedAuthority ga : au) {
				if (ga.getAuthority().equals("ROLE_LDAP_USER")) {
					rolefix = "LDAP:";
				}
			}
			return rolefix + role;
		}
		return principal.toString();
	}

	protected boolean isUserRole() {
		Object principal = SecurityContextHolder.getContext().getAuthentication().getPrincipal();
		if (principal instanceof UserDetails) {
			UserDetails details = (UserDetails) principal;
			Collection<? extends GrantedAuthority> au = details.getAuthorities();
			for (GrantedAuthority ga : au) {
				if (ga.getAuthority().equals("ROLE_USER") || ga.getAuthority().equals("ROLE_LDAP_USER")) {
					return true;
				}
			}
			return false;
		}
		return false;
	}

	protected boolean isAdminRole() {
		Object principal = SecurityContextHolder.getContext().getAuthentication().getPrincipal();
		if (principal instanceof UserDetails) {
			UserDetails details = (UserDetails) principal;
			Collection<? extends GrantedAuthority> au = details.getAuthorities();
			for (GrantedAuthority ga : au) {
				if (ga.getAuthority().equals("ROLE_ADMIN")) {
					return true;
				}
			}
			return false;
		}
		return false;
	}
	
}