TypeCodeFilter.java

package net.squiz.matrix.inspector;

import net.squiz.matrix.matrixtree.MatrixTreeNode;
import java.util.*;

public class TypeCodeFilter implements Filter {

    private List conditions;

    public TypeCodeFilter() {
        conditions = new ArrayList();
    }

    public TypeCodeFilter(Object condition) {
        conditions = new ArrayList();
        addCondition(condition);
    }

    public void addCondition(Object condition) {
        if (!conditions.contains(condition))
            conditions.add(condition);
    }

    public void removeCondition(Object condition) {
            conditions.remove(condition);
    }

    public boolean allowsNode(MatrixTreeNode node) {
        String typeCode = node.getAsset().getType().getTypeCode();
        Iterator iterator = conditions.iterator();
        while (iterator.hasNext()) {
            String filterTypeCode = (String) iterator.next();
            if (typeCode.equals(filterTypeCode)) {
                return true;
            }
        }
        return false;
    }
}