// This Source Code is in the Public Domain per: http://litesoft.org/License.txt
package org.litesoft.orsup.transact.nonpublic;

import org.litesoft.orsup.transact.*;

public class TransactionAugmentationManagerImpl implements TransactionAugmentationManagerInternalExtension
{
    private HistoryAugmentorFactory mHistoryAugmentorFactory = null;
    private TransactionAugmentorFactory[] mAugmentorFactories = TransactionAugmentorFactory.EMPTY_ARRAY;

    @Override
    public synchronized HistoryAugmentorFactory getHistoryAugmentorFactory()
    {
        return mHistoryAugmentorFactory;
    }

    @Override
    public synchronized void setHistoryAugmentorFactory( HistoryAugmentorFactory pFactory )
    {
        mHistoryAugmentorFactory = pFactory;
    }

    @Override
    public synchronized TransactionAugmentorFactory[] getTransactionAugmentors()
    {
        return (mAugmentorFactories.length == 0) ? mAugmentorFactories : mAugmentorFactories.clone();
    }

    @Override
    public synchronized void addTransactionAugmentor( TransactionAugmentorFactory pFactory )
    {
        if ( pFactory != null )
        {
            int zCurLen = mAugmentorFactories.length;
            if ( zCurLen == 0 )
            {
                mAugmentorFactories = new TransactionAugmentorFactory[]{pFactory};
            }
            else if ( -1 == findTransactionAugmentor( pFactory ) )
            {
                TransactionAugmentorFactory[] zFactories = new TransactionAugmentorFactory[zCurLen + 1];
                System.arraycopy( mAugmentorFactories, 0, zFactories, 1, zCurLen );
                zFactories[0] = pFactory;
                mAugmentorFactories = zFactories;
            }
        }
    }

    @Override
    public synchronized boolean removeTransactionAugmentor( Class<? extends TransactionAugmentorFactory> pFactoryClass )
    {
        boolean rv = false;
        for ( int at; -1 != (at = findTransactionAugmentor( pFactoryClass )); )
        {
            rv = true;
            removeTransactionAugmentor( at );
        }
        return rv;
    }

    @Override
    public synchronized boolean removeTransactionAugmentor( TransactionAugmentorFactory pFactory )
    {
        int at = findTransactionAugmentor( pFactory );
        if ( at != -1 )
        {
            removeTransactionAugmentor( at );
            return true;
        }
        return false;
    }

    @Override
    public synchronized void clearTransactionAugmentors()
    {
        mAugmentorFactories = TransactionAugmentorFactory.EMPTY_ARRAY;
    }

    @Override
    public synchronized TransactionAugmentorFactory[] getRawTransactionAugmentorsArray()
    {
        return mAugmentorFactories;
    }

    private int findTransactionAugmentor( Class<? extends TransactionAugmentorFactory> pFactoryClass )
    {
        if ( pFactoryClass != null )
        {
            for ( int i = mAugmentorFactories.length; --i >= 0; )
            {
                if ( pFactoryClass == mAugmentorFactories[i].getClass() )
                {
                    return i;
                }
            }
        }
        return -1;
    }

    private int findTransactionAugmentor( TransactionAugmentorFactory pFactory )
    {
        if ( pFactory != null )
        {
            for ( int i = mAugmentorFactories.length; --i >= 0; )
            {
                if ( pFactory == mAugmentorFactories[i] )
                {
                    return i;
                }
            }
        }
        return -1;
    }

    private void removeTransactionAugmentor( int pIndex )
    {
        int ndxTo, curLength = mAugmentorFactories.length;
        if ( curLength <= 1 )
        {
            mAugmentorFactories = TransactionAugmentorFactory.EMPTY_ARRAY;
            return;
        }
        TransactionAugmentorFactory[] newAFs = new TransactionAugmentorFactory[curLength - 1];
        for ( int ndxFrom = ndxTo = 0; ndxFrom < mAugmentorFactories.length; ndxFrom++ )
        {
            if ( ndxFrom != pIndex )
            {
                newAFs[ndxTo++] = mAugmentorFactories[ndxFrom];
            }
        }
        mAugmentorFactories = newAFs;
    }
}