first commit

Author: sslime336

.gitignore

@@ -0,0 +1,4 @@
+build/
+log/
+config.yaml
+*.exe

README.md

@@ -0,0 +1,3 @@
+# qqbot-demo
+
+将 `config.example.yaml` 改为 `config.yaml` 并填入相关信息

config.example.yaml

@@ -0,0 +1,8 @@
+bot:
+  uin: 123 # 机器人QQ号
+  appId: 123
+  token: string1
+  secret: string2
+
+log:
+  path: ./log/bot.log

config/config.go

@@ -0,0 +1,47 @@
+package config
+
+import (
+	"log"
+
+	"github.com/spf13/viper"
+)
+
+var appConfig AppConfig
+
+type AppConfig struct {
+	Bot BotConf `yaml:"bot"`
+	Log LogConf `yaml:"log"`
+}
+
+type BotConf struct {
+	Token string `yaml:"token"`
+	Uin   uint64 `yaml:"uin"`
+	AppId uint64 `yaml:"appId"`
+}
+
+type LogConf struct {
+	Path string `yaml:"path"`
+}
+
+func Init() {
+	viper.SetConfigName("config")
+	viper.SetConfigType("yaml")
+	viper.AddConfigPath("./")
+	err := viper.ReadInConfig()
+	if err != nil {
+		log.Fatal("read in config failed:", err)
+	}
+	viper.Unmarshal(&appConfig)
+}
+
+func Bot() BotConf {
+	return appConfig.Bot
+}
+
+func Log() LogConf {
+	return appConfig.Log
+}
+
+func App() AppConfig {
+	return appConfig
+}

go.mod

@@ -0,0 +1,36 @@
+module github.com/hduhelp/qqbot-demo
+
+go 1.22.1
+
+require (
+	github.com/spf13/viper v1.19.0
+	github.com/tencent-connect/botgo v0.1.6
+	go.uber.org/zap v1.27.0
+)
+
+require (
+	github.com/fsnotify/fsnotify v1.7.0 // indirect
+	github.com/go-resty/resty/v2 v2.6.0 // indirect
+	github.com/gorilla/websocket v1.5.3 // indirect
+	github.com/hashicorp/hcl v1.0.0 // indirect
+	github.com/magiconair/properties v1.8.7 // indirect
+	github.com/mitchellh/mapstructure v1.5.0 // indirect
+	github.com/pelletier/go-toml/v2 v2.2.2 // indirect
+	github.com/sagikazarmark/locafero v0.6.0 // indirect
+	github.com/sagikazarmark/slog-shim v0.1.0 // indirect
+	github.com/sourcegraph/conc v0.3.0 // indirect
+	github.com/spf13/afero v1.11.0 // indirect
+	github.com/spf13/cast v1.6.0 // indirect
+	github.com/spf13/pflag v1.0.5 // indirect
+	github.com/subosito/gotenv v1.6.0 // indirect
+	github.com/tidwall/gjson v1.9.3 // indirect
+	github.com/tidwall/match v1.1.1 // indirect
+	github.com/tidwall/pretty v1.2.0 // indirect
+	go.uber.org/multierr v1.11.0 // indirect
+	golang.org/x/exp v0.0.0-20240719175910-8a7402abbf56 // indirect
+	golang.org/x/net v0.27.0 // indirect
+	golang.org/x/sys v0.22.0 // indirect
+	golang.org/x/text v0.16.0 // indirect
+	gopkg.in/ini.v1 v1.67.0 // indirect
+	gopkg.in/yaml.v3 v3.0.1 // indirect
+)

go.sum

@@ -0,0 +1,39 @@
+package handler
+
+const (
+	EventGuildCreate           = "GUILD_CREATE"
+	EventGuildUpdate           = "GUILD_UPDATE"
+	EventGuildDelete           = "GUILD_DELETE"
+	EventChannelCreate         = "CHANNEL_CREATE"
+	EventChannelUpdate         = "CHANNEL_UPDATE"
+	EventChannelDelete         = "CHANNEL_DELETE"
+	EventGuildMemberAdd        = "GUILD_MEMBER_ADD"
+	EventGuildMemberUpdate     = "GUILD_MEMBER_UPDATE"
+	EventGuildMemberRemove     = "GUILD_MEMBER_REMOVE"
+	EventMessageCreate         = "MESSAGE_CREATE"
+	EventMessageReactionAdd    = "MESSAGE_REACTION_ADD"
+	EventMessageReactionRemove = "MESSAGE_REACTION_REMOVE"
+	EventAtMessageCreate       = "AT_MESSAGE_CREATE"
+	EventPublicMessageDelete   = "PUBLIC_MESSAGE_DELETE"
+	EventDirectMessageCreate   = "DIRECT_MESSAGE_CREATE"
+	EventDirectMessageDelete   = "DIRECT_MESSAGE_DELETE"
+	EventAudioStart            = "AUDIO_START"
+	EventAudioFinish           = "AUDIO_FINISH"
+	EventAudioOnMic            = "AUDIO_ON_MIC"
+	EventAudioOffMic           = "AUDIO_OFF_MIC"
+	EventMessageAuditPass      = "MESSAGE_AUDIT_PASS"
+	EventMessageAuditReject    = "MESSAGE_AUDIT_REJECT"
+	EventMessageDelete         = "MESSAGE_DELETE"
+	EventForumThreadCreate     = "FORUM_THREAD_CREATE"
+	EventForumThreadUpdate     = "FORUM_THREAD_UPDATE"
+	EventForumThreadDelete     = "FORUM_THREAD_DELETE"
+	EventForumPostCreate       = "FORUM_POST_CREATE"
+	EventForumPostDelete       = "FORUM_POST_DELETE"
+	EventForumReplyCreate      = "FORUM_REPLY_CREATE"
+	EventForumReplyDelete      = "FORUM_REPLY_DELETE"
+	EventForumAuditResult      = "FORUM_PUBLISH_AUDIT_RESULT"
+	EventInteractionCreate     = "INTERACTION_CREATE"
+
+	EventC2CMessageCreate     = "C2C_MESSAGE_CREATE"      // 用户单聊发消息给机器人时候
+	EventGroupAtMessageCreate = "GROUP_AT_MESSAGE_CREATE" // 用户在群里@机器人时收到的消息
+)

