com.harium.suneidesis.task.planning.FindTask Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of core Show documentation
Project to represent knowledge
The newest version!
package com.harium.suneidesis.task.planning;

import com.harium.suneidesis.concept.Action;
import com.harium.suneidesis.concept.Concept;
import com.harium.suneidesis.repository.KnowledgeBase;

import java.util.Iterator;

public class FindTask extends ComplexKnowledgeTask {

    public static final String ATTRIBUTE_TARGET = "target";

    public FindTask(String name, Concept target, KnowledgeBase database) {
        super(name, database);
        setTarget(target);
    }

    @Override
    public boolean start() {
        boolean found = false;
        Iterator iterator = database.iterator();

        while (iterator.hasNext()) {
            Concept concept = iterator.next();
            Action fact = new Action(concept.getName()).wrap(concept);
            Concept factSubject = fact.getSubject();
            if (isInventory(factSubject)) {
                Inventory inventory = new Inventory().wrap(factSubject);

                Concept subject = subject();
                canAccess(subject, inventory);
                Iterator inventoryIterator = inventory.getAttributes().iterator();
                while (inventoryIterator.hasNext()) {
                    Concept item = inventoryIterator.next();
                    Concept target = target();
                    if (target.getName().equals(item.getName())) {
                        found = true;
                        break;
                    }
                }
            }
        }

        if (!found) {
            return false;
        }

        // Execute subTasks
        return super.start();
    }

    public Concept setTarget(Concept target) {
        return getAttributes().save(ATTRIBUTE_TARGET, target);
    }

    public Concept target() {
        return getAttributes().get(ATTRIBUTE_TARGET);
    }

    @Override
    public boolean canStart() {
        Concept target = target();
        return !target.isUnknown();
    }

    private boolean isInventory(Concept factSubject) {
        return "inventory".equals(factSubject.getName());
    }

    private boolean canAccess(Concept subject, Inventory inventory) {
        return inventory.getOwner().equals(subject);
    }

}