org.yamcs.templating.VarStatement Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of yamcs-core Show documentation

There is a newer version: 5.10.9

package org.yamcs.templating;

import java.util.List;
import java.util.Map;

public class VarStatement implements Statement {

    private String identifier;
    private List filterIdentifiers;

    public VarStatement(String identifier, List filterIdentifiers) {
        this.identifier = identifier;
        this.filterIdentifiers = filterIdentifiers;
    }

    @Override
    public void render(StringBuilder buf, Map vars, Map filters) {
        var value = getValue(identifier, vars);
        for (var filterIdentifier : filterIdentifiers) {
            var filter = filters.get(filterIdentifier);
            if (filter == null) {
                throw new IllegalArgumentException(String.format("Unknown filter '%s'", filterIdentifier));
            }
            value = filter.applyFilter(value);
        }
        if (value == null) {
            throw new IllegalArgumentException(String.format("Variable '%s' is not set", identifier));
        }
        buf.append(value);
    }

    private Object getValue(String name, Map args) {
        return getValue(name, args, "");
    }

    @SuppressWarnings("unchecked")
    private Object getValue(String name, Map args, String nameContext) {
        int dotIndex = name.indexOf('.');
        if (dotIndex == -1) {
            return args.get(name);
        } else {
            String parentName = name.substring(0, dotIndex);
            Object parentValue = args.get(parentName);
            if (parentValue == null) {
                throw new IllegalArgumentException(String.format(
                        "Variable '%s%s' is not set", nameContext, parentName));
            }
            if (!(parentValue instanceof Map)) {
                throw new IllegalArgumentException(String.format(
                        "Variable '%s%s' is not a map", nameContext, parentName));
            }
            Map parentArgs = (Map) parentValue;
            return getValue(name.substring(dotIndex + 1), parentArgs);
        }
    }
}