From e0635b47c2f30719e1ea026812af85c988632c0e Mon Sep 17 00:00:00 2001 From: Runxi Yu Date: Sat, 5 Apr 2025 21:27:17 +0800 Subject: Move things to internal/ --- internal/render/chroma.go | 35 +++++++++++++++++++++++++++++++++++ 1 file changed, 35 insertions(+) create mode 100644 internal/render/chroma.go (limited to 'internal/render/chroma.go') diff --git a/internal/render/chroma.go b/internal/render/chroma.go new file mode 100644 index 0000000..c7d64ec --- /dev/null +++ b/internal/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("
Error tokenizing file: " + err.Error() + "
") //#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("
Error formatting file: " + err.Error() + "
") //#nosec G203 + } + + return template.HTML(buf.Bytes()) //#nosec G203 +} -- cgit v1.2.3