60 lines
1.5 KiB
Go
60 lines
1.5 KiB
Go
|
package cmd
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"fmt"
|
|||
|
|
"log"
|
|||
|
|
"net/http"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"github.com/tencent-connect/botgo"
|
|||
|
|
"github.com/tencent-connect/botgo/dto"
|
|||
|
|
"github.com/tencent-connect/botgo/event"
|
|||
|
|
"github.com/tencent-connect/botgo/interaction/webhook"
|
|||
|
|
"github.com/tencent-connect/botgo/token"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
const (
|
|||
|
|
host_ = "0.0.0.0"
|
|||
|
|
port_ = 9000
|
|||
|
|
path_ = "/qqbot"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func robot() {
|
|||
|
|
|
|||
|
|
credentials := &token.QQBotCredentials{
|
|||
|
|
AppID: "102815288",
|
|||
|
|
AppSecret: "faVQLHD951xurolifcaYWUSQPONMLKKK",
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
log.Println("credentials:", credentials)
|
|||
|
|
tokenSource := token.NewQQBotTokenSource(credentials)
|
|||
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|||
|
|
defer cancel() //释放刷新协程
|
|||
|
|
if err := token.StartRefreshAccessToken(ctx, tokenSource); err != nil {
|
|||
|
|
log.Fatalln(err)
|
|||
|
|
}
|
|||
|
|
// 初始化 openapi，正式环境
|
|||
|
|
botgo.NewOpenAPI(credentials.AppID, tokenSource).WithTimeout(5 * time.Second).SetDebug(true)
|
|||
|
|
|
|||
|
|
// 注册处理函数
|
|||
|
|
_ = event.RegisterHandlers(
|
|||
|
|
|
|||
|
|
GuildDirectMessageHandler(),
|
|||
|
|
)
|
|||
|
|
http.HandleFunc(path_, func(writer http.ResponseWriter, request *http.Request) {
|
|||
|
|
webhook.HTTPHandler(writer, request, credentials)
|
|||
|
|
})
|
|||
|
|
if err := http.ListenAndServe(fmt.Sprintf("%s:%d", host_, port_), nil); err != nil {
|
|||
|
|
log.Fatal("setup server fatal:", err)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GuildDirectMessageHandler 处理频道私信事件
|
|||
|
|
func GuildDirectMessageHandler() event.DirectMessageEventHandler {
|
|||
|
|
return func(event *dto.WSPayload, data *dto.WSDirectMessageData) error {
|
|||
|
|
fmt.Println(data)
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
}
|