Refactored sequence diagram generator

src/common/generators/plantuml/generator.cc

@@ -56,4 +56,16 @@ std::string to_plantuml(scope_t scope)
     }
 }
 
+std::string to_plantuml(message_t r)
+{
+    switch (r) {
+    case message_t::kCall:
+        return "->";
+    case message_t::kReturn:
+        return "-->";
+    default:
+        return "";
+    }
+}
+
 }

src/common/generators/plantuml/generator.h

@@ -18,7 +18,6 @@
 #pragma once
 
 #include "config/config.h"
-#include "generator.h"
 #include "util/error.h"
 #include "util/util.h"
 
@@ -26,12 +25,13 @@
 
 namespace clanguml::common::generators::plantuml {
 
+using clanguml::common::model::message_t;
 using clanguml::common::model::relationship_t;
 using clanguml::common::model::scope_t;
 
 std::string to_plantuml(relationship_t r, std::string style);
-
 std::string to_plantuml(scope_t scope);
+std::string to_plantuml(message_t r);
 
 template <typename ConfigType, typename DiagramType> class generator {
 public:

src/common/model/enums.cc

@@ -78,4 +78,16 @@ std::string to_string(access_t a)
         assert(false);
     }
 }
+
+std::string to_string(message_t r)
+{
+    switch (r) {
+    case message_t::kCall:
+        return "call";
+    case message_t::kReturn:
+        return "return";
+    default:
+        assert(false);
+    }
+}
 }

src/common/model/enums.h

@@ -38,9 +38,14 @@ enum class relationship_t {
     kDependency
 };
 
+enum class message_t { kCall, kReturn };
+
 std::string to_string(relationship_t r);
 
 std::string to_string(scope_t r);
 
 std::string to_string(access_t r);
+
+std::string to_string(message_t r);
+
 }