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

final class RefParser extends AbstractParser {

    private static final String GRAMMAR = "%s ";

    private final static int IDENTIFIER_INDEX = 1;

    ModelItem parse(DslContext context, Tokens tokens) {
        // !ref 

        if (tokens.hasMoreThan(IDENTIFIER_INDEX)) {
            throw new RuntimeException("Too many tokens, expected: " + String.format(GRAMMAR, tokens.get(0)));
        }

        if (!tokens.includes(IDENTIFIER_INDEX)) {
            throw new RuntimeException("Expected: " + String.format(GRAMMAR, tokens.get(0)));
        }

        String s = tokens.get(IDENTIFIER_INDEX);

        ModelItem modelItem;

        if (s.contains("://")) {
            modelItem = context.getWorkspace().getModel().getElementWithCanonicalName(s);
        } else {
            modelItem = context.getElement(s);

            if (modelItem == null) {
                modelItem = context.getRelationship(s);
            }
        }

        if (modelItem == null) {
            throw new RuntimeException("An element/relationship identified by \"" + s + "\" could not be found");
        }

        if (context instanceof GroupableDslContext && modelItem instanceof StaticStructureElement) {
            GroupableDslContext groupableDslContext = (GroupableDslContext)context;
            StaticStructureElement staticStructureElement = (StaticStructureElement)modelItem;
            if (groupableDslContext.hasGroup()) {
                staticStructureElement.setGroup(groupableDslContext.getGroup().getName());
            }
        }

        return modelItem;
    }

}