handler/eventType.go

@@ -0,0 +1,45 @@
+package handler
+
+import (
+	"encoding/json"
+	"strings"
+
+	"github.com/hduhelp/qqbot-demo/logging"
+	"github.com/hduhelp/qqbot-demo/model"
+	"github.com/tencent-connect/botgo/dto"
+	"github.com/tencent-connect/botgo/event"
+	"github.com/tencent-connect/botgo/openapi"
+	"go.uber.org/zap"
+)
+
+var api openapi.OpenAPI
+
+func RegisterApi(openapi openapi.OpenAPI) {
+	api = openapi
+}
+
+func Get() event.PlainEventHandler {
+	return func(event *dto.WSPayload, data []byte) error {
+		var g model.General
+		if err := json.Unmarshal(data, &g); err != nil {
+			logging.Error("failed to decode json", zap.Error(err))
+			return nil
+		}
+		switch g.T {
+		// 用户单聊发消息给机器人时候
+		case EventC2CMessageCreate:
+			do := ExtractPrivateChatMessage(data)
+			msgId := do.Data.ID
+			return SendTextMessagePrivate(do.Data.Author.UserOpenid, msgId, "不是，哥们儿")
+		// 用户在群里@机器人时收到的消息
+		case EventGroupAtMessageCreate:
+			do := ExtractGroupMessage(data)
+			msgId := do.Data.ID
+			if strings.HasPrefix(strings.TrimSpace(do.Data.Content), "/ping") {
+				return SendTextMessageGroup(do.Data.GroupOpenid, msgId, "pong")
+			}
+			return SendTextMessageGroup(do.Data.GroupOpenid, msgId, "不是，哥们儿")
+		}
+		return nil
+	}
+}

handler/handler.go

@@ -0,0 +1,54 @@
+package handler
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+
+	"github.com/hduhelp/qqbot-demo/logging"
+	"github.com/hduhelp/qqbot-demo/model"
+	"go.uber.org/zap"
+)
+
+func SendTextMessagePrivate(userOpenId, prvMsgId, msg string) (err error) {
+	url := fmt.Sprintf("https://sandbox.api.sgroup.qq.com/v2/users/%s/messages", userOpenId)
+	_, err = api.Transport(context.Background(), "POST", url, SendMessage{
+		Content: msg,
+		MsgType: 0,
+		MsgId:   prvMsgId,
+	})
+	return
+}
+
+func SendTextMessageGroup(groupId, prvMsgId, msg string) (err error) {
+	url := fmt.Sprintf("https://sandbox.api.sgroup.qq.com/v2/groups/%s/messages", groupId)
+	_, err = api.Transport(context.Background(), "POST", url, SendMessage{
+		Content: msg,
+		MsgType: 0,
+		MsgId:   prvMsgId,
+	})
+	return
+}
+
+type SendMessage struct {
+	Content string `json:"content"`
+	MsgType int    `json:"msg_type"`
+	MsgId   string `json:"msg_id"` // 带上该字段后发送消息变成被动消息，限制比主动消息少
+}
+
+func ExtractPrivateChatMessage(data []byte) *model.PrivateChatMessage {
+	var msg model.PrivateChatMessage
+	if err := json.Unmarshal(data, &msg); err != nil {
+		logging.Error("failed to unmarshal PrivateChatMessage", zap.Error(err))
+		return nil
+	}
+	return &msg
+}
+func ExtractGroupMessage(data []byte) *model.GroupAtMessage {
+	var msg model.GroupAtMessage
+	if err := json.Unmarshal(data, &msg); err != nil {
+		logging.Error("failed to unmarshal GroupAtMessage", zap.Error(err))
+		return nil
+	}
+	return &msg
+}

