com.structurizr.dsl.CustomViewExpressionParser Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of structurizr-dsl Show documentation

Structurizr for Java

The newest version!

package com.structurizr.dsl;

import com.structurizr.model.CustomElement;
import com.structurizr.model.Element;

import java.util.LinkedHashSet;
import java.util.Set;

import static com.structurizr.dsl.StructurizrDslExpressions.ELEMENT_TYPE_EQUALS_EXPRESSION;

final class CustomViewExpressionParser extends ExpressionParser {

    @Override
    protected Set evaluateElementTypeExpression(String expr, DslContext context) {
        Set elements = new LinkedHashSet<>();

        String type = expr.substring(ELEMENT_TYPE_EQUALS_EXPRESSION.length());
        switch (type.toLowerCase()) {
            case "custom":
                context.getWorkspace().getModel().getElements().stream().filter(e -> e instanceof CustomElement).forEach(elements::add);
                break;
            default:
                throw new RuntimeException("The element type of \"" + type + "\" is not valid for this view");
        }

        return elements;
    }

    protected Set findAfferentCouplings(Element element) {
        Set elements = new LinkedHashSet<>();

        elements.addAll(findAfferentCouplings(element, CustomElement.class));

        return elements;
    }

    protected Set findEfferentCouplings(Element element) {
        Set elements = new LinkedHashSet<>();

        elements.addAll(findEfferentCouplings(element, CustomElement.class));

        return elements;
    }

}