From 371d0d11a5c9efb6ac1a3f5996525927995122b1 Mon Sep 17 00:00:00 2001
From: nsnail <taokeu@gmail.com>
Date: Sun, 8 Jan 2023 19:58:16 +0800
Subject: [PATCH] =?UTF-8?q?<feat>=20=E4=BB=8E=E8=B5=84=E6=BA=90=E6=96=87?=
 =?UTF-8?q?=E4=BB=B6=E8=AF=BB=E5=8F=96Description?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/Attributes/LocalizationAttribute.cs | 24 ++++++++++++++++++++++++
 src/Extensions/EnumExtensions.cs        | 17 +++++++++++++----
 2 files changed, 37 insertions(+), 4 deletions(-)
 create mode 100644 src/Attributes/LocalizationAttribute.cs

diff --git a/src/Attributes/LocalizationAttribute.cs b/src/Attributes/LocalizationAttribute.cs
new file mode 100644
index 0000000..5658339
--- /dev/null
+++ b/src/Attributes/LocalizationAttribute.cs
@@ -0,0 +1,24 @@
+namespace NSExt.Attributes;
+
+/// <summary>
+///     指定本地化资源类型
+/// </summary>
+[AttributeUsage(AttributeTargets.Property | AttributeTargets.Class | AttributeTargets.Field)]
+public class LocalizationAttribute : Attribute
+{
+    /// <summary>
+    ///     Initializes a new instance of the <see cref="LocalizationAttribute" /> class.
+    /// </summary>
+    public LocalizationAttribute(Type resourceClass)
+    {
+        ResourceClass = resourceClass;
+    }
+
+    /// <summary>
+    ///     Gets or sets 资源类型
+    /// </summary>
+    /// <value>
+    ///     资源类型
+    /// </value>
+    public Type ResourceClass { get; set; }
+}
\ No newline at end of file
diff --git a/src/Extensions/EnumExtensions.cs b/src/Extensions/EnumExtensions.cs
index 80ae7bd..0c75c95 100644
--- a/src/Extensions/EnumExtensions.cs
+++ b/src/Extensions/EnumExtensions.cs
@@ -1,3 +1,6 @@
+using System.Reflection;
+using NSExt.Attributes;
+
 namespace NSExt.Extensions;
 
 /// <summary>
@@ -12,9 +15,15 @@ public static class EnumExtensions
     /// <returns>description属性</returns>
     public static string Desc(this Enum e)
     {
-        var t     = e.GetType();
-        var fi    = t.GetField(Enum.GetName(t, e)!);
-        var attrs = (DescriptionAttribute[])fi!.GetCustomAttributes(typeof(DescriptionAttribute), false);
-        return (attrs.Length != 0 ? attrs[0].Description : Enum.GetName(t, e)) ?? string.Empty;
+        var t        = e.GetType();
+        var fi       = t.GetField(Enum.GetName(t, e)!);
+        var descAttr = fi!.GetCustomAttribute<DescriptionAttribute>(false);
+        if (descAttr is null) {
+            return Enum.GetName(t, e);
+        }
+
+        var str     = descAttr.Description;
+        var locAttr = fi!.GetCustomAttribute<LocalizationAttribute>(false);
+        return locAttr is null ? str : locAttr.ResourceClass.GetProperty(str)?.GetValue(default) as string ?? str;
     }
 }
\ No newline at end of file