lixiangwuxian/qq_bot
0
0
Fork 0
Code Issues 3 Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 添加关键词回复

Browse Source
This commit is contained in:
lixiangwuxian 2024-10-20 15:47:00 +08:00
parent 003c9b6f90
commit 5a56f6fdc7
9 changed files with 340 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
handler/handler.go
View File

@ -9,7 +9,8 @@ import (
"git.lxtend.com/qqbot/model"
)
var handlers = make(map[string]model.HandlerInfo[model.Handler])
var allGroupHandlers = make(map[string]model.HandlerInfo[model.Handler])
var GroupHandlers = make(map[int64]map[string]model.HandlerInfo[model.Handler])
var frontMatchHandlers = make(map[string]model.HandlerInfo[model.Handler])
var liveHandlers = make(map[int64]map[int64]model.TryCatchHandler)
var livePrivateHandlers = make(map[int64]model.TryCatchHandler)
@ -21,7 +22,7 @@ func RegisterPrivateHandler(handler model.Handler) {
}
func RegisterHandler(trigger string, handler model.Handler, level constants.PermissionLevel) {
handlers[trigger] = model.HandlerInfo[model.Handler]{
allGroupHandlers[trigger] = model.HandlerInfo[model.Handler]{
Trigger: trigger,
Handler: handler,
Level: level,
@ -47,6 +48,23 @@ func RegisterLiveHandler(groupID int64, userID int64, handler model.TryCatchHand
liveHandlers[groupID][userID] = handler
}
func RegisterGroupHandler(groupID int64, trigger string, handler model.Handler, level constants.PermissionLevel) {
if _, ok := GroupHandlers[groupID]; !ok {
GroupHandlers[groupID] = make(map[string]model.HandlerInfo[model.Handler])
}
GroupHandlers[groupID][trigger] = model.HandlerInfo[model.Handler]{
Trigger: trigger,
Handler: handler,
Level: level,
}
}
func UnRegisterGroupHandler(groupID int64, trigger string) {
if _, ok := GroupHandlers[groupID]; ok {
delete(GroupHandlers[groupID], trigger)
}
}
func MsgInHandler(msg model.Message) (reply model.Reply) {
defer func() {
if r := recover(); r != nil {
@ -85,9 +103,14 @@ func MsgInHandler(msg model.Message) (reply model.Reply) {
}
}
msgArray := strings.Split(msg.RawMsg, " ")
if handler, ok := handlers[msgArray[0]]; ok {
if handler, ok := allGroupHandlers[msgArray[0]]; ok {
return auth.TryExecHandler(msg, handler.Level, handler.Handler)
}
if groupHandler, ok := GroupHandlers[msg.GroupInfo.GroupId]; ok {
if handler, ok := groupHandler[msgArray[0]]; ok {
return auth.TryExecHandler(msg, handler.Level, handler.Handler)
}
}
if !msg.GroupInfo.IsGroupMsg && privateAllHandler != nil {
return privateAllHandler(msg)
}

155
handler/kw/kw.go Normal file
View File

@ -0,0 +1,155 @@
package say
import (
"math/rand/v2"
"strconv"
"git.lxtend.com/qqbot/constants"
"git.lxtend.com/qqbot/handler"
"git.lxtend.com/qqbot/model"
"git.lxtend.com/qqbot/service/kw"
"git.lxtend.com/qqbot/util"
)
func init() {
handler.RegisterHandler("kw", setKw, constants.LEVEL_TRUSTED)
handler.RegisterHelpInform("kw", "kw [关键词] [回复内容] 添加关键词回复,kw [关键词] 删除全部关键词回复")
handler.RegisterHandler("kwd", deleteKw, constants.LEVEL_TRUSTED)
handler.RegisterHandler("kwall", getGroupAllKW, constants.LEVEL_TRUSTED)
handler.RegisterHelpInform("kwall", "kwall 查看所有关键词回复")
initKw()
}
func initKw() {
kwList, err := kw.GetAllKW()
if err != nil {
return
}
for _, v := range kwList {
group, _ := strconv.Atoi(v.GroupID)
regKw(v.Kw, int64(group))
}
}
func setKw(msg model.Message) (reply model.Reply) {
tokens := util.SplitN(msg.RawMsg, 3)
if len(tokens) < 2 {
return model.Reply{
ReplyMsg: "参数不足, 请使用\"kw [关键词] [回复内容]\"添加关键词回复",
ReferOriginMsg: true,
FromMsg: msg,
}
}
if _, err := kw.AddKW(tokens[1], tokens[2], strconv.Itoa(int(msg.GroupInfo.GroupId)), strconv.Itoa(int(msg.UserId))); err != nil {
return model.Reply{
ReplyMsg: "添加失败",
ReferOriginMsg: true,
FromMsg: msg,
}
}
regKw(tokens[1], msg.GroupInfo.GroupId)
return model.Reply{
ReplyMsg: "记下了",
ReferOriginMsg: true,
FromMsg: msg,
}
}
func deleteKw(msg model.Message) (reply model.Reply) {
tokens := util.SplitN(msg.RawMsg, 3)
if len(tokens) == 2 {
if err := kw.DeleteKW(tokens[1], strconv.Itoa(int(msg.GroupInfo.GroupId))); err != nil {
return model.Reply{
ReplyMsg: "删除失败",
ReferOriginMsg: true,
FromMsg: msg,
}
}
unRegKw(tokens[1], msg.GroupInfo.GroupId)
return model.Reply{
ReplyMsg: "清空了",
ReferOriginMsg: true,
FromMsg: msg,
}
}
if len(tokens) < 2 {
return model.Reply{
ReplyMsg: "参数不足, 请使用\"kwd [关键词] [回复内容]\"删除对应的关键词回复",
ReferOriginMsg: true,
FromMsg: msg,
}
}
if err := kw.DeleteKWReply(tokens[1], tokens[2], strconv.Itoa(int(msg.GroupInfo.GroupId))); err != nil {
return model.Reply{
ReplyMsg: "删除失败",
ReferOriginMsg: true,
FromMsg: msg,
}
}
return model.Reply{
ReplyMsg: "删除了",
ReferOriginMsg: true,
FromMsg: msg,
}
}
func getGroupAllKW(msg model.Message) (reply model.Reply) {
kwList, err := kw.GetGroupAllKW(strconv.Itoa(int(msg.GroupInfo.GroupId)))
if err != nil {
return model.Reply{
ReplyMsg: "获取失败",
ReferOriginMsg: true,
FromMsg: msg,
}
}
kws := ""
for k, v := range kwList {
kws += v.Kw + " -> " + v.Reply
if k != len(kwList)-1 {
kws += "\n"
}
}
return model.Reply{
ReplyMsg: kws,
ReferOriginMsg: true,
FromMsg: msg,
}
}
func regKw(keyword string, groupId int64) {
handler.RegisterGroupHandler(groupId, keyword, kwReply, constants.LEVEL_USER)
}
func unRegKw(keyword string, groupId int64) {
handler.UnRegisterGroupHandler(groupId, keyword)
}
func kwReply(msg model.Message) (reply model.Reply) {
w, err := kw.GetKW(msg.RawMsg, strconv.Itoa(int(msg.GroupInfo.GroupId)))
if err != nil {
return model.Reply{
ReplyMsg: "",
ReferOriginMsg: false,
FromMsg: msg,
}
}
wlist := []string{}
for _, v := range w {
wlist = append(wlist, v.Reply)
}
pick := rand.IntN(len(wlist))
if util.Fail(0.1) {
return model.Reply{
ReplyMsg: "",
ReferOriginMsg: false,
FromMsg: msg,
}
}
return model.Reply{
ReplyMsg: wlist[pick],
ReferOriginMsg: false,
FromMsg: msg,
}
}

22
handler/say/say.go
View File

@ -1,22 +0,0 @@
package say
import (
"git.lxtend.com/qqbot/constants"
"git.lxtend.com/qqbot/handler"
"git.lxtend.com/qqbot/model"
)
func init() {
handler.RegisterHandler("kw", say, constants.LEVEL_TRUSTED)
}
func say(msg model.Message) (reply model.Reply) {
if len(msg.RawMsg) <= len("kw ") {
return model.Reply{}
}
return model.Reply{
ReplyMsg: "记下了",
ReferOriginMsg: true,
FromMsg: msg,
}
}

28
handler/wordle/wordle.go Normal file
View File

@ -0,0 +1,28 @@
package wordle
import (
"git.lxtend.com/qqbot/constants"
"git.lxtend.com/qqbot/handler"
"git.lxtend.com/qqbot/model"
)
func init() {
handler.RegisterHandler("wordle", wordle, constants.LEVEL_USER)
}
func wordle(msg model.Message) (reply model.Reply) {
handler.RegisterLiveHandler(msg.GroupInfo.GroupId, msg.UserId, tempTrigger)
return model.Reply{
ReplyMsg: "",
ReferOriginMsg: true,
FromMsg: msg,
}
}
func tempTrigger(msg model.Message) (reply model.Reply, isTrigger bool) {
return model.Reply{
ReplyMsg: " ",
ReferOriginMsg: true,
FromMsg: msg,
}, true
}

2
register.go
View File

@ -8,11 +8,11 @@ import (
_ "git.lxtend.com/qqbot/handler/headmaster"
_ "git.lxtend.com/qqbot/handler/help"
_ "git.lxtend.com/qqbot/handler/jrrp"
_ "git.lxtend.com/qqbot/handler/kw"
_ "git.lxtend.com/qqbot/handler/restart"
_ "git.lxtend.com/qqbot/handler/roll"
_ "git.lxtend.com/qqbot/handler/scoresaber"
_ "git.lxtend.com/qqbot/handler/ticket"
_ "git.lxtend.com/qqbot/handler/urlparser"
_ "git.lxtend.com/qqbot/handler/xibao"
// _ "git.lxtend.com/qqbot/handler/wordle"
)

111
service/kw/kw.go Normal file
View File

@ -0,0 +1,111 @@
package kw
import "git.lxtend.com/qqbot/sqlite3"
func init() {
kwTable := `
CREATE TABLE IF NOT EXISTS kw (
id INTEGER PRIMARY KEY AUTOINCREMENT,
kw TEXT,
reply TEXT,
group_id TEXT,
created_by TEXT,
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
);
`
sqlite3.TryCreateTable(kwTable)
}
func AddKW(kw string, reply string, groupID string, createdBy string) (int64, error) {
kwData := Kw{
Kw: kw,
Reply: reply,
GroupID: groupID,
CreatedBy: createdBy,
}
tx, err := sqlite3.GetTran()
if err != nil {
return 0, err
}
defer tx.Rollback()
res, err := tx.Exec("INSERT INTO kw (kw, reply, group_id, created_by) VALUES (:kw, :reply, :group_id, :created_by)", kwData.Kw, kwData.Reply, kwData.GroupID, kwData.CreatedBy)
if err != nil {
return 0, err
}
id, _ := res.LastInsertId()
tx.Commit()
return id, nil
}
func DeleteKW(kw string, groupID string) error {
tx, err := sqlite3.GetTran()
if err != nil {
return err
}
defer tx.Rollback()
_, err = tx.Exec("DELETE FROM kw WHERE kw = ? AND group_id = ?", kw, groupID)
if err != nil {
return err
}
tx.Commit()
return nil
}
func DeleteKWReply(kw string, reply string, groupID string) error {
tx, err := sqlite3.GetTran()
if err != nil {
return err
}
defer tx.Rollback()
_, err = tx.Exec("DELETE FROM kw WHERE kw = ? AND reply = ? AND group_id = ?", kw, reply, groupID)
if err != nil {
return err
}
tx.Commit()
return nil
}
func GetKW(kw string, groupId string) ([]Kw, error) {
var kws []Kw
tx, err := sqlite3.GetTran()
if err != nil {
return nil, err
}
defer tx.Rollback()
err = tx.Select(&kws, "SELECT * FROM kw WHERE kw = ? AND group_id = ?", kw, groupId)
if err != nil {
return nil, err
}
tx.Commit()
return kws, nil
}
func GetGroupAllKW(groupID string) ([]Kw, error) {
var kws []Kw
tx, err := sqlite3.GetTran()
if err != nil {
return nil, err
}
defer tx.Rollback()
err = tx.Select(&kws, "SELECT * FROM kw WHERE group_id = ?", groupID)
if err != nil {
return nil, err
}
tx.Commit()
return kws, nil
}
func GetAllKW() ([]Kw, error) {
var kws []Kw
tx, err := sqlite3.GetTran()
if err != nil {
return nil, err
}
defer tx.Rollback()
err = tx.Select(&kws, "SELECT * FROM kw")
if err != nil {
return nil, err
}
tx.Commit()
return kws, nil
}

10
service/kw/model.go Normal file
View File

@ -0,0 +1,10 @@
package kw
type Kw struct {
ID int64 `json:"id" db:"id"`
Kw string `json:"kw" db:"kw"`
Reply string `json:"reply" db:"reply"`
GroupID string `json:"group_id" db:"group_id"`
CreatedBy string `json:"created_by" db:"created_by"`
CreatedAt string `json:"created_at" db:"created_at"`
}

7
util/random_fail.go Normal file
View File

@ -0,0 +1,7 @@
package util
import "math/rand/v2"
func Fail(chance float64) bool {
return rand.Float64() < chance
}

8
util/split.go
View File

@ -4,12 +4,8 @@ import (
"regexp"
)
func SplitN(text string, n int) ([]string, int) {
if n == 0 {
return []string{text}, 1
}
func SplitN(text string, n int) []string {
re := regexp.MustCompile(`\s+`)
tokens := re.Split(text, n)
return tokens, len(tokens)
return tokens
}