From 7dfbf3b527f816d21ea22ea80453043ab0411173 Mon Sep 17 00:00:00 2001 From: Yiwen-Chan Date: Thu, 15 Apr 2021 20:12:06 +0800 Subject: [PATCH] =?UTF-8?q?=F0=9F=8E=A8=20=E6=94=B9=E8=BF=9B=E8=AF=8D?= =?UTF-8?q?=E5=BA=93=E4=BB=A3=E7=A0=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- chat/chat.go | 59 +--------- chat/learn.go | 306 ++++++++++++++++++++++++++++++++++++++++++++++++++ main.go | 2 +- 3 files changed, 312 insertions(+), 55 deletions(-) create mode 100644 chat/learn.go diff --git a/chat/chat.go b/chat/chat.go index 91274f67..c2ea26ee 100644 --- a/chat/chat.go +++ b/chat/chat.go @@ -1,12 +1,11 @@ package chat import ( - "gopkg.in/yaml.v2" "io/ioutil" "math/rand" "os" - "strings" "time" + zero "github.com/wdvxdr1123/ZeroBot" "github.com/wdvxdr1123/ZeroBot/extension/rate" "github.com/wdvxdr1123/ZeroBot/message" @@ -15,54 +14,6 @@ import ( var poke = rate.NewManager(time.Minute*5, 8) // 戳一戳 func init() { // 插件主体 - myData := map[string][]string{} - yaml.Unmarshal(FileRead("chat\\myData.yaml"),&myData) - - zero.OnRegex("^(.+?)$").Handle(func(ctx *zero.Ctx) { - var nickname string = zero.BotConfig.NickName[0] - text := ctx.State["regex_matched"].([]string)[1] - - if strings.Index(text,nickname+"跟我学 ")==0{ - kv := strings.Split(text[len(nickname+"跟我学 "):len(text)]," ") - if len(kv)>0 && kv[1]!=""{ - myData[kv[0]] = kv[1:len(kv)] - content,_ := yaml.Marshal(myData) - FileWrite("chat\\myData.yaml",content) - ctx.Send(nickname + "学会了有关["+kv[0]+"]的新知识呢：\n- "+strings.Join(kv[1:len(kv)],"\n- ")) - }else { - ctx.Send("你想让"+nickname+"学些什么呀？") - } - return - } - - if strings.Index(text,nickname+"请忘掉 ")==0{ - keys := strings.Split(text[len(nickname+"请忘掉 "):len(text)]," ") - if len(keys)>0 && keys[0]!=""{ - for _, key := range keys { - if _,ok := myData[key];ok{ - delete(myData,key) - ctx.Send("["+key+"]是什么呀？"+nickname+"已经忘光光了哦~") - }else { - ctx.Send(nickname+"的脑袋里可没有与["+key+"]有关的内容呢~") - } - content,_ := yaml.Marshal(myData) - FileWrite("chat\\myData.yaml",content) - } - }else { - ctx.Send("你想让"+nickname+"忘掉什么呀？") - } - return - } - - for k,vs := range myData{ - if strings.Index(text,k) != -1{ - rand.Seed(time.Now().Unix()) - ctx.Send(vs[rand.Intn(len(vs))]) - return - } - } - }) - // 被喊名字 zero.OnFullMatch("", zero.OnlyToMe).SetBlock(false).FirstPriority(). Handle(func(ctx *zero.Ctx) { @@ -99,14 +50,14 @@ func init() { // 插件主体 func FileRead(path string) []byte { //读取文件数据 - file,_ := os.Open(path) + file, _ := os.Open(path) defer file.Close() - data,_ := ioutil.ReadAll(file) + data, _ := ioutil.ReadAll(file) return data } -func FileWrite(path string,content []byte) int { +func FileWrite(path string, content []byte) int { //写入文件数据 - ioutil.WriteFile(path,content,0644) + ioutil.WriteFile(path, content, 0644) return len(content) } diff --git a/chat/learn.go b/chat/learn.go new file mode 100644 index 00000000..a49d6056 --- /dev/null +++ b/chat/learn.go @@ -0,0 +1,306 @@ +package chat + +import ( + "crypto/md5" + "encoding/hex" + "encoding/json" + "fmt" + "io/ioutil" + "math/rand" + "net/http" + "os" + "regexp" + "strings" + "sync" + + zero "github.com/wdvxdr1123/ZeroBot" + "github.com/wdvxdr1123/ZeroBot/message" +) + +type QA struct { + Mutex sync.Mutex + Data map[int64]map[string]string + Path string +} + +var ( + BotPath = botPath() + ImagePath = "data\\chat\\image\\" + + Char = map[int64]string{} + CharIndex = map[string]int64{"椛椛": 0, "ATRI": 1} + + QACharPool = &QA{Data: map[int64]map[string]string{}, Path: BotPath + "data\\chat\\char.json"} + QAGroupPool = &QA{Data: map[int64]map[string]string{}, Path: BotPath + "data\\chat\\group.json"} + QAUserPool = &QA{Data: map[int64]map[string]string{}, Path: BotPath + "data\\chat\\user.json"} +) + +func init() { + QACharPool.load() + QAGroupPool.load() + QAUserPool.load() + zero.OnRegex(`切换角色(.*)`, zero.AdminPermission).SetBlock(true).FirstPriority(). + Handle(func(ctx *zero.Ctx) { + if _, ok := CharIndex[ctx.State["regex_matched"].([]string)[1]]; !ok { + ctx.SendChain(message.Text("???")) + return + } + Char[ctx.Event.GroupID] = ctx.State["regex_matched"].([]string)[1] + ctx.SendChain(message.Text("已经切换了哦~")) + }) + zero.OnRegex(`(.{1,2})问(.*)你答(.*)`, QAMatch(), QAPermission()).SetBlock(true).FirstPriority(). + Handle(func(ctx *zero.Ctx) { + question := ctx.State["regex_matched"].([]string)[2] + answer := ctx.State["regex_matched"].([]string)[3] + // 根据匹配使用不同池子对象 + pool := ctx.State["qa_pool"].(*QA) + user := ctx.State["qa_user"].(int64) + // 保存图片，重组图片信息 + r := message.ParseMessageFromString(answer) + for i := range r { + if r[i].Type != "image" { + continue + } + if filename, err := down(r[i].Data["url"], BotPath+ImagePath); err == nil { + r[i].Data["file"] = "file:///BOTPATH\\" + ImagePath + filename + delete(r[i].Data, "url") + } else { // 下载图片发生错误 + ctx.SendChain(message.Text("ERROR: ", err)) + return + } + } + answer = r.CQString() + // 如果是BOT主人，则可以CQ码注入 + if zero.AdminPermission(ctx) { + answer = strings.ReplaceAll(answer, "[", "[") + answer = strings.ReplaceAll(answer, "]", "]") + } + // 添加到池子 + pool.add(user, question, answer) + ctx.SendChain(message.Text("好的我记住了~")) + }) + zero.OnRegex(`删除(.{1,2})问(.*)`, QAMatch(), QAPermission()).SetBlock(true).FirstPriority(). + Handle(func(ctx *zero.Ctx) { + question := ctx.State["regex_matched"].([]string)[2] + // 根据匹配使用不同池子对象 + pool := ctx.State["qa_pool"].(*QA) + user := ctx.State["qa_user"].(int64) + if answer := pool.del(user, question); answer != "" { + ctx.SendChain(message.Text("我不会再回答[", answer, "]了")) + } else { + ctx.SendChain(message.Text("啊咧[", question, "]是什么？")) + } + }) + zero.OnRegex(`看看(.{1,2})问`, QAMatch()).SetBlock(true).FirstPriority(). + Handle(func(ctx *zero.Ctx) { + title := "" + r := []string{} + switch ctx.State["regex_matched"].([]string)[1] { + case "角色": + char := zero.BotConfig.NickName[0] + if Char[ctx.Event.GroupID] != "" { + char = Char[ctx.Event.GroupID] + } + title = "当前角色[" + char + "] 设置的关键词有：\n" + case "有人": + title = "本群设置的关键词有：\n" + case "我": + title = "你设置的关键词有：\n" + } + // 根据匹配使用不同池子对象 + pool := ctx.State["qa_pool"].(*QA) + user := ctx.State["qa_user"].(int64) + for question := range pool.Data[user] { + r = append(r, question) + } + if len(r) == 0 { + ctx.SendChain(message.Text("啊咧？我忘掉什么了吗")) + return + } + ctx.SendChain(message.Text( + title, + strings.Join(r, " | "), + )) + }) + zero.OnMessage().SetBlock(false).SetPriority(9999). + Handle(func(ctx *zero.Ctx) { + m := ctx.Event.RawEvent.Get("message").Str + // 角色问 + if answer := QACharPool.get(CharIndex[Char[ctx.Event.GroupID]], m); answer != "" { + ctx.Send(answer) + return + } + // 有人问 + if answer := QAGroupPool.get(ctx.Event.GroupID, m); answer != "" { + ctx.Send(strings.ReplaceAll(answer, "BOTPATH", BotPath)) + return + } + // 我问 + if answer := QAUserPool.get(ctx.Event.UserID, m); answer != "" { + ctx.Send(answer) + return + } + }) + +} + +func botPath() string { + dir, _ := os.Getwd() + return dir + "\\" +} + +func (qa *QA) load() { + path := qa.Path + idx := strings.LastIndex(qa.Path, "\\") + if idx != -1 { + path = path[:idx] + } + _, err := os.Stat(path) + if err != nil && !os.IsExist(err) { + os.MkdirAll(path, 0644) + } + f, _ := os.Open(qa.Path) + defer f.Close() + data, _ := ioutil.ReadAll(f) + qa.Mutex.Lock() + defer qa.Mutex.Unlock() + if data == nil { + data = []byte("0: \n test: ok") + } + json.Unmarshal(data, &qa.Data) +} + +func (qa *QA) save() { + path := qa.Path + idx := strings.LastIndex(qa.Path, "\\") + if idx != -1 { + path = path[:idx] + } + _, err := os.Stat(path) + if err != nil && !os.IsExist(err) { + os.MkdirAll(path, 0644) + } + data, _ := json.MarshalIndent(&qa.Data, "", "\t") + ioutil.WriteFile(qa.Path, data, 0644) +} + +func (qa *QA) add(user int64, question, answer string) { + qa.Mutex.Lock() + defer qa.Mutex.Unlock() + if qa.Data[user] == nil { // 防止未初始化 + qa.Data[user] = make(map[string]string) + } + qa.Data[user][question] = answer + qa.save() +} + +func (qa *QA) del(user int64, question string) (answer string) { + qa.Mutex.Lock() + defer qa.Mutex.Unlock() + if answer, ok := qa.Data[user][question]; ok { + delete(qa.Data[user], question) + qa.save() + return answer + } + return "" +} + +func (qa *QA) get(user int64, msg string) (answer string) { + for question, answer := range qa.Data[user] { + r := regexp.MustCompile("^" + question + "$") + if r.MatchString(msg) { + match := r.FindStringSubmatch(msg) + // 正则替换参数 + for i, p := range match { + if p == "[" || p == "]" { + continue + } + answer = strings.ReplaceAll(answer, fmt.Sprintf("$%d", i), p) + } + // 随机回复 + if strings.Contains(answer, "*") { + s := strings.Split(answer, "*") + return s[rand.Intn(len(s))] + } + return answer + } + } + return "" +} + +func down(url, path string) (filename string, err error) { + client := &http.Client{} + reqest, _ := http.NewRequest("GET", url, nil) + reqest.Header.Set("User-Agent", "QQ/8.2.0.1296 CFNetwork/1126") + reqest.Header.Set("Net-Type", "Wifi") + resp, err := client.Do(reqest) + if err != nil { + return "", err + } + defer resp.Body.Close() + if resp.StatusCode != http.StatusOK { + return "", fmt.Errorf("code %d", resp.StatusCode) + } + data, _ := ioutil.ReadAll(resp.Body) + // 获取文件MD5值 + m := md5.New() + m.Write(data) + filename = strings.ToUpper(hex.EncodeToString(m.Sum(nil))) + // 判断文件类型 + switch resp.Header.Get("Content-Type") { + case "image/jpeg": + filename = filename + ".jpg" + case "image/png": + filename = filename + ".png" + case "image/gif": + filename = filename + ".gif" + } + // 保存文件 + _, err = os.Stat(path) + if err != nil && !os.IsExist(err) { + os.MkdirAll(path, 0644) + } + f, err := os.OpenFile(path+"\\"+filename, os.O_WRONLY|os.O_TRUNC|os.O_CREATE, 0644) + if err != nil { + return "", err + } + defer f.Close() + f.Write(data) + return filename, nil +} + +// QAMatch 返回对应的权限 +func QAMatch() zero.Rule { + return func(ctx *zero.Ctx) bool { + switch ctx.State["regex_matched"].([]string)[1] { + case "角色": + ctx.State["qa_pool"] = QACharPool + ctx.State["qa_user"] = CharIndex[Char[ctx.Event.GroupID]] + return true + case "有人": + ctx.State["qa_pool"] = QAGroupPool + ctx.State["qa_user"] = ctx.Event.GroupID + return true + case "我": + ctx.State["qa_pool"] = QAUserPool + ctx.State["qa_user"] = ctx.Event.UserID + return true + } + return false + } +} + +// QAPermission 返回对应的权限 +func QAPermission() zero.Rule { + return func(ctx *zero.Ctx) bool { + switch ctx.State["regex_matched"].([]string)[1] { + case "角色": + return zero.AdminPermission(ctx) + case "有人": + return zero.AdminPermission(ctx) + case "我": + return true + } + return false + } +} diff --git a/main.go b/main.go index 20c87bf0..e86d977c 100644 --- a/main.go +++ b/main.go @@ -27,7 +27,7 @@ func main() { fmt.Printf(` ====================[ZeroBot-Plugin]==================== * OneBot + ZeroBot + Golang -* Copyright © 2018-2020 Kanri, All Rights Reserved +* Copyright © 2020-2021 Kanri, All Rights Reserved * Project: https://github.com/Yiwen-Chan/ZeroBot-Plugin ======================================================== `)