path: root/src/main/java/org/openslx/imagemaster/session/SessionManager.java

package org.openslx.imagemaster.session;

import java.sql.SQLException;
import java.util.Iterator;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.UUID;
import java.util.concurrent.TimeUnit;

import org.apache.log4j.Logger;
import org.openslx.bwlp.thrift.iface.ClientSessionData;
import org.openslx.bwlp.thrift.iface.Satellite;
import org.openslx.bwlp.thrift.iface.UserInfo;
import org.openslx.imagemaster.db.mappers.DbSatellite;
import org.openslx.imagemaster.util.Hash;
import org.openslx.util.QuickTimer;
import org.openslx.util.QuickTimer.Task;

/**
 * Class for managing active user sessions. This class and all its function are
 * (supposed to be) thread-safe.
 */
public class SessionManager
{
	private static Logger log = Logger.getLogger( SessionManager.class );

	// Map of currently known sessions
	private static final Map<String, Session> sessions = new LinkedHashMap<>();

	public static ClientSessionData addSession( Session session )
	{
		final String authToken = Hash.md5( UUID.randomUUID().toString() );
		final String sessionId = Hash.sha256( UUID.randomUUID().toString() );

		synchronized ( sessions ) {
			sessions.put( authToken, session );
			sessions.put( sessionId, session );
		}
		UserInfo ui = session.getUserInfo();
		List<Satellite> sats;
		try {
			sats = DbSatellite.getSatellites( ui );
		} catch ( SQLException e ) {
			sats = null;
		}
		return new ClientSessionData( sessionId, authToken, sats, ui );
	}

	public static Session getSessionFromToken( String token )
	{
		if ( token == null || token.length() != 32 ) {
			log.debug( "invalid token format: " + token );
			return null;
		}
		final Session session;
		synchronized ( sessions ) {
			session = sessions.get( token );
		}
		if ( session == null || session.timedOut() ) {
			return null;
		}
		return session;
	}

	static {
		QuickTimer.scheduleAtFixedDelay( new Task() {
			@Override
			public void fire()
			{
				synchronized ( sessions ) {
					Iterator<Session> it = sessions.values().iterator();
					while ( it.hasNext() ) {
						final Session s = it.next();
						if ( s.timedOut() ) {
							it.remove();
						}
					}
				}
			}
		}, 123, TimeUnit.MINUTES.toMillis( 13 ) );
	}

	public static Object getSessionFromSessionId( String sessionId )
	{
		if ( sessionId == null || sessionId.length() != 64 ) {
			log.debug( "invalid sessionid format: " + sessionId );
			return null;
		}
		final Session session;
		synchronized ( sessions ) {
			session = sessions.get( sessionId );
		}
		if ( session == null || session.timedOut() ) {
			return null;
		}
		return session;
	}

	public static void invalidate( String sessionId )
	{
		if ( sessionId == null || sessionId.length() != 64 ) {
			log.debug( "invalidate: invalid sessionid format: " + sessionId );
			return;
		}
		synchronized ( sessions ) {
			Session session = sessions.get( sessionId );
			if ( session != null ) {
				session.invalidate();
			}
		}
	}

}