- public void onStatementDefinitionDeclared(final StmtContext.Mutable<QName, ExtensionStatement, EffectiveStatement<QName, ExtensionStatement>> stmt) throws SourceException {
- stmt.addContext(ExtensionNamespace.class, stmt.getStatementArgument(), stmt);
+ public void onStatementDefinitionDeclared(
+ final Mutable<QName, ExtensionStatement, EffectiveStatement<QName, ExtensionStatement>> stmt) {
+ super.onStatementDefinitionDeclared(stmt);
+
+ QName stmtName = stmt.getStatementArgument();
+ if (SupportedExtensionsMapping.OPENCONFIG_VERSION.getStatementName().isEqualWithoutRevision(stmtName)) {
+ stmtName = stmtName.withoutRevision();
+ }
+
+ stmt.addContext(ExtensionNamespace.class, stmtName, stmt);
+
+ final StmtContext<QName, ?, ?> argument = StmtContextUtils.findFirstDeclaredSubstatement(stmt,
+ ArgumentStatement.class);
+ final StmtContext<Boolean, ?, ?> yinElement = StmtContextUtils.findFirstDeclaredSubstatement(stmt,
+ YinElementStatement.class);
+
+ stmt.addToNs(StatementDefinitionNamespace.class, stmt.getStatementArgument(),
+ new ModelDefinedStatementSupport(new ModelDefinedStatementDefinition(stmt.getStatementArgument(),
+ argument != null ? argument.getStatementArgument() : null,
+ yinElement != null ? yinElement.getStatementArgument() : false)));