2 * Copyright (c) 2014 Cisco Systems, Inc. and others. All rights reserved.
4 * This program and the accompanying materials are made available under the
5 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6 * and is available at http://www.eclipse.org/legal/epl-v10.html
8 package org.opendaylight.mdsal.binding.generator.util;
10 import static com.google.common.base.Preconditions.checkState;
11 import static java.util.Objects.requireNonNull;
13 import com.google.common.base.Preconditions;
15 import java.util.Iterator;
16 import java.util.Optional;
17 import org.eclipse.jdt.annotation.Nullable;
18 import org.opendaylight.yangtools.yang.common.QName;
19 import org.opendaylight.yangtools.yang.common.Revision;
20 import org.opendaylight.yangtools.yang.model.api.AugmentationSchemaNode;
21 import org.opendaylight.yangtools.yang.model.api.ChoiceSchemaNode;
22 import org.opendaylight.yangtools.yang.model.api.DataNodeContainer;
23 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
24 import org.opendaylight.yangtools.yang.model.api.GroupingDefinition;
25 import org.opendaylight.yangtools.yang.model.api.NamespaceRevisionAware;
26 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
27 import org.opendaylight.yangtools.yang.model.api.SchemaPath;
28 import org.opendaylight.yangtools.yang.model.api.TypeDefinition;
29 import org.opendaylight.yangtools.yang.model.api.UnknownSchemaNode;
31 public final class YangSchemaUtils {
32 public static final String AUGMENT_IDENTIFIER = "augment-identifier";
34 private YangSchemaUtils() {
35 throw new UnsupportedOperationException("Helper class. Instantiation is prohibited");
38 public static QName getAugmentationQName(final AugmentationSchemaNode augmentation) {
39 requireNonNull(augmentation, "Augmentation must not be null.");
40 final QName identifier = getAugmentationIdentifier(augmentation);
41 if (identifier != null) {
45 Optional<Revision> revision = null;
46 if (augmentation instanceof NamespaceRevisionAware) {
47 namespace = ((NamespaceRevisionAware) augmentation).getNamespace();
48 revision = ((NamespaceRevisionAware) augmentation).getRevision();
50 if (namespace == null || revision == null) {
51 for (DataSchemaNode child : augmentation.getChildNodes()) {
52 // Derive QName from child nodes
53 if (!child.isAugmenting()) {
54 namespace = child.getQName().getNamespace();
55 revision = child.getQName().getRevision();
60 checkState(namespace != null, "Augmentation namespace must not be null");
61 checkState(revision != null, "Augmentation revision must not be null");
62 // FIXME: Always return a qname with module namespace.
63 return QName.create(namespace, revision, "foo_augment");
66 public static QName getAugmentationIdentifier(final AugmentationSchemaNode augmentation) {
67 for (final UnknownSchemaNode extension : augmentation.getUnknownSchemaNodes()) {
68 if (AUGMENT_IDENTIFIER.equals(extension.getNodeType().getLocalName())) {
69 return extension.getQName();
75 public static @Nullable TypeDefinition<?> findTypeDefinition(final SchemaContext context, final SchemaPath path) {
76 final Iterator<QName> arguments = path.getPathFromRoot().iterator();
77 Preconditions.checkArgument(arguments.hasNext(), "Type Definition path must contain at least one element.");
79 QName currentArg = arguments.next();
80 DataNodeContainer currentNode = context.findModule(currentArg.getModule()).orElse(null);
81 if (currentNode == null) {
84 // Last argument is type definition, so we need to cycle until we hit last argument.
85 while (arguments.hasNext()) {
86 // Nested private type - we need to find container/grouping to which type belongs.
87 final DataSchemaNode child = currentNode.getDataChildByName(currentArg);
88 if (child instanceof DataNodeContainer) {
89 currentNode = (DataNodeContainer) child;
90 } else if (child instanceof ChoiceSchemaNode) {
91 final QName caseQName = arguments.next();
92 Preconditions.checkArgument(arguments.hasNext(), "Path must not refer case only.");
93 currentNode = ((ChoiceSchemaNode) child).getCaseNodeByName(caseQName);
96 for (GroupingDefinition grouping : currentNode.getGroupings()) {
97 if (currentArg.equals(grouping.getQName())) {
98 currentNode = grouping;
103 currentArg = arguments.next();
106 for (TypeDefinition<?> typedef : currentNode.getTypeDefinitions()) {
107 if (typedef.getQName().equals(currentArg)) {