From 451681e2b650661aa28694697d72ebfbc3efb9d1 Mon Sep 17 00:00:00 2001 From: Runxi Yu Date: Mon, 10 Feb 2025 15:33:21 +0800 Subject: repo_tree: Handle error from formatter.Format --- handle_repo_tree.go | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/handle_repo_tree.go b/handle_repo_tree.go index 47d02b4..ea596b5 100644 --- a/handle_repo_tree.go +++ b/handle_repo_tree.go @@ -61,13 +61,17 @@ func handle_repo_tree(w http.ResponseWriter, r *http.Request) { } iterator, err := lexer.Tokenise(nil, file_contents) if err != nil { - _, _ = w.Write([]byte("Error rendering code: " + err.Error())) + _, _ = w.Write([]byte("Error tokenizing code: " + err.Error())) return } var formatted_unencapsulated bytes.Buffer style := chroma_styles.Get("autumn") formatter := chroma_formatters_html.New(chroma_formatters_html.WithClasses(true), chroma_formatters_html.TabWidth(8)) - formatter.Format(&formatted_unencapsulated, style, iterator) + err = formatter.Format(&formatted_unencapsulated, style, iterator) + if err != nil { + _, _ = w.Write([]byte("Error formatting code: " + err.Error())) + return + } formatted_encapsulated := template.HTML(formatted_unencapsulated.Bytes()) data["file_contents"] = formatted_encapsulated -- cgit v1.2.3