[All] new firend group

LagrangeDev · Jul 31, 2024 · 9dd7216 · 9dd7216
1 parent 69c7b62
commit 9dd7216
Show file tree

Hide file tree

Showing 11 changed files with 96 additions and 49 deletions.
diff --git a/Lagrange.Core/Common/Entity/BotFriend.cs b/Lagrange.Core/Common/Entity/BotFriend.cs
@@ -13,29 +13,37 @@ internal BotFriend()
         Remarks = string.Empty;
         PersonalSign = string.Empty;
         Qid = string.Empty;
+        GroupId = 0;
+        GroupName = string.Empty;
     }
-    
-    internal BotFriend(uint uin,string uid, string nickname, string remarks, string personalSign, string qid)
+
+    internal BotFriend(uint uin, string uid, string nickname, string remarks, string personalSign, string qid, uint groupId, string groupName)
     {
         Uin = uin;
         Uid = uid;
         Nickname = nickname;
         Remarks = remarks;
         PersonalSign = personalSign;
         Qid = qid;
+        GroupId = groupId;
+        GroupName = groupName;
     }
-    
+
     public uint Uin { get; set; }
-    
+
     internal string Uid { get; set; }
-    
+
     public string Nickname { get; set; }
-    
+
     public string Remarks { get; set; }
-    
+
     public string PersonalSign { get; set; }
-    
+
     public string Qid { get; set; }
 
+    public uint GroupId { get; }
+
+    public string GroupName { get; set; }
+
     public string Avatar => $"https://q1.qlogo.cn/g?b=qq&nk={Uin}&s=640";
 }
diff --git a/Lagrange.Core/Internal/Packets/Service/Oidb/Generics/OidbGroup.cs b/Lagrange.Core/Internal/Packets/Service/Oidb/Generics/OidbGroup.cs
@@ -0,0 +1,12 @@
+using ProtoBuf;
+#pragma warning disable CS8618
+
+namespace Lagrange.Core.Internal.Packets.Service.Oidb.Generics;
+
+[ProtoContract]
+internal class OidbGroup
+{
+    [ProtoMember(1)] public uint GroupId { get; set; }
+
+    [ProtoMember(2)] public string GroupName { get; set; }
+}
diff --git a/Lagrange.Core/Internal/Packets/Service/Oidb/Response/OidbSvcTrpcTcp0xFD4_1Response.cs b/Lagrange.Core/Internal/Packets/Service/Oidb/Response/OidbSvcTrpcTcp0xFD4_1Response.cs
@@ -18,6 +18,8 @@ internal class OidbSvcTrpcTcp0xFD4_1Response
     [ProtoMember(7)] public uint SelfUin { get; set; }
 
     [ProtoMember(101)] public List<OidbFriend> Friends { get; set; }
+
+    [ProtoMember(102)] public List<OidbGroup> Groups { get; set; }
 }
 
 [ProtoContract]

