[Core & Milky] Implement Core's GetNTV2RichMediaUrl and Milky's get_resource_temp_url

Author: NoirHare

Lagrange.Core/Common/Interface/OperationExt.cs

@@ -41,4 +41,7 @@ public static Task SetGroupNotification(this BotContext context, long groupUin,
 
     public static Task SetGroupReaction(this BotContext context, long groupUin, ulong sequence, string code, bool isAdd) =>
         context.EventContext.GetLogic<OperationLogic>().SetGroupReaction(groupUin, sequence, code, isAdd);
+
+    public static Task<string> GetNTV2RichMediaUrl(this BotContext context, string fileUuid) =>
+        context.EventContext.GetLogic<OperationLogic>().GetNTV2RichMediaUrl(fileUuid);
 }

Lagrange.Core/Internal/Logic/OperationLogic.cs

@@ -5,9 +5,14 @@
 using Lagrange.Core.Internal.Events.Message;
 using Lagrange.Core.Internal.Events.System;
 using Lagrange.Core.Internal.Packets.Service;
+using Lagrange.Core.Message;
+using Lagrange.Core.Message.Entities;
 using Lagrange.Core.Utility;
 using Lagrange.Core.Utility.Cryptography;
 using Lagrange.Core.Utility.Extension;
+using Lagrange.Proto.Primitives;
+using Lagrange.Proto.Serialization;
+using FileInfo = Lagrange.Core.Internal.Packets.Service.FileInfo;
 
 namespace Lagrange.Core.Internal.Logic;
 
@@ -284,4 +289,93 @@ public async Task SetGroupReaction(long groupUin, ulong sequence, string code, b
             new ReduceGroupReactionEventReq(groupUin, sequence, code)
         );
     }
+
+    public async Task<string> GetNTV2RichMediaUrl(string uuid)
+    {
+        int remainder = uuid.Length % 4;
+        int length = remainder == 0 ? uuid.Length : uuid.Length + (4 - remainder);
+        string base64 = uuid.Replace('-', '+').Replace('_', '/').PadRight(length, '=');
+
+        ulong appId = 0;
+        uint ttl = 0;
+        var reader = new ProtoReader(Convert.FromBase64String(base64));
+        while (!reader.IsCompleted)
+        {
+            uint tag = reader.DecodeVarIntUnsafe<uint>();
+            switch (tag >>> 3)
+            {
+                case 4:
+                {
+                    appId = reader.DecodeVarInt<ulong>();
+                    break;
+                }
+                case 10:
+                {
+                    ttl = reader.DecodeVarInt<uint>();
+                    break;
+                }
+                default:
+                {
+                    reader.SkipField((WireType)(tag & 0b111));
+                    break;
+                }
+            }
+        }
+
+        NTV2RichMediaDownloadEventResp response = appId switch
+        {
+            // Record
+            1402 => await context.EventContext.SendEvent<RecordDownloadEventResp>(new RecordDownloadEventReq(
+                CreateFakeFriendMessage(context.BotUin),
+                CreateFakeEntity<RecordEntity>(uuid, ttl)
+            )),
+            1403 => await context.EventContext.SendEvent<RecordGroupDownloadEventResp>(new RecordGroupDownloadEventReq(
+                CreateFakeGroupMessage(context.BotUin),
+                CreateFakeEntity<RecordEntity>(uuid, ttl)
+            )),
+            // Video
+            1413 => await context.EventContext.SendEvent<VideoDownloadEventResp>(new VideoDownloadEventReq(
+                CreateFakeFriendMessage(context.BotUin),
+                CreateFakeEntity<VideoEntity>(uuid, ttl)
+            )),
+            1415 => await context.EventContext.SendEvent<VideoGroupDownloadEventResp>(new VideoGroupDownloadEventReq(
+                CreateFakeGroupMessage(context.BotUin),
+                CreateFakeEntity<VideoEntity>(uuid, ttl)
+            )),
+            // Image
+            1406 => await context.EventContext.SendEvent<ImageDownloadEventResp>(new ImageDownloadEventReq(
+                CreateFakeFriendMessage(context.BotUin),
+                CreateFakeEntity<ImageEntity>(uuid, ttl)
+            )),
+            1407 => await context.EventContext.SendEvent<ImageGroupDownloadEventResp>(new ImageGroupDownloadEventReq(
+                CreateFakeGroupMessage(context.BotUin),
+                CreateFakeEntity<ImageEntity>(uuid, ttl)
+            )),
+            _ => throw new NotSupportedException($"Unsupported AppId: {appId}"),
+        };
+
+        return response.Url;
+
+        static BotMessage CreateFakeFriendMessage(long uin) => BotMessage.CreateCustomFriend(uin, string.Empty, uin, string.Empty, DateTime.Now, []);
+        static BotMessage CreateFakeGroupMessage(long uin) => BotMessage.CreateCustomGroup(uin, 0, string.Empty, DateTime.Now, []);
+        static RichMediaEntityBase CreateFakeEntity<T>(string fileUuid, uint ttl) where T : RichMediaEntityBase, new()
+        {
+            return new T
+            {
+                MsgInfo = new MsgInfo
+                {
+                    MsgInfoBody = [new MsgInfoBody {
+                        Index = new IndexNode {
+                            Info = new FileInfo {},
+                            FileUuid = fileUuid,
+                            StoreId = 1,
+                            UploadTime = 0,
+                            Ttl = ttl,
+                            SubType = 0,
+                        }
+                    }]
+                }
+            };
+        }
+    }
 }

