From 451681e2b650661aa28694697d72ebfbc3efb9d1 Mon Sep 17 00:00:00 2001 From: Runxi Yu Date: Mon, 10 Feb 2025 15:33:21 +0800 Subject: [PATCH] repo_tree: Handle error from formatter.Format --- handle_repo_tree.go | 8 ++++++-- diff --git a/handle_repo_tree.go b/handle_repo_tree.go index 47d02b4bf72d0a180d01cbf27a1e0eca82999cd9..ea596b5d7c2386ead84f0f6e91d974847cf4a3fd 100644 --- a/handle_repo_tree.go +++ b/handle_repo_tree.go @@ -61,13 +61,17 @@ lexer = chroma_lexers.Fallback } 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 -- 2.48.1