c++ FEATURE allow constructing Submodule from Module (#1085)

CESNET · May 13, 2020 · 9267156 · 9267156
1 parent 43b04d1
commit 9267156
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/swig/cpp/src/Tree_Schema.cpp b/swig/cpp/src/Tree_Schema.cpp
@@ -39,6 +39,14 @@ Submodule::Submodule(struct lys_submodule *submodule, S_Deleter deleter):
     submodule(submodule),
     deleter(deleter)
 {};
+Submodule::Submodule(S_Module module):
+    submodule((lys_submodule*)module->module),
+    deleter(module->deleter)
+{
+    if (module->type() != 1) {
+        throw std::invalid_argument("Attempted to cast a YANG module into a YANG submodule");
+    }
+}
 std::vector<S_Schema_Node> Module::data_instantiables(int options) {
     std::vector<S_Schema_Node> s_vector;
     struct lys_node *iter = NULL;

diff --git a/swig/cpp/src/Tree_Schema.hpp b/swig/cpp/src/Tree_Schema.hpp
@@ -111,6 +111,7 @@ class Module
 
     friend Context;
     friend Data_Node;
+    friend Submodule;
 
 private:
     struct lys_module *module;
@@ -126,6 +127,8 @@ class Submodule
 public:
     /** wrapper for struct [lys_submodule](@ref lys_submodule), for internal use only */
     Submodule(struct lys_submodule *submodule, S_Deleter deleter);
+    /** creates a Submodule from `module` if it is a submodule */
+    Submodule(S_Module module);
     ~Submodule();
     /** get ctx variable from [lys_submodule](@ref lys_submodule)*/
     S_Context ctx() LY_NEW(submodule, ctx, Context);