/**
 * 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.policy.js.consumer;

import org.candlepin.controller.PoolManager;
import org.candlepin.model.Consumer;
import org.candlepin.model.Pool;
import org.candlepin.model.PoolCurator;

import com.google.inject.Inject;

import java.util.List;

/**
 * ConsumerRules
 */
public class ConsumerRules {

    private PoolCurator poolCurator;
    private PoolManager poolManager;

    @Inject
    public ConsumerRules(PoolManager poolManager, PoolCurator poolCurator) {
        this.poolManager = poolManager;
        this.poolCurator = poolCurator;
    }

    public void onConsumerDelete(Consumer consumer) {

        // Cleanup user restricted pools:
        if (consumer.getType().getLabel().equals("person")) {
            List<Pool> userRestrictedPools = poolCurator
                .listPoolsRestrictedToUser(consumer.getUsername());

            for (Pool pool : userRestrictedPools) {
                poolManager.deletePool(pool);
            }

        }
    }
}