Fixed issue #227.

3 files changed, 22 insertions, 19 deletions

diff --git a/src/yuescript/yue_ast.cpp b/src/yuescript/yue_ast.cpp
index 89b8d3f..2fccd35 100644
--- a/src/yuescript/yue_ast.cpp
+++ b/src/yuescript/yue_ast.cpp
@@ -215,11 +215,11 @@ std::string YueMultilineComment_t::to_string(void* ud) const {
         return "--[["s + info->convert(this) + "]]"s;
 }
 std::string YueComment_t::to_string(void* ud) const {
-        if (comment) {
-                return comment->to_string(ud);
-        } else {
-                return {};
+        str_list coms;
+        for (auto comment : comments.objects()) {
+                coms.emplace_back(comment->to_string(ud));
         }
+        return join(coms, " "sv);
 }
 std::string Variable_t::to_string(void* ud) const {
         return name->to_string(ud);
diff --git a/src/yuescript/yue_ast.h b/src/yuescript/yue_ast.h
index 5faeed5..98f9014 100644
--- a/src/yuescript/yue_ast.h
+++ b/src/yuescript/yue_ast.h
@@ -940,8 +940,8 @@ AST_LEAF(YueMultilineComment)
 AST_END(YueMultilineComment)
 
 AST_NODE(YueComment)
-        ast_sel<true, YueLineComment_t, YueMultilineComment_t> comment;
-        AST_MEMBER(YueComment, &comment)
+        ast_sel_list<true, YueLineComment_t, YueMultilineComment_t> comments;
+        AST_MEMBER(YueComment, &comments)
 AST_END(YueComment)
 
 AST_LEAF(EmptyLine)
diff --git a/src/yuescript/yue_compiler.cpp b/src/yuescript/yue_compiler.cpp
index 6aa7cfd..bcc44d2 100644
--- a/src/yuescript/yue_compiler.cpp
+++ b/src/yuescript/yue_compiler.cpp
@@ -78,7 +78,7 @@ static std::unordered_set<std::string> Metamethods = {
         "close"s // Lua 5.4
 };
 
-const std::string_view version = "0.32.7"sv;
+const std::string_view version = "0.32.8"sv;
 const std::string_view extension = "yue"sv;
 
 class CompileError : public std::logic_error {
@@ -1742,23 +1742,26 @@ private:
                 if (!_config.reserveComment) {
                         return;
                 }
-                auto node = comment->comment.get();
-                if (!node) {
+                if (comment->comments.empty()) {
                         out.push_back("\n"s);
                         return;
                 }
-                switch (node->get_id()) {
-                        case id<YueLineComment_t>(): {
-                                auto content = static_cast<YueLineComment_t*>(node);
-                                out.push_back(indent() + "--"s + _parser.toString(content) + '\n');
-                                break;
-                        }
-                        case id<YueMultilineComment_t>(): {
-                                auto content = static_cast<YueMultilineComment_t*>(node);
-                                out.push_back(indent() + "--[["s + _parser.toString(content) + "]]\n"s);
-                                break;
+                str_list temp;
+                for (auto node : comment->comments.objects()) {
+                        switch (node->get_id()) {
+                                case id<YueLineComment_t>(): {
+                                        auto content = static_cast<YueLineComment_t*>(node);
+                                        temp.emplace_back(indent() + "--"s + _parser.toString(content));
+                                        break;
+                                }
+                                case id<YueMultilineComment_t>(): {
+                                        auto content = static_cast<YueMultilineComment_t*>(node);
+                                        temp.emplace_back(indent() + "--[["s + _parser.toString(content) + "]]"s);
+                                        break;
+                                }
                         }
                 }
+                out.push_back(join(temp, " "sv) + '\n');
         }
 
         void transformStatement(Statement_t* statement, str_list& out) {