/*
 * Copyright 2014 MovingBlocks
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.terasology.logic.ai;

import org.terasology.entitySystem.entity.EntityManager;
import org.terasology.entitySystem.entity.EntityRef;
import org.terasology.entitySystem.systems.BaseComponentSystem;
import org.terasology.entitySystem.systems.RegisterSystem;
import org.terasology.logic.console.commandSystem.annotations.Command;
import org.terasology.registry.In;

import com.google.common.collect.Iterables;

/**
 * Artificial Intelligence commands for counting and destroying AIs used by entities with AI related components attached
 */
@RegisterSystem
public class AICommands extends BaseComponentSystem {
    @In
    private EntityManager entityManager;

    /**
     * Counts all AIs in the world
     * @return String string containing number of simple AIs and hierarchical AIs
     */
    @Command(runOnServer = true, shortDescription = "Count all AIs in the world")
    public String countAI() {
        int simpleAIs = Iterables.size(entityManager.getEntitiesWith(SimpleAIComponent.class));
        int hierarchical = Iterables.size(entityManager.getEntitiesWith(HierarchicalAIComponent.class));
        return "Simple AIs: " + simpleAIs + ", Hierarchical AIs: " + hierarchical;
    }

    /**
     * Destroys all entities with attached SimpleAIComponent or HierarchicalAIComponent in the world
     * @return String string containing number of simple AIs and hierarchical AIs destroyed
     */
    @Command(runOnServer = true, shortDescription = "Destroys all AIs in the world")
    public String destroyAI() {
        int simpleAI = 0;
        for (EntityRef ref : entityManager.getEntitiesWith(SimpleAIComponent.class)) {
            ref.destroy();
            simpleAI++;
        }

        int hierarchicalAI = 0;
        for (EntityRef ref : entityManager.getEntitiesWith(HierarchicalAIComponent.class)) {
            ref.destroy();
            hierarchicalAI++;
        }
        return "Simple AIs (" + simpleAI + ") Destroyed, Hierarchical AIs (" + hierarchicalAI + ") Destroyed ";
    }
}