Lagrange.Core/Message/Entities/RichMediaEntityBase.cs

@@ -11,53 +11,53 @@ public abstract class RichMediaEntityBase : IMessageEntity
     protected byte[]? _compat;
 
     private protected bool IsGroup;
-    
+
     internal MsgInfo? MsgInfo
     {
         get => _msgInfo;
-        private protected set
+        set
         {
             Debug.Assert(value != null);
-            
+
             var fileInfo = value.MsgInfoBody[0].Index.Info;
             FileUuid = value.MsgInfoBody[0].Index.FileUuid;
             FileMd5 = fileInfo.FileHash;
             FileSha1 = fileInfo.FileSha1;
             FileSize = fileInfo.FileSize;
             FileName = fileInfo.FileName;
             _msgInfo = value;
-        } 
+        }
     }
-    
+
     private protected bool DisposeOnCompletion;
 
     internal abstract Lazy<Stream>? Stream { get; }
 
     public string FileUuid { get; internal set; } = string.Empty;
-    
+
     public string FileUrl { get; internal set; } = string.Empty;
 
     public string FileName { get; internal set; } = string.Empty;
-    
+
     public string FileSha1 { get; internal set; } = string.Empty;
-    
+
     public uint FileSize { get; internal set; }
 
     public string FileMd5 { get; internal set; } = string.Empty;
-    
+
     public abstract Task Preprocess(BotContext context, BotMessage message);
 
     public abstract Task Postprocess(BotContext context, BotMessage message);
-    
+
     internal abstract Elem[] Build();
-    
+
     internal abstract IMessageEntity? Parse(List<Elem> elements, Elem target);
-    
+
     public abstract string ToPreviewString();
-    
+
     Elem[] IMessageEntity.Build() => Build();
 
     IMessageEntity? IMessageEntity.Parse(List<Elem> elements, Elem target) => Parse(elements, target);
-    
+
     string IMessageEntity.ToPreviewString() => ToPreviewString();
 }

Lagrange.Milky/Api/Handler/Message/GetResourceTempUrlHandler.cs

@@ -0,0 +1,30 @@
+
+using System.Text.Json.Serialization;
+using Lagrange.Core;
+using Lagrange.Core.Common.Interface;
+
+namespace Lagrange.Milky.Api.Handler.Message;
+
+[Api("get_resource_temp_url")]
+public class GetResourceTempUrlHandler(BotContext bot) : IApiHandler<GetResourceTempUrlParameter, GetResourceTempUrlResult>
+{
+    private readonly BotContext _bot = bot;
+
+    public async Task<GetResourceTempUrlResult> HandleAsync(GetResourceTempUrlParameter parameter, CancellationToken token)
+    {
+        return new GetResourceTempUrlResult(await _bot.GetNTV2RichMediaUrl(parameter.ResourceId));
+    }
+}
+
+public class GetResourceTempUrlParameter(string resourceId)
+{
+    [JsonRequired]
+    [JsonPropertyName("resource_id")]
+    public string ResourceId { get; init; } = resourceId;
+}
+
+public class GetResourceTempUrlResult(string url)
+{
+    [JsonPropertyName("url")]
+    public string Url { get; } = url;
+}

Lagrange.Milky/README.md

@@ -12,8 +12,11 @@ https://lagrangedev.github.io/Lagrange.Milky.Document
 
 ## Feature List
 
+Current Milky version: [77c6cbd](https://github.com/SaltifyDev/milky/tree/77c6cbd6fb131d581fbb4c128a540eadfbcbd475)
+
 ### communication
 
+- [ ] SSE (wontimpl)
 - [x] WebSocket
 - [x] WebHook
 
@@ -39,7 +42,7 @@ https://lagrangedev.github.io/Lagrange.Milky.Document
 - [x] /send_group_message
 - [x] /get_message
 - [x] /get_history_messages - not supported start_message_seq eql null
-- [ ] /get_resource_temp_url
+- [x] /get_resource_temp_url
 - [ ] /get_forwarded_messages
 - [ ] /recall_private_message
 - [ ] /recall_group_message
@@ -125,6 +128,7 @@ https://lagrangedev.github.io/Lagrange.Milky.Document
 - [x] image
 - [x] record
 - [x] video
+- [ ] file
 - [x] forward
 - [ ] market_face
 - [ ] light_app
@@ -145,8 +149,8 @@ https://lagrangedev.github.io/Lagrange.Milky.Document
 #### forward
 
 - [x] text
-<!-- - [ ] mention -->
-<!-- - [ ] mention_all -->
+- [ ] mention - (wontimpl) forward not supported
+- [ ] mention_all - (wontimpl) forward not supported
 - [ ] face
 - [ ] reply
 - [ ] image

Lagrange.Milky/Utility/JsonUtility.cs

@@ -75,6 +75,9 @@ public static partial class JsonUtility
     // get_history_messages
     [JsonSerializable(typeof(GetHistoryMessagesParameter))]
     [JsonSerializable(typeof(GetHistoryMessagesResult))]
+    // get_resource_temp_url
+    [JsonSerializable(typeof(GetResourceTempUrlParameter))]
+    [JsonSerializable(typeof(GetResourceTempUrlResult))]
     // == friend ==
     // send_friend_nudge
     [JsonSerializable(typeof(SendFriendNudgeParameter))]