source: VirtualCollectionRegistry/trunk/VirtualCollectionRegistry/src/main/java/eu/clarin/cmdi/virtualcollectionregistry/gui/table/Provider.java @ 963

package eu.clarin.cmdi.virtualcollectionregistry.gui.table;

import java.util.Iterator;
import java.util.List;

import org.apache.wicket.WicketRuntimeException;
import org.apache.wicket.extensions.markup.html.repeater.data.table.filter.IFilterStateLocator;
import org.apache.wicket.extensions.markup.html.repeater.util.SortParam;
import org.apache.wicket.extensions.markup.html.repeater.util.SortableDataProvider;
import org.apache.wicket.model.IModel;

import eu.clarin.cmdi.virtualcollectionregistry.QueryOptions;
import eu.clarin.cmdi.virtualcollectionregistry.QueryOptions.Property;
import eu.clarin.cmdi.virtualcollectionregistry.VirtualCollectionRegistry;
import eu.clarin.cmdi.virtualcollectionregistry.VirtualCollectionRegistryException;
import eu.clarin.cmdi.virtualcollectionregistry.gui.ApplicationSession;
import eu.clarin.cmdi.virtualcollectionregistry.gui.DetachableVirtualCollectionModel;
import eu.clarin.cmdi.virtualcollectionregistry.model.VirtualCollection;

@SuppressWarnings("serial")
class Provider extends
        SortableDataProvider<VirtualCollection> implements
        IFilterStateLocator<FilterState> {
    private final boolean privateMode;
    private FilterState filterstate = new FilterState();

    public Provider(boolean privateMode) {
        this.privateMode = privateMode;
        setSort("created", false);
    }

    @Override
    public FilterState getFilterState() {
        return filterstate;
    }

    @Override
    public void setFilterState(FilterState state) {
        this.filterstate = state;
    }

    @Override
    public IModel<VirtualCollection> model(VirtualCollection vc) {
        return new DetachableVirtualCollectionModel(vc);
    }

    @Override
    public int size() {
        try {
            final VirtualCollectionRegistry vcr =
                VirtualCollectionRegistry.instance();
            return vcr.getVirtualCollectionCount(getFilter());
        } catch (VirtualCollectionRegistryException e) {
            throw new WicketRuntimeException(e);
        }
    }

    @Override
    public Iterator<? extends VirtualCollection> iterator(int first,
            int count) {
        try {
            final VirtualCollectionRegistry vcr =
                VirtualCollectionRegistry.instance();
            final List<VirtualCollection> results =
                vcr.getVirtualCollections(first, count, getFilter());
            return results.iterator();
        } catch (VirtualCollectionRegistryException e) {
            throw new WicketRuntimeException(e);
        }
    }

    private QueryOptions getFilter() {
        QueryOptions options = new QueryOptions();

        QueryOptions.Filter filter = options.and();
        if (privateMode) {
            ApplicationSession session = ApplicationSession.get();
            filter.add(QueryOptions.Property.VC_OWNER,
                       QueryOptions.Relation.EQ,
                       session.getUser());
        } else {
            filter.add(QueryOptions.Property.VC_STATE,
                       QueryOptions.Relation.EQ,
                       VirtualCollection.State.PUBLIC);
        }
        if (filterstate.hasName()) {
            filter.add(QueryOptions.Property.VC_NAME,
                       QueryOptions.Relation.EQ,
                       filterstate.getName());
        }
        if (filterstate.hasType()) {
            filter.add(QueryOptions.Property.VC_TYPE,
                       QueryOptions.Relation.EQ,
                       filterstate.getType());
        }
        if (filterstate.hasState()) {
            filter.add(QueryOptions.Property.VC_STATE,
                       QueryOptions.Relation.EQ,
                       filterstate.getState());
        }
        if (filterstate.hasDescription()) {
            filter.add(QueryOptions.Property.VC_DESCRIPTION,
                       QueryOptions.Relation.EQ,
                       filterstate.getDescription());
        }
        if (filterstate.hasCreated()) {
            filter.add(QueryOptions.Property.VC_CREATION_DATE,
                       filterstate.getCreatedRelation(),
                       filterstate.getCreated());
        }
        options.setFilter(filter);

        final SortParam s = getSort();
        if (s != null) {
            final String p = s.getProperty();
            Property property = null;
            if ("name".equals(p)) {
                property = Property.VC_NAME;
            } else if ("type".equals(p)) {
                property = Property.VC_TYPE;
            } else if ("state".equals(p)) {
                property = Property.VC_STATE;
            } else if ("description".equals(p)) {
                property = Property.VC_DESCRIPTION;
            } else if ("created".equals(p)) {
                property = Property.VC_CREATION_DATE;
            }
            if (property != null) {
                options.addSortProperty(property, s.isAscending());
            }
        }
        return options;
    }

} // class VirtualCollectionProvider