From f70524b0a29eb60851d441ebefc46db2e6ece28c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=B5=B7=E4=B8=AA=E5=90=8D=E5=AD=97=E5=A5=BD=E9=9A=BE?=
 <25520958+MisakaAldrich@users.noreply.github.com>
Date: Fri, 28 Feb 2025 13:03:58 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E4=BC=81=E4=B8=9A=E5=BE=AE?=
 =?UTF-8?q?=E4=BF=A1=E5=9B=BE=E7=89=87=E9=80=9A=E7=9F=A5=20(#1222)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

* 添加企业微信图片通知

* 移除多余代码,修复CA1850规则
---
 .../Service/Notifier/WorkWeixinNotifier.cs    | 92 ++++++++++++++++---
 1 file changed, 80 insertions(+), 12 deletions(-)

diff --git a/BetterGenshinImpact/Service/Notifier/WorkWeixinNotifier.cs b/BetterGenshinImpact/Service/Notifier/WorkWeixinNotifier.cs
index 65a273e0..4d7d97f7 100644
--- a/BetterGenshinImpact/Service/Notifier/WorkWeixinNotifier.cs
+++ b/BetterGenshinImpact/Service/Notifier/WorkWeixinNotifier.cs
@@ -1,9 +1,13 @@
 using BetterGenshinImpact.Service.Notifier.Exception;
 using BetterGenshinImpact.Service.Notifier.Interface;
+using System;
+using System.IO;
 using System.Net.Http;
+using System.Security.Cryptography;
 using System.Text;
 using System.Text.Json;
 using System.Threading.Tasks;
+using BetterGenshinImpact.Helpers;
 using BetterGenshinImpact.Service.Notification.Model;
 
 namespace BetterGenshinImpact.Service.Notifier;
@@ -31,11 +35,28 @@ public class WorkWeixinNotifier : INotifier
         
         try
         {
-            var response = await _httpClient.PostAsync(Endpoint, TransformData(content));
-
-            if (!response.IsSuccessStatusCode)
+            // If there's a screenshot, send it first as an image
+            if (content.Screenshot != null)
             {
-                throw new NotifierException($"WorkWeixin webhook call failed with code: {response.StatusCode}");
+                var imagePayload = await TransformImageData(content.Screenshot);
+                var imageResponse = await _httpClient.PostAsync(Endpoint, imagePayload);
+
+                if (!imageResponse.IsSuccessStatusCode)
+                {
+                    throw new NotifierException($"WorkWeixin image webhook call failed with code: {imageResponse.StatusCode}");
+                }
+            }
+
+            // Then send the text message
+            if (!string.IsNullOrEmpty(content.Message))
+            {
+                var textPayload = await TransformTextData(content.Message);
+                var textResponse = await _httpClient.PostAsync(Endpoint, textPayload);
+
+                if (!textResponse.IsSuccessStatusCode)
+                {
+                    throw new NotifierException($"WorkWeixin text webhook call failed with code: {textResponse.StatusCode}");
+                }
             }
         }
         catch (NotifierException)
@@ -48,19 +69,66 @@ public class WorkWeixinNotifier : INotifier
         }
     }
 
-    private StringContent TransformData(BaseNotificationData notificationData)
+    private async Task<StringContent> TransformImageData(System.Drawing.Image screenshot)
     {
-        var workweixinMessage = new
+        var base64Image = ConvertImageToBase64(screenshot);
+        var md5Image = ComputeMd5Hash(screenshot); 
+        
+        var imageMessage = new
+        {
+            msgtype = "image",
+            image = new
+            {
+                base64 = base64Image,
+                md5 = md5Image
+            }
+        };
+        await Task.Yield();
+        var serializedImageData = JsonSerializer.Serialize(imageMessage);
+        return new StringContent(serializedImageData, Encoding.UTF8, "application/json");
+    }
+
+    private async Task<StringContent> TransformTextData(string message)
+    {
+        var textMessage = new
         {
             msgtype = "text",
             text = new
             {
-                content = notificationData.Message
+                content = message
             }
         };
-
-        var serializedData = JsonSerializer.Serialize(workweixinMessage);
-
-        return new StringContent(serializedData, Encoding.UTF8, "application/json");
+        await Task.Yield();
+        var serializedTextData = JsonSerializer.Serialize(textMessage);
+        return new StringContent(serializedTextData, Encoding.UTF8, "application/json");
     }
-}
\ No newline at end of file
+
+    private string ConvertImageToBase64(System.Drawing.Image image)
+    {
+        using (var ms = new MemoryStream())
+        {
+            // Save the image to a MemoryStream in JPEG format
+            image.Save(ms, System.Drawing.Imaging.ImageFormat.Jpeg);
+            byte[] imageBytes = ms.ToArray();
+            
+            // Convert to Base64 (this will produce a single-line string)
+            string base64String = Convert.ToBase64String(imageBytes);
+            return base64String;
+        }
+    }
+
+    private string ComputeMd5Hash(System.Drawing.Image image)
+    {
+        using (var ms = new MemoryStream())
+        {
+            image.Save(ms, System.Drawing.Imaging.ImageFormat.Jpeg);
+            byte[] imageBytes = ms.ToArray();
+
+            // Compute the MD5 hash of the raw byte data
+            byte[] hashBytes = MD5.HashData(imageBytes);
+            // Convert hash bytes to a hex string
+            string md5Hash = BitConverter.ToString(hashBytes).Replace("-", "").ToLower();
+            return md5Hash;
+        }
+    }
+}