EDIT

/**

* Copyright (C) 2007 EDIT

* European Distributed Institute of Taxonomy

* http://www.e-taxonomy.eu

*

* The contents of this file are subject to the Mozilla Public License Version 1.1

* See LICENSE.TXT at the top of this package for the full license terms.

*/

package eu.etaxonomy.taxeditor.store;

import java.util.Observable;

import java.util.Set;

import org.apache.logging.log4j.LogManager;

import org.apache.logging.log4j.Logger;

import org.eclipse.core.runtime.IProgressMonitor;

import org.eclipse.ui.IMemento;

import org.springframework.security.authentication.BadCredentialsException;

import org.springframework.security.authentication.LockedException;

import org.springframework.security.authentication.UsernamePasswordAuthenticationToken;

import org.springframework.security.core.Authentication;

import org.springframework.security.core.GrantedAuthority;

import org.springframework.security.core.context.SecurityContextHolder;

import eu.etaxonomy.cdm.api.application.CdmApplicationState;

import eu.etaxonomy.cdm.api.util.CdmUserHelper;

import eu.etaxonomy.cdm.api.util.RoleProberImpl;

import eu.etaxonomy.cdm.api.util.UserHelper;

import eu.etaxonomy.cdm.model.permission.Group;

import eu.etaxonomy.cdm.model.permission.User;

import eu.etaxonomy.cdm.persistence.permission.Role;

import eu.etaxonomy.taxeditor.model.IContextListener;

import eu.etaxonomy.taxeditor.model.MessagingUtils;

/**

 * <p>LoginManager class.</p>

 *

 * @author n.hoffmann

 * @created 03.07.2009

 */

public class LoginManager extends Observable implements IContextListener{

	public static final Logger logger = LogManager.getLogger();

	public static final String INCORRECT_CREDENTIALS_MESSAGE = "Login and/or Password incorrect";

	public static final String ACCOUNT_LOCKED_MESSAGE = "Account is locked";

	public static final String EMPTY_CREDENTIALS_MESSAGE = "Login and/or Password empty";

    private CdmUserHelper userHelper = null;

	public LoginManager(){

	    CdmStore.getContextManager().addContextListener(this);

	}

	private UserHelper userHelper() {

        if (userHelper == null){

            userHelper = new CdmUserHelper();

        }

        return userHelper;

	}

	/**

	 * <p>authenticate</p>

	 *

	 * @param token a {@link org.springframework.security.authentication.UsernamePasswordAuthenticationToken} object.

	 * @return true if the login attempt was successful even if the authentication has changed or not

	 */

	public boolean authenticate(String username, String password){

	    try{

	        doAuthenticate(username, password);

	    } catch (CdmAuthenticationException e) {

	        MessagingUtils.warningDialog("Could not authenticate", this, e.getMessage());

	        return false;

        }

	    return true;

	}

	public void doAuthenticate(String username, String password) throws CdmAuthenticationException {

	    try {

	        SecurityContextHolder.clearContext();

	        Authentication lastAuthentication = CdmStore.getCurrentAuthentiation();

	        UsernamePasswordAuthenticationToken token = new UsernamePasswordAuthenticationToken(username, password);

	        Authentication authentication = CdmStore.getAuthenticationManager().authenticate(token);

	        User user = (User) authentication.getPrincipal();

	        /* circumventing problem with hibernate not refreshing the transient collection authorities in this case,

	         * see http://dev.e-taxonomy.eu/trac/ticket/4053 */

	        user.initAuthorities();

	        if(logger.isDebugEnabled()){

	            StringBuilder gaText = new StringBuilder();

	            String indent = "    ";

	            Set<GrantedAuthority> gaSet = user.getGrantedAuthorities();

	            _logGrantedAuthotities(gaText, indent, gaSet);

	            for(Group gr : user.getGroups()){

	                gaText.append(indent).append("gr[").append(gr.hashCode()).append("] \"").append(gr.getName()).append("\" ").append(gr.toString()).append("\n");

	                _logGrantedAuthotities(gaText, indent + indent, gr.getGrantedAuthorities());

	            }

	            logger.debug("User authenticated: " + user.getUsername() + "\n" + gaText.toString());

	        }

	        authentication = new UsernamePasswordAuthenticationToken(user,password, authentication.getAuthorities());

	        SecurityContextHolder.getContext().setAuthentication(authentication);

	        CdmApplicationState.setCurrentSecurityContext(SecurityContextHolder.getContext());

	        if(!authentication.equals(lastAuthentication)){

	            this.setChanged();

	            this.notifyObservers();

	        }

	    } catch(BadCredentialsException e){

	        throw new CdmAuthenticationException(INCORRECT_CREDENTIALS_MESSAGE, e);

	    } catch(LockedException e){

	        throw new CdmAuthenticationException(ACCOUNT_LOCKED_MESSAGE, e);

	    } catch(IllegalArgumentException e){

	        e.printStackTrace();

	        throw new CdmAuthenticationException(EMPTY_CREDENTIALS_MESSAGE, e);

	    }

	}

	private void _logGrantedAuthotities(StringBuilder gaText, String indent,

			Set<GrantedAuthority> gaSet) {

		for(GrantedAuthority ga : gaSet){

			gaText.append(indent).append("ga[").append(ga.hashCode()).append("] ").append(ga.toString()).append("\n");

		}

	}

	public User getAuthenticatedUser(){

 		Authentication authentication = SecurityContextHolder.getContext().getAuthentication();

		if(authentication != null

				&& authentication.getPrincipal() != null

				&& authentication.getPrincipal() instanceof User){

			return (User)authentication.getPrincipal();

		}

		return null;

	}

	public void logoutAll(){

		SecurityContextHolder.clearContext();

		notifyObservers();

	}

	/**

	 * Whether the current user has the role admin

	 *

	 * @return

	 */

	public boolean isAdmin() {

	    boolean result = userHelper().userIs(new RoleProberImpl(Role.ROLE_ADMIN));

        return result;

	}

	/**

     * Whether the current user has the role user manager

     *

     * @return

     */

    public boolean isUserManager() {

        boolean result = userHelper().userIs(new RoleProberImpl(Role.ROLE_USER_MANAGER));

        return result;

    }

	@Override

	public void contextAboutToStop(IMemento memento, IProgressMonitor monitor) {

	}

	@Override

	public void contextStop(IMemento memento, IProgressMonitor monitor) {

	}

	@Override

	public void contextStart(IMemento memento, IProgressMonitor monitor){

	}

	@Override

	public void contextRefresh(IProgressMonitor monitor) {

	}

	@Override

	public void workbenchShutdown(IMemento memento, IProgressMonitor monitor) {

	}

}