/*
 * This is eMonocot, a global online biodiversity information resource.
 *
 * Copyright © 2011–2015 The Board of Trustees of the Royal Botanic Gardens, Kew and The University of Oxford
 *
 * eMonocot 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.
 *
 * eMonocot 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.
 *
 * The complete text of the GNU Affero General Public License is in the source repository as the file
 * ‘COPYING’.  It is also available from <http://www.gnu.org/licenses/>.
 */
package org.emonocot.persistence.dao.hibernate;

import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.emonocot.model.hibernate.Fetch;
import org.emonocot.model.registry.Resource;
import org.emonocot.persistence.dao.ResourceDao;
import org.hibernate.Criteria;
import org.hibernate.FetchMode;
import org.hibernate.criterion.Projections;
import org.hibernate.criterion.Property;
import org.hibernate.criterion.Restrictions;
import org.joda.time.DateTime;
import org.springframework.batch.core.BatchStatus;
import org.springframework.stereotype.Repository;

/**
 *
 * @author ben
 *
 */
@Repository
public class ResourceDaoImpl extends SearchableDaoImpl<Resource> implements ResourceDao {

	/**
	 *
	 */
	private static Map<String, Fetch[]> FETCH_PROFILES;

	static {
		FETCH_PROFILES = new HashMap<String, Fetch[]>();
		FETCH_PROFILES.put("job-with-source",
				new Fetch[] {
				new Fetch("organisation", FetchMode.JOIN),
				new Fetch("parameters", FetchMode.SELECT)
		});
	}

	/**
	 *
	 */
	public ResourceDaoImpl() {
		super(Resource.class);
	}

	/**
	 * @param profile Set the profile name
	 * @return the fetch profile
	 */
	@Override
	public final Fetch[] getProfile(final String profile) {
		return ResourceDaoImpl.FETCH_PROFILES.get(profile);
	}

	/**
	 * @param sourceId Set the source identifier
	 * @return the total number of jobs for a given source
	 */
	public final Long count(final String sourceId) {
		Criteria criteria = getSession().createCriteria(type);
		criteria.createAlias("source", "src").add(
				Restrictions.eq("src.identifier", sourceId));
		criteria.setProjection(Projections.rowCount());
		return (Long) criteria.uniqueResult();
	}

	/**
	 * @param sourceId Set the source identifier
	 * @param page Set the offset (in size chunks, 0-based), optional
	 * @param size Set the page size
	 * @return A list of jobs
	 */
	public final List<Resource> list(final String sourceId, final Integer page, final Integer size) {
		Criteria criteria = getSession().createCriteria(type);
		criteria.createAlias("organisation", "org").add(Restrictions.eq("org.identifier", sourceId));

		if (size != null) {
			criteria.setMaxResults(size);
			if (page != null) {
				criteria.setFirstResult(page * size);
			}
		}
		return (List<Resource>) criteria.list();
	}

	/**
	 * @param id Set the job id
	 * @return the job
	 */
	public final Resource findByJobId(final Long id) {
		Criteria criteria = getSession().createCriteria(type);
		criteria.add(Restrictions.eq("jobId", id));
		return (Resource) criteria.uniqueResult();
	}

	@Override
	public boolean isHarvesting() {
		Criteria criteria = getSession().createCriteria(type);
		criteria.add(Restrictions.isNotNull("resourceType"));
		criteria.add(Restrictions.not(Restrictions.in("status", Arrays.asList(new BatchStatus[] {BatchStatus.COMPLETED, BatchStatus.FAILED,BatchStatus.ABANDONED, BatchStatus.STOPPED}) )));
		criteria.setProjection(Projections.rowCount());
		return (Long) criteria.uniqueResult() > 0;
	}

	@Override
	public List<Resource> listResourcesToHarvest(Integer limit, DateTime now, String fetch) {
		Criteria criteria = getSession().createCriteria(type);
		criteria.add(Restrictions.isNotNull("resourceType"));
		criteria.add(Restrictions.in("status", Arrays.asList(new BatchStatus[] {BatchStatus.COMPLETED, BatchStatus.FAILED,BatchStatus.ABANDONED, BatchStatus.STOPPED})));
		criteria.add(Restrictions.eq("scheduled", Boolean.TRUE));
		criteria.add(Restrictions.disjunction().add(Restrictions.lt("nextAvailableDate", now)).add(Restrictions.isNull("nextAvailableDate")));

		if (limit != null) {
			criteria.setMaxResults(limit);
		}
		enableProfilePreQuery(criteria, fetch);
		criteria.addOrder( Property.forName("nextAvailableDate").asc() );
		List<Resource> result = (List<Resource>) criteria.list();
		for(Resource t : result) {
			enableProfilePostQuery(t, fetch);
		}
		return result;
	}

	@Override
	protected boolean isSearchableObject() {
		return false;
	}

	@Override
	public Resource findResourceByUri(String identifier) {
		Criteria criteria = getSession().createCriteria(type);
		criteria.add(Restrictions.eq("uri", identifier));
		return (Resource) criteria.uniqueResult();
	}

}