BUG-6497: Do not lose augmentation statement order
[yangtools.git] / yang / yang-parser-impl / src / main / java / org / opendaylight / yangtools / yang / parser / stmt / rfc6020 / effective / UnknownEffectiveStatementImpl.java
index 2027a1b9cc714a1150ce1010ba5939d395398457..2ee3f8283dc7c087ddd69b8f0fc3ffca30199039 100644 (file)
 
 package org.opendaylight.yangtools.yang.parser.stmt.rfc6020.effective;
 
-import java.util.ArrayList;
-import java.util.List;
 import java.util.Objects;
 import org.opendaylight.yangtools.yang.common.QName;
-import org.opendaylight.yangtools.yang.model.api.ExtensionDefinition;
 import org.opendaylight.yangtools.yang.model.api.SchemaPath;
-import org.opendaylight.yangtools.yang.model.api.Status;
 import org.opendaylight.yangtools.yang.model.api.UnknownSchemaNode;
-import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
-import org.opendaylight.yangtools.yang.model.api.stmt.ExtensionStatement;
 import org.opendaylight.yangtools.yang.model.api.stmt.UnknownStatement;
-import org.opendaylight.yangtools.yang.parser.spi.ExtensionNamespace;
 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
-import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext.TypeOfCopy;
 import org.opendaylight.yangtools.yang.parser.stmt.rfc6020.Utils;
 
