From 94894d29fac677c7f3e272b63043c0da98f74d0b Mon Sep 17 00:00:00 2001
From: kookxiang <kookxiang@gmail.com>
Date: Sun, 21 Jul 2024 19:30:02 +0800
Subject: [PATCH] add infobox support; add translation setting for person #143

---
 Jellyfin.Plugin.Bangumi.Test/Person.cs        |  7 +++--
 Jellyfin.Plugin.Bangumi.Test/Series.cs        |  1 +
 .../Configuration/ConfigPage.html             |  7 +++++
 .../Configuration/PluginConfiguration.cs      |  2 ++
 Jellyfin.Plugin.Bangumi/Model/InfoBox.cs      | 23 +++++++++++++++
 Jellyfin.Plugin.Bangumi/Model/PersonDetail.cs | 28 +++++++++++++++++++
 Jellyfin.Plugin.Bangumi/Model/Subject.cs      | 23 +++++++++++++--
 .../Providers/MovieProvider.cs                |  2 ++
 .../Providers/PersonProvider.cs               |  7 +++--
 .../Providers/SeasonProvider.cs               |  3 ++
 .../Providers/SeriesProvider.cs               |  4 ++-
 11 files changed, 100 insertions(+), 7 deletions(-)
 create mode 100644 Jellyfin.Plugin.Bangumi/Model/InfoBox.cs

diff --git a/Jellyfin.Plugin.Bangumi.Test/Person.cs b/Jellyfin.Plugin.Bangumi.Test/Person.cs
index cc8d1cb..1d5ad50 100644
--- a/Jellyfin.Plugin.Bangumi.Test/Person.cs
+++ b/Jellyfin.Plugin.Bangumi.Test/Person.cs
@@ -2,6 +2,7 @@
 using System.Linq;
 using System.Threading;
 using System.Threading.Tasks;
+using Jellyfin.Plugin.Bangumi.Configuration;
 using Jellyfin.Plugin.Bangumi.Providers;
 using Jellyfin.Plugin.Bangumi.Test.Util;
 using MediaBrowser.Controller.Providers;
@@ -28,12 +29,14 @@ public void ProviderInfo()
     [TestMethod]
     public async Task GetById()
     {
+        Bangumi.Plugin.Instance!.Configuration.PersonTranslationPreference = TranslationPreferenceType.Chinese;
         var result = await _provider.GetMetadata(new PersonLookupInfo
         {
-            ProviderIds = new Dictionary<string, string> { { Constants.ProviderName, "5847" } }
+            ProviderIds = new Dictionary<string, string> { { Constants.ProviderName, "7307" } }
         }, _token);
         Assert.IsNotNull(result.Item, "person info should not be null");
-        Assert.AreEqual("茅野愛衣", result.Item.Name, "should return correct name");
+        Assert.AreEqual("上坂すみれ", result.Item.OriginalTitle, "should return correct name");
+        Assert.AreEqual("上坂堇", result.Item.Name, "should return translated name");
         Assert.IsNotNull(result.Item.ProviderIds[Constants.ProviderName], "should have plugin provider id");
     }
 
