X-Git-Url: https://git.madduck.net/etc/neomutt.git/blobdiff_plain/a2978b6ed0902fda8c5ab38029428ae2637ce4a6..107b6e08e1a8603ee06f95eaf1586f452c7da024:/.config/neomutt/buildmimetree.py
diff --git a/.config/neomutt/buildmimetree.py b/.config/neomutt/buildmimetree.py
index f7d7c56..02d668b 100755
--- a/.config/neomutt/buildmimetree.py
+++ b/.config/neomutt/buildmimetree.py
@@ -23,8 +23,8 @@
# - python3-markdown
# Optional:
# - pytest
-# - Pynliner
-# - Pygments, if installed, then syntax highlighting is enabled
+# - Pynliner, provides --css-file and thus inline styling of HTML output
+# - Pygments, then syntax highlighting for fenced code is enabled
#
# Latest version:
# https://git.madduck.net/etc/neomutt.git/blob_plain/HEAD:/.config/neomutt/buildmimetree.py
@@ -63,9 +63,19 @@ def parse_cli_args(*args, **kwargs):
"--extensions",
type=str,
default="",
- help="Markdown extension to use (comma-separated list)"
+ help="Markdown extension to use (comma-separated list)",
)
+ if _PYNLINER:
+ parser.add_argument(
+ "--css-file",
+ type=str,
+ default="",
+ help="CSS file to merge with the final HTML",
+ )
+ else:
+ parser.set_defaults(css_file=None)
+
parser.add_argument(
"--only-build",
action="store_true",
@@ -85,12 +95,16 @@ def parse_cli_args(*args, **kwargs):
)
subp = parser.add_subparsers(help="Sub-command parsers", dest="mode")
- massage_p = subp.add_parser("massage", help="Massaging phase (internal use)")
+ massage_p = subp.add_parser(
+ "massage", help="Massaging phase (internal use)"
+ )
massage_p.add_argument(
"--write-commands-to",
+ "-o",
metavar="PATH",
dest="cmdpath",
+ required=True,
help="Temporary file path to write commands to",
)
@@ -158,11 +172,15 @@ class InlineImageExtension(Extension):
return self._images
-def markdown_with_inline_image_support(text, *, extensions=None):
+def markdown_with_inline_image_support(
+ text, *, extensions=None, extension_configs=None
+):
inline_image_handler = InlineImageExtension()
extensions = extensions or []
extensions.append(inline_image_handler)
- mdwn = markdown.Markdown(extensions=extensions)
+ mdwn = markdown.Markdown(
+ extensions=extensions, extension_configs=extension_configs
+ )
htmltext = mdwn.convert(text)
images = inline_image_handler.get_images()
@@ -177,6 +195,38 @@ def markdown_with_inline_image_support(text, *, extensions=None):
return text, htmltext, images
+# [ CSS STYLING ] #############################################################
+
+try:
+ import pynliner
+
+ _PYNLINER = True
+
+except ImportError:
+ _PYNLINER = False
+
+try:
+ from pygments.formatters import get_formatter_by_name
+
+ _CODEHILITE_CLASS = "codehilite"
+
+ _PYGMENTS_CSS = get_formatter_by_name(
+ "html", style="default"
+ ).get_style_defs(f".{_CODEHILITE_CLASS}")
+
+except ImportError:
+ _PYGMENTS_CSS = None
+
+
+def apply_styling(html, css):
+ return (
+ pynliner.Pynliner()
+ .from_string(html)
+ .with_cssString("\n".join(s for s in [_PYGMENTS_CSS, css] if s))
+ .run()
+ )
+
+
# [ PARTS GENERATION ] ########################################################
@@ -223,7 +273,12 @@ def collect_inline_images(
filewriter_fn(path, data.read(), "w+b")
relparts.append(
- Part(*mimetype.split("/"), path, cid=info.cid, desc=f"Image: {info.desc}")
+ Part(
+ *mimetype.split("/"),
+ path,
+ cid=info.cid,
+ desc=f"Image: {info.desc}",
+ )
)
return relparts
@@ -233,12 +288,19 @@ def convert_markdown_to_html(
origtext,
draftpath,
*,
+ cssfile=None,
filewriter_fn=filewriter_fn,
tempdir=None,
extensions=None,
+ extension_configs=None,
):
+ # TODO extension_configs need to be handled differently
+ extension_configs = extension_configs or {}
+ extension_configs.setdefault("pymdownx.highlight", {})
+ extension_configs["pymdownx.highlight"]["css_class"] = _CODEHILITE_CLASS
+
origtext, htmltext, images = markdown_with_inline_image_support(
- origtext, extensions=extensions
+ origtext, extensions=extensions, extension_configs=extension_configs
)
filewriter_fn(draftpath, origtext, encoding="utf-8")
@@ -246,6 +308,8 @@ def convert_markdown_to_html(
"text", "plain", draftpath, "Plain-text version", orig=True
)
+ htmltext = apply_styling(htmltext, cssfile)
+
htmlpath = draftpath.with_suffix(".html")
filewriter_fn(
htmlpath, htmltext, encoding="utf-8", errors="xmlcharrefreplace"
@@ -401,6 +465,7 @@ def do_massage(
cmd_f,
*,
extensions=None,
+ cssfile=None,
converter=convert_markdown_to_html,
only_build=False,
tempdir=None,
@@ -426,7 +491,13 @@ def do_massage(
cmds.flush()
extensions = extensions.split(",") if extensions else []
- tree = converter(draft_f.read(), draftpath, tempdir=tempdir, extensions=extensions)
+ tree = converter(
+ draft_f.read(),
+ draftpath,
+ cssfile=cssfile,
+ tempdir=tempdir,
+ extensions=extensions,
+ )
mimetree = MIMETreeDFWalker(debug=debug_walk)
@@ -436,7 +507,7 @@ def do_massage(
depth-first, and responsible for telling NeoMutt how to assemble
the tree.
"""
- KILL_LINE=r'\Ca\Ck'
+ KILL_LINE = r"\Ca\Ck"
if isinstance(item, Part):
# We've hit a leaf-node, i.e. an alternative or a related part
@@ -527,6 +598,7 @@ if __name__ == "__main__":
pathlib.Path(args.MAILDRAFT),
cmd_f,
extensions=args.extensions,
+ cssfile=args.css_file,
only_build=args.only_build,
tempdir=args.tempdir,
debug_commands=args.debug_commands,
@@ -669,7 +741,7 @@ try:
return StringIO(text or const1)
def test_do_massage_basic(self, const1, string_io, capsys):
- def converter(drafttext, draftpath, extensions, tempdir):
+ def converter(drafttext, draftpath, cssfile, extensions, tempdir):
return Part("text", "plain", draftpath, orig=True)
do_massage(
@@ -693,7 +765,7 @@ try:
def test_do_massage_fulltree(
self, string_io, const1, basic_mime_tree, capsys
):
- def converter(drafttext, draftpath, extensions, tempdir):
+ def converter(drafttext, draftpath, cssfile, extensions, tempdir):
return basic_mime_tree
do_massage(
@@ -889,5 +961,22 @@ try:
assert relparts[0].cid == const1
assert relparts[0].desc.endswith(const2)
+ def test_apply_stylesheet(self):
+ if _PYNLINER:
+ html = "Hello, world!
"
+ css = "p { color:red }"
+ out = apply_styling(html, css)
+ assert 'p style="color' in out
+
+ def test_apply_stylesheet_pygments(self):
+ if _PYGMENTS_CSS:
+ html = (
+ f'"
+ )
+ out = apply_styling(html, _PYGMENTS_CSS)
+ assert f'{_CODEHILITE_CLASS}" style="' in out
+
+
except ImportError:
pass