From 58d2ab4718f197ae39010329a636e3d6cc47ef0b Mon Sep 17 00:00:00 2001
From: Kevin Newton <kddnewton@gmail.com>
Date: Mon, 27 Feb 2023 15:27:05 -0500
Subject: [PATCH] Fix AST output for Command and CommandCall nodes

---
 CHANGELOG.md                     | 4 ++++
 lib/syntax_tree/field_visitor.rb | 2 ++
 2 files changed, 6 insertions(+)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 018d5b25..b06032f7 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -6,6 +6,10 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/) a
 
 ## [Unreleased]
 
+### Changed
+
+- Fixed the AST output by adding blocks to `Command` and `CommandCall` nodes in the `FieldVisitor`.
+
 ## [6.0.1] - 2023-02-26
 
 ### Added
diff --git a/lib/syntax_tree/field_visitor.rb b/lib/syntax_tree/field_visitor.rb
index ca1df55b..f5607c67 100644
--- a/lib/syntax_tree/field_visitor.rb
+++ b/lib/syntax_tree/field_visitor.rb
@@ -263,6 +263,7 @@ def visit_command(node)
         node(node, "command") do
           field("message", node.message)
           field("arguments", node.arguments)
+          field("block", node.block) if node.block
           comments(node)
         end
       end
@@ -273,6 +274,7 @@ def visit_command_call(node)
           field("operator", node.operator)
           field("message", node.message)
           field("arguments", node.arguments) if node.arguments
+          field("block", node.block) if node.block
           comments(node)
         end
       end