diff --git a/Jellyfin.Plugin.Bangumi.Test/Series.cs b/Jellyfin.Plugin.Bangumi.Test/Series.cs
index 86f0563..04c089c 100644
--- a/Jellyfin.Plugin.Bangumi.Test/Series.cs
+++ b/Jellyfin.Plugin.Bangumi.Test/Series.cs
@@ -192,6 +192,7 @@ private static void AssertSeries(MetadataResult<MediaBrowser.Controller.Entities
         Assert.AreEqual("2013-10-05", result.Item.AirTime, "should return correct air time info");
         Assert.AreEqual(DayOfWeek.Saturday, result.Item.AirDays?[0], "should return correct air day info");
         Assert.IsTrue(result.Item.CommunityRating is > 0 and <= 10, "should return rating info");
+        Assert.IsNotNull(result.Item.HomePageUrl, "should return official website link");
         Assert.IsNotNull(result.People.Find(x => x.IsType(PersonKind.Actor)), "should have at least one actor");
         Assert.IsNotNull(result.People.Find(x => x.IsType(PersonKind.Director)), "should have at least one director");
         Assert.IsNotNull(result.People.Find(x => x.IsType(PersonKind.Writer)), "should have at least one writer");
diff --git a/Jellyfin.Plugin.Bangumi/Configuration/ConfigPage.html b/Jellyfin.Plugin.Bangumi/Configuration/ConfigPage.html
index ce98b75..2099a27 100644
--- a/Jellyfin.Plugin.Bangumi/Configuration/ConfigPage.html
+++ b/Jellyfin.Plugin.Bangumi/Configuration/ConfigPage.html
@@ -195,6 +195,13 @@ <h2 class="sectionTitle">元数据</h2>
                         <option value="Chinese">优先使用中文翻译</option>
                     </select>
                 </div>
+                <div class="selectContainer">
+                    <label class="selectLabel" for="PersonTranslationPreference">人物名称翻译</label>
+                    <select class="emby-select-withcolor emby-select" id="PersonTranslationPreference" is="emby-select">
+                        <option value="Original">优先使用日文</option>
+                        <option value="Chinese">优先使用中文</option>
+                    </select>
+                </div>
                 <div class="checkboxContainer checkboxContainer-withDescription">
                     <label class="emby-checkbox-label">
                         <input id="UseBangumiSeasonTitle" is="emby-checkbox" type="checkbox"/>
diff --git a/Jellyfin.Plugin.Bangumi/Configuration/PluginConfiguration.cs b/Jellyfin.Plugin.Bangumi/Configuration/PluginConfiguration.cs
index 6636590..c551e01 100644
--- a/Jellyfin.Plugin.Bangumi/Configuration/PluginConfiguration.cs
+++ b/Jellyfin.Plugin.Bangumi/Configuration/PluginConfiguration.cs
@@ -12,6 +12,8 @@ public class PluginConfiguration : BasePluginConfiguration
 {
     public TranslationPreferenceType TranslationPreference { get; set; } = TranslationPreferenceType.Chinese;
 
+    public TranslationPreferenceType PersonTranslationPreference { get; set; } = TranslationPreferenceType.Original;
+
     public int RequestTimeout { get; set; } = 5000;
 
     public string BaseServerUrl { get; set; } = "https://api.bgm.tv";
diff --git a/Jellyfin.Plugin.Bangumi/Model/InfoBox.cs b/Jellyfin.Plugin.Bangumi/Model/InfoBox.cs
new file mode 100644
index 0000000..dcebb57
--- /dev/null
+++ b/Jellyfin.Plugin.Bangumi/Model/InfoBox.cs
@@ -0,0 +1,23 @@
+using System.Collections.Generic;
+using System.Linq;
+using System.Text.Json;
+using System.Text.Json.Serialization;
+
+namespace Jellyfin.Plugin.Bangumi.Model;
+
+public class InfoBox : List<InfoBoxItem>
+{
+    public string? GetString(string key)
+    {
+        return this.FirstOrDefault(x => x.Key == key)?.Value.GetString();
+    }
+}
+
+public class InfoBoxItem
+{
+    [JsonPropertyName("key")]
+    public string Key { get; set; } = null!;
+
+    [JsonPropertyName("value")]
+    public JsonElement Value { get; set; } = default!;
+}
\ No newline at end of file
diff --git a/Jellyfin.Plugin.Bangumi/Model/PersonDetail.cs b/Jellyfin.Plugin.Bangumi/Model/PersonDetail.cs
index 2c815e3..aaeb187 100644
--- a/Jellyfin.Plugin.Bangumi/Model/PersonDetail.cs
+++ b/Jellyfin.Plugin.Bangumi/Model/PersonDetail.cs
@@ -1,4 +1,5 @@
 using System;
+using System.Globalization;
 using System.Text.Json.Serialization;
 using Jellyfin.Plugin.Bangumi.Configuration;
 
@@ -26,6 +27,33 @@ public class PersonDetail : Person
     [JsonPropertyName("birth_day")]
     public int? Birthday { get; set; }
 
+    [JsonIgnore]
     public DateTime? Birthdate =>
         BirthYear != null && BirthMonth != null && Birthday != null ? new DateTime((int)BirthYear, (int)BirthMonth, (int)Birthday) : null;
+
+    [JsonIgnore]
+    public string? BirthPlace => InfoBox?.GetString("出生地") ?? InfoBox?.GetString("出身地");
+
+    [JsonIgnore]
+    public DateTime? DeathDate
+    {
+        get
+        {
+            var dateStr = InfoBox?.GetString("卒日");
+            if (dateStr != null && DateTime.TryParseExact(dateStr, "yyyy年MM月dd日", CultureInfo.GetCultureInfo("zh-CN"), DateTimeStyles.None, out var date))
+                return date;
+            return null;
+        }
+    }
+
+    [JsonPropertyName("infobox")]
+    public InfoBox? InfoBox { get; set; }
+
+    [JsonIgnore]
+    public string TranslatedName => Configuration.PersonTranslationPreference switch
+    {
+        TranslationPreferenceType.Original => Name,
+        TranslationPreferenceType.Chinese => InfoBox?.GetString("简体中文名") ?? Name,
+        _ => Name
+    };
 }
\ No newline at end of file
diff --git a/Jellyfin.Plugin.Bangumi/Model/Subject.cs b/Jellyfin.Plugin.Bangumi/Model/Subject.cs
index b97fde4..b8db9e8 100644
--- a/Jellyfin.Plugin.Bangumi/Model/Subject.cs
+++ b/Jellyfin.Plugin.Bangumi/Model/Subject.cs
@@ -1,10 +1,11 @@
-using System.Collections.Generic;
+using System;
+using System.Collections.Generic;
+using System.Globalization;
 using System.Linq;
 using System.Net;
 using System.Text.Json.Serialization;
 using Jellyfin.Plugin.Bangumi.Configuration;
 #if !EMBY
-using System;
 using Fastenshtein;
 #endif
 
@@ -83,6 +84,24 @@ public string[] PopularTags
         }
     }
 
