/**
 * Copyright (c) 2009 - 2012 Red Hat, Inc.
 *
 * This software is licensed to you under the GNU General Public License,
 * version 2 (GPLv2). There is NO WARRANTY for this software, express or
 * implied, including the implied warranties of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE. You should have received a copy of GPLv2
 * along with this software; if not, see
 * http://www.gnu.org/licenses/old-licenses/gpl-2.0.txt.
 *
 * Red Hat trademarks are not licensed under GPLv2. No permission is
 * granted to use or replicate Red Hat trademarks that are incorporated
 * in this software or its documentation.
 */
package org.candlepin.model;

import org.hibernate.criterion.Restrictions;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

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



/**
 * EnvironmentCurator
 */
public class EnvironmentCurator extends AbstractHibernateCurator<Environment> {
    private static Logger log = LoggerFactory.getLogger(OwnerContentCurator.class);

    public EnvironmentCurator() {
        super(Environment.class);
    }

    @SuppressWarnings("unchecked")
    public List<Environment> listForOwner(Owner o) {
        return this.currentSession().createCriteria(Environment.class)
            .add(Restrictions.eq("owner", o)).list();
    }

    @SuppressWarnings("unchecked")
    public List<Environment> listForOwnerByName(Owner o, String envName) {
        return this.currentSession().createCriteria(Environment.class)
            .add(Restrictions.eq("owner", o)).add(Restrictions.eq("name", envName)).list();
    }

    @SuppressWarnings("unchecked")
    public List<Environment> listWithContent(Set<String> contentIds) {
        return currentSession().createCriteria(Environment.class)
            .createCriteria("environmentContent")
            .createCriteria("content")
            .add(Restrictions.in("id", contentIds))
            .list();
    }

    /**
     * Performs a bulk deletion of all environment objects for the given owner.
     *
     * @param owner
     *  The owner for which to delete environments
     *
     * @return
     *  the number of environments deleted
     */
    public int deleteEnvironmentsForOwner(Owner owner) {
        String jpql = "SELECT e.id FROM Environment e WHERE e.owner.id = :owner_id";

        List<String> ids = this.getEntityManager()
            .createQuery(jpql, String.class)
            .setParameter("owner_id", owner.getId())
            .getResultList();

        int count = 0;

        if (ids != null && !ids.isEmpty()) {
            // We have some matching content, delete environment content first...
            Map<String, Object> criteria = new HashMap<String, Object>();
            criteria.put("environment_id", owner.getId());

            count = this.bulkSQLDelete(EnvironmentContent.DB_TABLE, criteria);
            log.info("{} environment-content relations updated", count);

            // Cleanup the environment objects themselves...
            criteria.clear();
            criteria.put("id", ids);

            count = this.bulkSQLDelete(Environment.DB_TABLE, criteria);
            log.info("{} environments deleted", count);
        }
        else {
            log.info("0 environments deleted");
        }

        return count;
    }
}