handler/util.go

@@ -0,0 +1,43 @@
+package logging
+
+import (
+	"go.uber.org/zap/zapcore"
+)
+
+func Debug(msg string, fields ...zapcore.Field) {
+	logger.Debug(msg, fields...)
+}
+func Info(msg string, fields ...zapcore.Field) {
+	logger.Info(msg, fields...)
+}
+func Warn(msg string, fields ...zapcore.Field) {
+	logger.Warn(msg, fields...)
+}
+func Error(msg string, fields ...zapcore.Field) {
+	logger.Error(msg, fields...)
+}
+func DPanic(msg string, fields ...zapcore.Field) {
+	logger.DPanic(msg, fields...)
+}
+func Fatal(msg string, fields ...zapcore.Field) {
+	logger.Fatal(msg, fields...)
+}
+
+func Debugf(msg string, fields ...any) {
+	logger.Sugar().Debugf(msg, fields...)
+}
+func Infof(msg string, fields ...any) {
+	logger.Sugar().Infof(msg, fields...)
+}
+func Warnf(msg string, fields ...any) {
+	logger.Sugar().Warnf(msg, fields...)
+}
+func Errorf(msg string, fields ...any) {
+	logger.Sugar().Errorf(msg, fields...)
+}
+func DPanicf(msg string, fields ...any) {
+	logger.Sugar().DPanicf(msg, fields...)
+}
+func Fatalf(msg string, fields ...any) {
+	logger.Sugar().Fatalf(msg, fields...)
+}

logging/wrapper.go

@@ -0,0 +1,61 @@
+package logging
+
+import (
+	"log"
+	"os"
+
+	"github.com/hduhelp/qqbot-demo/config"
+	"go.uber.org/zap"
+	"go.uber.org/zap/zapcore"
+)
+
+var logger *zap.Logger
+
+func Init() {
+	localLogPath := config.Log().Path
+
+	serverMode := os.Getenv("AIRP_MODE")
+	development := serverMode == "release"
+
+	lv := zapcore.InfoLevel
+	if development {
+		lv = zapcore.DebugLevel
+	}
+
+	logLevel := zap.NewAtomicLevelAt(lv)
+
+	var zc = zap.Config{
+		Level:             logLevel,
+		Development:       development,
+		DisableCaller:     false,
+		DisableStacktrace: false,
+		Sampling:          nil,
+		Encoding:          "json",
+		EncoderConfig: zapcore.EncoderConfig{
+			MessageKey:     "message",
+			LevelKey:       "level",
+			TimeKey:        "time",
+			NameKey:        "name",
+			CallerKey:      "caller",
+			StacktraceKey:  "stacktrace",
+			LineEnding:     zapcore.DefaultLineEnding,
+			EncodeLevel:    zapcore.LowercaseLevelEncoder,
+			EncodeTime:     zapcore.ISO8601TimeEncoder,
+			EncodeDuration: zapcore.StringDurationEncoder,
+			EncodeCaller:   zapcore.ShortCallerEncoder,
+			EncodeName:     zapcore.FullNameEncoder,
+		},
+		OutputPaths:      []string{"stdout", localLogPath},
+		ErrorOutputPaths: []string{"stderr", localLogPath},
+		InitialFields:    map[string]any{"app": "qqbot-paper-airplane"},
+	}
+
+	var err error
+	if logger, err = zc.Build(zap.AddCallerSkip(1)); err != nil {
+		log.Fatal("init logger failed:", err)
+	}
+}
+
+func Logger() *zap.Logger {
+	return logger
+}

logging/zap.go

@@ -0,0 +1,39 @@
+package main
+
+import (
+	"context"
+	"time"
+
+	"github.com/hduhelp/qqbot-demo/config"
+	"github.com/hduhelp/qqbot-demo/handler"
+	"github.com/hduhelp/qqbot-demo/logging"
+	"github.com/tencent-connect/botgo"
+	"github.com/tencent-connect/botgo/token"
+	"github.com/tencent-connect/botgo/websocket"
+	"go.uber.org/zap"
+)
+
+func init() {
+	config.Init()
+	logging.Init()
+}
+
+func main() {
+	conf := config.Bot()
+	token := token.BotToken(conf.AppId, conf.Token)
+	api := botgo.NewSandboxOpenAPI(token).WithTimeout(3 * time.Second)
+	handler.RegisterApi(api)
+
+	botgo.SetLogger(logging.Logger().Sugar())
+
+	ws, err := api.WS(context.Background(), nil, "")
+	if err != nil {
+		logging.Fatalf("%+v, error:%v", ws, err)
+	}
+
+	intent := websocket.RegisterHandlers(handler.Get())
+	intent |= 1 << 25 // 注册群和私聊消息
+	if err := botgo.NewSessionManager().Start(ws, token, &intent); err != nil {
+		logging.Fatal("unexpected exit", zap.Error(err))
+	}
+}