From da1d8f4e7c332c7109427915e6459b10209cedce Mon Sep 17 00:00:00 2001
From: Runxi Yu <me@runxiyu.org>
Date: Sun, 6 Apr 2025 09:26:46 +0800
Subject: Move the Go stuff to ./forged/

---
 forged/internal/render/chroma.go | 38 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 38 insertions(+)
 create mode 100644 forged/internal/render/chroma.go

(limited to 'forged/internal/render/chroma.go')

diff --git a/forged/internal/render/chroma.go b/forged/internal/render/chroma.go
new file mode 100644
index 0000000..c1a1869
--- /dev/null
+++ b/forged/internal/render/chroma.go
@@ -0,0 +1,38 @@
+// SPDX-License-Identifier: AGPL-3.0-only
+// SPDX-FileCopyrightText: Copyright (c) 2025 Runxi Yu <https://runxiyu.org>
+
+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
+}
-- 
cgit v1.2.3