X-Git-Url: https://git.madduck.net/etc/neomutt.git/blobdiff_plain/8fab16fc7b9a3755ed4fac2622fa45b1b457cbe2..f28612af12038ec707b20a416984012605670ec2:/.config/neomutt/buildmimetree.py diff --git a/.config/neomutt/buildmimetree.py b/.config/neomutt/buildmimetree.py index a85ceb6..3528b06 100755 --- a/.config/neomutt/buildmimetree.py +++ b/.config/neomutt/buildmimetree.py @@ -7,7 +7,7 @@ # Configuration: # neomuttrc (needs to be a single line): # macro compose B "\ -# source '$my_confdir/buildmimetree.py setup|'\ +# source '$my_confdir/buildmimetree.py setup --tempdir $tempdir|'\ # sourc e \$my_mdwn_postprocess_cmd_file\ # " "Convert message into a modern MIME tree with inline images" # @@ -61,12 +61,6 @@ def parse_cli_args(*args, **kwargs): parser_setup = subp.add_parser("setup", help="Setup phase") parser_massage = subp.add_parser("massage", help="Massaging phase") - parser_setup.add_argument( - "--debug-commands", - action="store_true", - help="Turn on debug logging of commands generated to stderr", - ) - parser_setup.add_argument( "--extension", "-x", @@ -79,21 +73,28 @@ def parse_cli_args(*args, **kwargs): ) parser_setup.add_argument( - "--send-message", + "--only-build", action="store_true", - help="Generate command(s) to send the message after processing", + help="Only build, don't send the message", ) - parser_massage.add_argument( + parser_setup.add_argument( + "--tempdir", + default=None, + help="Specify temporary directory to use for attachments", + ) + + parser_setup.add_argument( "--debug-commands", action="store_true", help="Turn on debug logging of commands generated to stderr", ) parser_massage.add_argument( - "--debug-walk", - action="store_true", - help="Turn on debugging to stderr of the MIME tree walk", + "--write-commands-to", + metavar="PATH", + dest="cmdpath", + help="Temporary file path to write commands to", ) parser_massage.add_argument( @@ -105,10 +106,27 @@ def parse_cli_args(*args, **kwargs): ) parser_massage.add_argument( - "--write-commands-to", - metavar="PATH", - dest="cmdpath", - help="Temporary file path to write commands to", + "--only-build", + action="store_true", + help="Only build, don't send the message", + ) + + parser_massage.add_argument( + "--tempdir", + default=None, + help="Specify temporary directory to use for attachments", + ) + + parser_massage.add_argument( + "--debug-commands", + action="store_true", + help="Turn on debug logging of commands generated to stderr", + ) + + parser_massage.add_argument( + "--debug-walk", + action="store_true", + help="Turn on debugging to stderr of the MIME tree walk", ) parser_massage.add_argument( @@ -234,14 +252,19 @@ def collect_inline_images( filewriter_fn(path, data.read(), "w+b") relparts.append( - Part(*mimetype.split("/"), path, cid=info.cid, desc=info.desc) + Part(*mimetype.split("/"), path, cid=info.cid, desc=f"Image: {info.desc}") ) return relparts def convert_markdown_to_html( - origtext, draftpath, *, filewriter_fn=filewriter_fn, extensions=None + origtext, + draftpath, + *, + filewriter_fn=filewriter_fn, + tempdir=None, + extensions=None, ): origtext, htmltext, images = markdown_with_inline_image_support( origtext, extensions=extensions @@ -262,7 +285,9 @@ def convert_markdown_to_html( "alternative", [textpart, htmlpart], "Group of alternative content" ) - imgparts = collect_inline_images(images, filewriter_fn=filewriter_fn) + imgparts = collect_inline_images( + images, tempdir=tempdir, filewriter_fn=filewriter_fn + ) if imgparts: return Multipart( "relative", [altpart] + imgparts, "Group of related content" @@ -377,17 +402,27 @@ class MuttCommands: def do_setup( - extensions=None, *, out_f=sys.stdout, temppath=None, debug_commands=False + extensions=None, + *, + out_f=sys.stdout, + only_build=False, + temppath=None, + tempdir=None, + debug_commands=False, ): extensions = extensions or [] temppath = temppath or pathlib.Path( - tempfile.mkstemp(prefix="muttmdwn-")[1] + tempfile.mkstemp(prefix="muttmdwn-", dir=tempdir)[1] ) cmds = MuttCommands(out_f, debug=debug_commands) editor = f"{sys.argv[0]} massage --write-commands-to {temppath}" if extensions: editor = f'{editor} --extensions {",".join(extensions)}' + if only_build: + editor = f'{editor} --only-build' + if tempdir: + editor = f"{editor} --tempdir {tempdir}" if debug_commands: editor = f"{editor} --debug-commands" @@ -407,6 +442,8 @@ def do_massage( *, extensions=None, converter=convert_markdown_to_html, + only_build=False, + tempdir=None, debug_commands=False, debug_walk=False, ): @@ -429,7 +466,7 @@ def do_massage( cmds.flush() extensions = extensions.split(",") if extensions else [] - tree = converter(draft_f.read(), draftpath, extensions=extensions) + tree = converter(draft_f.read(), draftpath, tempdir=tempdir, extensions=extensions) mimetree = MIMETreeDFWalker(debug=debug_walk) @@ -439,6 +476,8 @@ def do_massage( depth-first, and responsible for telling NeoMutt how to assemble the tree. """ + KILL_LINE=r'\Ca\Ck' + if isinstance(item, Part): # We've hit a leaf-node, i.e. an alternative or a related part # with actual content. @@ -459,7 +498,7 @@ def do_massage( # information, then we might just as well update the NeoMutt # tree now: if item.cid: - cmds.push(f"\\Ca\\Ck{item.cid}") + cmds.push(f"{KILL_LINE}{item.cid}") elif isinstance(item, Multipart): # This node has children, but we already visited them (see @@ -468,7 +507,7 @@ def do_massage( # appropriate grouping: if item.subtype == "alternative": cmds.push("") - elif item.subtype == "relative": + elif item.subtype in ("relative", "related"): cmds.push("") elif item.subtype == "multilingual": cmds.push("") @@ -479,7 +518,7 @@ def do_massage( # If the item has a description, we might just as well add it if item.desc: - cmds.push(f"\\Ca\\Ck{item.desc}") + cmds.push(f"{KILL_LINE}{item.desc}") # Finally, if we're at non-root level, tag the new container, # as it might itself be part of a container, to be processed @@ -494,6 +533,9 @@ def do_massage( # function mimetree.walk(tree, visitor_fn=visitor_fn) + if not only_build: + cmds.push("") + # Finally, cleanup. Since we're responsible for removing the temporary # file, how's this for a little hack? try: @@ -511,10 +553,12 @@ if __name__ == "__main__": args = parse_cli_args() if args.mode == "setup": - if args.send_message: - raise NotImplementedError() - - do_setup(args.extensions, debug_commands=args.debug_commands) + do_setup( + args.extensions, + only_build=args.only_build, + tempdir=args.tempdir, + debug_commands=args.debug_commands, + ) elif args.mode == "massage": with open(args.MAILDRAFT, "r") as draft_f, open( @@ -525,6 +569,8 @@ if __name__ == "__main__": pathlib.Path(args.MAILDRAFT), cmd_f, extensions=args.extensions, + only_build=args.only_build, + tempdir=args.tempdir, debug_commands=args.debug_commands, debug_walk=args.debug_walk, ) @@ -686,7 +732,7 @@ try: return StringIO(text or const1) def test_do_massage_basic(self, const1, string_io, capsys): - def converter(drafttext, draftpath, extensions): + def converter(drafttext, draftpath, extensions, tempdir): return Part("text", "plain", draftpath, orig=True) do_massage( @@ -702,6 +748,7 @@ try: assert '="$my_edit_headers"' in lines.pop(0) assert "unset my_editor" == lines.pop(0) assert "unset my_edit_headers" == lines.pop(0) + assert "send-message" in lines.pop(0) assert "update-encoding" in lines.pop(0) assert "source 'rm -f " in lines.pop(0) assert "unset my_mdwn_postprocess_cmd_file" == lines.pop(0) @@ -709,7 +756,7 @@ try: def test_do_massage_fulltree( self, string_io, const1, basic_mime_tree, capsys ): - def converter(drafttext, draftpath, extensions): + def converter(drafttext, draftpath, extensions, tempdir): return basic_mime_tree do_massage( @@ -721,6 +768,7 @@ try: captured = capsys.readouterr() lines = captured.out.splitlines()[4:] + assert "send-message" in lines.pop(0) assert "Related" in lines.pop(0) assert "group-related" in lines.pop(0) assert "tag-entry" in lines.pop(0) @@ -857,27 +905,26 @@ try: assert k == f"file://{imgpath}" break - def test_markdown_inline_image_processor_base64(self): - img = ( + @pytest.fixture + def test_png(self): + return ( "data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAE" "AAAABCAAAAAA6fptVAAAACklEQVQI12P4DwABAQEAG7buVgAA" ) - text = f"![1px white inlined]({img})" + + def test_markdown_inline_image_processor_base64(self, test_png): + text = f"![1px white inlined]({test_png})" text, html, images = markdown_with_inline_image_support(text) assert 'src="cid:' in html assert "](cid:" in text assert len(images) == 1 - assert img in images + assert test_png in images def test_converter_tree_inline_image_base64( - self, const1, fake_filewriter + self, test_png, const1, fake_filewriter ): - img = ( - "data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAE" - "AAAABCAAAAAA6fptVAAAACklEQVQI12P4DwABAQEAG7buVgAA" - ) - text = f"![inline base64 image]({img})" + text = f"![inline base64 image]({test_png})" path = pathlib.Path(const1) tree = convert_markdown_to_html( text, path, filewriter_fn=fake_filewriter @@ -887,7 +934,23 @@ try: assert tree.children[1].subtype == "png" written = fake_filewriter.pop() assert tree.children[1].path == written[0] - assert written[1] == request.urlopen(img).read() + assert written[1] == request.urlopen(test_png).read() + + def test_inline_image_collection( + self, test_png, const1, const2, fake_filewriter + ): + test_images = {test_png: InlineImageInfo(cid=const1, desc=const2)} + relparts = collect_inline_images( + test_images, filewriter_fn=fake_filewriter + ) + + written = fake_filewriter.pop() + assert b"PNG" in written[1] + + assert relparts[0].subtype == "png" + assert relparts[0].path == written[0] + assert relparts[0].cid == const1 + assert relparts[0].desc.endswith(const2) except ImportError: pass