/*
 * HeadsUp Agile
 * Copyright 2009-2012 Heads Up Development Ltd.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package org.headsupdev.agile.web.wicket;

import org.apache.wicket.extensions.markup.html.repeater.util.SortableDataProvider;
import org.apache.wicket.model.CompoundPropertyModel;
import org.apache.wicket.model.IModel;
import org.hibernate.Criteria;
import org.hibernate.criterion.Order;
import org.hibernate.criterion.Projections;

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

/**
 * A data provider that supports sorting of entities retrieved from hibernate.
 * <p/>
 * Created: 22/01/2012
 *
 * @author Andrew Williams
 * @since 2.0
 */
public abstract class SortableEntityProvider<T> extends SortableDataProvider<T>
{
    protected abstract Criteria createCriteria();

    protected abstract List<Order> getDefaultOrder();

    public Iterator<T> iterator( int start, int limit )
    {
        Criteria criteria = createCriteria();
        if ( getSort() != null )
        {
            if ( getSort().isAscending() )
            {
                criteria.addOrder( Order.asc( getSort().getProperty() ).ignoreCase() );
            }
            else
            {
                criteria.addOrder( Order.desc( getSort().getProperty() ).ignoreCase() );
            }
        }
        else
        {
            for ( Order order : getDefaultOrder() )
            {
                criteria.addOrder( order );
            }
        }
        criteria.setFirstResult( start ).setMaxResults( limit );
        return (Iterator<T>) criteria.list().iterator();
    }

    public int size()
    {
        Criteria criteria = createCriteria();
        criteria.setProjection( Projections.count( getCountProperty() ) );
        return ((Number) criteria.uniqueResult() ).intValue();
    }

    public String getCountProperty()
    {
        return "id.id";
    }

    public IModel model( Object o )
    {
        return new CompoundPropertyModel( o );
    }
}

/*
 * HeadsUp Agile
 * Copyright 2009-2012 Heads Up Development Ltd.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package org.headsupdev.agile.web.wicket;

import org.apache.wicket.extensions.markup.html.repeater.util.SortableDataProvider;
import org.apache.wicket.model.CompoundPropertyModel;
import org.apache.wicket.model.IModel;
import org.hibernate.Criteria;
import org.hibernate.criterion.Order;
import org.hibernate.criterion.Projections;

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

/**
 * A data provider that supports sorting of entities retrieved from hibernate.
 * <p/>
 * Created: 22/01/2012
 *
 * @author Andrew Williams
 * @since 2.0
 */
public abstract class SortableEntityProvider<T> extends SortableDataProvider<T>
{
    protected abstract Criteria createCriteria();

    protected abstract List<Order> getDefaultOrder();

    public Iterator<T> iterator( int start, int limit )
    {
        Criteria criteria = createCriteria();
        if ( getSort() != null )
        {
            if ( getSort().isAscending() )
            {
                criteria.addOrder( Order.asc( getSort().getProperty() ).ignoreCase() );
            }
            else
            {
                criteria.addOrder( Order.desc( getSort().getProperty() ).ignoreCase() );
            }
        }
        else
        {
            for ( Order order : getDefaultOrder() )
            {
                criteria.addOrder( order );
            }
        }
        criteria.setFirstResult( start ).setMaxResults( limit );
        return (Iterator<T>) criteria.list().iterator();
    }

    public int size()
    {
        Criteria criteria = createCriteria();
        criteria.setProjection( Projections.count( getCountProperty() ) );
        return ((Number) criteria.uniqueResult() ).intValue();
    }

    public String getCountProperty()
    {
        return "id.id";
    }

    public IModel<T> model( Object o )
    {
        return new CompoundPropertyModel<T>( o );
    }
}

Fix error with dataprovider generics - thanks mart