package dao.entities;

import java.util.List;
import java.util.Set;
import java.util.UUID;
import java.util.stream.Collectors;

import javax.persistence.Query;
import javax.persistence.TypedQuery;
import javax.transaction.Transactional;

import dao.JpaDao;
import entities.Dataset;
import entities.responses.IDLabelPair;

public class DatasetDaoImpl extends JpaDao<Dataset, UUID> implements DatasetDao {

	public Dataset loadDetails(Dataset t) {
		// TODO Auto-generated method stub
		return null;
	}
	
	@Override
	public List<UUID> listAllIDs() {
		String queryString = "SELECT dataset.id FROM Dataset dataset where dataset.status>=0";
		TypedQuery<UUID> typedQuery = entityManager.createQuery(queryString, UUID.class);
		return typedQuery.getResultList();
	}
	
	
	@Override
	public List<IDLabelPair> listAllIDsLabels() {
		String queryString = "SELECT dataset.id, dataset.label FROM Dataset dataset where dataset.status>=0";
		Query query = (Query) entityManager.createQuery(queryString);
		List<Object[]> rows = query.getResultList();
		return rows.stream().map(row -> {
			return new IDLabelPair(row[0].toString(), row[1].toString());
		})
		.collect(Collectors.toList());
	}

	@Override
	public List<Dataset> getDatasetsOfDmp(UUID dmpID) {
		String queryString = "FROM Dataset dataset where dataset.dmp.id=:dmpID and dataset.status>=0";
		Query query = (Query) entityManager.createQuery(queryString);
		query.setParameter("dmpID", dmpID);
		List<Dataset> datasets = (List<Dataset>) query.getResultList();
		return datasets;
	}

	
}