fix bug for capability

yang-central · Aug 16, 2024 · de24540 · de24540
1 parent 1122110
commit de24540
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/yangkit-parser/src/main/java/org/yangcentral/yangkit/parser/YangYinParser.java b/yangkit-parser/src/main/java/org/yangcentral/yangkit/parser/YangYinParser.java
@@ -20,6 +20,7 @@
 import java.util.ArrayList;
 import java.util.List;
 import java.util.Map;
+import java.util.Optional;
 
 public class YangYinParser {
    static {
@@ -211,6 +212,7 @@ public static YangSchemaContext filter(@Nonnull YangSchemaContext schemaContext,
                continue;
             }
          }
+
          unMatchedModules.add(module);
       }
 
@@ -222,6 +224,16 @@ public static YangSchemaContext filter(@Nonnull YangSchemaContext schemaContext,
          schemaContext.addImportOnlyModule(importOnlyModule);
       }
 
+      for (Module module : schemaContext.getModules()){
+         List<Module> dependencies = module.getDependencies();
+         for(Module dependency: dependencies){
+             Optional<Module> moduleOp = schemaContext.getModule(dependency.getModuleId());
+             if(!moduleOp.isPresent()){
+                schemaContext.addImportOnlyModule(dependency);
+                moduleSet.addImportOnlyModule(new YangModuleDescription(dependency.getModuleId()));
+             }
+         }
+      }
 
       YangSchema yangSchema = schemaContext.getYangSchema();
       if (yangSchema == null) {