dkforest

commit 61f47aff099ff72b3145519c01de732f0f11546d
parent 3b9e99c655b6cca72d17c5f8036c4f8300bfbfaf
Author: n0tr1v <n0tr1v@protonmail.com>
Date:   Thu,  8 Jun 2023 22:51:41 -0700

cleanup

Diffstat:
M
pkg/web/handlers/streamModals/codeModal.go
 | 
20
+++++++++++---------

1 file changed, 11 insertions(+), 9 deletions(-)
diff --git a/pkg/web/handlers/streamModals/codeModal.go b/pkg/web/handlers/streamModals/codeModal.go
@@ -145,15 +145,8 @@ func (_ CodeModal) InterceptMsg(cmd *command.Command) {
 	}
 
 	CodeModal{}.Hide(cmd.AuthUser.ID, cmd.Room.ID)
-
-	found := false
-	for _, l := range langs {
-		if lang == l[0] {
-			found = true
-			break
-		}
-	}
-	if !found {
+	
+	if !isValidLang(lang) {
 		lang = ""
 	}
 	cmd.AuthUser.SyntaxHighlightCode = lang
@@ -186,3 +179,12 @@ func (_ CodeModal) InterceptMsg(cmd *command.Command) {
 func codeFenceWrap(lang, msg string) string {
 	return "\n```" + lang + "\n" + msg + "\n```\n"
 }
+
+func isValidLang(lang string) (found bool) {
+	for _, l := range langs {
+		if lang == l[0] {
+			return true
+		}
+	}
+	return
+}