go-bot/utils/router.go

package utils

import (
	"context"
	"encoding/json"
	"fmt"
	"go-bot/config"
	"go-bot/tools"
	"go-bot/workers"
	"regexp"
	"strings"
	"time"
)

type GroupMessage struct {
	GroupID float64 `json:"group_id"`
	UserID  float64 `json:"user_id"`
	Message string  `json:"raw_message"`
	Role    string  `json:"role"`
	Time    string  `json:"time"`
}

func Router(data map[string]interface{}) {

	// 输出格式化后的JSON
	// fmt.Println(string(jsonString))
	// 读取字段值
	uid := data["user_id"].(float64)
	gid := data["group_id"].(float64)
	sender := data["sender"].(map[string]interface{})
	role := sender["role"].(string)
	mid := data["message_id"].(float64)

	// 将消息存入redis
	redisClient := tools.GetRedisClient()
	if redisClient != nil {
		var group_message GroupMessage
		loc, err := time.LoadLocation("Asia/Shanghai")
		if err != nil {
			return
		}
		group_message.GroupID = gid
		group_message.UserID = uid
		group_message.Message = data["raw_message"].(string)
		group_message.Role = role
		group_message.Time = (time.Unix(int64(data["time"].(float64)), 0).In(loc)).Format("2006-01-02 15:04:05")
		jsonString, err := json.Marshal(group_message)
		if err != nil {
			return
		}
		key := fmt.Sprintf("group_message:%d:%d", int64(gid), int64(uid))
		exists, err := tools.CheckKeyExists(key)
		if err != nil || !exists {
			redisClient.RPush(context.Background(), key, jsonString, 0)
			// tools.SetValue(key, string(jsonString), 10*time.Second)

		} else {
			tools.AddToContext(key, jsonString, 0)
		}
	}

	//包含发送消息的'!'
	raw_msg := data["raw_message"].(string)

	// 匹配回复消息
	if strings.HasPrefix(raw_msg, "[CQ:reply,id=") {
		pattern := `^\[CQ:reply,id=(-?\d+)\]`
		re := regexp.MustCompile(pattern)
		matches := re.FindStringSubmatch(raw_msg)
		if len(matches) > 0 {
			fullMatch := matches[0]
			raw_msg = re.ReplaceAllString(raw_msg, "")
			raw_msg = raw_msg + " " + fullMatch
		}
	}
	prefix, err := config.GetConfig()["Prefix"].(string)
	if !err {
		panic("Prefix参数设置异常！")

	}
	// fmt.Println("raw_msg:", string(raw_msg))
	if len(raw_msg) > 1 {
		if string(raw_msg[0]) == prefix {
			// 去除'!'
			raw_msg = raw_msg[1:]
			parms := strings.Fields(raw_msg)

			worker := workers.NewWorker(parms, fmt.Sprintf("%d", int(uid)), fmt.Sprintf("%d", int(gid)), role, fmt.Sprintf("%d", int(mid)), raw_msg)

			// fmt.Println("router:", parms[0])
			// fmt.Println("CheckPermission:", worker.CheckPermission())
			// ans := NewStdAns(parms[0], fmt.Sprintf("%d", int(uid)), fmt.Sprintf("%d", int(gid)), role, fmt.Sprintf("%d", int(mid)), raw_msg)
			message := worker.CheckPermission()
			if message == "ok" {

				message = worker.GetMsg()
				worker.SendMsg(message)

			} else {
				println("权限校验失败")
			}
		} else {
			// 实现触发器功能

		}
	}

}