get msg

Author: lz1998

service/bot/api_handler.go

@@ -67,9 +67,57 @@ func HandleSendGroupMsg(cli *client.QQClient, req *onebot.SendGroupMsgReq) *oneb
 func HandleDeleteMsg(cli *client.QQClient, req *onebot.DeleteMsgReq) *onebot.DeleteMsgResp {
 	eventInterface, ok := cache.GroupMessageLru.Get(req.MessageId)
 	if !ok {
-		return &onebot.DeleteMsgResp{}
+		return nil
+	}
+	event, ok := eventInterface.(*message.GroupMessage)
+	if !ok {
+		return nil
 	}
-	event := eventInterface.(*message.GroupMessage)
 	cli.RecallGroupMessage(event.GroupCode, event.Id, event.InternalId)
 	return &onebot.DeleteMsgResp{}
 }
+
+func HandleGetMsg(cli *client.QQClient, req *onebot.GetMsgReq) *onebot.GetMsgResp {
+	eventInterface, isGroup := cache.GroupMessageLru.Get(req.MessageId)
+	if isGroup {
+		event := eventInterface.(*message.GroupMessage)
+		messageType := "group"
+		if event.Sender.Uin == cli.Uin {
+			messageType = "self"
+		}
+		return &onebot.GetMsgResp{
+			Time:        event.Time,
+			MessageType: messageType,
+			MessageId:   req.MessageId,
+			RealId:      event.InternalId, // 不知道是什么？
+			Message:     MiraiMsgToProtoMsg(event.Elements),
+			RawMessage:  MiraiMsgToRawMsg(event.Elements),
+			Sender: &onebot.GetMsgResp_Sender{
+				UserId:   event.Sender.Uin,
+				Nickname: event.Sender.Nickname,
+			},
+		}
+
+	}
+	eventInterface, isPrivate := cache.PrivateMessageLru.Get(req.MessageId)
+	if isPrivate {
+		event := eventInterface.(*message.PrivateMessage)
+		messageType := "private"
+		if event.Sender.Uin == cli.Uin {
+			messageType = "self"
+		}
+		return &onebot.GetMsgResp{
+			Time:        event.Time,
+			MessageType: messageType,
+			MessageId:   req.MessageId,
+			RealId:      event.InternalId, // 不知道是什么？
+			Message:     MiraiMsgToProtoMsg(event.Elements),
+			RawMessage:  MiraiMsgToRawMsg(event.Elements),
+			Sender: &onebot.GetMsgResp_Sender{
+				UserId:   event.Sender.Uin,
+				Nickname: event.Sender.Nickname,
+			},
+		}
+	}
+	return nil
+}

service/bot/remote.go

@@ -103,6 +103,11 @@ func handleApiFrame(cli *client.QQClient, req *onebot.Frame) *onebot.Frame {
 		resp.Data = &onebot.Frame_DeleteMsgResp{
 			DeleteMsgResp: HandleDeleteMsg(cli, data.DeleteMsgReq),
 		}
+	case *onebot.Frame_GetMsgReq:
+		resp.FrameType = onebot.Frame_TGetMsgReq
+		resp.Data = &onebot.Frame_GetMsgResp{
+			GetMsgResp: HandleGetMsg(cli, data.GetMsgReq),
+		}
 	default:
 		return resp
 	}