From e8d502966b089e9a4d1cc960aca8a982e97ce601 Mon Sep 17 00:00:00 2001
From: Matthew Brown <49692338+matthewcbrown@users.noreply.github.com>
Date: Tue, 16 Feb 2021 06:21:39 -0800
Subject: [PATCH] Buildozer PrintComment now also returns 'Suffix' and 'After'
 comments on rules (#958)

---
 CONTRIBUTORS                |  3 ++-
 buildozer/buildozer_test.sh | 10 +++++++++-
 edit/buildozer.go           |  2 ++
 3 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/CONTRIBUTORS b/CONTRIBUTORS
index 2d919a47f..bbf86f0d5 100644
--- a/CONTRIBUTORS
+++ b/CONTRIBUTORS
@@ -12,4 +12,5 @@ Paul Bethe <pbethe@google.com>
 Russ Cox <rsc@google.com>
 Laurent Le Brun <laurentlb@google.com>
 Justine Alexandra Roberts Tunney <jart@google.com>
-Nilton Volpato <nilton@google.com>
\ No newline at end of file
+Nilton Volpato <nilton@google.com>
+Matthew C Brown <matthewbr@google.com>
diff --git a/buildozer/buildozer_test.sh b/buildozer/buildozer_test.sh
index 44695eb80..20cb08585 100755
--- a/buildozer/buildozer_test.sh
+++ b/buildozer/buildozer_test.sh
@@ -1384,7 +1384,15 @@ function test_rule_print_comment() {
   in='# Hello
 cc_library(name = "a")'
   run "$in" 'print_comment' //pkg:a
-  assert_output 'Hello'
+  assert_output 'Hello  '
+}
+
+function test_rule_print_comment_with_suffix_and_after() {
+  in='# Hello Before
+cc_library(name = "a") # Hello Suffix
+# Hello After'
+  run "$in" 'print_comment' //pkg:a
+  assert_output 'Hello Before Hello Suffix Hello After'
 }
 
 function test_attribute_print_comment() {
diff --git a/edit/buildozer.go b/edit/buildozer.go
index e2dac0655..b502e1df3 100644
--- a/edit/buildozer.go
+++ b/edit/buildozer.go
@@ -158,6 +158,8 @@ func cmdPrintComment(opts *Options, env CmdEnvironment) (*build.File, error) {
 	case 0: // Print rule comment.
 		env.output.Fields = []*apipb.Output_Record_Field{
 			{Value: &apipb.Output_Record_Field_Text{commentsText(env.Rule.Call.Comments.Before)}},
+			{Value: &apipb.Output_Record_Field_Text{commentsText(env.Rule.Call.Comments.Suffix)}},
+			{Value: &apipb.Output_Record_Field_Text{commentsText(env.Rule.Call.Comments.After)}},
 		}
 	case 1: // Print attribute comment.
 		attr := env.Rule.AttrDefn(env.Args[0])