feat: add pygments support

1 files changed, 35 insertions, 0 deletions

diff --git a/syntax-highlighting.py b/syntax-highlighting.py
new file mode 100644
index 0000000..2186c35
--- /dev/null
+++ b/syntax-highlighting.py
@@ -0,0 +1,35 @@
+#!/usr/bin/env python3
+
+"""Render cgit source blobs as syntax-highlighted, HTML-escaped markup."""
+
+import io
+import sys
+
+from pygments import highlight
+from pygments.formatters import HtmlFormatter
+from pygments.lexers import TextLexer, guess_lexer, guess_lexer_for_filename
+from pygments.util import ClassNotFound
+
+
+sys.stdin = io.TextIOWrapper(sys.stdin.buffer, encoding="utf-8", errors="replace")
+sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding="utf-8", errors="replace")
+
+source = sys.stdin.read()
+filename = sys.argv[1]
+formatter = HtmlFormatter(style="monokai", nobackground=True)
+
+try:
+    lexer = guess_lexer_for_filename(filename, source)
+except (ClassNotFound, TypeError):
+    if source.startswith("#!"):
+        try:
+            lexer = guess_lexer(source)
+        except ClassNotFound:
+            lexer = TextLexer()
+    else:
+        lexer = TextLexer()
+
+sys.stdout.write("<style>")
+sys.stdout.write(formatter.get_style_defs(".highlight"))
+sys.stdout.write("</style>")
+sys.stdout.write(highlight(source, lexer, formatter))