diff --git a/Lagrange.Core/Internal/Service/System/FetchFriendsService.cs b/Lagrange.Core/Internal/Service/System/FetchFriendsService.cs
@@ -17,7 +17,7 @@ namespace Lagrange.Core.Internal.Service.System;
 internal class FetchFriendsService : BaseService<FetchFriendsEvent>
 {
     private const int MaxFriendCount = 300;
-    
+
     protected override bool Build(FetchFriendsEvent input, BotKeystore keystore, BotAppInfo appInfo, BotDeviceInfo device,
         out BinaryPacket output, out List<BinaryPacket>? extraPackets)
     {
@@ -32,15 +32,15 @@ protected override bool Build(FetchFriendsEvent input, BotKeystore keystore, Bot
         });
 
         if (input.NextUin != null) packet.Body.NextUin = new OidbSvcTrpcTcp0xFD4_1Uin { Uin = input.NextUin.Value };
-        
+
         /*
          * OidbNumber里面的东西代表你想要拿到的Property，这些Property将会在返回的数据里面的Preserve的Field，
          * 102：个性签名
          * 103：备注
          * 20002：昵称
          * 27394：QID
          */
-        
+
         output = packet.Serialize();
         extraPackets = null;
         return true;
@@ -51,23 +51,34 @@ protected override bool Parse(Span<byte> input, BotKeystore keystore, BotAppInfo
     {
         var packet = Serializer.Deserialize<OidbSvcTrpcTcpBase<OidbSvcTrpcTcp0xFD4_1Response>>(input);
 
-        var friends = new List<BotFriend>();
-        foreach (var raw in packet.Body.Friends)
+        var groups = Group(packet.Body.Groups);
+        var friends = packet.Body.Friends.Select(f =>
         {
-            var additional = raw.Additional.First(x => x.Type == 1);
-            var properties = Property(additional.Layer1.Properties);
-            friends.Add(new BotFriend(raw.Uin, raw.Uid, properties[20002], properties[103], properties[102], properties[27394]));
-        }
-
+            var properties = Property(f.Additional.First(a => a.Type == 1).Layer1.Properties);
+            return new BotFriend(
+                f.Uin,
+                f.Uid,
+                properties[20002],
+                properties[102],
+                properties[102],
+                properties[27394],
+                f.CustomGroup,
+                groups[f.CustomGroup]
+            );
+        }).ToList();
+
         output = FetchFriendsEvent.Result(0, friends, packet.Body.Next?.Uin); // 全家4完了才能想出来这种分页的逻辑
         extraEvents = null;
         return true;
     }
 
     private static Dictionary<uint, string> Property(List<OidbFriendProperty> properties)
     {
-        var dict = new Dictionary<uint, string>(properties.Capacity);
-        foreach (var property in properties) dict[property.Code] = property.Value;
-        return dict;
+        return properties.ToDictionary(p => p.Code, p => p.Value);
+    }
+
+    private static Dictionary<uint, string> Group(List<OidbGroup> groups)
+    {
+        return groups.ToDictionary(g => g.GroupId, g => g.GroupName);
     }
 }
