From 0512e1a33e7d140d8f8d03d15a2818ce8a85f70c Mon Sep 17 00:00:00 2001 From: Rusty Russell Date: Sat, 2 May 2020 20:11:16 +0930 Subject: [PATCH] tools/generate-wire.py: add --include argument for putting #includes in spec-generated files. We need this for bolt13. Signed-off-by: Rusty Russell --- tools/generate-wire.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tools/generate-wire.py b/tools/generate-wire.py index 8dd9cea70..9144b6e2c 100755 --- a/tools/generate-wire.py +++ b/tools/generate-wire.py @@ -547,6 +547,9 @@ def main(options, args=None, output=sys.stdout, lines=None): # Create a new 'master' that serves as the coordinator for the file generation master = Master() + for i in options.include: + master.add_include('#include <{}>'.format(i)) + try: while True: ln, line = next(genline) @@ -689,6 +692,7 @@ if __name__ == "__main__": action="store_true", default=False) parser.add_argument("--page", choices=['header', 'impl'], help="page to print") parser.add_argument('--expose-tlv-type', action='append', default=[]) + parser.add_argument('--include', action='append', default=[]) parser.add_argument('header_filename', help='The filename of the header') parser.add_argument('enum_name', help='The name of the enum to produce') parser.add_argument("files", help='Files to read in (or stdin)', nargs=REMAINDER)