com.structurizr.dsl.FindElementsParser 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.Element;
import com.structurizr.model.ModelItem;

import java.util.Set;
import java.util.stream.Collectors;

final class FindElementsParser extends AbstractParser {

    private static final String GRAMMAR = "!elements ";

    private final static int EXPRESSION_INDEX = 1;

    Set parse(DslContext context, Tokens tokens) {
        // !elements 

        if (tokens.hasMoreThan(EXPRESSION_INDEX)) {
            throw new RuntimeException("Too many tokens, expected: " + GRAMMAR);
        }

        String expression = tokens.get(1);
        Set modelItems = new ExpressionParser().parseExpression(expression, context);

        Set elements = modelItems.stream().filter(mi -> mi instanceof Element).map(mi -> (Element)mi).collect(Collectors.toSet());

        if (elements.isEmpty()) {
            throw new RuntimeException("No elements found for expression \"" + expression + "\"");
        }

        return elements;
    }

}