Files
goodBaby/drivers/onebot/tool.go
T

64 lines
1.8 KiB
Go
Raw Normal View History

2026-04-05 10:36:22 +08:00
package onebot
import (
"encoding/json"
2026-07-26 22:19:41 +08:00
"fmt"
2026-04-05 10:36:22 +08:00
napcat "github.com/ssdomei232/napcat-http-go-sdk"
)
2026-09-12 15:32:27 +08:00
// ParseAccountConfig 解析 OneBot 账号配置
func ParseAccountConfig(config string) (*OneBotAccount, error) {
if config == "" {
return nil, fmt.Errorf("解析OneBot账号配置失败: 规则没有关联账号")
}
var account OneBotAccount
if err := json.Unmarshal([]byte(config), &account); err != nil {
return nil, fmt.Errorf("解析OneBot账号配置失败: %v", err)
}
return &account, nil
}
// ParseRuleConfig 解析 OneBot 规则配置
func ParseRuleConfig(configJSON string) (*OneBotConfig, error) {
var config OneBotConfig
if err := json.Unmarshal([]byte(configJSON), &config); err != nil {
return nil, fmt.Errorf("解析OneBot规则配置失败: %v", err)
}
return &config, nil
}
// newClient 创建 OneBot HTTP 客户端
2026-07-26 22:19:41 +08:00
func newClient(account *OneBotAccount) *napcat.Client {
return napcat.NewClient(account.Token, account.URL)
}
2026-09-12 15:32:27 +08:00
// sendGroupMsg 发送一条群消息
2026-07-26 22:19:41 +08:00
func sendGroupMsg(account *OneBotAccount, groupID int64, msg string) error {
return call(account, "/send_group_msg", map[string]any{
"group_id": groupID,
"message": msg,
})
}
2026-04-05 10:36:22 +08:00
2026-09-12 15:32:27 +08:00
// sendPrivateMsg 发送一条私聊消息
2026-07-26 22:19:41 +08:00
func sendPrivateMsg(account *OneBotAccount, userID int64, msg string) error {
return call(account, "/send_private_msg", map[string]any{
"user_id": userID,
"message": msg,
})
}
2026-09-12 15:32:27 +08:00
// call 调用一次 OneBot HTTP API。
//
// SDK 只返回传输层错误,这里自己解析 OneBot 的响应信封,
// 否则 token 错误、群号不存在这类失败会被当成发送成功。
2026-07-26 22:19:41 +08:00
func call(account *OneBotAccount, endpoint string, payload map[string]any) error {
var resp apiResponse
if err := newClient(account).DoRequest("POST", endpoint, payload, &resp); err != nil {
return fmt.Errorf("请求 OneBot 失败: %w", err)
}
return resp.err()
2026-04-05 10:36:22 +08:00
}