LoginContext.java

/**
 *  Copyright 2012 Universitat Pompeu Fabra.
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 *
 *
 */
package org.onexus.resource.api.session;

import java.io.Serializable;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

public class LoginContext implements Serializable {

    private static Map<String, LoginContext> contexts = new HashMap<String, LoginContext>();

    public static LoginContext ANONYMOUS_CONTEXT = new LoginContext();
    public static LoginContext SERVICE_CONTEXT = new LoginContext("service");

    private static ThreadLocal<LoginContext> loginContext = new ThreadLocal<LoginContext>() {
        @Override
        protected LoginContext initialValue() {
            return ANONYMOUS_CONTEXT;
        }
    };

    public static LoginContext get() {
        return loginContext.get();
    }

    public static LoginContext get(String sessionId) {
        return contexts.get(sessionId);
    }

    public static void set(LoginContext ctx, String sessionId) {
        loginContext.set(ctx);

        if (sessionId != null) {
            contexts.put(sessionId, ctx);
        }
    }

    private String userName;
    private Set<String> roles = new HashSet<String>();

    public LoginContext() {
        this.userName = null;
    }

    public LoginContext(String userName) {
        this.userName = userName;
    }

    public void logout() {
        this.userName = null;
    }

    public boolean isAnonymous() {
        return userName==null;
    }

    public String getUserName() {
        return userName;
    }

    public Set<String> getRoles() {
        return roles;
    }

    public void addRole(String role) {
        roles.add(role);
    }

    /**
     * Whether this context object containes the provided role.
     *
     * @param role the role to check
     * @return true if it contains the role, false otherwise
     */
    public boolean hasRole(final String role) {
        if (userName != null && role != null) {
            return roles.contains(role);
        }
        return false;
    }


}