/*
 * Copyright 2006-2008 Sxip Identity Corporation
 */

package org.openid4java.consumer;

import org.openid4java.association.Association;

import java.util.*;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

/**
 * @author Marius Scurtescu, Johnny Bufu
 */
public class InMemoryConsumerAssociationStore implements ConsumerAssociationStore
{
    private static Log _log = LogFactory.getLog(InMemoryConsumerAssociationStore.class);
    private static final boolean DEBUG = _log.isDebugEnabled();

    private Map _opMap = new HashMap();

    public synchronized void save(String opUrl, Association association)
    {
        removeExpired();

        Map handleMap = (Map) _opMap.get(opUrl);

        if (handleMap == null)
        {
            handleMap = new HashMap();


            _opMap.put(opUrl, handleMap);
        }

        String handle = association.getHandle();

        if(DEBUG)
            _log.debug("Adding association to the in-memory store: " + handle +
                       " with OP: " + opUrl);

        handleMap.put(association.getHandle(), association);
    }

    public synchronized Association load(String opUrl, String handle)
    {
        removeExpired();

        if (_opMap.containsKey(opUrl))
        {
            Map handleMap = (Map) _opMap.get(opUrl);

            if (handleMap.containsKey(handle))
            {
                return (Association) handleMap.get(handle);
            }
        }

        return null;
    }


    public synchronized Association load(String opUrl)
    {
        removeExpired();

        Association latest = null;

        if (_opMap.containsKey(opUrl))
        {
            Map handleMap = (Map) _opMap.get(opUrl);

            Iterator handles = handleMap.keySet().iterator();
            while (handles.hasNext())
            {
                String handle = (String) handles.next();

                Association association = (Association) handleMap.get(handle);

                if (latest == null ||
                        latest.getExpiry().before(association.getExpiry()))
                    latest = association;
            }
        }

        return latest;
    }

    public synchronized void remove(String opUrl, String handle)
    {
        removeExpired();

        if (_opMap.containsKey(opUrl))
        {
            Map handleMap = (Map) _opMap.get(opUrl);

            _log.info("Removing association: " + handle + " widh OP: " + opUrl);

            handleMap.remove(handle);

            if (handleMap.size() == 0)
                _opMap.remove(opUrl);
        }
    }


    private synchronized void removeExpired()
    {
        Set opToRemove = new HashSet();
        Iterator opUrls = _opMap.keySet().iterator();
        while (opUrls.hasNext())
        {
            String opUrl = (String) opUrls.next();

            Map handleMap = (Map) _opMap.get(opUrl);

            Set handleToRemove = new HashSet();
            Iterator handles = handleMap.keySet().iterator();
            while (handles.hasNext())
            {
                String handle = (String) handles.next();

                Association association = (Association) handleMap.get(handle);

                if (association.hasExpired())
                {
                    handleToRemove.add(handle);
                }
            }

            handles = handleToRemove.iterator();
            while (handles.hasNext())
            {
                String handle = (String) handles.next();

                _log.info("Removing expired association: " + handle +
                          " with OP: " + opUrl);

                handleMap.remove(handle);
            }

            if (handleMap.size() == 0)
                opToRemove.add(opUrl);
        }

        opUrls = opToRemove.iterator();
        while (opUrls.hasNext())
        {
            String opUrl = (String) opUrls.next();

            _opMap.remove(opUrl);
        }
    }

    protected synchronized int size()
    {
        int total = 0;

        Iterator opUrls = _opMap.keySet().iterator();
        while (opUrls.hasNext())
        {
            String opUrl = (String) opUrls.next();

            Map handleMap = (Map) _opMap.get(opUrl);

            total += handleMap.size();
        }

        return total;
    }
}