+    [JsonPropertyName("infobox")]
+    public InfoBox? InfoBox { get; set; }
+
+    [JsonIgnore]
+    public string? OfficialWebSite => InfoBox?.GetString("官方网站");
+
+    [JsonIgnore]
+    public DateTime? EndDate
+    {
+        get
+        {
+            var dateStr = InfoBox?.GetString("播放结束");
+            if (dateStr != null && DateTime.TryParseExact(dateStr, "yyyy年MM月dd日", CultureInfo.GetCultureInfo("zh-CN"), DateTimeStyles.None, out var date))
+                return date;
+            return null;
+        }
+    }
+
     public static List<Subject> SortBySimilarity(IEnumerable<Subject> list, string keyword)
     {
 #if EMBY
diff --git a/Jellyfin.Plugin.Bangumi/Providers/MovieProvider.cs b/Jellyfin.Plugin.Bangumi/Providers/MovieProvider.cs
index a83cf8e..47d6ffe 100644
--- a/Jellyfin.Plugin.Bangumi/Providers/MovieProvider.cs
+++ b/Jellyfin.Plugin.Bangumi/Providers/MovieProvider.cs
@@ -83,6 +83,8 @@ public async Task<MetadataResult<Movie>> GetMetadata(MovieInfo info, Cancellatio
         result.Item.OriginalTitle = subject.OriginalName;
         result.Item.Overview = string.IsNullOrEmpty(subject.Summary) ? null : subject.Summary;
         result.Item.Tags = subject.PopularTags;
+        result.Item.HomePageUrl = subject.OfficialWebSite;
+        result.Item.EndDate = subject.EndDate;
 
         if (DateTime.TryParse(subject.AirDate, out var airDate))
             result.Item.PremiereDate = airDate;
diff --git a/Jellyfin.Plugin.Bangumi/Providers/PersonProvider.cs b/Jellyfin.Plugin.Bangumi/Providers/PersonProvider.cs
index 28a6756..580a70e 100644
--- a/Jellyfin.Plugin.Bangumi/Providers/PersonProvider.cs
+++ b/Jellyfin.Plugin.Bangumi/Providers/PersonProvider.cs
@@ -28,10 +28,13 @@ public async Task<MetadataResult<Person>> GetMetadata(PersonLookupInfo info, Can
         result.HasMetadata = true;
         result.Item = new Person
         {
-            Name = person.Name,
+            Name = person.TranslatedName,
+            OriginalTitle = person.Name,
             Overview = person.Summary,
             PremiereDate = person.Birthdate,
-            ProductionYear = person.BirthYear
+            ProductionYear = person.BirthYear,
+            ProductionLocations = [person.BirthPlace],
+            EndDate = person.DeathDate
         };
         result.Item.ProviderIds.Add(Constants.ProviderName, $"{person.Id}");
         return result;
diff --git a/Jellyfin.Plugin.Bangumi/Providers/SeasonProvider.cs b/Jellyfin.Plugin.Bangumi/Providers/SeasonProvider.cs
index 8227de4..00cd986 100644
--- a/Jellyfin.Plugin.Bangumi/Providers/SeasonProvider.cs
+++ b/Jellyfin.Plugin.Bangumi/Providers/SeasonProvider.cs
@@ -107,6 +107,9 @@ public async Task<MetadataResult<Season>> GetMetadata(SeasonInfo info, Cancellat
         if (subject.ProductionYear?.Length == 4)
             result.Item.ProductionYear = int.Parse(subject.ProductionYear);
 
+        result.Item.HomePageUrl = subject.OfficialWebSite;
+        result.Item.EndDate = subject.EndDate;
+
         if (subject.IsNSFW)
             result.Item.OfficialRating = "X";
 
diff --git a/Jellyfin.Plugin.Bangumi/Providers/SeriesProvider.cs b/Jellyfin.Plugin.Bangumi/Providers/SeriesProvider.cs
index 74b0b01..388c58d 100644
--- a/Jellyfin.Plugin.Bangumi/Providers/SeriesProvider.cs
+++ b/Jellyfin.Plugin.Bangumi/Providers/SeriesProvider.cs
@@ -97,11 +97,13 @@ public async Task<MetadataResult<Series>> GetMetadata(SeriesInfo info, Cancellat
         result.Item.OriginalTitle = subject.OriginalName;
         result.Item.Overview = string.IsNullOrEmpty(subject.Summary) ? null : subject.Summary;
         result.Item.Tags = subject.PopularTags;
+        result.Item.HomePageUrl = subject.OfficialWebSite;
+        result.Item.EndDate = subject.EndDate;
 
         if (DateTime.TryParse(subject.AirDate, out var airDate))
         {
             result.Item.AirTime = subject.AirDate;
-            result.Item.AirDays = new[] { airDate.DayOfWeek };
+            result.Item.AirDays = [airDate.DayOfWeek];
             result.Item.PremiereDate = airDate;
             result.Item.ProductionYear = airDate.Year;
         }