| | | 1 | | using System; |
| | | 2 | | using System.Collections.Generic; |
| | | 3 | | using System.Collections.Immutable; |
| | | 4 | | using System.Linq; |
| | | 5 | | using System.Threading; |
| | | 6 | | using Jellyfin.Database.Implementations; |
| | | 7 | | using Jellyfin.Database.Implementations.Entities; |
| | | 8 | | using MediaBrowser.Controller; |
| | | 9 | | using MediaBrowser.Controller.Persistence; |
| | | 10 | | using MediaBrowser.Model.Entities; |
| | | 11 | | using MediaBrowser.Model.Globalization; |
| | | 12 | | using Microsoft.EntityFrameworkCore; |
| | | 13 | | |
| | | 14 | | namespace Jellyfin.Server.Implementations.Item; |
| | | 15 | | |
| | | 16 | | /// <summary> |
| | | 17 | | /// Repository for obtaining MediaStreams. |
| | | 18 | | /// </summary> |
| | | 19 | | public class MediaStreamRepository : IMediaStreamRepository |
| | | 20 | | { |
| | | 21 | | private readonly IDbContextFactory<JellyfinDbContext> _dbProvider; |
| | | 22 | | private readonly IServerApplicationHost _serverApplicationHost; |
| | | 23 | | private readonly ILocalizationManager _localization; |
| | | 24 | | |
| | | 25 | | /// <summary> |
| | | 26 | | /// Initializes a new instance of the <see cref="MediaStreamRepository"/> class. |
| | | 27 | | /// </summary> |
| | | 28 | | /// <param name="dbProvider">The EFCore db factory.</param> |
| | | 29 | | /// <param name="serverApplicationHost">The Application host.</param> |
| | | 30 | | /// <param name="localization">The Localisation Provider.</param> |
| | | 31 | | public MediaStreamRepository(IDbContextFactory<JellyfinDbContext> dbProvider, IServerApplicationHost serverApplicati |
| | | 32 | | { |
| | 21 | 33 | | _dbProvider = dbProvider; |
| | 21 | 34 | | _serverApplicationHost = serverApplicationHost; |
| | 21 | 35 | | _localization = localization; |
| | 21 | 36 | | } |
| | | 37 | | |
| | | 38 | | /// <inheritdoc /> |
| | | 39 | | public void SaveMediaStreams(Guid id, IReadOnlyList<MediaStream> streams, CancellationToken cancellationToken) |
| | | 40 | | { |
| | 0 | 41 | | using var context = _dbProvider.CreateDbContext(); |
| | 0 | 42 | | using var transaction = context.Database.BeginTransaction(); |
| | | 43 | | |
| | 0 | 44 | | context.MediaStreamInfos.Where(e => e.ItemId.Equals(id)).ExecuteDelete(); |
| | 0 | 45 | | context.MediaStreamInfos.AddRange(streams.Select(f => Map(f, id))); |
| | 0 | 46 | | context.SaveChanges(); |
| | | 47 | | |
| | 0 | 48 | | transaction.Commit(); |
| | 0 | 49 | | } |
| | | 50 | | |
| | | 51 | | /// <inheritdoc /> |
| | | 52 | | public IReadOnlyList<MediaStream> GetMediaStreams(MediaStreamQuery filter) |
| | | 53 | | { |
| | 0 | 54 | | using var context = _dbProvider.CreateDbContext(); |
| | 0 | 55 | | return TranslateQuery(context.MediaStreamInfos.AsNoTracking(), filter).AsEnumerable().Select(Map).ToArray(); |
| | 0 | 56 | | } |
| | | 57 | | |
| | | 58 | | /// <inheritdoc /> |
| | | 59 | | public IReadOnlyList<string> GetMediaStreamLanguages(MediaStreamType mediaStreamType) |
| | | 60 | | { |
| | 0 | 61 | | using var context = _dbProvider.CreateDbContext(); |
| | 0 | 62 | | return context.MediaStreamInfos |
| | 0 | 63 | | .Where(e => e.StreamType == (MediaStreamTypeEntity)mediaStreamType) |
| | 0 | 64 | | .Select(s => string.IsNullOrEmpty(s.Language) ? "und" : s.Language) // und = undetermined |
| | 0 | 65 | | .Distinct() |
| | 0 | 66 | | .ToArray(); |
| | 0 | 67 | | } |
| | | 68 | | |
| | | 69 | | private string? GetPathToSave(string? path) |
| | | 70 | | { |
| | 0 | 71 | | if (path is null) |
| | | 72 | | { |
| | 0 | 73 | | return null; |
| | | 74 | | } |
| | | 75 | | |
| | 0 | 76 | | return _serverApplicationHost.ReverseVirtualPath(path); |
| | | 77 | | } |
| | | 78 | | |
| | | 79 | | private string? RestorePath(string? path) |
| | | 80 | | { |
| | 0 | 81 | | if (path is null) |
| | | 82 | | { |
| | 0 | 83 | | return null; |
| | | 84 | | } |
| | | 85 | | |
| | 0 | 86 | | return _serverApplicationHost.ExpandVirtualPath(path); |
| | | 87 | | } |
| | | 88 | | |
| | | 89 | | private IQueryable<MediaStreamInfo> TranslateQuery(IQueryable<MediaStreamInfo> query, MediaStreamQuery filter) |
| | | 90 | | { |
| | 0 | 91 | | query = query.Where(e => e.ItemId.Equals(filter.ItemId)); |
| | 0 | 92 | | if (filter.Index.HasValue) |
| | | 93 | | { |
| | 0 | 94 | | query = query.Where(e => e.StreamIndex == filter.Index); |
| | | 95 | | } |
| | | 96 | | |
| | 0 | 97 | | if (filter.Type.HasValue) |
| | | 98 | | { |
| | 0 | 99 | | var typeValue = (MediaStreamTypeEntity)filter.Type.Value; |
| | 0 | 100 | | query = query.Where(e => e.StreamType == typeValue); |
| | | 101 | | } |
| | | 102 | | |
| | 0 | 103 | | return query.OrderBy(e => e.StreamIndex); |
| | | 104 | | } |
| | | 105 | | |
| | | 106 | | private MediaStream Map(MediaStreamInfo entity) |
| | | 107 | | { |
| | 0 | 108 | | var dto = new MediaStream(); |
| | 0 | 109 | | dto.Index = entity.StreamIndex; |
| | 0 | 110 | | dto.Type = (MediaStreamType)entity.StreamType; |
| | | 111 | | |
| | 0 | 112 | | dto.IsAVC = entity.IsAvc; |
| | 0 | 113 | | dto.Codec = entity.Codec; |
| | | 114 | | |
| | 0 | 115 | | var language = entity.Language; |
| | | 116 | | |
| | | 117 | | // Check if the language has multiple three letter ISO codes |
| | | 118 | | // if yes choose the first as that is the ISO 639-2/T code we're needing |
| | 0 | 119 | | if (language != null && _localization.TryGetISO6392TFromB(language, out string? isoT)) |
| | | 120 | | { |
| | 0 | 121 | | language = isoT; |
| | | 122 | | } |
| | | 123 | | |
| | 0 | 124 | | dto.Language = language; |
| | | 125 | | |
| | 0 | 126 | | dto.ChannelLayout = entity.ChannelLayout; |
| | 0 | 127 | | dto.Profile = entity.Profile; |
| | 0 | 128 | | dto.AspectRatio = entity.AspectRatio; |
| | 0 | 129 | | dto.Path = RestorePath(entity.Path); |
| | 0 | 130 | | dto.IsInterlaced = entity.IsInterlaced.GetValueOrDefault(); |
| | 0 | 131 | | dto.BitRate = entity.BitRate; |
| | 0 | 132 | | dto.Channels = entity.Channels; |
| | 0 | 133 | | dto.SampleRate = entity.SampleRate; |
| | 0 | 134 | | dto.IsDefault = entity.IsDefault; |
| | 0 | 135 | | dto.IsForced = entity.IsForced; |
| | 0 | 136 | | dto.IsExternal = entity.IsExternal; |
| | 0 | 137 | | dto.IsOriginal = entity.IsOriginal; |
| | 0 | 138 | | dto.Height = entity.Height; |
| | 0 | 139 | | dto.Width = entity.Width; |
| | 0 | 140 | | dto.AverageFrameRate = entity.AverageFrameRate; |
| | 0 | 141 | | dto.RealFrameRate = entity.RealFrameRate; |
| | 0 | 142 | | dto.Level = entity.Level; |
| | 0 | 143 | | dto.PixelFormat = entity.PixelFormat; |
| | 0 | 144 | | dto.BitDepth = entity.BitDepth; |
| | 0 | 145 | | dto.IsAnamorphic = entity.IsAnamorphic; |
| | 0 | 146 | | dto.RefFrames = entity.RefFrames; |
| | 0 | 147 | | dto.CodecTag = entity.CodecTag; |
| | 0 | 148 | | dto.Comment = entity.Comment; |
| | 0 | 149 | | dto.NalLengthSize = entity.NalLengthSize; |
| | 0 | 150 | | dto.Title = entity.Title; |
| | 0 | 151 | | dto.TimeBase = entity.TimeBase; |
| | 0 | 152 | | dto.CodecTimeBase = entity.CodecTimeBase; |
| | 0 | 153 | | dto.ColorPrimaries = entity.ColorPrimaries; |
| | 0 | 154 | | dto.ColorSpace = entity.ColorSpace; |
| | 0 | 155 | | dto.ColorTransfer = entity.ColorTransfer; |
| | 0 | 156 | | dto.DvVersionMajor = entity.DvVersionMajor; |
| | 0 | 157 | | dto.DvVersionMinor = entity.DvVersionMinor; |
| | 0 | 158 | | dto.DvProfile = entity.DvProfile; |
| | 0 | 159 | | dto.DvLevel = entity.DvLevel; |
| | 0 | 160 | | dto.RpuPresentFlag = entity.RpuPresentFlag; |
| | 0 | 161 | | dto.ElPresentFlag = entity.ElPresentFlag; |
| | 0 | 162 | | dto.BlPresentFlag = entity.BlPresentFlag; |
| | 0 | 163 | | dto.DvBlSignalCompatibilityId = entity.DvBlSignalCompatibilityId; |
| | 0 | 164 | | dto.IsHearingImpaired = entity.IsHearingImpaired.GetValueOrDefault(); |
| | 0 | 165 | | dto.Rotation = entity.Rotation; |
| | 0 | 166 | | dto.Hdr10PlusPresentFlag = entity.Hdr10PlusPresentFlag; |
| | | 167 | | |
| | 0 | 168 | | if (dto.Type is MediaStreamType.Audio or MediaStreamType.Subtitle) |
| | | 169 | | { |
| | 0 | 170 | | dto.LocalizedDefault = _localization.GetLocalizedString("Default"); |
| | 0 | 171 | | dto.LocalizedExternal = _localization.GetLocalizedString("External"); |
| | | 172 | | |
| | 0 | 173 | | if (!string.IsNullOrEmpty(dto.Language)) |
| | | 174 | | { |
| | 0 | 175 | | var culture = _localization.FindLanguageInfo(dto.Language); |
| | 0 | 176 | | dto.LocalizedLanguage = culture?.DisplayName; |
| | | 177 | | } |
| | | 178 | | |
| | 0 | 179 | | if (dto.Type is MediaStreamType.Audio) |
| | | 180 | | { |
| | 0 | 181 | | dto.LocalizedOriginal = _localization.GetLocalizedString("Original"); |
| | | 182 | | } |
| | | 183 | | |
| | 0 | 184 | | if (dto.Type is MediaStreamType.Subtitle) |
| | | 185 | | { |
| | 0 | 186 | | dto.LocalizedUndefined = _localization.GetLocalizedString("Undefined"); |
| | 0 | 187 | | dto.LocalizedForced = _localization.GetLocalizedString("Forced"); |
| | 0 | 188 | | dto.LocalizedHearingImpaired = _localization.GetLocalizedString("HearingImpaired"); |
| | | 189 | | } |
| | | 190 | | } |
| | | 191 | | |
| | 0 | 192 | | return dto; |
| | | 193 | | } |
| | | 194 | | |
| | | 195 | | private MediaStreamInfo Map(MediaStream dto, Guid itemId) |
| | | 196 | | { |
| | 0 | 197 | | var entity = new MediaStreamInfo |
| | 0 | 198 | | { |
| | 0 | 199 | | Item = null!, |
| | 0 | 200 | | ItemId = itemId, |
| | 0 | 201 | | StreamIndex = dto.Index, |
| | 0 | 202 | | StreamType = (MediaStreamTypeEntity)dto.Type, |
| | 0 | 203 | | IsAvc = dto.IsAVC, |
| | 0 | 204 | | |
| | 0 | 205 | | Codec = dto.Codec, |
| | 0 | 206 | | Language = dto.Language, |
| | 0 | 207 | | ChannelLayout = dto.ChannelLayout, |
| | 0 | 208 | | Profile = dto.Profile, |
| | 0 | 209 | | AspectRatio = dto.AspectRatio, |
| | 0 | 210 | | Path = GetPathToSave(dto.Path) ?? dto.Path, |
| | 0 | 211 | | IsInterlaced = dto.IsInterlaced, |
| | 0 | 212 | | BitRate = dto.BitRate, |
| | 0 | 213 | | Channels = dto.Channels, |
| | 0 | 214 | | SampleRate = dto.SampleRate, |
| | 0 | 215 | | IsDefault = dto.IsDefault, |
| | 0 | 216 | | IsForced = dto.IsForced, |
| | 0 | 217 | | IsExternal = dto.IsExternal, |
| | 0 | 218 | | IsOriginal = dto.IsOriginal, |
| | 0 | 219 | | Height = dto.Height, |
| | 0 | 220 | | Width = dto.Width, |
| | 0 | 221 | | AverageFrameRate = dto.AverageFrameRate, |
| | 0 | 222 | | RealFrameRate = dto.RealFrameRate, |
| | 0 | 223 | | Level = dto.Level.HasValue ? (float)dto.Level : null, |
| | 0 | 224 | | PixelFormat = dto.PixelFormat, |
| | 0 | 225 | | BitDepth = dto.BitDepth, |
| | 0 | 226 | | IsAnamorphic = dto.IsAnamorphic, |
| | 0 | 227 | | RefFrames = dto.RefFrames, |
| | 0 | 228 | | CodecTag = dto.CodecTag, |
| | 0 | 229 | | Comment = dto.Comment, |
| | 0 | 230 | | NalLengthSize = dto.NalLengthSize, |
| | 0 | 231 | | Title = dto.Title, |
| | 0 | 232 | | TimeBase = dto.TimeBase, |
| | 0 | 233 | | CodecTimeBase = dto.CodecTimeBase, |
| | 0 | 234 | | ColorPrimaries = dto.ColorPrimaries, |
| | 0 | 235 | | ColorSpace = dto.ColorSpace, |
| | 0 | 236 | | ColorTransfer = dto.ColorTransfer, |
| | 0 | 237 | | DvVersionMajor = dto.DvVersionMajor, |
| | 0 | 238 | | DvVersionMinor = dto.DvVersionMinor, |
| | 0 | 239 | | DvProfile = dto.DvProfile, |
| | 0 | 240 | | DvLevel = dto.DvLevel, |
| | 0 | 241 | | RpuPresentFlag = dto.RpuPresentFlag, |
| | 0 | 242 | | ElPresentFlag = dto.ElPresentFlag, |
| | 0 | 243 | | BlPresentFlag = dto.BlPresentFlag, |
| | 0 | 244 | | DvBlSignalCompatibilityId = dto.DvBlSignalCompatibilityId, |
| | 0 | 245 | | IsHearingImpaired = dto.IsHearingImpaired, |
| | 0 | 246 | | Rotation = dto.Rotation, |
| | 0 | 247 | | Hdr10PlusPresentFlag = dto.Hdr10PlusPresentFlag, |
| | 0 | 248 | | }; |
| | 0 | 249 | | return entity; |
| | | 250 | | } |
| | | 251 | | } |