package org.erlide.cover.views.model;

import java.util.LinkedList;
import java.util.List;

/**
 * Basic element of the EUnit result tree
 *
 * @author Aleksandra Lipiec
 *
 */
public class TestTreeObject {

    public static final int WARN = 0;
    public static final int SUCCESS = 1;
    public static final int FAILURE = 2;
    public static final int DESCR = 3;

    private final String name;
    private String description = "";
    private final List<TestTreeObject> children = new LinkedList<>();
    private TestTreeObject parent = null;
    private int type; // type of node

    public TestTreeObject(final String name, final int type) {
        this.name = name;
        description = name.replace("\"", "").replace("\\'", "'").replaceFirst("^'", "")
                .replaceFirst("'$", "");
        this.type = type;
    }

    public void setDescription(final String description) {
        this.description = description;
    }

    public String getDescription() {
        return description;
    }

    /**
     * Adds time to the description
     *
     * @param time
     */
    public void setTime(final int time) {
        description = String.format("%s     [%.3f]", description, time / 1000.0);
    }

    public void addChild(final TestTreeObject child) {
        child.setParent(this);
        children.add(child);
    }

    public List<TestTreeObject> getChildren() {
        return children;
    }

    public boolean hasChildren() {
        return !children.isEmpty();
    }

    public TestTreeObject getParent() {
        return parent;
    }

    private void setParent(final TestTreeObject parent) {
        this.parent = parent;
    }

    public String getName() {
        return name;
    }

    public TestTreeObject findNode(final String myName) {
        if (name.equals(myName)) {
            return this;
        }
        TestTreeObject result;
        for (final TestTreeObject obj : children) {
            if ((result = obj.findNode(myName)) != null) {
                return result;
            }
        }
        return null;
    }

    public int getType() {
        return type;
    }

    // updates type
    public void updateType() {
        if (children.size() == 0) {
            return;
        }

        type = TestTreeObject.SUCCESS;

        for (final TestTreeObject child : children) {
            if (child.getType() != TestTreeObject.SUCCESS) {
                type = TestTreeObject.WARN;
            }
        }
    }

}