-public class UnknownEffectiveStatementImpl extends EffectiveStatementBase<QName, UnknownStatement<QName>> implements
-        UnknownSchemaNode {
+public final class UnknownEffectiveStatementImpl extends UnknownEffectiveStatementBase<String> {
 
-    private boolean augmenting;
-    private boolean addedByUses;
-    private UnknownSchemaNode original;
-
-    private final QName qName;
+    private final QName maybeQNameArgument;
     private final SchemaPath path;
-    private ExtensionDefinition extension;
-    private String description;
-    private String reference;
-    private final Status status = Status.CURRENT;
-    private final List<UnknownSchemaNode> unknownNodes = new ArrayList<>();
-    private QName nodeType;
-    private final String nodeParameter;
 
-    public UnknownEffectiveStatementImpl(final StmtContext<QName, UnknownStatement<QName>, ?> ctx) {
+    public UnknownEffectiveStatementImpl(final StmtContext<String, UnknownStatement<String>, ?> ctx) {
         super(ctx);
 
-        final StmtContext<?, ExtensionStatement, EffectiveStatement<QName, ExtensionStatement>> extensionInit = ctx
-                .getAllFromNamespace(ExtensionNamespace.class).get(ctx.getPublicDefinition().getStatementName());
-
-        if (extensionInit == null) {
-            extension = null;
-            nodeType = ctx.getPublicDefinition().getArgumentName();
+        // FIXME: Remove following section after fixing 4380
+        final UnknownSchemaNode original = ctx.getOriginalCtx() == null ? null : (UnknownSchemaNode) ctx
+                .getOriginalCtx().buildEffective();
+        if (original != null) {
+            this.maybeQNameArgument = original.getQName();
         } else {
-            extension = (ExtensionEffectiveStatementImpl) extensionInit.buildEffective();
-            nodeType = extension.getQName();
-        }
-
-        qName = ctx.getStatementArgument();
-        path = Utils.getSchemaPath(ctx);
-
-        nodeParameter = (ctx.rawStatementArgument() == null) ? "" : ctx.rawStatementArgument();
-
-        // TODO init other fields (see Bug1412Test)
-
-        for (final EffectiveStatement<?, ?> effectiveStatement : effectiveSubstatements()) {
-            if (effectiveStatement instanceof DescriptionEffectiveStatementImpl) {
-                description = ((DescriptionEffectiveStatementImpl) effectiveStatement).argument();
-            }
-            if (effectiveStatement instanceof ReferenceEffectiveStatementImpl) {
-                reference = ((ReferenceEffectiveStatementImpl) effectiveStatement).argument();
-            }
-            if (effectiveStatement instanceof UnknownEffectiveStatementImpl) {
-                unknownNodes.add((UnknownEffectiveStatementImpl) effectiveStatement);
+            QName maybeQNameArgumentInit = null;
+            try {
+                maybeQNameArgumentInit = Utils.qNameFromArgument(ctx, argument());
+            } catch (IllegalArgumentException e) {
+                maybeQNameArgumentInit = getNodeType();
             }
+            this.maybeQNameArgument = maybeQNameArgumentInit;
         }
-
-        initCopyType(ctx);
-    }
-
-    private void initCopyType(final StmtContext<QName, UnknownStatement<QName>, ?> ctx) {
-
-        List<TypeOfCopy> copyTypesFromOriginal = ctx.getCopyHistory();
-
-        if (copyTypesFromOriginal.contains(TypeOfCopy.ADDED_BY_AUGMENTATION)) {
-            augmenting = true;
-        }
-        if (copyTypesFromOriginal.contains(TypeOfCopy.ADDED_BY_USES)) {
-            addedByUses = true;
-        }
-        if (copyTypesFromOriginal.contains(TypeOfCopy.ADDED_BY_USES_AUGMENTATION)) {
-            addedByUses = augmenting = true;
-        }
-
-        if (ctx.getOriginalCtx() != null) {
-            original = (UnknownSchemaNode) ctx.getOriginalCtx().buildEffective();
-        }
-    }
-
-    @Override
-    public QName getNodeType() {
-        return nodeType;
-    }
-
-    @Override
-    public String getNodeParameter() {
-        return nodeParameter;
-    }
-
-    @Override
-    public boolean isAddedByUses() {
-        return addedByUses;
-    }
-
-    @Override
-    public ExtensionDefinition getExtensionDefinition() {
-        return extension;
+        path = ctx.getParentContext().getSchemaPath().get().createChild(maybeQNameArgument);
     }
 
     @Override
     public QName getQName() {
-        return qName;
+        return maybeQNameArgument;
     }
 
     @Override
@@ -126,34 +51,14 @@ public class UnknownEffectiveStatementImpl extends EffectiveStatementBase<QName,
         return path;
     }
 
-    @Override
-    public List<UnknownSchemaNode> getUnknownSchemaNodes() {
-        return unknownNodes;
-    }
-
-    @Override
-    public String getDescription() {
-        return description;
-    }
-
-    @Override
-    public String getReference() {
-        return reference;
-    }
-
-    @Override
-    public Status getStatus() {
-        return status;
-    }
-
     @Override
     public int hashCode() {
         final int prime = 31;
         int result = 1;
-        result = prime * result + Objects.hashCode(qName);
+        result = prime * result + Objects.hashCode(maybeQNameArgument);
         result = prime * result + Objects.hashCode(path);
-        result = prime * result + Objects.hashCode(nodeType);
-        result = prime * result + Objects.hashCode(nodeParameter);
+        result = prime * result + Objects.hashCode(getNodeType());
+        result = prime * result + Objects.hashCode(getNodeParameter());
         return result;
     }
 
@@ -169,29 +74,18 @@ public class UnknownEffectiveStatementImpl extends EffectiveStatementBase<QName,
             return false;
         }
         UnknownEffectiveStatementImpl other = (UnknownEffectiveStatementImpl) obj;
-        if (!Objects.equals(qName, other.qName)) {
+        if (!Objects.equals(maybeQNameArgument, other.maybeQNameArgument)) {
             return false;
         }
         if (!Objects.equals(path, other.path)) {
             return false;
         }
-        if (!Objects.equals(nodeType, other.nodeType)) {
+        if (!Objects.equals(getNodeType(), other.getNodeType())) {
             return false;
         }
-        if (!Objects.equals(nodeParameter, other.nodeParameter)) {
+        if (!Objects.equals(getNodeParameter(), other.getNodeParameter())) {
             return false;
         }
         return true;
     }
-
-    @Override
-    public String toString() {
-        StringBuilder sb = new StringBuilder();
-        sb.append(nodeType.getNamespace());
-        sb.append(":");
-        sb.append(nodeType.getLocalName());
-        sb.append(" ");
-        sb.append(nodeParameter);
-        return sb.toString();
-    }
 }