Added support for iterating over diagram elements from inja templates (fixes #54)

src/common/generators/plantuml/generator.h

@@ -69,6 +69,8 @@ public:
     template <typename E>
     void generate_link(std::ostream &ostr, const E &e) const;
 
+    void update_context() const;
+
 protected:
     const inja::json &context() const;
 
@@ -85,7 +87,7 @@ protected:
     ConfigType &m_config;
     DiagramType &m_model;
     mutable std::set<std::string> m_generated_aliases;
-    inja::json m_context;
+    mutable inja::json m_context;
     mutable inja::Environment m_env;
 };
 
@@ -372,9 +374,11 @@ template <typename C, typename D> void generator<C, D>::init_context()
         m_context["git"]["commit"] = m_config.git().commit;
         m_context["git"]["toplevel"] = m_config.git().toplevel;
     }
+}
 
-    m_context["diagram"]["name"] = m_config.name;
-    m_context["diagram"]["type"] = to_string(m_config.type());
+template <typename C, typename D> void generator<C, D>::update_context() const
+{
+    m_context["diagram"] = m_model.context();
 }
 
 template <typename C, typename D> void generator<C, D>::init_env()

src/common/model/diagram.h

@@ -74,6 +74,8 @@ public:
     virtual bool should_include(
         const namespace_ &ns, const std::string &name) const;
 
+    virtual inja::json context() const = 0;
+
 private:
     std::string name_;
     std::unique_ptr<diagram_filter> filter_;

src/common/model/diagram_element.h

@@ -48,6 +48,8 @@ public:
 
     std::string name() const { return name_; }
 
+    virtual std::string type_name() const { return "__undefined__"; };
+
     virtual std::string full_name(bool /*relative*/) const { return name(); }
 
     std::vector<relationship> &relationships();

src/common/model/element.cc

@@ -40,6 +40,7 @@ inja::json element::context() const
 {
     inja::json ctx;
     ctx["name"] = name();
+    ctx["type"] = type_name();
     ctx["alias"] = alias();
     ctx["full_name"] = full_name(false);
     ctx["namespace"] = get_namespace().to_string();

src/common/model/nested_trait.h

@@ -71,7 +71,7 @@ public:
             return dynamic_cast<nested_trait<T, Path> &>(parent.value())
                 .template add_element<V>(std::move(p));
         else {
-            spdlog::error("No parent element found at: {}", path.to_string());
+            spdlog::info("No parent element found at: {}", path.to_string());
             throw std::runtime_error(
                 "No parent element found for " + path.to_string());
         }

src/common/model/package.h

@@ -42,6 +42,8 @@ public:
     package &operator=(const package &) = delete;
     package &operator=(package &&) = delete;
 
+    std::string type_name() const override { return "package"; }
+
     std::string full_name(bool relative) const override;
 
     bool is_deprecated() const;