dkforest

commit 003f83376951f69d6b7ca76c50936ed55896ba16
parent 3a1b387030541fd01d43e97f46570246473897fd
Author: n0tr1v <n0tr1v@protonmail.com>
Date:   Wed,  7 Jun 2023 22:45:57 -0700

cleanup

Diffstat:
M
pkg/web/handlers/api/v1/codeModalInterceptor.go
 | 
8
++++++--

1 file changed, 6 insertions(+), 2 deletions(-)
diff --git a/pkg/web/handlers/api/v1/codeModalInterceptor.go b/pkg/web/handlers/api/v1/codeModalInterceptor.go
@@ -42,6 +42,10 @@ func (i CodeModalInterceptor) InterceptMsg(cmd *Command) {
 	if !utils.InArr(lang, []string{"go", "c", "cpp", "py", "js", "php", "css", "sql", "rs", "c#", "rb"}) {
 		lang = ""
 	}
-	cmd.origMessage = "\n```" + lang + "\n" + cmd.origMessage + "\n```\n"
-	cmd.message = "\n```" + lang + "\n" + cmd.message + "\n```\n"
+	cmd.origMessage = codeFenceWrap(lang, cmd.origMessage)
+	cmd.message = codeFenceWrap(lang, cmd.message)
+}
+
+func codeFenceWrap(lang, msg string) string {
+	return "\n```" + lang + "\n" + msg + "\n```\n"
 }