fix: 修复词云中可能存在panic的地方 (#33)

yqchilde · Feb 26, 2023 · fdcca88 · fdcca88
1 parent aaa9245
commit fdcca88
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/plugins/wordcloud/main.go b/plugins/wordcloud/main.go
@@ -59,7 +59,11 @@ func init() {
 			}
 			// 剔除消息中的艾特
 			if strings.HasPrefix(msg.Content, "@") {
-				msg.Content = msg.Content[strings.Index(msg.Content, " "):]
+				if strings.Contains(msg.Content, " ") {
+					msg.Content = msg.Content[strings.Index(msg.Content, " "):]
+				} else {
+					msg.Content = msg.Content[1:]
+				}
 			}
 			words += msg.Content + " "
 		}