Add OriginalLanguage as option to PreferredAudioLanguage (#12579)

* Add OriginalLanguage as option to PreferredAudioLanguage

* Support for multiple original languages

* Add original audio stream indicator

* Fetch OriginalLanguage from TMDB

* Adapt to EFCore refactor

* Fix PlayDefaultAudioTrack OriginalLanguage behavior

* Fix better PlayDefaultAudioTrack OriginalLanguage behavior

* Add comment to ItemFields

* Improved PlayDefaultAudioTrack behavior

* Add migration for original language

* Use sting.Equals for string comparisons

* Always set dto OriginalLanguage

* Remove OriginalLanguage from ItemFields

---------

Co-authored-by: Lampan-git <lampan-git@users.noreply.github.com>

1 files changed, 116 insertions, 0 deletions

diff --git a/tests/Jellyfin.Server.Implementations.Tests/Library/MediaSourceManagerTests.cs b/tests/Jellyfin.Server.Implementations.Tests/Library/MediaSourceManagerTests.cs
index 8ed3d8b944..facdb2bc2e 100644
--- a/tests/Jellyfin.Server.Implementations.Tests/Library/MediaSourceManagerTests.cs
+++ b/tests/Jellyfin.Server.Implementations.Tests/Library/MediaSourceManagerTests.cs
@@ -1,9 +1,18 @@
+using System;
 using AutoFixture;
 using AutoFixture.AutoMoq;
+using Castle.Components.DictionaryAdapter;
 using Emby.Server.Implementations.IO;
 using Emby.Server.Implementations.Library;
+using Jellyfin.Database.Implementations.Entities;
+using MediaBrowser.Controller.Entities;
+using MediaBrowser.Controller.Library;
+using MediaBrowser.Model.Dto;
+using MediaBrowser.Model.Entities;
+using MediaBrowser.Model.Globalization;
 using MediaBrowser.Model.IO;
 using MediaBrowser.Model.MediaInfo;
+using Moq;
 using Xunit;
 
 namespace Jellyfin.Server.Implementations.Tests.Library
@@ -11,12 +20,28 @@ namespace Jellyfin.Server.Implementations.Tests.Library
     public class MediaSourceManagerTests
     {
         private readonly MediaSourceManager _mediaSourceManager;
+        private readonly Mock<IUserDataManager> _mockUserDataManager;
+        private readonly Mock<ILocalizationManager> _mockLocalizationManager;
+        private Video _item;
+        private User _user;
 
         public MediaSourceManagerTests()
         {
             IFixture fixture = new Fixture().Customize(new AutoMoqCustomization { ConfigureMembers = true });
             fixture.Inject<IFileSystem>(fixture.Create<ManagedFileSystem>());
+
+            _mockUserDataManager = fixture.Freeze<Mock<IUserDataManager>>();
+            _mockUserDataManager.Setup(m => m.GetUserData(It.IsAny<User>(), It.IsAny<BaseItem>())).Returns(new UserItemData() { Key = "key" });
+
+            _mockLocalizationManager = fixture.Create<Mock<ILocalizationManager>>();
+            _mockLocalizationManager.Setup(m => m.FindLanguageInfo(It.IsAny<string>())).Returns((string s) => string.IsNullOrEmpty(s) ? null : new CultureDto(s, s, s, new EditableList<string> { s }));
+            fixture.Inject(_mockLocalizationManager.Object);
+
             _mediaSourceManager = fixture.Create<MediaSourceManager>();
+
+            _item = new Video { Id = Guid.NewGuid(), OwnerId = Guid.Empty, ParentId = Guid.Empty };
+
+            _user = fixture.Create<User>();
         }
 
         [Theory]
@@ -28,5 +53,96 @@ namespace Jellyfin.Server.Implementations.Tests.Library
         [InlineData("rtsp://media.example.com:554/twister/audiotrack", MediaProtocol.Rtsp)]
         public void GetPathProtocol_ValidArg_Correct(string path, MediaProtocol expected)
             => Assert.Equal(expected, _mediaSourceManager.GetPathProtocol(path));
+
+        [Theory]
+        [InlineData(5, "eng", "eng", false, true)]
+        [InlineData(5, "eng", "eng", true, true)]
+        [InlineData(2, "ger", "eng", false, true)]
+        [InlineData(2, "ger", "eng", true, true)]
+        [InlineData(1, "fre", "eng", false, true)]
+        [InlineData(2, "fre", "eng", true, true)]
+        [InlineData(5, "OriginalLanguage", "eng", false, false)]
+        [InlineData(4, "OriginalLanguage", "eng", false, true)]
+        [InlineData(5, "OriginalLanguage", "eng", true, false)]
+        [InlineData(5, "OriginalLanguage", "eng", true, true)]
+        [InlineData(2, "OriginalLanguage", "jpn", true, true)]
+        [InlineData(2, "OriginalLanguage", "jpn", false, true)]
+        [InlineData(2, "OriginalLanguage", "jpn,eng", false, true)]
+        [InlineData(4, "OriginalLanguage", null, false, true)]
+        [InlineData(2, "OriginalLanguage", null, true, true)]
+        [InlineData(4, "OriginalLanguage", "", false, true)]
+        [InlineData(2, "OriginalLanguage", "", false, false)]
+        [InlineData(2, "OriginalLanguage", "ger", false, true)]
+        [InlineData(2, "OriginalLanguage", "ger", false, false)]
+        [InlineData(1, "OriginalLanguage", "fre", false, false)]
+        [InlineData(2, "OriginalLanguage", "fre", true, true)]
+        [InlineData(2, "OriginalLanguage", "fre", true, false)]
+        public void SetDefaultAudioStreamIndex_Index_Correct(
+            int expectedIndex,
+            string prefferedLanguage,
+            string? originalLanguage,
+            bool playDefault,
+            bool originalExist)
+        {
+            var streams = new MediaStream[]
+            {
+                new()
+                {
+                    Index = 0,
+                    Type = MediaStreamType.Video,
+                    IsDefault = true
+                },
+                new()
+                {
+                    Index = 1,
+                    Type = MediaStreamType.Audio,
+                    Language = "fre",
+                    IsDefault = false,
+                    IsOriginal = false
+                },
+                new()
+                {
+                    Index = 2,
+                    Type = MediaStreamType.Audio,
+                    Language = "jpn",
+                    IsDefault = true,
+                    IsOriginal = false
+                },
+                new()
+                {
+                    Index = 3,
+                    Type = MediaStreamType.Audio,
+                    Language = "eng",
+                    IsDefault = false,
+                    IsOriginal = false
+                },
+                new()
+                {
+                    Index = 4,
+                    Type = MediaStreamType.Audio,
+                    Language = "eng",
+                    IsDefault = false,
+                    IsOriginal = originalExist,
+                },
+                new()
+                {
+                    Index = 5,
+                    Type = MediaStreamType.Audio,
+                    Language = "eng",
+                    IsDefault = true,
+                    IsOriginal = false,
+                }
+            };
+            var mediaInfo = new MediaSourceInfo
+            {
+                MediaStreams = streams
+            };
+            _user.AudioLanguagePreference = prefferedLanguage;
+            _user.PlayDefaultAudioTrack = playDefault;
+            _item.OriginalLanguage = originalLanguage;
+
+            _mediaSourceManager.SetDefaultAudioAndSubtitleStreamIndices(_item, mediaInfo, _user);
+            Assert.Equal(expectedIndex, mediaInfo.DefaultAudioStreamIndex);
+        }
     }
 }