diff --git a/Lagrange.Core/Message/MessagePacker.cs b/Lagrange.Core/Message/MessagePacker.cs
@@ -25,7 +25,7 @@ internal static class MessagePacker
 {
     private static readonly Dictionary<Type, List<PropertyInfo>> EntityToElem;
     private static readonly Dictionary<Type, IMessageEntity> Factory;
-    
+
     static MessagePacker()
     {
         EntityToElem = new Dictionary<Type, List<PropertyInfo>>();
@@ -34,7 +34,7 @@ static MessagePacker()
         var assembly = Assembly.GetExecutingAssembly();
         var types = assembly.GetTypeWithMultipleAttributes<MessageElementAttribute>(out var attributeArrays);
         var elemType = typeof(Elem);
-        
+
         for (int i = 0; i < types.Count; i++)
         {
             var type = types[i];
@@ -73,12 +73,12 @@ public static Internal.Packets.Message.Message Build(MessageChain chain, string
                 message.Body.MsgContent = stream.ToArray();
             }
         }
-        
+
         BuildAdditional(chain, message);
 
         return message;
     }
-    
+
     public static PushMsgBody BuildFake(MessageChain chain, string selfUid)
     {
         var message = BuildFakePacketBase(chain, selfUid);
@@ -87,7 +87,7 @@ public static PushMsgBody BuildFake(MessageChain chain, string selfUid)
         {
             entity.SetSelfUid(selfUid);
             message.Body?.RichText?.Elems.AddRange(entity.PackElement());
-            
+
             if (message.Body != null)
             {
                 if (entity.PackMessageContent() is not { } content) continue;
@@ -104,34 +104,34 @@ public static PushMsgBody BuildFake(MessageChain chain, string selfUid)
     private static void BuildAdditional(MessageChain chain, Internal.Packets.Message.Message message)
     {
         if (message.Body?.RichText == null) return;
-        
+
         foreach (var entity in chain)
         {
             switch (entity)
             {
                 case RecordEntity { Compat: { } compat }:  // Append Tag 04 -> Ptt
-                {
-                    message.Body.RichText.Ptt = compat.Ptt;
-                    message.Body.RichText.Elems.AddRange(compat.Elems);
-                    break;
-                }
+                    {
+                        message.Body.RichText.Ptt = compat.Ptt;
+                        message.Body.RichText.Elems.AddRange(compat.Elems);
+                        break;
+                    }
             }
         }
     }
-    
+
     public static MessageChain Parse(PushMsgBody message, bool isFake = false)
     {
         var chain = isFake ? ParseFakeChain(message) : ParseChain(message);
 
-        if (message.Body?.RichText is { Elems: { } elements}) // 怎么Body还能是null的
+        if (message.Body?.RichText is { Elems: { } elements }) // 怎么Body还能是null的
         {
             foreach (var element in elements)
             {
                 foreach (var (entityType, expectElems) in EntityToElem)
                 {
                     foreach (var expectElem in expectElems)
                     {
-                        if (expectElem.GetValueByExpr(element) is not null && 
+                        if (expectElem.GetValueByExpr(element) is not null &&
                             Factory[entityType].UnpackElement(element) is { } entity)
                         {
                             chain.Add(entity);
@@ -147,8 +147,8 @@ public static MessageChain Parse(PushMsgBody message, bool isFake = false)
             case { } groupPtt when chain.IsGroup && groupPtt.FileId != 0:  //  for legacy ptt
                 chain.Add(new RecordEntity(groupPtt.GroupFileKey, groupPtt.FileName));
                 break;
-            case { } privatePtt when !chain.IsGroup: 
-                if (chain.OfType<RecordEntity>().FirstOrDefault(x => x.AudioName == privatePtt.FileName) == null) 
+            case { } privatePtt when !chain.IsGroup:
+                if (chain.OfType<RecordEntity>().FirstOrDefault(x => x.AudioName == privatePtt.FileName) == null)
                     chain.Add(new RecordEntity(privatePtt.FileUuid, privatePtt.FileName));
                 break;
         }
@@ -159,13 +159,13 @@ public static MessageChain Parse(PushMsgBody message, bool isFake = false)
     public static MessageChain ParsePrivateFile(PushMsgBody message)
     {
         if (message.Body?.MsgContent == null) throw new Exception();
-        
+
         var chain = ParseChain(message);
 
         var extra = Serializer.Deserialize<FileExtra>(message.Body.MsgContent.AsSpan());
         var file = extra.File;
 
-        if ( file is { FileSize: not null, FileName: not null, FileMd5: not null, FileUuid: not null, FileHash: not null })
+        if (file is { FileSize: not null, FileName: not null, FileMd5: not null, FileUuid: not null, FileHash: not null })
         {
             chain.Add(new FileEntity((long)file.FileSize, file.FileName, file.FileMd5, file.FileUuid, file.FileHash));
             return chain;
@@ -243,29 +243,29 @@ public static MessageChain ParsePrivateFile(PushMsgBody message)
         },
         Body = new MessageBody { RichText = new RichText { Elems = new List<Elem>() } }
     };
-    
+
     private static MessageChain ParseChain(PushMsgBody message)
     {
         var chain = message.ResponseHead.Grp == null
             ? new MessageChain(
                 message.ResponseHead.FromUin,
-                message.ResponseHead.ToUid ?? string.Empty , 
-                message.ResponseHead.FromUid ?? string.Empty, 
+                message.ResponseHead.ToUid ?? string.Empty,
+                message.ResponseHead.FromUid ?? string.Empty,
                 message.ResponseHead.ToUin,
                 message.ContentHead.Sequence ?? 0,
                 message.ContentHead.NewId ?? 0,
                 message.ContentHead.Type == 141 ? MessageChain.MessageType.Temp : MessageChain.MessageType.Friend)
-            
+
             : new MessageChain(
-                message.ResponseHead.Grp.GroupUin, 
-                message.ResponseHead.FromUin, 
+                message.ResponseHead.Grp.GroupUin,
+                message.ResponseHead.FromUin,
                 message.ContentHead.Sequence ?? 0,
                 message.ContentHead.NewId ?? 0);
 
         if (message.Body?.RichText?.Elems is { } elems) chain.Elements.AddRange(elems);
 
         chain.Time = DateTimeOffset.FromUnixTimeSeconds(message.ContentHead.Timestamp ?? 0).DateTime;
-        
+
         return chain;
     }
 
@@ -284,9 +284,9 @@ private static MessageChain ParseFakeChain(PushMsgBody message)
         }
         else
         {
-            @base.FriendInfo = new BotFriend(0, message.ResponseHead.FromUid ?? string.Empty, message.ResponseHead.Forward?.FriendName ?? string.Empty, string.Empty, string.Empty, string.Empty);
+            @base.FriendInfo = new BotFriend(0, message.ResponseHead.FromUid ?? string.Empty, message.ResponseHead.Forward?.FriendName ?? string.Empty, string.Empty, string.Empty, string.Empty, 0, string.Empty);
         }
-        
+
         return @base;
     }
 

diff --git a/Lagrange.OneBot/Core/Entity/OneBotFriend.cs b/Lagrange.OneBot/Core/Entity/OneBotFriend.cs
@@ -12,4 +12,8 @@ public class OneBotFriend
     [JsonPropertyName("nickname")] public string NickName { get; set; } = "";
 
     [JsonPropertyName("remark")] public string Remark { get; set; } = "";
+
+    [JsonPropertyName("group_id")] public uint GroupId { get; set; } = 0;
+
+    [JsonPropertyName("group_name")] public string GroupName { get; set; } = "";
 }
diff --git a/Lagrange.OneBot/Core/Operation/Info/GetFriendListOperation.cs b/Lagrange.OneBot/Core/Operation/Info/GetFriendListOperation.cs
@@ -18,7 +18,9 @@ public async Task<OneBotResult> HandleOperation(BotContext context, JsonNode? pa
                 UserId = friend.Uin,
                 QId = friend.Qid,
                 NickName = friend.Nickname,
-                Remark = friend.Remarks
+                Remark = friend.Remarks,
+                GroupId = friend.GroupId,
+                GroupName = friend.GroupName
             };
         }).ToArray(), 0, "ok");
 }
diff --git a/Lagrange.OneBot/Core/Operation/Message/MessageCommon.cs b/Lagrange.OneBot/Core/Operation/Message/MessageCommon.cs
@@ -255,6 +255,8 @@ public List<MessageChain> BuildForwardChains(BotContext context, OneBotForward f
                     element.Name,
                     string.Empty,
                     string.Empty,
+                    string.Empty,
+                    0,
                     string.Empty
                 );
                 chains.Add(chain);  // as fake is constructed, use uid from bot itself to upload image

diff --git a/Lagrange.OneBot/Core/Operation/Message/SendMessageOperation.cs b/Lagrange.OneBot/Core/Operation/Message/SendMessageOperation.cs
@@ -39,6 +39,8 @@ public async Task<OneBotResult> HandleOperation(BotContext context, JsonNode? pa
                 context.BotName ?? string.Empty,
                 string.Empty,
                 string.Empty,
+                string.Empty,
+                0,
                 string.Empty
             ),
             GroupMemberInfo = null,

diff --git a/Lagrange.OneBot/Core/Operation/Message/SendPrivateForwardOperation.cs b/Lagrange.OneBot/Core/Operation/Message/SendPrivateForwardOperation.cs
@@ -39,6 +39,8 @@ public async Task<OneBotResult> HandleOperation(BotContext context, JsonNode? pa
                     context.BotName ?? string.Empty,
                     string.Empty,
                     string.Empty,
+                    string.Empty,
+                    0,
                     string.Empty
                 ),
                 GroupMemberInfo = null,

diff --git a/Lagrange.OneBot/Core/Operation/Message/SendPrivateMessageOperation.cs b/Lagrange.OneBot/Core/Operation/Message/SendPrivateMessageOperation.cs
@@ -39,6 +39,8 @@ public async Task<OneBotResult> HandleOperation(BotContext context, JsonNode? pa
                 context.BotName ?? string.Empty,
                 string.Empty,
                 string.Empty,
+                string.Empty,
+                0,
                 string.Empty
             ),
             GroupMemberInfo = null,