Separate code/README rendering and unsafe to their own packages

1 files changed, 35 insertions, 0 deletions

diff --git a/render/chroma.go b/render/chroma.go
new file mode 100644
index 0000000..c7d64ec
--- /dev/null
+++ b/render/chroma.go
@@ -0,0 +1,35 @@
+package render
+
+import (
+	"bytes"
+	"html/template"
+
+	chromaHTML "github.com/alecthomas/chroma/v2/formatters/html"
+	chromaLexers "github.com/alecthomas/chroma/v2/lexers"
+	chromaStyles "github.com/alecthomas/chroma/v2/styles"
+)
+
+func Highlight(filename, content string) template.HTML {
+	lexer := chromaLexers.Match(filename)
+	if lexer == nil {
+		lexer = chromaLexers.Fallback
+	}
+
+	iterator, err := lexer.Tokenise(nil, content)
+	if err != nil {
+		return template.HTML("<pre>Error tokenizing file: " + err.Error() + "</pre>") //#nosec G203`
+	}
+
+	var buf bytes.Buffer
+	style := chromaStyles.Get("autumn")
+	formatter := chromaHTML.New(
+		chromaHTML.WithClasses(true),
+		chromaHTML.TabWidth(8),
+	)
+
+	if err := formatter.Format(&buf, style, iterator); err != nil {
+		return template.HTML("<pre>Error formatting file: " + err.Error() + "</pre>") //#nosec G203
+	}
+
+	return template.HTML(buf.Bytes()) //#nosec G203
+}