Buildozer PrintComment now also returns 'Suffix' and 'After' comments…

… on rules (#958)

CONTRIBUTORS

@@ -12,4 +12,5 @@ Paul Bethe <[email protected]>
 Russ Cox <[email protected]>
 Laurent Le Brun <[email protected]>
 Justine Alexandra Roberts Tunney <[email protected]>
-Nilton Volpato <[email protected]>
+Nilton Volpato <[email protected]>
+Matthew C Brown <[email protected]>

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() {

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])