/* * Copyright (c) 2013 Cisco Systems, Inc. and others. All rights reserved. * * This program and the accompanying materials are made available under the * terms of the Eclipse Public License v1.0 which accompanies this distribution, * and is available at http://www.eclipse.org/legal/eplv10.html */ package org.opendaylight.controller.yang.model.parser.util; import java.net.URI; import java.util.ArrayList; import java.util.Collections; import java.util.Date; import java.util.List; import java.util.Stack; import org.antlr.v4.runtime.tree.ParseTree; import org.opendaylight.controller.antlrv4.code.gen.YangParser; import org.opendaylight.controller.antlrv4.code.gen.YangParser.Argument_stmtContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.Bit_stmtContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.Bits_specificationContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.Config_argContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.Config_stmtContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.Decimal64_specificationContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.Description_stmtContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.Enum_specificationContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.Enum_stmtContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.Fraction_digits_stmtContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.Leafref_specificationContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.Length_stmtContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.Mandatory_argContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.Mandatory_stmtContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.Max_elements_stmtContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.Min_elements_stmtContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.Must_stmtContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.Numerical_restrictionsContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.Ordered_by_argContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.Ordered_by_stmtContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.Path_stmtContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.Pattern_stmtContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.Position_stmtContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.Range_stmtContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.Reference_stmtContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.Require_instance_argContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.Require_instance_stmtContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.Status_argContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.Status_stmtContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.StringContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.String_restrictionsContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.Type_body_stmtsContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.Units_stmtContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.When_stmtContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.Yin_element_argContext; import org.opendaylight.controller.antlrv4.code.gen.YangParser.Yin_element_stmtContext; import org.opendaylight.controller.yang.common.QName; import org.opendaylight.controller.yang.model.api.RevisionAwareXPath; import org.opendaylight.controller.yang.model.api.SchemaPath; import org.opendaylight.controller.yang.model.api.Status; import org.opendaylight.controller.yang.model.api.TypeDefinition; import org.opendaylight.controller.yang.model.api.UnknownSchemaNode; import org.opendaylight.controller.yang.model.api.type.BitsTypeDefinition; import org.opendaylight.controller.yang.model.api.type.BitsTypeDefinition.Bit; import org.opendaylight.controller.yang.model.api.type.EnumTypeDefinition; import org.opendaylight.controller.yang.model.api.type.LengthConstraint; import org.opendaylight.controller.yang.model.api.type.PatternConstraint; import org.opendaylight.controller.yang.model.api.type.RangeConstraint; import org.opendaylight.controller.yang.model.parser.builder.api.SchemaNodeBuilder; import org.opendaylight.controller.yang.model.parser.builder.impl.ConstraintsBuilder; import org.opendaylight.controller.yang.model.util.BaseConstraints; import org.opendaylight.controller.yang.model.util.BinaryType; import org.opendaylight.controller.yang.model.util.BitsType; import org.opendaylight.controller.yang.model.util.EnumerationType; import org.opendaylight.controller.yang.model.util.InstanceIdentifier; import org.opendaylight.controller.yang.model.util.Leafref; import org.opendaylight.controller.yang.model.util.RevisionAwareXPathImpl; import org.opendaylight.controller.yang.model.util.StringType; import org.opendaylight.controller.yang.model.util.UnknownType; import org.opendaylight.controller.yang.model.util.YangTypesConverter; import org.slf4j.Logger; import org.slf4j.LoggerFactory; public final class YangModelBuilderUtil { private static final Logger logger = LoggerFactory .getLogger(YangModelBuilderUtil.class); /** * Parse given tree and get first string value. * * @param treeNode * tree to parse * @return first string value from given tree */ public static String stringFromNode(final ParseTree treeNode) { final String result = ""; for (int i = 0; i < treeNode.getChildCount(); ++i) { if (treeNode.getChild(i) instanceof StringContext) { final StringContext context = (StringContext) treeNode .getChild(i); if (context != null) { return context.getChild(0).getText().replace("\"", ""); } } } return result; } /** * Parse 'description', 'reference' and 'status' statements and fill in * given builder. * * @param ctx * context to parse * @param builder * builder to fill in with parsed statements */ public static void parseSchemaNodeArgs(ParseTree ctx, SchemaNodeBuilder builder) { for (int i = 0; i < ctx.getChildCount(); i++) { ParseTree child = ctx.getChild(i); if (child instanceof Description_stmtContext) { String desc = stringFromNode(child); builder.setDescription(desc); } else if (child instanceof Reference_stmtContext) { String ref = stringFromNode(child); builder.setReference(ref); } else if (child instanceof Status_stmtContext) { Status status = parseStatus((Status_stmtContext) child); builder.setStatus(status); } } } /** * Parse given context and return its value; * * @param ctx * status context * @return value parsed from context */ public static Status parseStatus(Status_stmtContext ctx) { Status result = null; for (int i = 0; i < ctx.getChildCount(); i++) { ParseTree statusArg = ctx.getChild(i); if (statusArg instanceof Status_argContext) { String statusArgStr = stringFromNode(statusArg); if ("current".equals(statusArgStr)) { result = Status.CURRENT; } else if ("deprecated".equals(statusArgStr)) { result = Status.DEPRECATED; } else if ("obsolete".equals(statusArgStr)) { result = Status.OBSOLETE; } else { logger.warn("Invalid 'status' statement: " + statusArgStr); } } } return result; } /** * Parse given tree and returns units statement as string. * * @param ctx * context to parse * @return value of units statement as string or null if there is no units * statement */ public static String parseUnits(ParseTree ctx) { String units = null; for (int i = 0; i < ctx.getChildCount(); i++) { ParseTree child = ctx.getChild(i); if (child instanceof Units_stmtContext) { units = stringFromNode(child); break; } } return units; } /** * Create SchemaPath object from given path list with namespace, revision * and prefix based on given values. * * @param actualPath * current position in model * @param namespace * @param revision * @param prefix * @return SchemaPath object. */ public static SchemaPath createActualSchemaPath(List actualPath, URI namespace, Date revision, String prefix) { final List path = new ArrayList(); QName qname; // start from index 1 - module name omitted for (int i = 1; i < actualPath.size(); i++) { qname = new QName(namespace, revision, prefix, actualPath.get(i)); path.add(qname); } return new SchemaPath(path, true); } /** * Create SchemaPath from given string. * * @param augmentPath * string representation of path * @return SchemaPath object */ public static SchemaPath parseAugmentPath(String augmentPath) { boolean absolute = augmentPath.startsWith("/"); String[] splittedPath = augmentPath.split("/"); List path = new ArrayList(); QName name; for (String pathElement : splittedPath) { if (pathElement.length() > 0) { String[] splittedElement = pathElement.split(":"); if (splittedElement.length == 1) { name = new QName(null, null, null, splittedElement[0]); } else { name = new QName(null, null, splittedElement[0], splittedElement[1]); } path.add(name); } } return new SchemaPath(path, absolute); } /** * Create java.util.List of QName objects from given key definition as * string. * * @param keyDefinition * key definition as string * @param namespace * current namespace * @param revision * current revision * @param prefix * current prefix * @return YANG list key as java.util.List of QName objects */ public static List createListKey(String keyDefinition, URI namespace, Date revision, String prefix) { List key = new ArrayList(); String[] splittedKey = keyDefinition.split(" "); QName qname = null; for (String keyElement : splittedKey) { if (keyElement.length() != 0) { qname = new QName(namespace, revision, prefix, keyElement); key.add(qname); } } return key; } private static List getEnumConstants( Type_body_stmtsContext ctx, List path, URI namespace, Date revision, String prefix) { List enumConstants = new ArrayList(); out: for (int j = 0; j < ctx.getChildCount(); j++) { ParseTree enumSpecChild = ctx.getChild(j); if (enumSpecChild instanceof Enum_specificationContext) { for (int k = 0; k < enumSpecChild.getChildCount(); k++) { ParseTree enumChild = enumSpecChild.getChild(k); if (enumChild instanceof Enum_stmtContext) { enumConstants.add(createEnumPair( (Enum_stmtContext) enumChild, k, path, namespace, revision, prefix)); if (k == enumSpecChild.getChildCount() - 1) { break out; } } } } } return enumConstants; } private static EnumTypeDefinition.EnumPair createEnumPair( Enum_stmtContext ctx, final int value, List path, final URI namespace, final Date revision, final String prefix) { final String name = stringFromNode(ctx); final QName qname = new QName(namespace, revision, prefix, name); String description = null; String reference = null; Status status = null; List enumPairPath = new ArrayList(path); enumPairPath.add(name); for (int i = 0; i < ctx.getChildCount(); i++) { ParseTree child = ctx.getChild(i); if (child instanceof Description_stmtContext) { description = stringFromNode(child); } else if (child instanceof Reference_stmtContext) { reference = stringFromNode(child); } else if (child instanceof Status_stmtContext) { status = parseStatus((Status_stmtContext) child); } } EnumPairImpl result = new EnumPairImpl(); result.qname = qname; result.path = createActualSchemaPath(enumPairPath, namespace, revision, prefix); result.description = description; result.reference = reference; result.status = status; result.name = name; result.value = value; return result; } private static class EnumPairImpl implements EnumTypeDefinition.EnumPair { private QName qname; private SchemaPath path; private String description; private String reference; private Status status; private List extensionSchemaNodes = Collections .emptyList(); private String name; private Integer value; @Override public QName getQName() { return qname; } @Override public SchemaPath getPath() { return path; } @Override public String getDescription() { return description; } @Override public String getReference() { return reference; } @Override public Status getStatus() { return status; } @Override public List getUnknownSchemaNodes() { return extensionSchemaNodes; } @Override public String getName() { return name; } @Override public Integer getValue() { return value; } @Override public int hashCode() { final int prime = 31; int result = 1; result = prime * result + ((qname == null) ? 0 : qname.hashCode()); result = prime * result + ((path == null) ? 0 : path.hashCode()); result = prime * result + ((extensionSchemaNodes == null) ? 0 : extensionSchemaNodes.hashCode()); result = prime * result + ((name == null) ? 0 : name.hashCode()); result = prime * result + ((value == null) ? 0 : value.hashCode()); return result; } @Override public boolean equals(Object obj) { if (this == obj) { return true; } if (obj == null) { return false; } if (getClass() != obj.getClass()) { return false; } EnumPairImpl other = (EnumPairImpl) obj; if (qname == null) { if (other.qname != null) { return false; } } else if (!qname.equals(other.qname)) { return false; } if (path == null) { if (other.path != null) { return false; } } else if (!path.equals(other.path)) { return false; } if (extensionSchemaNodes == null) { if (other.extensionSchemaNodes != null) { return false; } } else if (!extensionSchemaNodes.equals(other.extensionSchemaNodes)) { return false; } if (name == null) { if (other.name != null) { return false; } } else if (!name.equals(other.name)) { return false; } if (value == null) { if (other.value != null) { return false; } } else if (!value.equals(other.value)) { return false; } return true; } @Override public String toString() { return EnumTypeDefinition.EnumPair.class.getSimpleName() + "[name=" + name + ", value=" + value + "]"; } }; private static List getRangeConstraints( Type_body_stmtsContext ctx) { final List rangeConstraints = new ArrayList(); for (int j = 0; j < ctx.getChildCount(); j++) { ParseTree numRestrChild = ctx.getChild(j); if (numRestrChild instanceof Numerical_restrictionsContext) { for (int k = 0; k < numRestrChild.getChildCount(); k++) { ParseTree rangeChild = numRestrChild.getChild(k); if (rangeChild instanceof Range_stmtContext) { rangeConstraints .addAll(parseRangeConstraints((Range_stmtContext) rangeChild)); break; } } } } return rangeConstraints; } private static List parseRangeConstraints( Range_stmtContext ctx) { List rangeConstraints = new ArrayList(); String description = null; String reference = null; for (int i = 0; i < ctx.getChildCount(); i++) { ParseTree child = ctx.getChild(i); if (child instanceof Description_stmtContext) { description = stringFromNode(child); } else if (child instanceof Reference_stmtContext) { reference = stringFromNode(child); } } String rangeStr = stringFromNode(ctx); String trimmed = rangeStr.replace(" ", ""); String[] splittedRange = trimmed.split("\\|"); for (String rangeDef : splittedRange) { String[] splittedRangeDef = rangeDef.split("\\.\\."); Number min; Number max; if (splittedRangeDef.length == 1) { min = max = parseRangeValue(splittedRangeDef[0]); } else { min = parseRangeValue(splittedRangeDef[0]); max = parseRangeValue(splittedRangeDef[1]); } RangeConstraint range = BaseConstraints.rangeConstraint(min, max, description, reference); rangeConstraints.add(range); } return rangeConstraints; } private static List getLengthConstraints( Type_body_stmtsContext ctx) { List lengthConstraints = new ArrayList(); for (int j = 0; j < ctx.getChildCount(); j++) { ParseTree stringRestrChild = ctx.getChild(j); if (stringRestrChild instanceof String_restrictionsContext) { for (int k = 0; k < stringRestrChild.getChildCount(); k++) { ParseTree lengthChild = stringRestrChild.getChild(k); if (lengthChild instanceof Length_stmtContext) { lengthConstraints .addAll(parseLengthConstraints((Length_stmtContext) lengthChild)); } } } } return lengthConstraints; } private static List parseLengthConstraints( Length_stmtContext ctx) { List lengthConstraints = new ArrayList(); String description = null; String reference = null; for (int i = 0; i < ctx.getChildCount(); i++) { ParseTree child = ctx.getChild(i); if (child instanceof Description_stmtContext) { description = stringFromNode(child); } else if (child instanceof Reference_stmtContext) { reference = stringFromNode(child); } } String lengthStr = stringFromNode(ctx); String trimmed = lengthStr.replace(" ", ""); String[] splittedRange = trimmed.split("\\|"); for (String rangeDef : splittedRange) { String[] splittedRangeDef = rangeDef.split("\\.\\."); Number min; Number max; if (splittedRangeDef.length == 1) { min = max = parseRangeValue(splittedRangeDef[0]); } else { min = parseRangeValue(splittedRangeDef[0]); max = parseRangeValue(splittedRangeDef[1]); } LengthConstraint range = BaseConstraints.lengthConstraint(min, max, description, reference); lengthConstraints.add(range); } return lengthConstraints; } private static Number parseRangeValue(String value) { Number result = null; if ("min".equals(value) || "max".equals(value)) { result = new UnknownBoundaryNumber(value); } else { try { result = Long.valueOf(value); } catch (NumberFormatException e) { throw new YangParseException("Unable to parse range value '" + value + "'.", e); } } return result; } private static List getPatternConstraint( Type_body_stmtsContext ctx) { List patterns = new ArrayList(); out: for (int j = 0; j < ctx.getChildCount(); j++) { ParseTree stringRestrChild = ctx.getChild(j); if (stringRestrChild instanceof String_restrictionsContext) { for (int k = 0; k < stringRestrChild.getChildCount(); k++) { ParseTree lengthChild = stringRestrChild.getChild(k); if (lengthChild instanceof Pattern_stmtContext) { patterns.add(parsePatternConstraint((Pattern_stmtContext) lengthChild)); if (k == lengthChild.getChildCount() - 1) { break out; } } } } } return patterns; } /** * Internal helper method. * * @param ctx * pattern context * @return PatternConstraint object */ private static PatternConstraint parsePatternConstraint( Pattern_stmtContext ctx) { String description = null; String reference = null; for (int i = 0; i < ctx.getChildCount(); i++) { ParseTree child = ctx.getChild(i); if (child instanceof Description_stmtContext) { description = stringFromNode(child); } else if (child instanceof Reference_stmtContext) { reference = stringFromNode(child); } } String pattern = patternStringFromNode(ctx); return BaseConstraints.patternConstraint(pattern, description, reference); } /** * Parse given context and return pattern value. * * @param ctx * context to parse * @return pattern value as String */ public static String patternStringFromNode(final Pattern_stmtContext ctx) { StringBuilder result = new StringBuilder(); for (int i = 0; i < ctx.getChildCount(); ++i) { ParseTree child = ctx.getChild(i); if (child instanceof StringContext) { for (int j = 0; j < child.getChildCount(); j++) { if (j % 2 == 0) { String patternToken = child.getChild(j).getText(); result.append(patternToken.substring(1, patternToken.length() - 1)); } } } } return result.toString(); } private static Integer getFractionDigits(Type_body_stmtsContext ctx) { Integer result = null; for (int j = 0; j < ctx.getChildCount(); j++) { ParseTree dec64specChild = ctx.getChild(j); if (dec64specChild instanceof Decimal64_specificationContext) { result = parseFractionDigits((Decimal64_specificationContext) dec64specChild); } } return result; } private static Integer parseFractionDigits( Decimal64_specificationContext ctx) { Integer result = null; for (int k = 0; k < ctx.getChildCount(); k++) { ParseTree fdChild = ctx.getChild(k); if (fdChild instanceof Fraction_digits_stmtContext) { String value = stringFromNode(fdChild); try { result = Integer.valueOf(value); } catch (NumberFormatException e) { throw new YangParseException( "Unable to parse fraction digits value '" + value + "'.", e); } } } return result; } private static List getBits( Type_body_stmtsContext ctx, List actualPath, URI namespace, Date revision, String prefix) { List bits = new ArrayList(); for (int j = 0; j < ctx.getChildCount(); j++) { ParseTree bitsSpecChild = ctx.getChild(j); if (bitsSpecChild instanceof Bits_specificationContext) { long highestPosition = -1; for (int k = 0; k < bitsSpecChild.getChildCount(); k++) { ParseTree bitChild = bitsSpecChild.getChild(k); if (bitChild instanceof Bit_stmtContext) { Bit bit = parseBit((Bit_stmtContext) bitChild, highestPosition, actualPath, namespace, revision, prefix); if (bit.getPosition() > highestPosition) { highestPosition = bit.getPosition(); } bits.add(bit); } } } } return bits; } private static BitsTypeDefinition.Bit parseBit(final Bit_stmtContext ctx, long highestPosition, List actualPath, final URI namespace, final Date revision, final String prefix) { String name = stringFromNode(ctx); final QName qname = new QName(namespace, revision, prefix, name); Long position = null; String description = null; String reference = null; Status status = Status.CURRENT; Stack bitPath = new Stack(); bitPath.addAll(actualPath); bitPath.add(name); SchemaPath schemaPath = createActualSchemaPath(bitPath, namespace, revision, prefix); for (int i = 0; i < ctx.getChildCount(); i++) { ParseTree child = ctx.getChild(i); if (child instanceof Position_stmtContext) { String positionStr = stringFromNode(child); position = Long.valueOf(positionStr); } else if (child instanceof Description_stmtContext) { description = stringFromNode(child); } else if (child instanceof Reference_stmtContext) { reference = stringFromNode(child); } else if (child instanceof Status_stmtContext) { status = parseStatus((Status_stmtContext) child); } } if (position == null) { position = highestPosition + 1; } if (position < 0 || position > 4294967295L) { throw new YangParseException( "Error on bit '" + name + "': the position value MUST be in the range 0 to 4294967295"); } final List extensionNodes = Collections.emptyList(); return createBit(qname, schemaPath, description, reference, status, extensionNodes, position); } private static BitsTypeDefinition.Bit createBit(final QName qname, final SchemaPath schemaPath, final String description, final String reference, final Status status, final List unknownNodes, final Long position) { return new BitsTypeDefinition.Bit() { @Override public QName getQName() { return qname; } @Override public SchemaPath getPath() { return schemaPath; } @Override public String getDescription() { return description; } @Override public String getReference() { return reference; } @Override public Status getStatus() { return status; } @Override public List getUnknownSchemaNodes() { return unknownNodes; } @Override public Long getPosition() { return position; } @Override public String getName() { return qname.getLocalName(); } @Override public int hashCode() { final int prime = 31; int result = 1; result = prime * result + ((qname == null) ? 0 : qname.hashCode()); result = prime * result + ((schemaPath == null) ? 0 : schemaPath.hashCode()); result = prime * result + ((position == null) ? 0 : position.hashCode()); result = prime * result + ((unknownNodes == null) ? 0 : unknownNodes.hashCode()); return result; } @Override public boolean equals(Object obj) { if (this == obj) { return true; } if (obj == null) { return false; } if (getClass() != obj.getClass()) { return false; } Bit other = (Bit) obj; if (qname == null) { if (other.getQName() != null) { return false; } } else if (!qname.equals(other.getQName())) { return false; } if (schemaPath == null) { if (other.getPath() != null) { return false; } } else if (!schemaPath.equals(other.getPath())) { return false; } if (unknownNodes == null) { if (other.getUnknownSchemaNodes() != null) { return false; } } else if (!unknownNodes.equals(other.getUnknownSchemaNodes())) { return false; } if (position == null) { if (other.getPosition() != null) { return false; } } else if (!position.equals(other.getPosition())) { return false; } return true; } @Override public String toString() { return Bit.class.getSimpleName() + "[name=" + qname.getLocalName() + ", position=" + position + "]"; } }; } /** * Parse orderedby statement. * * @param childNode * Ordered_by_stmtContext * @return true, if orderedby contains value 'user' or false otherwise */ public static boolean parseUserOrdered(Ordered_by_stmtContext childNode) { boolean result = false; for (int j = 0; j < childNode.getChildCount(); j++) { ParseTree orderArg = childNode.getChild(j); if (orderArg instanceof Ordered_by_argContext) { String orderStr = stringFromNode(orderArg); if ("system".equals(orderStr)) { result = false; } else if ("user".equals(orderStr)) { result = true; } else { logger.warn("Invalid 'orderedby' statement."); } } } return result; } /** * Parse given config context and return true if it contains string 'true', * false otherwise. * * @param ctx * config context to parse. * @return true if given context contains string 'true', false otherwise */ public static boolean parseConfig(final Config_stmtContext ctx) { boolean result = false; if (ctx != null) { for (int i = 0; i < ctx.getChildCount(); ++i) { final ParseTree configContext = ctx.getChild(i); if (configContext instanceof Config_argContext) { final String value = stringFromNode(configContext); if ("true".equals(value)) { result = true; break; } } } } return result; } /** * Parse given type body and creates UnknownType definition. * * @param typedefQName * qname of current type * @param ctx * type body * @return UnknownType object with constraints from parsed type body */ public static TypeDefinition parseUnknownTypeBody(QName typedefQName, Type_body_stmtsContext ctx) { UnknownType.Builder unknownType = new UnknownType.Builder(typedefQName); if (ctx != null) { List rangeStatements = getRangeConstraints(ctx); List lengthStatements = getLengthConstraints(ctx); List patternStatements = getPatternConstraint(ctx); Integer fractionDigits = getFractionDigits(ctx); unknownType.rangeStatements(rangeStatements); unknownType.lengthStatements(lengthStatements); unknownType.patterns(patternStatements); unknownType.fractionDigits(fractionDigits); } return unknownType.build(); } /** * Create TypeDefinition object based on given type name and type body. * * @param typeName * name of type * @param typeBody * type body * @param actualPath * current path in schema * @param namespace * current namespace * @param revision * current revision * @param prefix * current prefix * @return TypeDefinition object based on parsed values. */ public static TypeDefinition parseTypeBody(String typeName, Type_body_stmtsContext typeBody, List actualPath, URI namespace, Date revision, String prefix) { TypeDefinition type = null; List rangeStatements = getRangeConstraints(typeBody); Integer fractionDigits = getFractionDigits(typeBody); List lengthStatements = getLengthConstraints(typeBody); List patternStatements = getPatternConstraint(typeBody); List enumConstants = getEnumConstants( typeBody, actualPath, namespace, revision, prefix); if ("decimal64".equals(typeName)) { type = YangTypesConverter.javaTypeForBaseYangDecimal64Type( rangeStatements, fractionDigits); } else if (typeName.startsWith("int")) { type = YangTypesConverter.javaTypeForBaseYangSignedIntegerType( typeName, rangeStatements); } else if (typeName.startsWith("uint")) { type = YangTypesConverter.javaTypeForBaseYangUnsignedIntegerType( typeName, rangeStatements); } else if ("enumeration".equals(typeName)) { type = new EnumerationType(enumConstants); } else if ("string".equals(typeName)) { type = new StringType(lengthStatements, patternStatements); } else if ("bits".equals(typeName)) { type = new BitsType(getBits(typeBody, actualPath, namespace, revision, prefix)); } else if ("leafref".equals(typeName)) { final String path = parseLeafrefPath(typeBody); final boolean absolute = path.startsWith("/"); RevisionAwareXPath xpath = new RevisionAwareXPathImpl(path, absolute); type = new Leafref(actualPath, namespace, revision, xpath); } else if ("binary".equals(typeName)) { List bytes = Collections.emptyList(); type = new BinaryType(bytes, lengthStatements, null); } else if ("instance-identifier".equals(typeName)) { boolean requireInstance = isRequireInstance(typeBody); type = new InstanceIdentifier(null, requireInstance); } return type; } private static boolean isRequireInstance(Type_body_stmtsContext ctx) { for (int i = 0; i < ctx.getChildCount(); i++) { ParseTree child = ctx.getChild(i); if (child instanceof Require_instance_stmtContext) { for (int j = 0; j < child.getChildCount(); j++) { ParseTree reqArg = child.getChild(j); if (reqArg instanceof Require_instance_argContext) { return Boolean.valueOf(stringFromNode(reqArg)); } } } } return false; } private static String parseLeafrefPath(Type_body_stmtsContext ctx) { for (int i = 0; i < ctx.getChildCount(); i++) { ParseTree child = ctx.getChild(i); if (child instanceof Leafref_specificationContext) { for (int j = 0; j < child.getChildCount(); j++) { ParseTree leafRefSpec = child.getChild(j); if (leafRefSpec instanceof Path_stmtContext) { return stringFromNode(leafRefSpec); } } } } return null; } /** * Internal helper method for parsing Must_stmtContext. * * @param ctx * Must_stmtContext * @return an array of strings with following fields: [0] must text [1] * description [2] reference */ public static String[] parseMust(YangParser.Must_stmtContext ctx) { String[] params = new String[3]; StringBuilder mustText = new StringBuilder(); String description = null; String reference = null; for (int i = 0; i < ctx.getChildCount(); ++i) { ParseTree child = ctx.getChild(i); if (child instanceof StringContext) { final StringContext context = (StringContext) child; for (int j = 0; j < context.getChildCount(); j++) { String mustPart = context.getChild(j).getText(); if (j == 0) { mustText.append(mustPart.substring(0, mustPart.length() - 1)); continue; } if (j % 2 == 0) { mustText.append(mustPart.substring(1)); } } } else if (child instanceof Description_stmtContext) { description = stringFromNode(child); } else if (child instanceof Reference_stmtContext) { reference = stringFromNode(child); } } params[0] = mustText.toString(); params[1] = description; params[2] = reference; return params; } /** * Parse given tree and set constraints to given builder. * * @param ctx * Context to search. * @param constraintsBuilder * ConstraintsBuilder to fill. */ public static void parseConstraints(ParseTree ctx, ConstraintsBuilder constraintsBuilder) { for (int i = 0; i < ctx.getChildCount(); ++i) { final ParseTree childNode = ctx.getChild(i); if (childNode instanceof Max_elements_stmtContext) { Integer max = Integer.valueOf(stringFromNode(childNode)); constraintsBuilder.setMinElements(max); } else if (childNode instanceof Min_elements_stmtContext) { Integer min = Integer.valueOf(stringFromNode(childNode)); constraintsBuilder.setMinElements(min); } else if (childNode instanceof Must_stmtContext) { String[] mustParams = parseMust((Must_stmtContext) childNode); constraintsBuilder.addMustDefinition(mustParams[0], mustParams[1], mustParams[2]); } else if (childNode instanceof Mandatory_stmtContext) { for (int j = 0; j < childNode.getChildCount(); j++) { ParseTree mandatoryTree = ctx.getChild(j); if (mandatoryTree instanceof Mandatory_argContext) { Boolean mandatory = Boolean .valueOf(stringFromNode(mandatoryTree)); constraintsBuilder.setMandatory(mandatory); } } } else if (childNode instanceof When_stmtContext) { constraintsBuilder.addWhenCondition(stringFromNode(childNode)); } } } /** * Parse given context and return yin value. * * @param ctx * context to parse * @return true if value is 'true', false otherwise */ public static boolean parseYinValue(Argument_stmtContext ctx) { boolean yinValue = false; outer: for (int j = 0; j < ctx.getChildCount(); j++) { ParseTree yin = ctx.getChild(j); if (yin instanceof Yin_element_stmtContext) { for (int k = 0; k < yin.getChildCount(); k++) { ParseTree yinArg = yin.getChild(k); if (yinArg instanceof Yin_element_argContext) { String yinString = stringFromNode(yinArg); if ("true".equals(yinString)) { yinValue = true; break outer; } } } } } return yinValue; } }