From 6ca699d508d2b62ab51d290e6a38a1a24f7d83cf Mon Sep 17 00:00:00 2001
From: KiritoDv <kiritodev01@gmail.com>
Date: Sun, 9 Mar 2025 14:10:32 -0600
Subject: [PATCH] Implemented comment

---
 src/Companion.cpp | 8 ++++++++
 src/Companion.h   | 1 +
 2 files changed, 9 insertions(+)

diff --git a/src/Companion.cpp b/src/Companion.cpp
index 885ca2f..7e9744d 100644
--- a/src/Companion.cpp
+++ b/src/Companion.cpp
@@ -733,6 +733,7 @@ void Companion::ProcessFile(YAML::Node root) {
                     result.node["offset"].as<uint32_t>(),
                     alignment,
                     stream.str(),
+                    GetNode<std::string>(result.node, "comment"),
                     std::nullopt
                 };
             } else {
@@ -743,6 +744,7 @@ void Companion::ProcessFile(YAML::Node root) {
                             result.node["offset"].as<uint32_t>(),
                             alignment,
                             stream.str(),
+                            GetNode<std::string>(result.node, "comment"),
                             std::get<size_t>(endptr.value())
                         };
                         break;
@@ -753,6 +755,7 @@ void Companion::ProcessFile(YAML::Node root) {
                             oentry.start,
                             alignment,
                             stream.str(),
+                            GetNode<std::string>(result.node, "comment"),
                             oentry.end
                         };
                         break;
@@ -830,6 +833,11 @@ void Companion::ProcessFile(YAML::Node root) {
         for (size_t i = 0; i < entries.size(); i++) {
             const auto result = entries[i];
             const auto hasSize = result.endptr.has_value();
+
+            if(result.comment.has_value()){
+                stream << "// " << result.comment.value() << "\n";
+            }
+
             if (hasSize && this->IsDebug()) {
                 stream << "// 0x" << std::hex << std::uppercase << ASSET_PTR(result.addr) << "\n";
             }
diff --git a/src/Companion.h b/src/Companion.h
index de1c6dc..069a6e4 100644
--- a/src/Companion.h
+++ b/src/Companion.h
@@ -70,6 +70,7 @@ struct WriteEntry {
     uint32_t addr;
     uint32_t alignment;
     std::string buffer;
+    std::optional<std::string> comment;
     std::optional<uint32_t> endptr;
 };