pkgdev mask: Accept -b/--bug for referencing bugs

When masking a package - and especially last-riting - the author should
provide bug references. When --bug is given, the comment template is
expanded to include the bug refs.

Signed-off-by: Thomas Bracht Laumann Jespersen <t@laumann.xyz>

Author: laumann

src/pkgdev/scripts/pkgdev_mask.py

@@ -17,6 +17,7 @@
 from snakeoil.bash import iter_read_bash
 from snakeoil.cli import arghparse
 from snakeoil.osutils import pjoin
+from snakeoil.strings import pluralism
 
 from .. import git
 from .argparsers import cwd_repo_argparser, git_repo_argparser
@@ -43,6 +44,13 @@
         package removal but accepts an optional argument for the number of
         days.
     """)
+mask_opts.add_argument(
+    '-b', '--bug', dest='bugs', action='append', type=arghparse.positive_int,
+    help='reference bug in the mask comment',
+    docs="""
+        Add a reference to a bug in the mask comment. May be specified multiple
+        times to reference multiple bugs.
+    """)
 
 
 @mask.bind_final_check
@@ -166,9 +174,14 @@ def __str__(self):
         return ''.join(self.header) + '\n\n'.join(map(str, self.masks))
 
 
-def get_comment():
+def get_comment(bugs):
     """Spawn editor to get mask comment."""
     tmp = tempfile.NamedTemporaryFile(mode='w')
+    if bugs:
+        # Bug(s) #A, #B, #C
+        bug_list = ", ".join(f'#{b}' for b in bugs)
+        s = pluralism(bugs)
+        tmp.write(f'\nBug{s} {bug_list}')
     tmp.write(textwrap.dedent("""
 
         # Please enter the mask message. Lines starting with '#' will be ignored.
@@ -226,7 +239,7 @@ def _mask(options, out, err):
         'author': author,
         'email': email,
         'date': today.strftime('%Y-%m-%d'),
-        'comment': get_comment(),
+        'comment': get_comment(options.bugs),
         'atoms': options.atoms,
     }