Refactored package_trait to more generic nested_trait

src/package_diagram/model/diagram.cc

@@ -33,7 +33,7 @@ std::string diagram::to_alias(const std::string &full_name) const
         throw error::uml_alias_missing(
             fmt::format("Missing alias for '{}'", full_name));
 
-    auto package = get_package(fn);
+    auto package = get_element(fn);
 
     if (!package)
         throw error::uml_alias_missing(

src/package_diagram/model/diagram.h

@@ -28,7 +28,7 @@
 namespace clanguml::package_diagram::model {
 
 class diagram : public clanguml::common::model::diagram,
-                public detail::package_trait<package> {
+                public clanguml::common::model::nested_trait<package> {
 public:
     diagram() = default;
 

src/package_diagram/model/package.h

@@ -18,6 +18,7 @@
 #pragma once
 
 #include "common/model/element.h"
+#include "common/model/nested_trait.h"
 #include "common/model/stylable_element.h"
 #include "util/util.h"
 
@@ -30,108 +31,9 @@
 
 namespace clanguml::package_diagram::model {
 
-namespace detail {
-template <typename T> class package_trait {
-public:
-    package_trait() = default;
-
-    package_trait(const package_trait &) = delete;
-    package_trait(package_trait &&) = default;
-
-    package_trait &operator=(const package_trait &) = delete;
-    package_trait &operator=(package_trait &&) = default;
-
-    virtual ~package_trait() = default;
-
-    void add_package(std::unique_ptr<T> p)
-    {
-        auto it = std::find_if(packages_.begin(), packages_.end(),
-            [&p](const auto &e) { return *e == *p; });
-
-        if (it != packages_.end()) {
-            (*it)->append(*p);
-        }
-        else {
-            packages_.emplace_back(std::move(p));
-        }
-    }
-
-    void add_package(std::vector<std::string> path, std::unique_ptr<T> p)
-    {
-        assert(p);
-
-        LOG_DBG(
-            "Adding package {} at path '{}'", p->name(), fmt::join(path, "::"));
-
-        if (path.empty()) {
-            add_package(std::move(p));
-            return;
-        }
-
-        auto parent = get_package(path);
-
-        if (parent)
-            parent.value().add_package(std::move(p));
-        else
-            spdlog::error(
-                "No parent package found at: {}", fmt::join(path, "::"));
-    }
-
-    type_safe::optional_ref<T> get_package(std::vector<std::string> path) const
-    {
-        LOG_DBG("Getting package at path: {}", fmt::join(path, "::"));
-
-        if (path.empty() || !has_package(path.at(0))) {
-            LOG_WARN(
-                "Sub package {} not found in package", fmt::join(path, "::"));
-            return {};
-        }
-
-        auto p = get_package(path.at(0));
-        if (path.size() == 1)
-            return p;
-
-        return p.value().get_package(
-            std::vector<std::string>(path.begin() + 1, path.end()));
-    }
-
-    type_safe::optional_ref<T> get_package(const std::string &name) const
-    {
-        auto it = std::find_if(packages_.cbegin(), packages_.cend(),
-            [&](const auto &p) { return name == p->name(); });
-
-        if (it == packages_.end())
-            return {};
-
-        assert(it->get() != nullptr);
-
-        return type_safe::ref(*(it->get()));
-    }
-
-    bool has_package(const std::string &name) const
-    {
-        return std::find_if(packages_.cbegin(), packages_.cend(),
-                   [&](const auto &p) { return name == p->name(); }) !=
-            packages_.end();
-    }
-
-    auto begin() { return packages_.begin(); }
-    auto end() { return packages_.end(); }
-
-    auto cbegin() const { return packages_.cbegin(); }
-    auto cend() const { return packages_.cend(); }
-
-    auto begin() const { return packages_.begin(); }
-    auto end() const { return packages_.end(); }
-
-private:
-    std::vector<std::unique_ptr<T>> packages_;
-};
-}
-
 class package : public common::model::element,
                 public common::model::stylable_element,
-                public detail::package_trait<package> {
+                public common::model::nested_trait<package> {
 public:
     package(const std::vector<std::string> &using_namespaces);
 

src/package_diagram/visitor/translation_unit_visitor.cc

@@ -122,10 +122,10 @@ void translation_unit_visitor::operator()(const cppast::cpp_entity &file)
                             }
 
                             if (!p->skip()) {
-                                ctx.diagram().add_package(
+                                ctx.diagram().add_element(
                                     package_parent, std::move(p));
                                 ctx.set_current_package(
-                                    ctx.diagram().get_package(package_path));
+                                    ctx.diagram().get_element(package_path));
                             }
                         }