👍 添加聊天热词 (#192)

* ✨ 添加聊天热词 * ✨ 添加聊天热词 * ✨ 添加聊天热词 * 🚨 修lint * 🚨 修lint
2025-12-19 22:00:11 +08:00 · 2022-04-17 21:24:07 +08:00 · 2022-04-17 21:24:07 +08:00 · 51f3eda09a
commit 51f3eda09a
parent 6ec08a5227
3 changed files with 185 additions and 0 deletions
--- a/README.md
+++ b/README.md
@ -405,6 +405,8 @@ print("run[CQ:image,file="+j["img"]+"]")
 ```
 - **舔狗日记** `import _ "github.com/FloatTech/ZeroBot-Plugin/plugin/tiangou"`
    - [x] 舔狗日记
+- **聊天热词** `import _ "github.com/FloatTech/ZeroBot-Plugin/plugin/word_count"`
+    - [x] 热词 [群号] [消息数目]|热词 123456 1000
 - **TODO...**

 ## 使用方法
--- a/main.go
+++ b/main.go
@ -110,6 +110,7 @@ import (
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin/translation"    // 翻译
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin/vtb_quotation"  // vtb语录
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin/wangyiyun"      // 网易云音乐热评
+	_ "github.com/FloatTech/ZeroBot-Plugin/plugin/word_count"     // 聊天热词
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin/wordle"         // 猜单词
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin/ymgal"          // 月幕galgame
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin/zaobao"         // 早报
--- a/plugin/word_count/word_count.go
+++ b/plugin/word_count/word_count.go
@ -0,0 +1,182 @@
+// Package wordcount 聊天热词
+package wordcount
+
+import (
+	"fmt"
+	"github.com/FloatTech/zbputils/binary"
+	"github.com/FloatTech/zbputils/control"
+	"github.com/FloatTech/zbputils/ctxext"
+	"github.com/FloatTech/zbputils/file"
+	"github.com/FloatTech/zbputils/img/text"
+	"github.com/golang/freetype"
+	"github.com/sirupsen/logrus"
+	"github.com/wcharczuk/go-chart/v2"
+	zero "github.com/wdvxdr1123/ZeroBot"
+	"github.com/wdvxdr1123/ZeroBot/message"
+	"os"
+	"regexp"
+	"sort"
+	"strconv"
+	"strings"
+	"time"
+)
+
+var (
+	re        = regexp.MustCompile(`^[一-龥]+$`)
+	stopwords []string
+)
+
+func init() {
+	engine := control.Register("wordcount", &control.Options{
+		DisableOnDefault: false,
+		Help: "聊天热词\n" +
+			"- 热词 [群号] [消息数目]|热词 123456 1000",
+		PublicDataFolder: "WordCount",
+	})
+	cachePath := engine.DataFolder() + "cache/"
+	_ = os.RemoveAll(cachePath)
+	_ = os.MkdirAll(cachePath, 0755)
+	engine.OnRegex(`^热词\s?(\d*)\s?(\d*)$`, zero.OnlyGroup, ctxext.DoOnceOnSuccess(func(ctx *zero.Ctx) bool {
+		_, err := file.GetLazyData(engine.DataFolder()+"stopwords.txt", false, false)
+		if err != nil {
+			ctx.SendChain(message.Text("ERROR:", err))
+			return false
+		}
+		data, err := os.ReadFile(engine.DataFolder() + "stopwords.txt")
+		if err != nil {
+			ctx.SendChain(message.Text("ERROR:", err))
+			return false
+		}
+		stopwords = strings.Split(binary.BytesToString(data), "\r\n")
+		sort.Strings(stopwords)
+		logrus.Infoln("[wordcount]加载", len(stopwords), "条停用词")
+		return true
+	})).Limit(ctxext.LimitByUser).SetBlock(true).
+		Handle(func(ctx *zero.Ctx) {
+			ctx.SendChain(message.Text("少女祈祷中..."))
+			gid, _ := strconv.ParseInt(ctx.State["regex_matched"].([]string)[1], 10, 64)
+			p, _ := strconv.ParseInt(ctx.State["regex_matched"].([]string)[2], 10, 64)
+			if p > 10000 {
+				p = 10000
+			}
+			if p == 0 {
+				p = 1000
+			}
+			if gid == 0 {
+				gid = ctx.Event.GroupID
+			}
+			group := ctx.GetGroupInfo(gid, false)
+			if group.MemberCount == 0 {
+				ctx.SendChain(message.Text(zero.BotConfig.NickName[0], "未加入", group.Name, "(", gid, "),无法获得热词呢"))
+				return
+			}
+			today := time.Now().Format("20060102")
+			drawedFile := fmt.Sprintf("%s%d%s%dwordCount.png", cachePath, gid, today, p)
+			if file.IsExist(drawedFile) {
+				ctx.SendChain(message.Image("file:///" + file.BOTPATH + "/" + drawedFile))
+				return
+			}
+			messageMap := make(map[string]int)
+			h := ctx.CallAction("get_group_msg_history", zero.Params{"group_id": gid}).Data
+			messageSeq := h.Get("messages.0.message_seq").Int()
+			for i := 0; i < int(p/20) && messageSeq != 0; i++ {
+				if i != 0 {
+					h = ctx.CallAction("get_group_msg_history", zero.Params{"group_id": ctx.Event.GroupID, "message_seq": messageSeq}).Data
+				}
+				for _, v := range h.Get("messages.#.message").Array() {
+					tex := strings.TrimSpace(message.ParseMessageFromString(v.Str).ExtractPlainText())
+					if tex == "" {
+						continue
+					}
+					for _, t := range ctx.GetWordSlices(tex).Get("slices").Array() {
+						tex := strings.TrimSpace(t.Str)
+						i := sort.SearchStrings(stopwords, tex)
+						if re.MatchString(tex) && (i >= len(stopwords) || stopwords[i] != tex) {
+							messageMap[tex]++
+						}
+
+					}
+				}
+				messageSeq = h.Get("messages.0.message_seq").Int()
+			}
+			wc := rankByWordCount(messageMap)
+			if len(wc) > 20 {
+				wc = wc[:20]
+			}
+			// 绘图
+			if len(wc) == 0 {
+				ctx.SendChain(message.Text("ERROR:历史消息为空或者无法获得历史消息"))
+				return
+			}
+			_, err := file.GetLazyData(text.FontFile, false, true)
+			if err != nil {
+				ctx.SendChain(message.Text("ERROR:", err))
+				return
+			}
+			b, err := os.ReadFile(text.FontFile)
+			if err != nil {
+				ctx.SendChain(message.Text("ERROR:", err))
+				return
+			}
+			font, err := freetype.ParseFont(b)
+			if err != nil {
+				ctx.SendChain(message.Text("ERROR:", err))
+				return
+			}
+			bars := make([]chart.Value, len(wc))
+			for i, v := range wc {
+				bars[i] = chart.Value{
+					Value: float64(v.Value),
+					Label: v.Key,
+				}
+			}
+			graph := chart.BarChart{
+				Font:  font,
+				Title: fmt.Sprintf("%s(%d)在%s号的%d条消息的热词top20", group.Name, gid, time.Now().Format("2006-01-02"), p),
+				Background: chart.Style{
+					Padding: chart.Box{
+						Top: 40,
+					},
+				},
+				Height:   500,
+				BarWidth: 25,
+				Bars:     bars,
+			}
+			f, err := os.Create(drawedFile)
+			if err != nil {
+				ctx.SendChain(message.Text("ERROR:", err))
+				return
+			}
+			err = graph.Render(chart.PNG, f)
+			_ = f.Close()
+			if err != nil {
+				_ = os.Remove(drawedFile)
+				ctx.SendChain(message.Text("ERROR:", err))
+				return
+			}
+			ctx.SendChain(message.Image("file:///" + file.BOTPATH + "/" + drawedFile))
+
+		})
+}
+
+func rankByWordCount(wordFrequencies map[string]int) pairlist {
+	pl := make(pairlist, len(wordFrequencies))
+	i := 0
+	for k, v := range wordFrequencies {
+		pl[i] = pair{k, v}
+		i++
+	}
+	sort.Sort(sort.Reverse(pl))
+	return pl
+}
+
+type pair struct {
+	Key   string
+	Value int
+}
+
+type pairlist []pair
+
+func (p pairlist) Len() int           { return len(p) }
+func (p pairlist) Less(i, j int) bool { return p[i].Value < p[j].Value }
+func (p pairlist) Swap(i, j int)      { p[i], p[j] = p[j], p[i] }