package org.litesoft.GWT.client.context;

import org.litesoft.bo.views.*;
import org.litesoft.commonfoundation.base.*;
import org.litesoft.core.*;
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) && Currently.areEqual( 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) && Currently.areEqual( pRestrictedResource, zAccessor.getRestrictedResource() );
    }

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