From f912fe5742cf30d93851ad75a0dba9c2ebb05706 Mon Sep 17 00:00:00 2001 From: JustAnotherID Date: Thu, 19 Oct 2023 08:04:45 -0500 Subject: [PATCH] =?UTF-8?q?feat(red-protocol):=20red=20=E5=8D=8F=E8=AE=AE?= =?UTF-8?q?=E7=9A=84=E9=80=82=E9=85=8D=20(#341)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit * feat(red-protocol): red 协议的简单适配 * chore: make lint happy --- api/api_bind.go | 1 + api/conn.go | 28 ++ dice/im_session.go | 18 +- dice/platform_adapter_qq.go | 2 +- dice/platform_adapter_red.go | 887 +++++++++++++++++++++++++++++++++++ dice/qq_helper.go | 48 +- dice/red_helper.go | 36 ++ dice/utils_email.go | 8 +- main.go | 3 + tray_windows.go | 16 +- 10 files changed, 1019 insertions(+), 28 deletions(-) create mode 100644 dice/platform_adapter_red.go create mode 100644 dice/red_helper.go diff --git a/api/api_bind.go b/api/api_bind.go index 71ba99e6..aad9b4a8 100644 --- a/api/api_bind.go +++ b/api/api_bind.go @@ -344,6 +344,7 @@ func Bind(e *echo.Echo, _myDice *dice.DiceManager) { e.POST(prefix+"/im_connections/addDodo", ImConnectionsAddDodo) e.POST(prefix+"/im_connections/addWalleQ", ImConnectionsAddWalleQ) e.POST(prefix+"/im_connections/addGocqSeparate", ImConnectionsAddGocqSeparate) + e.POST(prefix+"/im_connections/addRed", ImConnectionsAddRed) e.POST(prefix+"/im_connections/del", ImConnectionsDel) e.POST(prefix+"/im_connections/set_enable", ImConnectionsSetEnable) e.POST(prefix+"/im_connections/set_data", ImConnectionsSetData) diff --git a/api/conn.go b/api/conn.go index dba499ec..9dfc655c 100644 --- a/api/conn.go +++ b/api/conn.go @@ -637,3 +637,31 @@ func ImConnectionsAddGocqSeparate(c echo.Context) error { } return c.String(430, "") } + +func ImConnectionsAddRed(c echo.Context) error { + if !doAuth(c) { + return c.JSON(http.StatusForbidden, nil) + } + if dm.JustForTest { + return Success(&c, Response{"testMode": true}) + } + + v := struct { + Host string `yaml:"host" json:"host"` + Port int `yaml:"port" json:"port"` + Token string `yaml:"token" json:"token"` + }{} + err := c.Bind(&v) + if err == nil { + conn := dice.NewRedConnItem(v.Host, v.Port, v.Token) + conn.Session = myDice.ImSession + pa := conn.Adapter.(*dice.PlatformAdapterRed) + pa.Session = myDice.ImSession + myDice.ImSession.EndPoints = append(myDice.ImSession.EndPoints, conn) + myDice.LastUpdatedTime = time.Now().Unix() + myDice.Save(false) + go dice.ServeRed(myDice, conn) + return Success(&c, Response{}) + } + return c.String(430, "") +} diff --git a/dice/im_session.go b/dice/im_session.go index 8b748d19..db017b8d 100644 --- a/dice/im_session.go +++ b/dice/im_session.go @@ -230,7 +230,7 @@ type EndPointInfoBase struct { Platform string `yaml:"platform" json:"platform" jsbind:"platform"` // 平台，如QQ等 RelWorkDir string `yaml:"relWorkDir" json:"relWorkDir"` // 工作目录 Enable bool `yaml:"enable" json:"enable" jsbind:"enable"` // 是否启用 - ProtocolType string `yaml:"protocolType"` // 协议类型，如onebot、koishi等 + ProtocolType string `yaml:"protocolType" json:"protocolType"` // 协议类型，如onebot、koishi等 IsPublic bool `yaml:"isPublic"` Session *IMSession `yaml:"-" json:"-"` @@ -274,6 +274,15 @@ func (ep *EndPointInfo) UnmarshalYAML(value *yaml.Node) error { Adapter *PlatformAdapterWalleQ `yaml:"adapter"` } + err = value.Decode(&val) + if err != nil { + return err + } + ep.Adapter = val.Adapter + case "red": + var val struct { + Adapter *PlatformAdapterRed `yaml:"adapter"` + } err = value.Decode(&val) if err != nil { return err @@ -407,7 +416,7 @@ type MsgContext struct { } // func (s *IMSession) GroupEnableCheck(ep *EndPointInfo, msg *Message, runInSync bool) { -//} +// } // fillPrivilege 填写MsgContext中的权限字段, 并返回填写的权限等级 // - msg 使用其中的msg.Sender.GroupRole @@ -1162,6 +1171,11 @@ func (ep *EndPointInfo) AdapterSetup() { pa.Session = ep.Session pa.EndPoint = ep } + if ep.ProtocolType == "red" { + pa := ep.Adapter.(*PlatformAdapterRed) + pa.Session = ep.Session + pa.EndPoint = ep + } case "DISCORD": pa := ep.Adapter.(*PlatformAdapterDiscord) pa.Session = ep.Session diff --git a/dice/platform_adapter_qq.go b/dice/platform_adapter_qq.go index 29765209..6f890b3a 100644 --- a/dice/platform_adapter_qq.go +++ b/dice/platform_adapter_qq.go @@ -855,7 +855,7 @@ func (pa *PlatformAdapterGocq) Serve() int { socket.OnDisconnected = func(err error, socket gowebsocket.Socket) { log.Info("onebot 服务的连接被对方关闭 ") - _ = pa.Session.Parent.SendMail("", MailTypeOnebotClose) + _ = pa.Session.Parent.SendMail("", MailTypeConnectClose) pa.InPackGoCqhttpDisconnectedCH <- 1 } diff --git a/dice/platform_adapter_red.go b/dice/platform_adapter_red.go new file mode 100644 index 00000000..17b91373 --- /dev/null +++ b/dice/platform_adapter_red.go @@ -0,0 +1,887 @@ +package dice + +import ( + "encoding/json" + "fmt" + "io" + "net/http" + "net/url" + "os" + "os/signal" + "path/filepath" + "strconv" + "strings" + "time" + + "github.com/gorilla/websocket" +) + +type PlatformAdapterRed struct { + Session *IMSession `yaml:"-" json:"-"` + EndPoint *EndPointInfo `yaml:"-" json:"-"` + DiceServing bool `yaml:"-"` // 是否正在连接中 + + Host string `yaml:"host" json:"host"` + Port int `yaml:"port" json:"port"` + Token string `yaml:"token" json:"token"` + RedVersion string `yaml:"-" json:"redVersion"` + + wsUrl *url.URL + httpUrl *url.URL + + conn *websocket.Conn +} + +type RedPack[T interface{}] struct { + Type string `json:"type"` + Payload *T `json:"payload"` +} + +type RedConnectReq struct { + Token string `json:"token"` +} + +type RedConnectResp struct { + Version string `json:"version"` + Name string `json:"name"` + AuthData *struct { + Account string `json:"account"` + MainAccount string `json:"mainAccount"` + Uin string `json:"uin"` + Uid string `json:"uid"` + NickName string `json:"nickName"` + Gender int `json:"gender"` + Age int `json:"age"` + FaceUrl string `json:"faceUrl"` + A2 string `json:"a2"` + D2 string `json:"d2"` + D2key string `json:"d2Key"` + } `json:"authData"` +} + +type RedMessageRecv []*RedMessage + +type RedChatType int + +const ( + PersonChat RedChatType = 1 + GroupChat RedChatType = 2 +) + +type RedPeer struct { + ChatType RedChatType `json:"chatType"` // Group: 2 + PeerUin string `json:"peerUin"` + GuildId string `json:"guildId,omitempty"` // 一直为 Null +} + +type RedMessageSend struct { + Peer *RedPeer `json:"peer"` + Elements []*RedElement `json:"elements"` +} + +type RedMessage struct { + MsgID string `json:"msgId"` + MsgRandom string `json:"msgRandom"` + MsgSeq string `json:"msgSeq"` + CntSeq string `json:"cntSeq"` + ChatType int `json:"chatType"` + MsgType int `json:"msgType"` + SubMsgType int `json:"subMsgType"` + SendType int `json:"sendType"` + SenderUid string `json:"senderUid"` + PeerUid string `json:"peerUid"` + ChannelId string `json:"channelId"` + GuildId string `json:"guildId"` + GuildCode string `json:"guildCode"` + FromUid string `json:"fromUid"` + FromAppid string `json:"fromAppid"` + MsgTime string `json:"msgTime"` + MsgMeta string `json:"msgMeta"` + SendStatus int `json:"sendStatus"` + SendMemberName string `json:"sendMemberName"` + SendNickName string `json:"sendNickName"` + GuildName string `json:"guildName"` + ChannelName string `json:"channelName"` + Elements []*RedElement `json:"elements"` + Records []interface{} `json:"records"` + EmojiLikesList []interface{} `json:"emojiLikesList"` + CommentCnt string `json:"commentCnt"` + DirectMsgFlag int `json:"directMsgFlag"` + DirectMsgMembers []interface{} `json:"directMsgMembers"` + PeerName string `json:"peerName"` + Editable bool `json:"editable"` + AvatarMeta string `json:"avatarMeta"` + AvatarPendant string `json:"avatarPendant"` + FeedId string `json:"feedId"` + RoleId string `json:"roleId"` + TimeStamp string `json:"timeStamp"` + IsImportMsg bool `json:"isImportMsg"` + AtType int `json:"atType"` + RoleType int `json:"roleType"` + FromChannelRoleInfo *RedRoleInfo `json:"fromChannelRoleInfo"` + FromGuildRoleInfo *RedRoleInfo `json:"fromGuildRoleInfo"` + LevelRoleInfo *RedRoleInfo `json:"levelRoleInfo"` + RecallTime string `json:"recallTime"` + IsOnlineMsg bool `json:"isOnlineMsg"` + GeneralFlags string `json:"generalFlags"` + ClientSeq string `json:"clientSeq"` + NameType int `json:"nameType"` + AvatarFlag int `json:"avatarFlag"` + SenderUin string `json:"senderUin"` + PeerUin string `json:"peerUin"` +} + +type RedElement struct { + ElementType int `json:"elementType,omitempty"` + ElementId string `json:"elementId,omitempty"` + ExtBufForUI string `json:"extBufForUI,omitempty"` + PicElement *RedPicElement `json:"picElement,omitempty"` + TextElement *RedTextElement `json:"textElement,omitempty"` + ArkElement interface{} `json:"arkElement,omitempty"` + AvRecordElement interface{} `json:"avRecordElement,omitempty"` + CalendarElement interface{} `json:"calendarElement,omitempty"` + FaceElement interface{} `json:"faceElement,omitempty"` + FileElement interface{} `json:"fileElement,omitempty"` + GiphyElement interface{} `json:"giphyElement,omitempty"` + GrayTipElement *struct { + XmlElement *RedXMLElement `json:"xmlElement,omitempty"` + AioOpGrayTipElement interface{} `json:"aioOpGrayTipElement,omitempty"` + BlockGrayTipElement interface{} `json:"blockGrayTipElement,omitempty"` + BuddyElement interface{} `json:"buddyElement,omitempty"` + BuddyNotifyElement interface{} `json:"buddyNotifyElement,omitempty"` + EmojiReplyElement interface{} `json:"emojiReplyElement,omitempty"` + EssenceElement interface{} `json:"essenceElement,omitempty"` + FeedMsgElement interface{} `json:"feedMsgElement,omitempty"` + FileReceiptElement interface{} `json:"fileReceiptElement,omitempty"` + GroupElement interface{} `json:"groupElement,omitempty"` + GroupNotifyElement interface{} `json:"groupNotifyElement,omitempty"` + JsonGrayTipElement interface{} `json:"jsonGrayTipElement,omitempty"` + LocalGrayTipElement interface{} `json:"localGrayTipElement,omitempty"` + ProclamationElement interface{} `json:"proclamationElement,omitempty"` + RevokeElement interface{} `json:"revokeElement,omitempty"` + SubElementType interface{} `json:"subElementType,omitempty"` + } `json:"grayTipElement,omitempty"` + InlineKeyboardElement interface{} `json:"inlineKeyboardElement,omitempty"` + LiveGiftElement interface{} `json:"liveGiftElement,omitempty"` + MarkdownElement interface{} `json:"markdownElement,omitempty"` + MarketFaceElement interface{} `json:"marketFaceElement,omitempty"` + MultiForwardMsgElement interface{} `json:"multiForwardMsgElement,omitempty"` + PttElement interface{} `json:"pttElement,omitempty"` + ReplyElement interface{} `json:"replyElement,omitempty"` + StructLongMsgElement interface{} `json:"structLongMsgElement,omitempty"` + TextGiftElement interface{} `json:"textGiftElement,omitempty"` + VideoElement interface{} `json:"videoElement,omitempty"` + WalletElement interface{} `json:"walletElement,omitempty"` + YoloGameResultElement interface{} `json:"yoloGameResultElement,omitempty"` +} + +type RedXMLElement struct { + BusiType string `json:"busiType,omitempty"` + BusiId string `json:"busiId,omitempty"` + C2cType int `json:"c2CType,omitempty"` + ServiceType int `json:"serviceType,omitempty"` + CtrlFlag int `json:"ctrlFlag,omitempty"` + Content string `json:"content,omitempty"` + TemplId string `json:"templId,omitempty"` + SeqId string `json:"seqId,omitempty"` + TemplParam any `json:"templParam,omitempty"` + PbReserv string `json:"pbReserv,omitempty"` + Members any `json:"members,omitempty"` +} + +type RedPicElement struct { + PicSubType int `json:"picSubType,omitempty"` + FileName string `json:"fileName,omitempty"` + FileSize string `json:"fileSize,omitempty"` + PicWidth int `json:"picWidth,omitempty"` + PicHeight int `json:"picHeight,omitempty"` + Original bool `json:"original,omitempty"` + Md5HexStr string `json:"md5HexStr,omitempty"` + SourcePath string `json:"sourcePath,omitempty"` + ThumbPath *RedThumbPath `json:"thumbPath,omitempty"` + TransferStatus int `json:"transferStatus,omitempty"` + Progress int `json:"progress,omitempty"` + PicType int `json:"picType,omitempty"` + InvalidState int `json:"invalidState,omitempty"` + FileUuid string `json:"fileUuid,omitempty"` + FileSubId string `json:"fileSubId,omitempty"` + ThumbFileSize int `json:"thumbFileSize,omitempty"` + Summary string `json:"summary,omitempty"` + EmojiAd *RedEmojiAd `json:"emojiAd,omitempty"` + EmojiMall *RedEmojiMall `json:"emojiMall,omitempty"` + EmojiZplan *RedEmojiZplan `json:"emojiZplan,omitempty"` + OriginImageUrl string `json:"originImageUrl,omitempty"` +} + +type RedEmojiAd struct { + Url string `json:"url,omitempty"` + Desc string `json:"desc,omitempty"` +} + +type RedEmojiMall struct { + PackageId int `json:"packageId,omitempty"` + EmojiId int `json:"emojiId,omitempty"` +} + +type RedEmojiZplan struct { + ActionId int `json:"actionId,omitempty"` + ActionName string `json:"actionName,omitempty"` + ActionType int `json:"actionType,omitempty"` + PlayerNumber int `json:"playerNumber,omitempty"` + PeerUid string `json:"peerUid,omitempty"` + BytesReserveInfo string `json:"bytesReserveInfo,omitempty"` +} + +type RedThumbPath struct { +} + +type RedTextElement struct { + Content string `json:"content,omitempty"` + AtType int `json:"atType,omitempty"` + AtUid string `json:"atUid,omitempty"` + AtTinyId string `json:"atTinyId,omitempty"` + AtNtUid string `json:"atNtUid,omitempty"` + SubElementType int `json:"subElementType,omitempty"` + AtChannelId string `json:"atChannelId,omitempty"` + AtRoleId string `json:"atRoleId,omitempty"` + AtRoleColor int `json:"atRoleColor,omitempty"` + AtRoleName string `json:"atRoleName,omitempty"` + NeedNotify int `json:"needNotify,omitempty"` + AtNtUin string `json:"atNtUin,omitempty"` +} + +type RedRoleInfo struct { + RoleId string `json:"roleId"` + Name string `json:"name"` + Color int `json:"color"` +} + +type Friend struct { + Qid string `json:"qid"` + Uin string `json:"uin"` // QQ 号 + Nick string `json:"nick"` + Remark string `json:"remark"` + LongNick string `json:"longNick"` + AvatarUrl string `json:"avatarUrl"` + Birthday_year int `json:"birthday_Year"` + Birthday_month int `json:"birthday_Month"` + Birthday_day int `json:"birthday_Day"` + Sex int `json:"sex"` // 性别 + TopTime string `json:"topTime"` + IsBlock bool `json:"isBlock"` // 是否拉黑 + IsMsgDisturb bool `json:"isMsgDisturb"` + IsSpecialCareOpen bool `json:"isSpecialCareOpen"` + IsSpecialCareZone bool `json:"isSpecialCareZone"` + RingId string `json:"ringId"` + Status int `json:"status"` + ExtStatus int `json:"extStatus"` + CategoryId int `json:"categoryId"` + OnlyChat bool `json:"onlyChat"` + QzoneNotWatch bool `json:"qzoneNotWatch"` + QzoneNotWatched bool `json:"qzoneNotWatched"` + VipFlag bool `json:"vipFlag"` + YearVipFlag bool `json:"yearVipFlag"` + SvipFlag bool `json:"svipFlag"` + VipLevel int `json:"vipLevel"` + Category string `json:"category"` // 分组信息 +} + +type Group struct { + GroupCode string `json:"groupCode"` // 群号 + MaxMember int `json:"maxMember"` // 最大人数 + MemberCount int `json:"memberCount"` // 成员人数 + GroupName string `json:"groupName"` // 群名 + GroupStatus int `json:"groupStatus"` + MemberRole int `json:"memberRole"` // 群成员角色 + IsTop bool `json:"isTop"` + ToppedTimestamp string `json:"toppedTimestamp"` + PrivilegeFlag int `json:"privilegeFlag"` // 群权限 + IsConf bool `json:"isConf"` + HasModifyConfGroupFace bool `json:"hasModifyConfGroupFace"` + HasModifyConfGroupName bool `json:"hasModifyConfGroupName"` + RemarkName string `json:"remarkName"` + HasMemo bool `json:"hasMemo"` + GroupShutupExpireTime string `json:"groupShutupExpireTime"` + PersonShutupExpireTime string `json:"personShutupExpireTime"` + DiscussToGroupUin string `json:"discussToGroupUin"` + DiscussToGroupMaxMsgSeq int `json:"discussToGroupMaxMsgSeq"` + DiscussToGroupTime int `json:"discussToGroupTime"` +} + +func (pa *PlatformAdapterRed) Serve() int { + ep := pa.EndPoint + s := pa.Session + log := s.Parent.Logger + dm := pa.Session.Parent.Parent + + interrupt := make(chan os.Signal, 1) + signal.Notify(interrupt, os.Interrupt) + + wsUrl := url.URL{ + Scheme: "ws", + Host: fmt.Sprintf("%s:%d", pa.Host, pa.Port), + } + httpUrl := url.URL{ + Scheme: "http", + Host: fmt.Sprintf("%s:%d", pa.Host, pa.Port), + Path: "/api", + } + log.Infof("connecting to %s", wsUrl.String()) + conn, resp, err := websocket.DefaultDialer.Dial(wsUrl.String(), nil) + if err != nil { + log.Error("dial:", err) + pa.EndPoint.State = 3 + return 1 + } + defer resp.Body.Close() + defer func(conn *websocket.Conn) { + _ = conn.Close() + }(conn) + pa.conn = conn + pa.EndPoint.State = 2 + + // 鉴权 + auth := &RedPack[RedConnectReq]{ + Type: "meta::connect", + Payload: &RedConnectReq{pa.Token}, + } + authData, _ := json.Marshal(auth) + err = conn.WriteMessage(websocket.TextMessage, authData) + if err != nil { + log.Error("auth failed:", err) + pa.EndPoint.State = 3 + return 1 + } + _, authRespData, err := conn.ReadMessage() + if err != nil { + log.Error("auth failed:", err) + pa.EndPoint.State = 3 + return 1 + } + var authResp RedPack[RedConnectResp] + err = json.Unmarshal(authRespData, &authResp) + if err != nil { + log.Error("auth failed:", err) + pa.EndPoint.State = 3 + return 1 + } + log.Debugf("red auth resp:%+v", authResp) + + pa.wsUrl = &wsUrl + pa.httpUrl = &httpUrl + pa.RedVersion = authResp.Payload.Version + pa.EndPoint.State = 1 + + // 获得用户信息 + botInfo := pa.getBotInfo() + ep.Nickname = botInfo.Name + ep.UserID = botInfo.SelfId + d := pa.Session.Parent + d.LastUpdatedTime = time.Now().Unix() + d.Save(false) + pa.Session.Parent.Logger.Infof("red 连接成功，账号<%s>(%s)", pa.EndPoint.Nickname, pa.EndPoint.UserID) + + // 获得好友列表 + refreshFriends := func() { + friends := pa.getFriends() + for _, friend := range friends { + dm.UserNameCache.Set(friend.Uin, &GroupNameCacheItem{ + Name: friend.Nick, + time: time.Now().Unix(), + }) + } + } + go refreshFriends() + // 获得群列表 + pa.GetGroupInfoAsync("") + + done := make(chan struct{}) + + go func() { + defer close(done) + for { + msgType, msgData, err := conn.ReadMessage() + if err != nil { + interrupt <- os.Interrupt + return + } + switch msgType { + case websocket.TextMessage: + log.Debugf("message=%s", msgData) + + var msgRowMap map[string]interface{} + err := json.Unmarshal(msgData, &msgRowMap) + if err != nil { + log.Errorf("recv parse error: %s, rowData: %s", err, msgData) + } + if msgType, ok := msgRowMap["type"]; ok && msgType != "message::send::reply" { + var msgRow RedPack[RedMessageRecv] + _ = json.Unmarshal(msgData, &msgRow) + log.Debug("recv: %+v", msgRow) + for _, msg := range *msgRow.Payload { + pa.Session.Execute(pa.EndPoint, pa.decodeMessage(msg), false) + } + } + case websocket.BinaryMessage: + case websocket.CloseMessage: + log.Debug("server close") + pa.conn = nil + done <- struct{}{} + case websocket.PingMessage: + case websocket.PongMessage: + } + } + }() + + for { + select { + case <-done: + case <-interrupt: + log.Debug("red interrupt") + + _ = conn.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, "")) + _ = pa.conn.Close() + + select { + case <-done: + case <-time.After(time.Second): + } + } + } +} + +func (pa *PlatformAdapterRed) DoRelogin() bool { + pa.Session.Parent.Logger.Infof("正在启用 red 连接……") + pa.EndPoint.State = 0 + pa.EndPoint.Enable = false + if pa.conn != nil { + _ = pa.conn.Close() + } + pa.conn = nil + return pa.Serve() == 0 +} + +func (pa *PlatformAdapterRed) SetEnable(enable bool) { + d := pa.Session.Parent + e := pa.EndPoint + if enable { + e.Enable = true + pa.DiceServing = false + + if pa.conn == nil { + go ServeQQ(d, e) + } + } else { + e.State = 0 + e.Enable = false + if pa.conn != nil { + _ = pa.conn.Close() + pa.conn = nil + } + } + d.LastUpdatedTime = time.Now().Unix() + d.Save(false) +} + +func (pa *PlatformAdapterRed) QuitGroup(_ *MsgContext, id string) { + log := pa.Session.Parent.Logger + log.Warnf("red: 尝试退出群组(%s)，但尚不支持该功能", id) +} + +func (pa *PlatformAdapterRed) SendToPerson(ctx *MsgContext, uid string, text string, flag string) { + rowId, chatType := pa.mustExtractId(uid) + if chatType != PersonChat { + return + } + + for _, i := range ctx.Dice.ExtList { + if i.OnMessageSend != nil { + i.callWithJsCheck(ctx.Dice, func() { + i.OnMessageSend(ctx, &Message{ + Message: text, + MessageType: "private", + Platform: pa.EndPoint.Platform, + Sender: SenderBase{ + Nickname: pa.EndPoint.Nickname, + UserID: pa.EndPoint.UserID, + }, + }, + flag) + }) + } + } + + texts := textSplit(text) + for _, subText := range texts { + doSleepQQ(ctx) + pa.sendRow(&RedMessageSend{ + Peer: &RedPeer{ + ChatType: PersonChat, + PeerUin: strconv.FormatInt(rowId, 10), + }, + Elements: []*RedElement{{ + ElementType: 1, + TextElement: &RedTextElement{Content: subText}, + }}, + }) + } +} + +func (pa *PlatformAdapterRed) SendToGroup(ctx *MsgContext, groupId string, text string, flag string) { + rowId, chatType := pa.mustExtractId(groupId) + if chatType != GroupChat { + return + } + + if ctx.Session.ServiceAtNew[groupId] != nil { + for _, i := range ctx.Session.ServiceAtNew[groupId].ActivatedExtList { + if i.OnMessageSend != nil { + i.callWithJsCheck(ctx.Dice, func() { + i.OnMessageSend(ctx, &Message{ + Message: text, + MessageType: "group", + Platform: pa.EndPoint.Platform, + GroupID: groupId, + Sender: SenderBase{ + Nickname: pa.EndPoint.Nickname, + UserID: pa.EndPoint.UserID, + }, + }, flag) + }) + } + } + } + + texts := textSplit(text) + for _, subText := range texts { + doSleepQQ(ctx) + pa.sendRow(&RedMessageSend{ + Peer: &RedPeer{ + ChatType: GroupChat, + PeerUin: strconv.FormatInt(rowId, 10), + }, + Elements: []*RedElement{{ + ElementType: 1, + TextElement: &RedTextElement{Content: subText}, + }}, + }) + } +} + +func (pa *PlatformAdapterRed) sendRow(redMsg *RedMessageSend) { + if pa.conn != nil { + log := pa.Session.Parent.Logger + conn := pa.conn + + param := &RedPack[RedMessageSend]{ + Type: "message::send", + Payload: redMsg, + } + data, _ := json.Marshal(param) + err := conn.WriteMessage(websocket.TextMessage, data) + if err != nil { + log.Errorf("send msg failed: %s", err.Error()) + } + } +} + +func (pa *PlatformAdapterRed) mustExtractId(id string) (int64, RedChatType) { + if strings.HasPrefix(id, "QQ:") { + num, _ := strconv.ParseInt(id[len("QQ:"):], 10, 64) + return num, PersonChat + } + if strings.HasPrefix(id, "QQ-Group:") { + num, _ := strconv.ParseInt(id[len("QQ-Group:"):], 10, 64) + return num, GroupChat + } + return 0, 0 +} + +func (pa *PlatformAdapterRed) SetGroupCardName(groupId string, userId string, name string) { + log := pa.Session.Parent.Logger + log.Warn("red: 尚未实现该功能") +} + +func (pa *PlatformAdapterRed) SendFileToPerson(ctx *MsgContext, uid string, path string, flag string) { + dice := pa.Session.Parent + fileElement, err := dice.FilepathToFileElement(path) + if err == nil { + pa.SendToPerson(ctx, uid, fmt.Sprintf("[尝试发送文件: %s，但不支持]", fileElement.File), flag) + } else { + pa.SendToPerson(ctx, uid, fmt.Sprintf("[尝试发送文件出错: %s]", err.Error()), flag) + } +} + +func (pa *PlatformAdapterRed) SendFileToGroup(ctx *MsgContext, uid string, path string, flag string) { + dice := pa.Session.Parent + fileElement, err := dice.FilepathToFileElement(path) + if err == nil { + pa.SendToGroup(ctx, uid, fmt.Sprintf("[尝试发送文件: %s，但不支持]", fileElement.File), flag) + } else { + pa.SendToGroup(ctx, uid, fmt.Sprintf("[尝试发送文件出错: %s]", err.Error()), flag) + } +} + +func (pa *PlatformAdapterRed) MemberBan(_ string, _ string, _ int64) {} + +func (pa *PlatformAdapterRed) MemberKick(_ string, _ string) {} + +func (pa *PlatformAdapterRed) GetGroupInfoAsync(_ string) { + // 触发更新群信息 + dm := pa.Session.Parent.Parent + ep := pa.EndPoint + s := pa.Session + session := s + + refresh := func() { + groups := pa.getGroups() + for _, group := range groups { + if group != nil { + groupId := group.GroupCode + dm.GroupNameCache.Set(groupId, &GroupNameCacheItem{ + Name: group.GroupName, + time: time.Now().Unix(), + }) + + groupRecord := session.ServiceAtNew[groupId] + if groupRecord != nil { + if group.MemberCount == 0 { + diceId := ep.UserID + if _, exists := groupRecord.DiceIDExistsMap.Load(diceId); exists { + // 不在群里了，更新信息 + groupRecord.DiceIDExistsMap.Delete(diceId) + groupRecord.UpdatedAtTime = time.Now().Unix() + } + } else if groupRecord.GroupName != group.GroupName { + // 更新群名 + groupRecord.GroupName = group.GroupName + groupRecord.UpdatedAtTime = time.Now().Unix() + } + + // TODO: 处理被强制拉群的情况 + } + } + } + } + go refresh() +} + +type BotInfo struct { + SelfId string + Name string +} + +func (pa *PlatformAdapterRed) getBotInfo() *BotInfo { + data, _ := pa.httpDo("GET", "getSelfProfile", nil, nil) + var body map[string]interface{} + _ = json.Unmarshal(data, &body) + + return &BotInfo{ + SelfId: formatDiceIDRed(body["uin"].(string)), + Name: body["nick"].(string), + } +} + +func (pa *PlatformAdapterRed) getFriends() []*Friend { + data, _ := pa.httpDo("GET", "bot/friends", nil, nil) + var friends []*Friend + _ = json.Unmarshal(data, &friends) + return friends +} + +func (pa *PlatformAdapterRed) getGroups() []*Group { + data, _ := pa.httpDo("GET", "bot/groups", nil, nil) + var groups []*Group + _ = json.Unmarshal(data, &groups) + return groups +} + +// func (pa *PlatformAdapterRed) getMemberList(group int, size int) { +// paramData, _ := json.Marshal(map[string]int{ +// "group": group, +// "size": size, +// }) +// data, _ := pa.httpDo("POST", "group/getMemberList", bytes.NewBuffer(paramData)) +// var body map[string]interface{} +// _ = json.Unmarshal(data, &body) +// } + +type RedRichMediaReq struct { +} + +// func (pa *PlatformAdapterRed) getFile(msgID string, chatType RedChatType, peerUid string, elementID string) { +// paramData, _ := json.Marshal(map[string]interface{}{ +// "msgId": msgID, +// "chatType": chatType, +// "peerUid": peerUid, +// "elementId": elementID, +// "thumbSize": 0, +// "downloadType": 2, +// }) +// data, _ := pa.httpDo("POST", "message/fetchRichMedia", nil, bytes.NewBuffer(paramData)) +// var body map[string]interface{} +// _ = json.Unmarshal(data, &body) +// } +// +// func (pa *PlatformAdapterRed) uploadFile(path string) *RedElement { +// _, err := os.Stat(path) +// if errors.Is(os.ErrNotExist, err) { +// return nil +// } +// +// file, _ := os.Open(path) +// body := &bytes.Buffer{} +// writer := multipart.NewWriter(body) +// part, _ := writer.CreateFormFile("file", filepath.Base(path)) +// _, _ = io.Copy(part, file) +// _ = writer.Close() +// +// data, _ := pa.httpDo("POST", "upload", map[string]string{"Content-Type": writer.FormDataContentType()}, body) +// var resp RedElement +// _ = json.Unmarshal(data, &resp) +// return &resp +// } + +func (pa *PlatformAdapterRed) httpDo(method, action string, headers map[string]string, body io.Reader) ([]byte, error) { + client := http.Client{} + request, _ := http.NewRequest(method, pa.httpUrl.String()+"/"+action, body) + request.Header.Add("Authorization", "Bearer "+pa.Token) + if len(headers) != 0 { + for key, value := range headers { + request.Header.Add(key, value) + } + } + resp, err := client.Do(request) + if err != nil { + return nil, err + } + defer resp.Body.Close() + + return io.ReadAll(resp.Body) +} + +func (pa *PlatformAdapterRed) decodeMessage(message *RedMessage) *Message { + log := pa.Session.Parent.Logger + msg := new(Message) + if t, err := strconv.ParseInt(message.MsgTime, 10, 64); err == nil { + msg.Time = t + } else { + log.Errorf("red 消息 msgTime 解析错误，err=%s, row str=%s", err, message.MsgTime) + } + msg.RawID = message.MsgID + var content string + uid := message.SenderUin + if uid == "" { + uid = message.SenderUid + } + for _, element := range message.Elements { + switch element.ElementType { + case 1: + // 文本 + switch element.TextElement.AtType { + case 1: + // at 全体 + content += "[CQ:at,qq=all]" + case 2: + // at 某人 + id := element.TextElement.AtNtUin + if id == "" { + id = element.TextElement.AtUid + } + content += fmt.Sprintf("[CQ:at,qq=%s]", id) + default: + content += element.TextElement.Content + } + case 2: + // 图片 + u := element.PicElement.OriginImageUrl + id := uid + if id == "" { + id = message.SenderUid + } + var gid string + if message.ChatType == int(GroupChat) { + gid = message.PeerUin + if gid == "" { + gid = message.PeerUin + } + } + fUuid := element.PicElement.FileUuid + + if u == "" { + if message.ChatType == int(GroupChat) { + content += fmt.Sprintf( + "[CQ:image,file=%s,subType=0,url=https://gchat.qpic.cn/gchatpic_new/%s/%s-%s-%s/0]", + filepath.Base(element.PicElement.SourcePath), + id, + gid, + fUuid, + strings.ToUpper(element.PicElement.Md5HexStr), + ) + } else { + content += fmt.Sprintf( + "[CQ:image,file=%s,url=https://c2cpicdw.qpic.cn/offpic_new/%s/%s/0]", + filepath.Base(element.PicElement.SourcePath), + id, + fUuid, + ) + } + } else if strings.Contains(u, "rkey") { + // TODO: 下载图片 + // content += fmt.Sprintf("[CQ:image,file=///%s,type=show,id=40000]", url) + } else { + content += fmt.Sprintf("[CQ:image,file=https://c2cpicdw.qpic.cn%s,type=show,id=40000]", u) + } + case 4: + // TODO: 语音 + case 6: + // 表情 + faceElement := element.FaceElement.(map[string]interface{}) + faceIndex := faceElement["faceIndex"].(float64) + content += fmt.Sprintf("[CQ:face,id=%d]", int(faceIndex)) + case 7: + // TODO: 引用 + } + } + msg.Message = content + msg.Platform = "QQ" + + send := SenderBase{} + send.UserID = formatDiceIDRed(uid) + if message.ChatType == 1 { + // 私聊消息 + msg.MessageType = "private" + dm := pa.Session.Parent.Parent + if nick, ok := dm.UserNameCache.Get(uid); ok { + nameInfo := nick.(*GroupNameCacheItem) + send.Nickname = nameInfo.Name + } + if send.Nickname == "" { + send.Nickname = "<未知用户>" + } + } else { + msg.MessageType = "group" + msg.GroupID = formatDiceIDRedGroup(message.PeerUid) + send.Nickname = message.SendNickName + if send.Nickname == "" { + send.Nickname = message.SendMemberName + } + // send.GroupRole = message.RoleType + if message.SendNickName != "" { + send.Nickname = message.SendNickName + } else if message.SendMemberName != "" { + send.Nickname = message.SendMemberName + } + } + msg.Sender = send + + return msg +} + +func formatDiceIDRed(diceRed string) string { + return fmt.Sprintf("QQ:%s", diceRed) +} + +func formatDiceIDRedGroup(diceRed string) string { + return fmt.Sprintf("QQ-Group:%s", diceRed) +} diff --git a/dice/qq_helper.go b/dice/qq_helper.go index d6cb1617..fd151479 100644 --- a/dice/qq_helper.go +++ b/dice/qq_helper.go @@ -35,7 +35,6 @@ func ServeQQ(d *Dice, ep *EndPointInfo) { return false } - // lastRetryTime := time.Now().Unix() waitTimes := 0 for { if checkQuit() { @@ -46,11 +45,6 @@ func ServeQQ(d *Dice, ep *EndPointInfo) { d.Logger.Infof("开始连接 onebot 服务，帐号 <%s>(%s)，重试计数[%d/%d]", ep.Nickname, ep.UserID, waitTimes, 5) ret := ep.Adapter.Serve() - // if time.Now().Unix()-lastRetryTime > 8*60 { - // lastRetryTime = 0 - // } - // lastRetryTime = time.Now().Unix() - if ret == 0 { break } @@ -66,7 +60,6 @@ func ServeQQ(d *Dice, ep *EndPointInfo) { break } - // d.Logger.Infof("onebot 连接失败[%d/%d]，将在15秒后重新连接，帐号 <%s>(%s)", waitTimes, 5, ep.Nickname, ep.UserId) time.Sleep(15 * time.Second) } } @@ -101,7 +94,6 @@ func ServeQQ(d *Dice, ep *EndPointInfo) { return false } - // lastRetryTime := time.Now().Unix() waitTimes := 0 for { if checkQuit() { @@ -112,11 +104,6 @@ func ServeQQ(d *Dice, ep *EndPointInfo) { d.Logger.Infof("开始连接 onebot 服务，帐号 <%s>(%s)，重试计数[%d/%d]", ep.Nickname, ep.UserID, waitTimes, 5) ret := ep.Adapter.Serve() - // if time.Now().Unix()-lastRetryTime > 8*60 { - // lastRetryTime = 0 - // } - // lastRetryTime = time.Now().Unix() - if ret == 0 { break } @@ -135,4 +122,39 @@ func ServeQQ(d *Dice, ep *EndPointInfo) { time.Sleep(15 * time.Second) } } + + if ep.Platform == "QQ" && ep.ProtocolType == "red" { + conn := ep.Adapter.(*PlatformAdapterRed) + + if !conn.DiceServing { + conn.DiceServing = true + } else { + return + } + + ep.Enable = true + ep.State = 2 // 连接中 + d.LastUpdatedTime = time.Now().Unix() + d.Save(false) + waitTimes := 0 + + for { + // 骰子开始连接 + d.Logger.Infof("开始连接 red 服务，帐号 <%s>(%s)，重试计数[%d/%d]", ep.Nickname, ep.UserID, waitTimes, 5) + ret := ep.Adapter.Serve() + + if ret == 0 { + break + } + + waitTimes += 1 + if waitTimes > 5 { + d.Logger.Infof("red 连接重试次数过多，先行中断: <%s>(%s)", ep.Nickname, ep.UserID) + conn.DiceServing = false + break + } + + time.Sleep(15 * time.Second) + } + } } diff --git a/dice/red_helper.go b/dice/red_helper.go new file mode 100644 index 00000000..bf6043d4 --- /dev/null +++ b/dice/red_helper.go @@ -0,0 +1,36 @@ +package dice + +import ( + "time" + + "github.com/google/uuid" +) + +func NewRedConnItem(host string, port int, token string) *EndPointInfo { + conn := new(EndPointInfo) + conn.ID = uuid.New().String() + conn.Platform = "QQ" + conn.ProtocolType = "red" + conn.Enable = false + conn.RelWorkDir = "extra/red-" + conn.ID + conn.Adapter = &PlatformAdapterRed{ + EndPoint: conn, + Host: host, + Port: port, + Token: token, + } + return conn +} + +func ServeRed(d *Dice, ep *EndPointInfo) { + defer CrashLog() + conn := ep.Adapter.(*PlatformAdapterRed) + d.Logger.Infof("red 尝试连接") + if conn.Serve() == 0 { + } else { + d.Logger.Errorf("连接 red 服务失败") + ep.State = 3 + d.LastUpdatedTime = time.Now().Unix() + d.Save(false) + } +} diff --git a/dice/utils_email.go b/dice/utils_email.go index 15695941..f23b0835 100644 --- a/dice/utils_email.go +++ b/dice/utils_email.go @@ -10,8 +10,8 @@ import ( type MailCode int const ( - // MailTypeOnebotClose 掉线 - MailTypeOnebotClose MailCode = iota + // MailTypeConnectClose 掉线 + MailTypeConnectClose MailCode = iota // MailTypeCIAMLock 风控 // tx 云把 Customer Identity Access Management 叫做 账号风控平台…… MailTypeCIAMLock // MailTypeNotice 通知 @@ -35,8 +35,8 @@ func (d *Dice) SendMail(body string, m MailCode) error { } sub := "Seal News: " switch m { - case MailTypeOnebotClose: - sub += "Onebot 连接中断" + case MailTypeConnectClose: + sub += "Connect 连接中断" case MailTypeCIAMLock: sub += "Bot 机器人被风控" case MailTypeNotice: diff --git a/main.go b/main.go index 00ea8df1..a46778d5 100644 --- a/main.go +++ b/main.go @@ -468,6 +468,9 @@ func diceServe(d *dice.Dice) { SignServerConfig: pa.SignServerConfig, }) } + if conn.EndPointInfoBase.ProtocolType == "red" { + dice.ServeRed(d, conn) + } time.Sleep(10 * time.Second) // 稍作等待再连接 dice.ServeQQ(d, conn) case "DISCORD": diff --git a/tray_windows.go b/tray_windows.go index afb59921..a9d82692 100644 --- a/tray_windows.go +++ b/tray_windows.go @@ -55,7 +55,7 @@ func CreateMutex(name string) (uintptr, error) { 0, uintptr(unsafe.Pointer(s)), ) - switch int(err.(syscall.Errno)) { + switch int(err.(syscall.Errno)) { //nolint:errorlint case 0: return ret, nil default: @@ -143,7 +143,7 @@ func onReady() { systray.Quit() systrayQuited = true cleanUpCreate(theDM)() - time.Sleep(time.Duration(3 * time.Second)) + time.Sleep(3 * time.Second) os.Exit(0) case <-mAutoBoot.ClickedCh: if mAutoBoot.Checked() { @@ -185,7 +185,7 @@ var _trayPortStr = "3211" func httpServe(e *echo.Echo, dm *dice.DiceManager, hideUI bool) { portStr := "3211" - //runtime.LockOSThread() + // runtime.LockOSThread() go func() { for { @@ -269,10 +269,10 @@ func executeWin(name string, arg ...string) *exec.Cmd { NoInheritHandles: true, } - //cmd.Dir, _ = os.Getwd() - //path, err := os.Executable() - //if err != nil { - // cmd.Dir, _ = filepath.Abs(filepath.Dir(path)) - //} + // cmd.Dir, _ = os.Getwd() + // path, err := os.Executable() + // if err != nil { + // cmd.Dir, _ = filepath.Abs(filepath.Dir(path)) + // } return cmd }