package org.litesoft.GWT.client.context;

import org.litesoft.bo.views.*;
import org.litesoft.core.*;
import org.litesoft.core.typeutils.Objects;
import org.litesoft.core.util.*;
import org.litesoft.security.*;

import java.util.*;

public abstract class SecurityTypedVoDataProvider<T extends IViewObject> extends TypedVoDataProvider<T>
{
    protected SecurityTypedVoDataProvider( VoMetaData<T> pMetaData )
    {
        super( pMetaData );
    }

    @Override
    public void commit( TransactionSet pTransactionSet, final CommitCallBack<T> pCallBack, ObjectURL... pReturnOnSuccess )
    {
        final SecurityUserView zCurrentUser = checkForCurrentUserInTransaction( pTransactionSet );
        final SecurityRestrictedResourceView zCurrentRestrictedResource = checkForCurrentRestrictedResourceInTransaction( pTransactionSet );
        if ( (zCurrentUser == null) && (zCurrentRestrictedResource == null) )
        {
            super.commit( pTransactionSet, pCallBack, pReturnOnSuccess );
            return;
        }
        if ( zCurrentRestrictedResource != null )
        {
            pReturnOnSuccess = ObjectURL.prepend( zCurrentRestrictedResource.getObjectURL(), pReturnOnSuccess );
        }
        if ( zCurrentUser != null )
        {
            pReturnOnSuccess = ObjectURL.prepend( zCurrentUser.getObjectURL(), pReturnOnSuccess );
        }
        super.commit( pTransactionSet, new CommitCallBack<T>()
        {
            @Override
            public void success( ImmutableArrayList<T> pRows )
            {
                if ( zCurrentUser != null )
                {
                    updateContext( (SecurityUserView) pRows.get( 0 ) );
                    pRows = removeFirst( pRows );
                }
                if ( zCurrentRestrictedResource != null )
                {
                    updateContext( (SecurityRestrictedResourceView) pRows.get( 0 ) );
                    pRows = removeFirst( pRows );
                }
                pCallBack.success( pRows );
            }

            @Override
            public void error( String pError )
            {
                pCallBack.error( pError );
            }
        }, pReturnOnSuccess );
    }

    private ImmutableArrayList<T> removeFirst( ImmutableArrayList<T> pRows )
    {
        if ( pRows.isEmpty() )
        {
            return pRows;
        }
        ArrayList<T> temp = new ArrayList<T>( pRows );
        temp.remove( 0 );
        return new ImmutableArrayList<T>( temp );
    }

    public static SecurityUserView getCurrentUser()
    {
        AbstractCurrentUserViewAccessor zAccessor = ClientContext.get().checkGet( AbstractCurrentUserViewAccessor.class );
        return zAccessor != null ? zAccessor.getUser() : null;
    }

    public static SecurityUserView checkForCurrentUserInTransaction( TransactionSet pTransactionSet )
    {
        SecurityUserView zCurrentUser = getCurrentUser();
        if ( zCurrentUser != null )
        {
            SecurityUserView zTransUser = (SecurityUserView) pTransactionSet.get( zCurrentUser.getObjectURL() );
            if ( zTransUser != null )
            {
                return zTransUser;
            }
        }
        return null;
    }

    public static boolean checkForCurrentUser( SecurityUserView pUser )
    {
        AbstractCurrentUserViewAccessor zAccessor = ClientContext.get().checkGet( AbstractCurrentUserViewAccessor.class );
        return (zAccessor != null) && Objects.areNonArraysEqual( pUser, zAccessor.getUser() );
    }

    public static void updateContext( SecurityUserView pUserView )
    {
        AbstractCurrentUserViewAccessor zAccessor = ClientContext.get().checkGet( AbstractCurrentUserViewAccessor.class );
        if ( zAccessor != null )
        {
            zAccessor.setUser( pUserView );
        }
        if ( pUserView != null )
        {
            SecurityRestrictedResourceView zRestrictedResourceView = pUserView.getCurrentRestrictedResource(); // !null on Login!
            if ( zRestrictedResourceView != null )
            {
                updateContext( zRestrictedResourceView );
                CurrentRestrictedResourceOptionsAccessor zOptionsAccessor = ClientContext.get().checkGet( CurrentRestrictedResourceOptionsAccessor.class );
                if ( zOptionsAccessor != null )
                {
                    zOptionsAccessor.setRestrictedResourceOptions( pUserView.getAlternateRestrictedResourceOptions() );
                }
            }
        }
    }

    public static SecurityRestrictedResourceView checkForCurrentRestrictedResourceInTransaction( TransactionSet pTransactionSet )
    {
        AbstractCurrentRestrictedResourceViewAccessor zAccessor = ClientContext.get().checkGet( AbstractCurrentRestrictedResourceViewAccessor.class );
        if ( zAccessor != null )
        {
            SecurityRestrictedResourceView zCurrentRestrictedResource = zAccessor.getRestrictedResource();
            if ( zCurrentRestrictedResource != null )
            {
                ViewObject zViewObject = pTransactionSet.get( zCurrentRestrictedResource.getObjectURL() );
                return (SecurityRestrictedResourceView) zViewObject;
            }
        }
        return null;
    }

    public static boolean checkForCurrentRestrictedResource( SecurityRestrictedResourceView pRestrictedResource )
    {
        AbstractCurrentRestrictedResourceViewAccessor zAccessor = ClientContext.get().checkGet( AbstractCurrentRestrictedResourceViewAccessor.class );
        return (zAccessor != null) && Objects.areNonArraysEqual( pRestrictedResource, zAccessor.getRestrictedResource() );
    }

    public static void updateContext( SecurityRestrictedResourceView pRestrictedResourceView )
    {
        AbstractCurrentRestrictedResourceViewAccessor zAccessor = ClientContext.get().checkGet( AbstractCurrentRestrictedResourceViewAccessor.class );
        if ( zAccessor != null )
        {
            zAccessor.setRestrictedResource( pRestrictedResourceView );
        }
    }
}