| | | 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 | | private string? GetPathToSave(string? path) |
| | | 59 | | { |
| | 0 | 60 | | if (path is null) |
| | | 61 | | { |
| | 0 | 62 | | return null; |
| | | 63 | | } |
| | | 64 | | |
| | 0 | 65 | | return _serverApplicationHost.ReverseVirtualPath(path); |
| | | 66 | | } |
| | | 67 | | |
| | | 68 | | private string? RestorePath(string? path) |
| | | 69 | | { |
| | 0 | 70 | | if (path is null) |
| | | 71 | | { |
| | 0 | 72 | | return null; |
| | | 73 | | } |
| | | 74 | | |
| | 0 | 75 | | return _serverApplicationHost.ExpandVirtualPath(path); |
| | | 76 | | } |
| | | 77 | | |
| | | 78 | | private IQueryable<MediaStreamInfo> TranslateQuery(IQueryable<MediaStreamInfo> query, MediaStreamQuery filter) |
| | | 79 | | { |
| | 0 | 80 | | query = query.Where(e => e.ItemId.Equals(filter.ItemId)); |
| | 0 | 81 | | if (filter.Index.HasValue) |
| | | 82 | | { |
| | 0 | 83 | | query = query.Where(e => e.StreamIndex == filter.Index); |
| | | 84 | | } |
| | | 85 | | |
| | 0 | 86 | | if (filter.Type.HasValue) |
| | | 87 | | { |
| | 0 | 88 | | var typeValue = (MediaStreamTypeEntity)filter.Type.Value; |
| | 0 | 89 | | query = query.Where(e => e.StreamType == typeValue); |
| | | 90 | | } |
| | | 91 | | |
| | 0 | 92 | | return query.OrderBy(e => e.StreamIndex); |
| | | 93 | | } |
| | | 94 | | |
| | | 95 | | private MediaStream Map(MediaStreamInfo entity) |
| | | 96 | | { |
| | 0 | 97 | | var dto = new MediaStream(); |
| | 0 | 98 | | dto.Index = entity.StreamIndex; |
| | 0 | 99 | | dto.Type = (MediaStreamType)entity.StreamType; |
| | | 100 | | |
| | 0 | 101 | | dto.IsAVC = entity.IsAvc; |
| | 0 | 102 | | dto.Codec = entity.Codec; |
| | 0 | 103 | | dto.Language = entity.Language; |
| | 0 | 104 | | dto.ChannelLayout = entity.ChannelLayout; |
| | 0 | 105 | | dto.Profile = entity.Profile; |
| | 0 | 106 | | dto.AspectRatio = entity.AspectRatio; |
| | 0 | 107 | | dto.Path = RestorePath(entity.Path); |
| | 0 | 108 | | dto.IsInterlaced = entity.IsInterlaced.GetValueOrDefault(); |
| | 0 | 109 | | dto.BitRate = entity.BitRate; |
| | 0 | 110 | | dto.Channels = entity.Channels; |
| | 0 | 111 | | dto.SampleRate = entity.SampleRate; |
| | 0 | 112 | | dto.IsDefault = entity.IsDefault; |
| | 0 | 113 | | dto.IsForced = entity.IsForced; |
| | 0 | 114 | | dto.IsExternal = entity.IsExternal; |
| | 0 | 115 | | dto.Height = entity.Height; |
| | 0 | 116 | | dto.Width = entity.Width; |
| | 0 | 117 | | dto.AverageFrameRate = entity.AverageFrameRate; |
| | 0 | 118 | | dto.RealFrameRate = entity.RealFrameRate; |
| | 0 | 119 | | dto.Level = entity.Level; |
| | 0 | 120 | | dto.PixelFormat = entity.PixelFormat; |
| | 0 | 121 | | dto.BitDepth = entity.BitDepth; |
| | 0 | 122 | | dto.IsAnamorphic = entity.IsAnamorphic; |
| | 0 | 123 | | dto.RefFrames = entity.RefFrames; |
| | 0 | 124 | | dto.CodecTag = entity.CodecTag; |
| | 0 | 125 | | dto.Comment = entity.Comment; |
| | 0 | 126 | | dto.NalLengthSize = entity.NalLengthSize; |
| | 0 | 127 | | dto.Title = entity.Title; |
| | 0 | 128 | | dto.TimeBase = entity.TimeBase; |
| | 0 | 129 | | dto.CodecTimeBase = entity.CodecTimeBase; |
| | 0 | 130 | | dto.ColorPrimaries = entity.ColorPrimaries; |
| | 0 | 131 | | dto.ColorSpace = entity.ColorSpace; |
| | 0 | 132 | | dto.ColorTransfer = entity.ColorTransfer; |
| | 0 | 133 | | dto.DvVersionMajor = entity.DvVersionMajor; |
| | 0 | 134 | | dto.DvVersionMinor = entity.DvVersionMinor; |
| | 0 | 135 | | dto.DvProfile = entity.DvProfile; |
| | 0 | 136 | | dto.DvLevel = entity.DvLevel; |
| | 0 | 137 | | dto.RpuPresentFlag = entity.RpuPresentFlag; |
| | 0 | 138 | | dto.ElPresentFlag = entity.ElPresentFlag; |
| | 0 | 139 | | dto.BlPresentFlag = entity.BlPresentFlag; |
| | 0 | 140 | | dto.DvBlSignalCompatibilityId = entity.DvBlSignalCompatibilityId; |
| | 0 | 141 | | dto.IsHearingImpaired = entity.IsHearingImpaired.GetValueOrDefault(); |
| | 0 | 142 | | dto.Rotation = entity.Rotation; |
| | | 143 | | |
| | 0 | 144 | | if (dto.Type is MediaStreamType.Audio or MediaStreamType.Subtitle) |
| | | 145 | | { |
| | 0 | 146 | | dto.LocalizedDefault = _localization.GetLocalizedString("Default"); |
| | 0 | 147 | | dto.LocalizedExternal = _localization.GetLocalizedString("External"); |
| | | 148 | | |
| | 0 | 149 | | if (dto.Type is MediaStreamType.Subtitle) |
| | | 150 | | { |
| | 0 | 151 | | dto.LocalizedUndefined = _localization.GetLocalizedString("Undefined"); |
| | 0 | 152 | | dto.LocalizedForced = _localization.GetLocalizedString("Forced"); |
| | 0 | 153 | | dto.LocalizedHearingImpaired = _localization.GetLocalizedString("HearingImpaired"); |
| | | 154 | | } |
| | | 155 | | } |
| | | 156 | | |
| | 0 | 157 | | return dto; |
| | | 158 | | } |
| | | 159 | | |
| | | 160 | | private MediaStreamInfo Map(MediaStream dto, Guid itemId) |
| | | 161 | | { |
| | 0 | 162 | | var entity = new MediaStreamInfo |
| | 0 | 163 | | { |
| | 0 | 164 | | Item = null!, |
| | 0 | 165 | | ItemId = itemId, |
| | 0 | 166 | | StreamIndex = dto.Index, |
| | 0 | 167 | | StreamType = (MediaStreamTypeEntity)dto.Type, |
| | 0 | 168 | | IsAvc = dto.IsAVC, |
| | 0 | 169 | | |
| | 0 | 170 | | Codec = dto.Codec, |
| | 0 | 171 | | Language = dto.Language, |
| | 0 | 172 | | ChannelLayout = dto.ChannelLayout, |
| | 0 | 173 | | Profile = dto.Profile, |
| | 0 | 174 | | AspectRatio = dto.AspectRatio, |
| | 0 | 175 | | Path = GetPathToSave(dto.Path) ?? dto.Path, |
| | 0 | 176 | | IsInterlaced = dto.IsInterlaced, |
| | 0 | 177 | | BitRate = dto.BitRate, |
| | 0 | 178 | | Channels = dto.Channels, |
| | 0 | 179 | | SampleRate = dto.SampleRate, |
| | 0 | 180 | | IsDefault = dto.IsDefault, |
| | 0 | 181 | | IsForced = dto.IsForced, |
| | 0 | 182 | | IsExternal = dto.IsExternal, |
| | 0 | 183 | | Height = dto.Height, |
| | 0 | 184 | | Width = dto.Width, |
| | 0 | 185 | | AverageFrameRate = dto.AverageFrameRate, |
| | 0 | 186 | | RealFrameRate = dto.RealFrameRate, |
| | 0 | 187 | | Level = dto.Level.HasValue ? (float)dto.Level : null, |
| | 0 | 188 | | PixelFormat = dto.PixelFormat, |
| | 0 | 189 | | BitDepth = dto.BitDepth, |
| | 0 | 190 | | IsAnamorphic = dto.IsAnamorphic, |
| | 0 | 191 | | RefFrames = dto.RefFrames, |
| | 0 | 192 | | CodecTag = dto.CodecTag, |
| | 0 | 193 | | Comment = dto.Comment, |
| | 0 | 194 | | NalLengthSize = dto.NalLengthSize, |
| | 0 | 195 | | Title = dto.Title, |
| | 0 | 196 | | TimeBase = dto.TimeBase, |
| | 0 | 197 | | CodecTimeBase = dto.CodecTimeBase, |
| | 0 | 198 | | ColorPrimaries = dto.ColorPrimaries, |
| | 0 | 199 | | ColorSpace = dto.ColorSpace, |
| | 0 | 200 | | ColorTransfer = dto.ColorTransfer, |
| | 0 | 201 | | DvVersionMajor = dto.DvVersionMajor, |
| | 0 | 202 | | DvVersionMinor = dto.DvVersionMinor, |
| | 0 | 203 | | DvProfile = dto.DvProfile, |
| | 0 | 204 | | DvLevel = dto.DvLevel, |
| | 0 | 205 | | RpuPresentFlag = dto.RpuPresentFlag, |
| | 0 | 206 | | ElPresentFlag = dto.ElPresentFlag, |
| | 0 | 207 | | BlPresentFlag = dto.BlPresentFlag, |
| | 0 | 208 | | DvBlSignalCompatibilityId = dto.DvBlSignalCompatibilityId, |
| | 0 | 209 | | IsHearingImpaired = dto.IsHearingImpaired, |
| | 0 | 210 | | Rotation = dto.Rotation |
| | 0 | 211 | | }; |
| | 0 | 212 | | return entity; |
| | | 213 | | } |
| | | 214 | | } |