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

src/class_diagram/generators/plantuml/class_diagram_generator.cc

@@ -489,6 +489,8 @@ void generator::generate_relationships(
 
 void generator::generate(std::ostream &ostr) const
 {
+    update_context();
+
     ostr << "@startuml" << '\n';
 
     generate_plantuml_directives(ostr, m_config.puml().before);

src/class_diagram/model/class.h

@@ -42,6 +42,8 @@ public:
     class_ &operator=(const class_ &) = delete;
     class_ &operator=(class_ &&) = delete;
 
+    std::string type_name() const override { return "class"; }
+
     bool is_struct() const;
     void is_struct(bool is_struct);
 

src/class_diagram/model/diagram.cc

@@ -271,6 +271,28 @@ std::string diagram::to_alias(
     throw error::uml_alias_missing(fmt::format("Missing alias for {}", id));
 }
 
+inja::json diagram::context() const {
+    inja::json ctx;
+    ctx["name"] = name();
+    ctx["type"] = "class";
+
+    inja::json::array_t elements{};
+
+    // Add classes
+    for(const auto &c : classes()) {
+        elements.emplace_back(c.get().context());
+    }
+
+    // Add enums
+    for(const auto &e : enums()) {
+        elements.emplace_back(e.get().context());
+    }
+
+    ctx["elements"] = elements;
+
+    return ctx;
+}
+
 }
 
 namespace clanguml::common::model {

src/class_diagram/model/diagram.h

@@ -87,6 +87,8 @@ public:
     bool has_element(
         const clanguml::common::model::diagram_element::id_t id) const override;
 
+    inja::json context() const override;
+
 private:
     common::reference_vector<class_> classes_;
 

src/class_diagram/model/enum.h

@@ -34,6 +34,8 @@ public:
     enum_ &operator=(const enum_ &) = delete;
     enum_ &operator=(enum_ &&) = delete;
 
+    std::string type_name() const override { return "enum"; }
+
     // TODO: Do we need this?
     friend bool operator==(const enum_ &l, const enum_ &r);
 

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;

src/include_diagram/model/diagram.cc

@@ -135,6 +135,22 @@ diagram::files() const
     return files_;
 }
 
+inja::json diagram::context() const {
+    inja::json ctx;
+    ctx["name"] = name();
+    ctx["type"] = "include";
+
+    inja::json::array_t elements{};
+
+    // Add files and directories
+    for(const auto &f : files()) {
+        elements.emplace_back(f.get().context());
+    }
+
+    ctx["elements"] = elements;
+
+    return ctx;
+}
 }
 
 namespace clanguml::common::model {

src/include_diagram/model/diagram.h

@@ -59,6 +59,8 @@ public:
 
     const common::reference_vector<common::model::source_file> &files() const;
 
+    inja::json context() const override;
+
 private:
     common::reference_vector<common::model::source_file> files_;
 };

src/package_diagram/model/diagram.cc

@@ -94,6 +94,22 @@ std::string diagram::to_alias(
 
     return {};
 }
+
+inja::json diagram::context() const {
+    inja::json ctx;
+    ctx["name"] = name();
+    ctx["type"] = "package";
+
+    inja::json::array_t elements{};
+
+    for(const auto &p : packages()) {
+        elements.emplace_back(p.get().context());
+    }
+
+    ctx["elements"] = elements;
+
+    return ctx;
+}
 }
 
 namespace clanguml::common::model {

src/package_diagram/model/diagram.h

@@ -59,6 +59,8 @@ public:
     std::string to_alias(
         const clanguml::common::model::diagram_element::id_t) const;
 
+    inja::json context() const override;
+
 private:
     common::reference_vector<clanguml::common::model::package> packages_;
 };

src/sequence_diagram/model/diagram.cc

@@ -45,6 +45,18 @@ std::string diagram::to_alias(const std::string &full_name) const
     return full_name;
 }
 
+inja::json diagram::context() const {
+    inja::json ctx;
+    ctx["name"] = name();
+    ctx["type"] = "sequence";
+
+    inja::json::array_t elements{};
+
+    ctx["elements"] = elements;
+
+    return ctx;
+}
+
 }
 
 namespace clanguml::common::model {

src/sequence_diagram/model/diagram.h

@@ -45,6 +45,8 @@ public:
 
     std::string to_alias(const std::string &full_name) const;
 
+    inja::json context() const override;
+
     bool started{false};
 
     std::map<int64_t, activity> sequences;