mirror of https://github.com/ErsatzTV/ErsatzTV.git
Browse Source
* catch health check cancellation * local library scanner cleanup * emby and jf library scanner cleanup * rework emby movie library scanner * refactor emby movie library scanner * refactor jellyfin movie library scanner * clear etag until after new media has been processed * refactor plex movie library scanner * update changelogpull/762/head
46 changed files with 2768 additions and 2454 deletions
@ -0,0 +1,3 @@ |
|||||||
|
namespace ErsatzTV.Core.Domain.MediaServer; |
||||||
|
|
||||||
|
public abstract record MediaServerConnectionParameters; |
@ -0,0 +1,8 @@ |
|||||||
|
namespace ErsatzTV.Core.Domain; |
||||||
|
|
||||||
|
public abstract class MediaServerItemEtag |
||||||
|
{ |
||||||
|
public abstract string MediaServerItemId { get; } |
||||||
|
public abstract string Etag { get; set; } |
||||||
|
public abstract MediaItemState State { get; set; } |
||||||
|
} |
@ -0,0 +1,5 @@ |
|||||||
|
using ErsatzTV.Core.Domain.MediaServer; |
||||||
|
|
||||||
|
namespace ErsatzTV.Core.Emby; |
||||||
|
|
||||||
|
public record EmbyConnectionParameters(string Address, string ApiKey) : MediaServerConnectionParameters; |
@ -1,7 +1,11 @@ |
|||||||
namespace ErsatzTV.Core.Emby; |
using ErsatzTV.Core.Domain; |
||||||
|
|
||||||
public class EmbyItemEtag |
namespace ErsatzTV.Core.Emby; |
||||||
|
|
||||||
|
public class EmbyItemEtag : MediaServerItemEtag |
||||||
{ |
{ |
||||||
public string ItemId { get; set; } |
public string ItemId { get; set; } |
||||||
public string Etag { get; set; } |
public override string MediaServerItemId => ItemId; |
||||||
|
public override string Etag { get; set; } |
||||||
|
public override MediaItemState State { get; set; } |
||||||
} |
} |
||||||
|
@ -0,0 +1,8 @@ |
|||||||
|
using ErsatzTV.Core.Domain; |
||||||
|
using ErsatzTV.Core.Emby; |
||||||
|
|
||||||
|
namespace ErsatzTV.Core.Interfaces.Repositories; |
||||||
|
|
||||||
|
public interface IEmbyMovieRepository : IMediaServerMovieRepository<EmbyLibrary, EmbyMovie, EmbyItemEtag> |
||||||
|
{ |
||||||
|
} |
@ -0,0 +1,9 @@ |
|||||||
|
using ErsatzTV.Core.Domain; |
||||||
|
using ErsatzTV.Core.Jellyfin; |
||||||
|
|
||||||
|
namespace ErsatzTV.Core.Interfaces.Repositories; |
||||||
|
|
||||||
|
public interface |
||||||
|
IJellyfinMovieRepository : IMediaServerMovieRepository<JellyfinLibrary, JellyfinMovie, JellyfinItemEtag> |
||||||
|
{ |
||||||
|
} |
@ -0,0 +1,16 @@ |
|||||||
|
using ErsatzTV.Core.Domain; |
||||||
|
using ErsatzTV.Core.Metadata; |
||||||
|
|
||||||
|
namespace ErsatzTV.Core.Interfaces.Repositories; |
||||||
|
|
||||||
|
public interface IMediaServerMovieRepository<in TLibrary, TMovie, TEtag> where TLibrary : Library |
||||||
|
where TMovie : Movie |
||||||
|
where TEtag : MediaServerItemEtag |
||||||
|
{ |
||||||
|
Task<List<TEtag>> GetExistingMovies(TLibrary library); |
||||||
|
Task<bool> FlagNormal(TLibrary library, TMovie movie); |
||||||
|
Task<Option<int>> FlagUnavailable(TLibrary library, TMovie movie); |
||||||
|
Task<List<int>> FlagFileNotFound(TLibrary library, List<string> movieItemIds); |
||||||
|
Task<Either<BaseError, MediaItemScanResult<TMovie>>> GetOrAdd(TLibrary library, TMovie item); |
||||||
|
Task<Unit> SetEtag(TMovie movie, string etag); |
||||||
|
} |
@ -1,10 +1,8 @@ |
|||||||
using ErsatzTV.Core.Domain; |
using ErsatzTV.Core.Domain; |
||||||
|
using ErsatzTV.Core.Plex; |
||||||
|
|
||||||
namespace ErsatzTV.Core.Interfaces.Repositories; |
namespace ErsatzTV.Core.Interfaces.Repositories; |
||||||
|
|
||||||
public interface IPlexMovieRepository |
public interface IPlexMovieRepository : IMediaServerMovieRepository<PlexLibrary, PlexMovie, PlexItemEtag> |
||||||
{ |
{ |
||||||
Task<bool> FlagNormal(PlexLibrary library, PlexMovie movie); |
|
||||||
Task<Option<int>> FlagUnavailable(PlexLibrary library, PlexMovie movie); |
|
||||||
Task<List<int>> FlagFileNotFound(PlexLibrary library, List<string> plexMovieKeys); |
|
||||||
} |
} |
||||||
|
@ -0,0 +1,6 @@ |
|||||||
|
using ErsatzTV.Core.Domain.MediaServer; |
||||||
|
|
||||||
|
namespace ErsatzTV.Core.Jellyfin; |
||||||
|
|
||||||
|
public record JellyfinConnectionParameters |
||||||
|
(string Address, string ApiKey, int MediaSourceId) : MediaServerConnectionParameters; |
@ -1,7 +1,11 @@ |
|||||||
namespace ErsatzTV.Core.Jellyfin; |
using ErsatzTV.Core.Domain; |
||||||
|
|
||||||
public class JellyfinItemEtag |
namespace ErsatzTV.Core.Jellyfin; |
||||||
|
|
||||||
|
public class JellyfinItemEtag : MediaServerItemEtag |
||||||
{ |
{ |
||||||
public string ItemId { get; set; } |
public string ItemId { get; set; } |
||||||
public string Etag { get; set; } |
public override string MediaServerItemId => ItemId; |
||||||
|
public override string Etag { get; set; } |
||||||
|
public override MediaItemState State { get; set; } |
||||||
} |
} |
||||||
|
@ -0,0 +1,338 @@ |
|||||||
|
using ErsatzTV.Core.Domain; |
||||||
|
using ErsatzTV.Core.Domain.MediaServer; |
||||||
|
using ErsatzTV.Core.Errors; |
||||||
|
using ErsatzTV.Core.Interfaces.Metadata; |
||||||
|
using ErsatzTV.Core.Interfaces.Repositories; |
||||||
|
using ErsatzTV.Core.Interfaces.Search; |
||||||
|
using MediatR; |
||||||
|
using Microsoft.Extensions.Logging; |
||||||
|
|
||||||
|
namespace ErsatzTV.Core.Metadata; |
||||||
|
|
||||||
|
public abstract class MediaServerMovieLibraryScanner<TConnectionParameters, TLibrary, TMovie, TEtag> |
||||||
|
where TConnectionParameters : MediaServerConnectionParameters |
||||||
|
where TLibrary : Library |
||||||
|
where TMovie : Movie |
||||||
|
where TEtag : MediaServerItemEtag |
||||||
|
{ |
||||||
|
private readonly ILocalFileSystem _localFileSystem; |
||||||
|
private readonly ILocalStatisticsProvider _localStatisticsProvider; |
||||||
|
private readonly ILocalSubtitlesProvider _localSubtitlesProvider; |
||||||
|
private readonly ILogger _logger; |
||||||
|
private readonly IMediator _mediator; |
||||||
|
private readonly ISearchIndex _searchIndex; |
||||||
|
private readonly ISearchRepository _searchRepository; |
||||||
|
|
||||||
|
protected MediaServerMovieLibraryScanner( |
||||||
|
ILocalStatisticsProvider localStatisticsProvider, |
||||||
|
ILocalSubtitlesProvider localSubtitlesProvider, |
||||||
|
ILocalFileSystem localFileSystem, |
||||||
|
IMediator mediator, |
||||||
|
ISearchIndex searchIndex, |
||||||
|
ISearchRepository searchRepository, |
||||||
|
ILogger logger) |
||||||
|
{ |
||||||
|
_localStatisticsProvider = localStatisticsProvider; |
||||||
|
_localSubtitlesProvider = localSubtitlesProvider; |
||||||
|
_localFileSystem = localFileSystem; |
||||||
|
_mediator = mediator; |
||||||
|
_searchIndex = searchIndex; |
||||||
|
_searchRepository = searchRepository; |
||||||
|
_logger = logger; |
||||||
|
} |
||||||
|
|
||||||
|
protected async Task<Either<BaseError, Unit>> ScanLibrary( |
||||||
|
IMediaServerMovieRepository<TLibrary, TMovie, TEtag> movieRepository, |
||||||
|
TConnectionParameters connectionParameters, |
||||||
|
TLibrary library, |
||||||
|
Func<TMovie, string> getLocalPath, |
||||||
|
string ffmpegPath, |
||||||
|
string ffprobePath, |
||||||
|
bool deepScan, |
||||||
|
CancellationToken cancellationToken) |
||||||
|
{ |
||||||
|
try |
||||||
|
{ |
||||||
|
Either<BaseError, List<TMovie>> entries = await GetMovieLibraryItems(connectionParameters, library); |
||||||
|
|
||||||
|
foreach (BaseError error in entries.LeftToSeq()) |
||||||
|
{ |
||||||
|
return error; |
||||||
|
} |
||||||
|
|
||||||
|
return await ScanLibrary( |
||||||
|
movieRepository, |
||||||
|
connectionParameters, |
||||||
|
library, |
||||||
|
getLocalPath, |
||||||
|
ffmpegPath, |
||||||
|
ffprobePath, |
||||||
|
entries.RightToSeq().Flatten().ToList(), |
||||||
|
deepScan, |
||||||
|
cancellationToken); |
||||||
|
} |
||||||
|
catch (Exception ex) when (ex is TaskCanceledException or OperationCanceledException) |
||||||
|
{ |
||||||
|
return new ScanCanceled(); |
||||||
|
} |
||||||
|
finally |
||||||
|
{ |
||||||
|
_searchIndex.Commit(); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
private async Task<Either<BaseError, Unit>> ScanLibrary( |
||||||
|
IMediaServerMovieRepository<TLibrary, TMovie, TEtag> movieRepository, |
||||||
|
TConnectionParameters connectionParameters, |
||||||
|
TLibrary library, |
||||||
|
Func<TMovie, string> getLocalPath, |
||||||
|
string ffmpegPath, |
||||||
|
string ffprobePath, |
||||||
|
List<TMovie> movieEntries, |
||||||
|
bool deepScan, |
||||||
|
CancellationToken cancellationToken) |
||||||
|
{ |
||||||
|
List<TEtag> existingMovies = await movieRepository.GetExistingMovies(library); |
||||||
|
|
||||||
|
foreach (TMovie incoming in movieEntries) |
||||||
|
{ |
||||||
|
if (cancellationToken.IsCancellationRequested) |
||||||
|
{ |
||||||
|
return new ScanCanceled(); |
||||||
|
} |
||||||
|
|
||||||
|
decimal percentCompletion = (decimal)movieEntries.IndexOf(incoming) / movieEntries.Count; |
||||||
|
await _mediator.Publish(new LibraryScanProgress(library.Id, percentCompletion), cancellationToken); |
||||||
|
|
||||||
|
string localPath = getLocalPath(incoming); |
||||||
|
|
||||||
|
if (await ShouldScanItem(movieRepository, library, existingMovies, incoming, localPath, deepScan) == false) |
||||||
|
{ |
||||||
|
continue; |
||||||
|
} |
||||||
|
|
||||||
|
Either<BaseError, MediaItemScanResult<TMovie>> maybeMovie = await movieRepository |
||||||
|
.GetOrAdd(library, incoming) |
||||||
|
.MapT( |
||||||
|
result => |
||||||
|
{ |
||||||
|
result.LocalPath = localPath; |
||||||
|
return result; |
||||||
|
}) |
||||||
|
.BindT(existing => UpdateMetadata(connectionParameters, library, existing, incoming, deepScan)) |
||||||
|
.BindT(existing => UpdateStatistics(existing, incoming, ffmpegPath, ffprobePath)) |
||||||
|
.BindT(UpdateSubtitles); |
||||||
|
|
||||||
|
if (maybeMovie.IsLeft) |
||||||
|
{ |
||||||
|
foreach (BaseError error in maybeMovie.LeftToSeq()) |
||||||
|
{ |
||||||
|
_logger.LogWarning( |
||||||
|
"Error processing movie {Title}: {Error}", |
||||||
|
incoming.MovieMetadata.Head().Title, |
||||||
|
error.Value); |
||||||
|
} |
||||||
|
|
||||||
|
continue; |
||||||
|
} |
||||||
|
|
||||||
|
foreach (MediaItemScanResult<TMovie> result in maybeMovie.RightToSeq()) |
||||||
|
{ |
||||||
|
await movieRepository.SetEtag(result.Item, MediaServerEtag(incoming)); |
||||||
|
|
||||||
|
if (_localFileSystem.FileExists(result.LocalPath)) |
||||||
|
{ |
||||||
|
if (await movieRepository.FlagNormal(library, result.Item)) |
||||||
|
{ |
||||||
|
result.IsUpdated = true; |
||||||
|
} |
||||||
|
} |
||||||
|
else |
||||||
|
{ |
||||||
|
Option<int> flagResult = await movieRepository.FlagUnavailable(library, result.Item); |
||||||
|
if (flagResult.IsSome) |
||||||
|
{ |
||||||
|
result.IsUpdated = true; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
if (result.IsAdded) |
||||||
|
{ |
||||||
|
await _searchIndex.AddItems(_searchRepository, new List<MediaItem> { result.Item }); |
||||||
|
} |
||||||
|
else if (result.IsUpdated) |
||||||
|
{ |
||||||
|
await _searchIndex.UpdateItems(_searchRepository, new List<MediaItem> { result.Item }); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// trash items that are no longer present on the media server
|
||||||
|
var fileNotFoundItemIds = existingMovies.Map(m => m.MediaServerItemId) |
||||||
|
.Except(movieEntries.Map(MediaServerItemId)).ToList(); |
||||||
|
List<int> ids = await movieRepository.FlagFileNotFound(library, fileNotFoundItemIds); |
||||||
|
await _searchIndex.RebuildItems(_searchRepository, ids); |
||||||
|
|
||||||
|
await _mediator.Publish(new LibraryScanProgress(library.Id, 0), cancellationToken); |
||||||
|
|
||||||
|
return Unit.Default; |
||||||
|
} |
||||||
|
|
||||||
|
protected abstract string MediaServerItemId(TMovie movie); |
||||||
|
protected abstract string MediaServerEtag(TMovie movie); |
||||||
|
|
||||||
|
protected abstract Task<Either<BaseError, List<TMovie>>> GetMovieLibraryItems( |
||||||
|
TConnectionParameters connectionParameters, |
||||||
|
TLibrary library); |
||||||
|
|
||||||
|
protected abstract Task<Option<MovieMetadata>> GetFullMetadata( |
||||||
|
TConnectionParameters connectionParameters, |
||||||
|
TLibrary library, |
||||||
|
MediaItemScanResult<TMovie> result, |
||||||
|
TMovie incoming, |
||||||
|
bool deepScan); |
||||||
|
|
||||||
|
protected abstract Task<Either<BaseError, MediaItemScanResult<TMovie>>> UpdateMetadata( |
||||||
|
MediaItemScanResult<TMovie> result, |
||||||
|
MovieMetadata fullMetadata); |
||||||
|
|
||||||
|
private async Task<bool> ShouldScanItem( |
||||||
|
IMediaServerMovieRepository<TLibrary, TMovie, TEtag> movieRepository, |
||||||
|
TLibrary library, |
||||||
|
List<TEtag> existingMovies, |
||||||
|
TMovie incoming, |
||||||
|
string localPath, |
||||||
|
bool deepScan) |
||||||
|
{ |
||||||
|
// deep scan will always pull every movie
|
||||||
|
if (deepScan) |
||||||
|
{ |
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
Option<TEtag> maybeExisting = |
||||||
|
existingMovies.Find(m => m.MediaServerItemId == MediaServerItemId(incoming)); |
||||||
|
string existingItemId = await maybeExisting.Map(e => e.MediaServerItemId).IfNoneAsync(string.Empty); |
||||||
|
MediaItemState existingState = await maybeExisting.Map(e => e.State).IfNoneAsync(MediaItemState.Normal); |
||||||
|
|
||||||
|
if (existingState == MediaItemState.Unavailable) |
||||||
|
{ |
||||||
|
// skip scanning unavailable items that still don't exist locally
|
||||||
|
if (!_localFileSystem.FileExists(localPath)) |
||||||
|
{ |
||||||
|
return false; |
||||||
|
} |
||||||
|
} |
||||||
|
else if (existingItemId == MediaServerItemId(incoming)) |
||||||
|
{ |
||||||
|
// item is unchanged, but file does not exist
|
||||||
|
// don't scan, but mark as unavailable
|
||||||
|
if (!_localFileSystem.FileExists(localPath)) |
||||||
|
{ |
||||||
|
foreach (int id in await movieRepository.FlagUnavailable(library, incoming)) |
||||||
|
{ |
||||||
|
await _searchIndex.RebuildItems(_searchRepository, new List<int> { id }); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
if (maybeExisting.IsNone) |
||||||
|
{ |
||||||
|
_logger.LogDebug("INSERT: new movie {Movie}", incoming.MovieMetadata.Head().Title); |
||||||
|
} |
||||||
|
else |
||||||
|
{ |
||||||
|
_logger.LogDebug("UPDATE: Etag has changed for movie {Movie}", incoming.MovieMetadata.Head().Title); |
||||||
|
} |
||||||
|
|
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
private async Task<Either<BaseError, MediaItemScanResult<TMovie>>> UpdateMetadata( |
||||||
|
TConnectionParameters connectionParameters, |
||||||
|
TLibrary library, |
||||||
|
MediaItemScanResult<TMovie> result, |
||||||
|
TMovie incoming, |
||||||
|
bool deepScan) |
||||||
|
{ |
||||||
|
foreach (MovieMetadata fullMetadata in await GetFullMetadata( |
||||||
|
connectionParameters, |
||||||
|
library, |
||||||
|
result, |
||||||
|
incoming, |
||||||
|
deepScan)) |
||||||
|
{ |
||||||
|
// TODO: move some of this code into this scanner
|
||||||
|
// will have to merge JF, Emby, Plex logic
|
||||||
|
return await UpdateMetadata(result, fullMetadata); |
||||||
|
} |
||||||
|
|
||||||
|
return result; |
||||||
|
} |
||||||
|
|
||||||
|
private async Task<Either<BaseError, MediaItemScanResult<TMovie>>> UpdateStatistics( |
||||||
|
MediaItemScanResult<TMovie> result, |
||||||
|
TMovie incoming, |
||||||
|
string ffmpegPath, |
||||||
|
string ffprobePath) |
||||||
|
{ |
||||||
|
TMovie existing = result.Item; |
||||||
|
|
||||||
|
if (result.IsAdded || MediaServerItemId(existing) != MediaServerItemId(incoming) || |
||||||
|
existing.MediaVersions.Head().Streams.Count == 0) |
||||||
|
{ |
||||||
|
if (_localFileSystem.FileExists(result.LocalPath)) |
||||||
|
{ |
||||||
|
_logger.LogDebug("Refreshing {Attribute} for {Path}", "Statistics", result.LocalPath); |
||||||
|
Either<BaseError, bool> refreshResult = |
||||||
|
await _localStatisticsProvider.RefreshStatistics( |
||||||
|
ffmpegPath, |
||||||
|
ffprobePath, |
||||||
|
existing, |
||||||
|
result.LocalPath); |
||||||
|
|
||||||
|
foreach (BaseError error in refreshResult.LeftToSeq()) |
||||||
|
{ |
||||||
|
_logger.LogWarning( |
||||||
|
"Unable to refresh {Attribute} for media item {Path}. Error: {Error}", |
||||||
|
"Statistics", |
||||||
|
result.LocalPath, |
||||||
|
error.Value); |
||||||
|
} |
||||||
|
|
||||||
|
foreach (bool _ in refreshResult.RightToSeq()) |
||||||
|
{ |
||||||
|
result.IsUpdated = true; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
return result; |
||||||
|
} |
||||||
|
|
||||||
|
private async Task<Either<BaseError, MediaItemScanResult<TMovie>>> UpdateSubtitles( |
||||||
|
MediaItemScanResult<TMovie> existing) |
||||||
|
{ |
||||||
|
try |
||||||
|
{ |
||||||
|
// skip checking subtitles for files that don't exist locally
|
||||||
|
if (!_localFileSystem.FileExists(existing.LocalPath)) |
||||||
|
{ |
||||||
|
return existing; |
||||||
|
} |
||||||
|
|
||||||
|
if (await _localSubtitlesProvider.UpdateSubtitles(existing.Item, existing.LocalPath, false)) |
||||||
|
{ |
||||||
|
return existing; |
||||||
|
} |
||||||
|
|
||||||
|
return BaseError.New("Failed to update local subtitles"); |
||||||
|
} |
||||||
|
catch (Exception ex) |
||||||
|
{ |
||||||
|
return BaseError.New(ex.ToString()); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
@ -0,0 +1,7 @@ |
|||||||
|
using ErsatzTV.Core.Domain; |
||||||
|
using ErsatzTV.Core.Domain.MediaServer; |
||||||
|
|
||||||
|
namespace ErsatzTV.Core.Plex; |
||||||
|
|
||||||
|
public record PlexConnectionParameters |
||||||
|
(PlexConnection Connection, PlexServerAuthToken Token) : MediaServerConnectionParameters; |
@ -0,0 +1,348 @@ |
|||||||
|
using Dapper; |
||||||
|
using ErsatzTV.Core; |
||||||
|
using ErsatzTV.Core.Domain; |
||||||
|
using ErsatzTV.Core.Emby; |
||||||
|
using ErsatzTV.Core.Interfaces.Repositories; |
||||||
|
using ErsatzTV.Core.Metadata; |
||||||
|
using ErsatzTV.Infrastructure.Extensions; |
||||||
|
using Microsoft.EntityFrameworkCore; |
||||||
|
|
||||||
|
namespace ErsatzTV.Infrastructure.Data.Repositories; |
||||||
|
|
||||||
|
public class EmbyMovieRepository : IEmbyMovieRepository |
||||||
|
{ |
||||||
|
private readonly IDbContextFactory<TvContext> _dbContextFactory; |
||||||
|
|
||||||
|
public EmbyMovieRepository(IDbContextFactory<TvContext> dbContextFactory) => _dbContextFactory = dbContextFactory; |
||||||
|
|
||||||
|
public async Task<List<EmbyItemEtag>> GetExistingMovies(EmbyLibrary library) |
||||||
|
{ |
||||||
|
await using TvContext dbContext = await _dbContextFactory.CreateDbContextAsync(); |
||||||
|
return await dbContext.Connection.QueryAsync<EmbyItemEtag>( |
||||||
|
@"SELECT ItemId, Etag, MI.State FROM EmbyMovie
|
||||||
|
INNER JOIN Movie M on EmbyMovie.Id = M.Id |
||||||
|
INNER JOIN MediaItem MI on M.Id = MI.Id |
||||||
|
INNER JOIN LibraryPath LP on MI.LibraryPathId = LP.Id |
||||||
|
WHERE LP.LibraryId = @LibraryId",
|
||||||
|
new { LibraryId = library.Id }) |
||||||
|
.Map(result => result.ToList()); |
||||||
|
} |
||||||
|
|
||||||
|
public async Task<bool> FlagNormal(EmbyLibrary library, EmbyMovie movie) |
||||||
|
{ |
||||||
|
await using TvContext dbContext = await _dbContextFactory.CreateDbContextAsync(); |
||||||
|
|
||||||
|
movie.State = MediaItemState.Normal; |
||||||
|
|
||||||
|
return await dbContext.Connection.ExecuteAsync( |
||||||
|
@"UPDATE MediaItem SET State = 0 WHERE Id IN
|
||||||
|
(SELECT EmbyMovie.Id FROM EmbyMovie |
||||||
|
INNER JOIN MediaItem MI ON MI.Id = EmbyMovie.Id |
||||||
|
INNER JOIN LibraryPath LP on MI.LibraryPathId = LP.Id AND LibraryId = @LibraryId |
||||||
|
WHERE EmbyMovie.ItemId = @ItemId)",
|
||||||
|
new { LibraryId = library.Id, movie.ItemId }).Map(count => count > 0); |
||||||
|
} |
||||||
|
|
||||||
|
public async Task<Option<int>> FlagUnavailable(EmbyLibrary library, EmbyMovie movie) |
||||||
|
{ |
||||||
|
await using TvContext dbContext = await _dbContextFactory.CreateDbContextAsync(); |
||||||
|
|
||||||
|
movie.State = MediaItemState.Unavailable; |
||||||
|
|
||||||
|
Option<int> maybeId = await dbContext.Connection.ExecuteScalarAsync<int>( |
||||||
|
@"SELECT EmbyMovie.Id FROM EmbyMovie
|
||||||
|
INNER JOIN MediaItem MI ON MI.Id = EmbyMovie.Id |
||||||
|
INNER JOIN LibraryPath LP on MI.LibraryPathId = LP.Id AND LibraryId = @LibraryId |
||||||
|
WHERE EmbyMovie.ItemId = @ItemId",
|
||||||
|
new { LibraryId = library.Id, movie.ItemId }); |
||||||
|
|
||||||
|
foreach (int id in maybeId) |
||||||
|
{ |
||||||
|
return await dbContext.Connection.ExecuteAsync( |
||||||
|
@"UPDATE MediaItem SET State = 2 WHERE Id = @Id", |
||||||
|
new { Id = id }).Map(count => count > 0 ? Some(id) : None); |
||||||
|
} |
||||||
|
|
||||||
|
return None; |
||||||
|
} |
||||||
|
|
||||||
|
public async Task<List<int>> FlagFileNotFound(EmbyLibrary library, List<string> movieItemIds) |
||||||
|
{ |
||||||
|
if (movieItemIds.Count == 0) |
||||||
|
{ |
||||||
|
return new List<int>(); |
||||||
|
} |
||||||
|
|
||||||
|
await using TvContext dbContext = await _dbContextFactory.CreateDbContextAsync(); |
||||||
|
|
||||||
|
List<int> ids = await dbContext.Connection.QueryAsync<int>( |
||||||
|
@"SELECT M.Id
|
||||||
|
FROM MediaItem M |
||||||
|
INNER JOIN EmbyMovie ON EmbyMovie.Id = M.Id |
||||||
|
INNER JOIN LibraryPath LP on M.LibraryPathId = LP.Id AND LP.LibraryId = @LibraryId |
||||||
|
WHERE EmbyMovie.ItemId IN @MovieItemIds",
|
||||||
|
new { LibraryId = library.Id, MovieItemIds = movieItemIds }) |
||||||
|
.Map(result => result.ToList()); |
||||||
|
|
||||||
|
await dbContext.Connection.ExecuteAsync( |
||||||
|
@"UPDATE MediaItem SET State = 1 WHERE Id IN @Ids", |
||||||
|
new { Ids = ids }); |
||||||
|
|
||||||
|
return ids; |
||||||
|
} |
||||||
|
|
||||||
|
public async Task<Either<BaseError, MediaItemScanResult<EmbyMovie>>> GetOrAdd(EmbyLibrary library, EmbyMovie item) |
||||||
|
{ |
||||||
|
await using TvContext dbContext = await _dbContextFactory.CreateDbContextAsync(); |
||||||
|
Option<EmbyMovie> maybeExisting = await dbContext.EmbyMovies |
||||||
|
.Include(m => m.LibraryPath) |
||||||
|
.ThenInclude(lp => lp.Library) |
||||||
|
.Include(m => m.MediaVersions) |
||||||
|
.ThenInclude(mv => mv.MediaFiles) |
||||||
|
.Include(m => m.MediaVersions) |
||||||
|
.ThenInclude(mv => mv.Streams) |
||||||
|
.Include(m => m.MovieMetadata) |
||||||
|
.ThenInclude(mm => mm.Genres) |
||||||
|
.Include(m => m.MovieMetadata) |
||||||
|
.ThenInclude(mm => mm.Tags) |
||||||
|
.Include(m => m.MovieMetadata) |
||||||
|
.ThenInclude(mm => mm.Studios) |
||||||
|
.Include(m => m.MovieMetadata) |
||||||
|
.ThenInclude(mm => mm.Actors) |
||||||
|
.Include(m => m.MovieMetadata) |
||||||
|
.ThenInclude(mm => mm.Directors) |
||||||
|
.Include(m => m.MovieMetadata) |
||||||
|
.ThenInclude(mm => mm.Writers) |
||||||
|
.Include(m => m.MovieMetadata) |
||||||
|
.ThenInclude(mm => mm.Artwork) |
||||||
|
.Include(m => m.MovieMetadata) |
||||||
|
.ThenInclude(mm => mm.Guids) |
||||||
|
.Include(m => m.TraktListItems) |
||||||
|
.ThenInclude(tli => tli.TraktList) |
||||||
|
.SelectOneAsync(m => m.ItemId, m => m.ItemId == item.ItemId); |
||||||
|
|
||||||
|
foreach (EmbyMovie embyMovie in maybeExisting) |
||||||
|
{ |
||||||
|
var result = new MediaItemScanResult<EmbyMovie>(embyMovie) { IsAdded = false }; |
||||||
|
if (embyMovie.Etag != item.Etag) |
||||||
|
{ |
||||||
|
await UpdateMovie(dbContext, embyMovie, item); |
||||||
|
result.IsUpdated = true; |
||||||
|
} |
||||||
|
|
||||||
|
return result; |
||||||
|
} |
||||||
|
|
||||||
|
return await AddMovie(dbContext, library, item); |
||||||
|
} |
||||||
|
|
||||||
|
public async Task<Unit> SetEtag(EmbyMovie movie, string etag) |
||||||
|
{ |
||||||
|
await using TvContext dbContext = await _dbContextFactory.CreateDbContextAsync(); |
||||||
|
return await dbContext.Connection.ExecuteAsync( |
||||||
|
"UPDATE EmbyMovie SET Etag = @Etag WHERE Id = @Id", |
||||||
|
new { Etag = etag, movie.Id }).Map(_ => Unit.Default); |
||||||
|
} |
||||||
|
|
||||||
|
private async Task<Either<BaseError, MediaItemScanResult<EmbyMovie>>> AddMovie( |
||||||
|
TvContext dbContext, |
||||||
|
EmbyLibrary library, |
||||||
|
EmbyMovie movie) |
||||||
|
{ |
||||||
|
try |
||||||
|
{ |
||||||
|
// blank out etag for initial save in case other updates fail
|
||||||
|
movie.Etag = string.Empty; |
||||||
|
|
||||||
|
movie.LibraryPathId = library.Paths.Head().Id; |
||||||
|
|
||||||
|
await dbContext.AddAsync(movie); |
||||||
|
await dbContext.SaveChangesAsync(); |
||||||
|
|
||||||
|
await dbContext.Entry(movie).Reference(m => m.LibraryPath).LoadAsync(); |
||||||
|
await dbContext.Entry(movie.LibraryPath).Reference(lp => lp.Library).LoadAsync(); |
||||||
|
return new MediaItemScanResult<EmbyMovie>(movie) { IsAdded = true }; |
||||||
|
} |
||||||
|
catch (Exception ex) |
||||||
|
{ |
||||||
|
return BaseError.New(ex.ToString()); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
private async Task UpdateMovie(TvContext dbContext, EmbyMovie existing, EmbyMovie incoming) |
||||||
|
{ |
||||||
|
// library path is used for search indexing later
|
||||||
|
incoming.LibraryPath = existing.LibraryPath; |
||||||
|
incoming.Id = existing.Id; |
||||||
|
|
||||||
|
// metadata
|
||||||
|
MovieMetadata metadata = existing.MovieMetadata.Head(); |
||||||
|
MovieMetadata incomingMetadata = incoming.MovieMetadata.Head(); |
||||||
|
metadata.MetadataKind = incomingMetadata.MetadataKind; |
||||||
|
metadata.ContentRating = incomingMetadata.ContentRating; |
||||||
|
metadata.Title = incomingMetadata.Title; |
||||||
|
metadata.SortTitle = incomingMetadata.SortTitle; |
||||||
|
metadata.Plot = incomingMetadata.Plot; |
||||||
|
metadata.Year = incomingMetadata.Year; |
||||||
|
metadata.Tagline = incomingMetadata.Tagline; |
||||||
|
metadata.DateAdded = incomingMetadata.DateAdded; |
||||||
|
metadata.DateUpdated = DateTime.UtcNow; |
||||||
|
|
||||||
|
// genres
|
||||||
|
foreach (Genre genre in metadata.Genres |
||||||
|
.Filter(g => incomingMetadata.Genres.All(g2 => g2.Name != g.Name)) |
||||||
|
.ToList()) |
||||||
|
{ |
||||||
|
metadata.Genres.Remove(genre); |
||||||
|
} |
||||||
|
|
||||||
|
foreach (Genre genre in incomingMetadata.Genres |
||||||
|
.Filter(g => metadata.Genres.All(g2 => g2.Name != g.Name)) |
||||||
|
.ToList()) |
||||||
|
{ |
||||||
|
metadata.Genres.Add(genre); |
||||||
|
} |
||||||
|
|
||||||
|
// tags
|
||||||
|
foreach (Tag tag in metadata.Tags |
||||||
|
.Filter(g => incomingMetadata.Tags.All(g2 => g2.Name != g.Name)) |
||||||
|
.Filter(g => g.ExternalCollectionId is null) |
||||||
|
.ToList()) |
||||||
|
{ |
||||||
|
metadata.Tags.Remove(tag); |
||||||
|
} |
||||||
|
|
||||||
|
foreach (Tag tag in incomingMetadata.Tags |
||||||
|
.Filter(g => metadata.Tags.All(g2 => g2.Name != g.Name)) |
||||||
|
.ToList()) |
||||||
|
{ |
||||||
|
metadata.Tags.Add(tag); |
||||||
|
} |
||||||
|
|
||||||
|
// studios
|
||||||
|
foreach (Studio studio in metadata.Studios |
||||||
|
.Filter(g => incomingMetadata.Studios.All(g2 => g2.Name != g.Name)) |
||||||
|
.ToList()) |
||||||
|
{ |
||||||
|
metadata.Studios.Remove(studio); |
||||||
|
} |
||||||
|
|
||||||
|
foreach (Studio studio in incomingMetadata.Studios |
||||||
|
.Filter(g => metadata.Studios.All(g2 => g2.Name != g.Name)) |
||||||
|
.ToList()) |
||||||
|
{ |
||||||
|
metadata.Studios.Add(studio); |
||||||
|
} |
||||||
|
|
||||||
|
// actors
|
||||||
|
foreach (Actor actor in metadata.Actors |
||||||
|
.Filter( |
||||||
|
a => incomingMetadata.Actors.All( |
||||||
|
a2 => a2.Name != a.Name || a.Artwork == null && a2.Artwork != null)) |
||||||
|
.ToList()) |
||||||
|
{ |
||||||
|
metadata.Actors.Remove(actor); |
||||||
|
} |
||||||
|
|
||||||
|
foreach (Actor actor in incomingMetadata.Actors |
||||||
|
.Filter(a => metadata.Actors.All(a2 => a2.Name != a.Name)) |
||||||
|
.ToList()) |
||||||
|
{ |
||||||
|
metadata.Actors.Add(actor); |
||||||
|
} |
||||||
|
|
||||||
|
// directors
|
||||||
|
foreach (Director director in metadata.Directors |
||||||
|
.Filter(d => incomingMetadata.Directors.All(d2 => d2.Name != d.Name)) |
||||||
|
.ToList()) |
||||||
|
{ |
||||||
|
metadata.Directors.Remove(director); |
||||||
|
} |
||||||
|
|
||||||
|
foreach (Director director in incomingMetadata.Directors |
||||||
|
.Filter(d => metadata.Directors.All(d2 => d2.Name != d.Name)) |
||||||
|
.ToList()) |
||||||
|
{ |
||||||
|
metadata.Directors.Add(director); |
||||||
|
} |
||||||
|
|
||||||
|
// writers
|
||||||
|
foreach (Writer writer in metadata.Writers |
||||||
|
.Filter(w => incomingMetadata.Writers.All(w2 => w2.Name != w.Name)) |
||||||
|
.ToList()) |
||||||
|
{ |
||||||
|
metadata.Writers.Remove(writer); |
||||||
|
} |
||||||
|
|
||||||
|
foreach (Writer writer in incomingMetadata.Writers |
||||||
|
.Filter(w => metadata.Writers.All(w2 => w2.Name != w.Name)) |
||||||
|
.ToList()) |
||||||
|
{ |
||||||
|
metadata.Writers.Add(writer); |
||||||
|
} |
||||||
|
|
||||||
|
// guids
|
||||||
|
foreach (MetadataGuid guid in metadata.Guids |
||||||
|
.Filter(g => incomingMetadata.Guids.All(g2 => g2.Guid != g.Guid)) |
||||||
|
.ToList()) |
||||||
|
{ |
||||||
|
metadata.Guids.Remove(guid); |
||||||
|
} |
||||||
|
|
||||||
|
foreach (MetadataGuid guid in incomingMetadata.Guids |
||||||
|
.Filter(g => metadata.Guids.All(g2 => g2.Guid != g.Guid)) |
||||||
|
.ToList()) |
||||||
|
{ |
||||||
|
metadata.Guids.Add(guid); |
||||||
|
} |
||||||
|
|
||||||
|
metadata.ReleaseDate = incomingMetadata.ReleaseDate; |
||||||
|
|
||||||
|
// poster
|
||||||
|
Artwork incomingPoster = |
||||||
|
incomingMetadata.Artwork.FirstOrDefault(a => a.ArtworkKind == ArtworkKind.Poster); |
||||||
|
if (incomingPoster != null) |
||||||
|
{ |
||||||
|
Artwork poster = metadata.Artwork.FirstOrDefault(a => a.ArtworkKind == ArtworkKind.Poster); |
||||||
|
if (poster == null) |
||||||
|
{ |
||||||
|
poster = new Artwork { ArtworkKind = ArtworkKind.Poster }; |
||||||
|
metadata.Artwork.Add(poster); |
||||||
|
} |
||||||
|
|
||||||
|
poster.Path = incomingPoster.Path; |
||||||
|
poster.DateAdded = incomingPoster.DateAdded; |
||||||
|
poster.DateUpdated = incomingPoster.DateUpdated; |
||||||
|
} |
||||||
|
|
||||||
|
// fan art
|
||||||
|
Artwork incomingFanArt = |
||||||
|
incomingMetadata.Artwork.FirstOrDefault(a => a.ArtworkKind == ArtworkKind.FanArt); |
||||||
|
if (incomingFanArt != null) |
||||||
|
{ |
||||||
|
Artwork fanArt = metadata.Artwork.FirstOrDefault(a => a.ArtworkKind == ArtworkKind.FanArt); |
||||||
|
if (fanArt == null) |
||||||
|
{ |
||||||
|
fanArt = new Artwork { ArtworkKind = ArtworkKind.FanArt }; |
||||||
|
metadata.Artwork.Add(fanArt); |
||||||
|
} |
||||||
|
|
||||||
|
fanArt.Path = incomingFanArt.Path; |
||||||
|
fanArt.DateAdded = incomingFanArt.DateAdded; |
||||||
|
fanArt.DateUpdated = incomingFanArt.DateUpdated; |
||||||
|
} |
||||||
|
|
||||||
|
// version
|
||||||
|
MediaVersion version = existing.MediaVersions.Head(); |
||||||
|
MediaVersion incomingVersion = incoming.MediaVersions.Head(); |
||||||
|
version.Name = incomingVersion.Name; |
||||||
|
version.DateAdded = incomingVersion.DateAdded; |
||||||
|
|
||||||
|
// media file
|
||||||
|
MediaFile file = version.MediaFiles.Head(); |
||||||
|
MediaFile incomingFile = incomingVersion.MediaFiles.Head(); |
||||||
|
file.Path = incomingFile.Path; |
||||||
|
|
||||||
|
await dbContext.SaveChangesAsync(); |
||||||
|
} |
||||||
|
} |
@ -0,0 +1,353 @@ |
|||||||
|
using Dapper; |
||||||
|
using ErsatzTV.Core; |
||||||
|
using ErsatzTV.Core.Domain; |
||||||
|
using ErsatzTV.Core.Interfaces.Repositories; |
||||||
|
using ErsatzTV.Core.Jellyfin; |
||||||
|
using ErsatzTV.Core.Metadata; |
||||||
|
using ErsatzTV.Infrastructure.Extensions; |
||||||
|
using Microsoft.EntityFrameworkCore; |
||||||
|
|
||||||
|
namespace ErsatzTV.Infrastructure.Data.Repositories; |
||||||
|
|
||||||
|
public class JellyfinMovieRepository : IJellyfinMovieRepository |
||||||
|
{ |
||||||
|
private readonly IDbContextFactory<TvContext> _dbContextFactory; |
||||||
|
|
||||||
|
public JellyfinMovieRepository(IDbContextFactory<TvContext> dbContextFactory) => |
||||||
|
_dbContextFactory = dbContextFactory; |
||||||
|
|
||||||
|
public async Task<List<JellyfinItemEtag>> GetExistingMovies(JellyfinLibrary library) |
||||||
|
{ |
||||||
|
await using TvContext dbContext = await _dbContextFactory.CreateDbContextAsync(); |
||||||
|
return await dbContext.Connection.QueryAsync<JellyfinItemEtag>( |
||||||
|
@"SELECT ItemId, Etag, MI.State FROM JellyfinMovie
|
||||||
|
INNER JOIN Movie M on JellyfinMovie.Id = M.Id |
||||||
|
INNER JOIN MediaItem MI on M.Id = MI.Id |
||||||
|
INNER JOIN LibraryPath LP on MI.LibraryPathId = LP.Id |
||||||
|
WHERE LP.LibraryId = @LibraryId",
|
||||||
|
new { LibraryId = library.Id }) |
||||||
|
.Map(result => result.ToList()); |
||||||
|
} |
||||||
|
|
||||||
|
public async Task<bool> FlagNormal(JellyfinLibrary library, JellyfinMovie movie) |
||||||
|
{ |
||||||
|
await using TvContext dbContext = await _dbContextFactory.CreateDbContextAsync(); |
||||||
|
|
||||||
|
movie.State = MediaItemState.Normal; |
||||||
|
|
||||||
|
return await dbContext.Connection.ExecuteAsync( |
||||||
|
@"UPDATE MediaItem SET State = 0 WHERE Id IN
|
||||||
|
(SELECT JellyfinMovie.Id FROM JellyfinMovie |
||||||
|
INNER JOIN MediaItem MI ON MI.Id = JellyfinMovie.Id |
||||||
|
INNER JOIN LibraryPath LP on MI.LibraryPathId = LP.Id AND LibraryId = @LibraryId |
||||||
|
WHERE JellyfinMovie.ItemId = @ItemId)",
|
||||||
|
new { LibraryId = library.Id, movie.ItemId }).Map(count => count > 0); |
||||||
|
} |
||||||
|
|
||||||
|
public async Task<Option<int>> FlagUnavailable(JellyfinLibrary library, JellyfinMovie movie) |
||||||
|
{ |
||||||
|
await using TvContext dbContext = await _dbContextFactory.CreateDbContextAsync(); |
||||||
|
|
||||||
|
movie.State = MediaItemState.Unavailable; |
||||||
|
|
||||||
|
Option<int> maybeId = await dbContext.Connection.ExecuteScalarAsync<int>( |
||||||
|
@"SELECT JellyfinMovie.Id FROM JellyfinMovie
|
||||||
|
INNER JOIN MediaItem MI ON MI.Id = JellyfinMovie.Id |
||||||
|
INNER JOIN LibraryPath LP on MI.LibraryPathId = LP.Id AND LibraryId = @LibraryId |
||||||
|
WHERE JellyfinMovie.ItemId = @ItemId",
|
||||||
|
new { LibraryId = library.Id, movie.ItemId }); |
||||||
|
|
||||||
|
foreach (int id in maybeId) |
||||||
|
{ |
||||||
|
return await dbContext.Connection.ExecuteAsync( |
||||||
|
@"UPDATE MediaItem SET State = 2 WHERE Id = @Id", |
||||||
|
new { Id = id }).Map(count => count > 0 ? Some(id) : None); |
||||||
|
} |
||||||
|
|
||||||
|
return None; |
||||||
|
} |
||||||
|
|
||||||
|
public async Task<List<int>> FlagFileNotFound(JellyfinLibrary library, List<string> movieItemIds) |
||||||
|
{ |
||||||
|
if (movieItemIds.Count == 0) |
||||||
|
{ |
||||||
|
return new List<int>(); |
||||||
|
} |
||||||
|
|
||||||
|
await using TvContext dbContext = await _dbContextFactory.CreateDbContextAsync(); |
||||||
|
|
||||||
|
List<int> ids = await dbContext.Connection.QueryAsync<int>( |
||||||
|
@"SELECT M.Id
|
||||||
|
FROM MediaItem M |
||||||
|
INNER JOIN JellyfinMovie ON JellyfinMovie.Id = M.Id |
||||||
|
INNER JOIN LibraryPath LP on M.LibraryPathId = LP.Id AND LP.LibraryId = @LibraryId |
||||||
|
WHERE JellyfinMovie.ItemId IN @MovieItemIds",
|
||||||
|
new { LibraryId = library.Id, MovieItemIds = movieItemIds }) |
||||||
|
.Map(result => result.ToList()); |
||||||
|
|
||||||
|
await dbContext.Connection.ExecuteAsync( |
||||||
|
@"UPDATE MediaItem SET State = 1 WHERE Id IN @Ids", |
||||||
|
new { Ids = ids }); |
||||||
|
|
||||||
|
return ids; |
||||||
|
} |
||||||
|
|
||||||
|
public async Task<Either<BaseError, MediaItemScanResult<JellyfinMovie>>> GetOrAdd( |
||||||
|
JellyfinLibrary library, |
||||||
|
JellyfinMovie item) |
||||||
|
{ |
||||||
|
await using TvContext dbContext = await _dbContextFactory.CreateDbContextAsync(); |
||||||
|
Option<JellyfinMovie> maybeExisting = await dbContext.JellyfinMovies |
||||||
|
.Include(m => m.LibraryPath) |
||||||
|
.ThenInclude(lp => lp.Library) |
||||||
|
.Include(m => m.MediaVersions) |
||||||
|
.ThenInclude(mv => mv.MediaFiles) |
||||||
|
.Include(m => m.MediaVersions) |
||||||
|
.ThenInclude(mv => mv.Streams) |
||||||
|
.Include(m => m.MovieMetadata) |
||||||
|
.ThenInclude(mm => mm.Genres) |
||||||
|
.Include(m => m.MovieMetadata) |
||||||
|
.ThenInclude(mm => mm.Tags) |
||||||
|
.Include(m => m.MovieMetadata) |
||||||
|
.ThenInclude(mm => mm.Studios) |
||||||
|
.Include(m => m.MovieMetadata) |
||||||
|
.ThenInclude(mm => mm.Actors) |
||||||
|
.Include(m => m.MovieMetadata) |
||||||
|
.ThenInclude(mm => mm.Directors) |
||||||
|
.Include(m => m.MovieMetadata) |
||||||
|
.ThenInclude(mm => mm.Writers) |
||||||
|
.Include(m => m.MovieMetadata) |
||||||
|
.ThenInclude(mm => mm.Artwork) |
||||||
|
.Include(m => m.MovieMetadata) |
||||||
|
.ThenInclude(mm => mm.Guids) |
||||||
|
.Include(m => m.TraktListItems) |
||||||
|
.ThenInclude(tli => tli.TraktList) |
||||||
|
.SelectOneAsync(m => m.ItemId, m => m.ItemId == item.ItemId); |
||||||
|
|
||||||
|
foreach (JellyfinMovie jellyfinMovie in maybeExisting) |
||||||
|
{ |
||||||
|
var result = new MediaItemScanResult<JellyfinMovie>(jellyfinMovie) { IsAdded = false }; |
||||||
|
if (jellyfinMovie.Etag != item.Etag) |
||||||
|
{ |
||||||
|
await UpdateMovie(dbContext, jellyfinMovie, item); |
||||||
|
result.IsUpdated = true; |
||||||
|
} |
||||||
|
|
||||||
|
return result; |
||||||
|
} |
||||||
|
|
||||||
|
return await AddMovie(dbContext, library, item); |
||||||
|
} |
||||||
|
|
||||||
|
public async Task<Unit> SetEtag(JellyfinMovie movie, string etag) |
||||||
|
{ |
||||||
|
await using TvContext dbContext = await _dbContextFactory.CreateDbContextAsync(); |
||||||
|
return await dbContext.Connection.ExecuteAsync( |
||||||
|
"UPDATE JellyfinMovie SET Etag = @Etag WHERE Id = @Id", |
||||||
|
new { Etag = etag, movie.Id }).Map(_ => Unit.Default); |
||||||
|
} |
||||||
|
|
||||||
|
private async Task UpdateMovie(TvContext dbContext, JellyfinMovie existing, JellyfinMovie incoming) |
||||||
|
{ |
||||||
|
// library path is used for search indexing later
|
||||||
|
incoming.LibraryPath = existing.LibraryPath; |
||||||
|
incoming.Id = existing.Id; |
||||||
|
|
||||||
|
existing.Etag = incoming.Etag; |
||||||
|
|
||||||
|
// metadata
|
||||||
|
MovieMetadata metadata = existing.MovieMetadata.Head(); |
||||||
|
MovieMetadata incomingMetadata = incoming.MovieMetadata.Head(); |
||||||
|
metadata.MetadataKind = incomingMetadata.MetadataKind; |
||||||
|
metadata.ContentRating = incomingMetadata.ContentRating; |
||||||
|
metadata.Title = incomingMetadata.Title; |
||||||
|
metadata.SortTitle = incomingMetadata.SortTitle; |
||||||
|
metadata.Plot = incomingMetadata.Plot; |
||||||
|
metadata.Year = incomingMetadata.Year; |
||||||
|
metadata.Tagline = incomingMetadata.Tagline; |
||||||
|
metadata.DateAdded = incomingMetadata.DateAdded; |
||||||
|
metadata.DateUpdated = DateTime.UtcNow; |
||||||
|
|
||||||
|
// genres
|
||||||
|
foreach (Genre genre in metadata.Genres |
||||||
|
.Filter(g => incomingMetadata.Genres.All(g2 => g2.Name != g.Name)) |
||||||
|
.ToList()) |
||||||
|
{ |
||||||
|
metadata.Genres.Remove(genre); |
||||||
|
} |
||||||
|
|
||||||
|
foreach (Genre genre in incomingMetadata.Genres |
||||||
|
.Filter(g => metadata.Genres.All(g2 => g2.Name != g.Name)) |
||||||
|
.ToList()) |
||||||
|
{ |
||||||
|
metadata.Genres.Add(genre); |
||||||
|
} |
||||||
|
|
||||||
|
// tags
|
||||||
|
foreach (Tag tag in metadata.Tags |
||||||
|
.Filter(g => incomingMetadata.Tags.All(g2 => g2.Name != g.Name)) |
||||||
|
.Filter(g => g.ExternalCollectionId is null) |
||||||
|
.ToList()) |
||||||
|
{ |
||||||
|
metadata.Tags.Remove(tag); |
||||||
|
} |
||||||
|
|
||||||
|
foreach (Tag tag in incomingMetadata.Tags |
||||||
|
.Filter(g => metadata.Tags.All(g2 => g2.Name != g.Name)) |
||||||
|
.ToList()) |
||||||
|
{ |
||||||
|
metadata.Tags.Add(tag); |
||||||
|
} |
||||||
|
|
||||||
|
// studios
|
||||||
|
foreach (Studio studio in metadata.Studios |
||||||
|
.Filter(g => incomingMetadata.Studios.All(g2 => g2.Name != g.Name)) |
||||||
|
.ToList()) |
||||||
|
{ |
||||||
|
metadata.Studios.Remove(studio); |
||||||
|
} |
||||||
|
|
||||||
|
foreach (Studio studio in incomingMetadata.Studios |
||||||
|
.Filter(g => metadata.Studios.All(g2 => g2.Name != g.Name)) |
||||||
|
.ToList()) |
||||||
|
{ |
||||||
|
metadata.Studios.Add(studio); |
||||||
|
} |
||||||
|
|
||||||
|
// actors
|
||||||
|
foreach (Actor actor in metadata.Actors |
||||||
|
.Filter( |
||||||
|
a => incomingMetadata.Actors.All( |
||||||
|
a2 => a2.Name != a.Name || a.Artwork == null && a2.Artwork != null)) |
||||||
|
.ToList()) |
||||||
|
{ |
||||||
|
metadata.Actors.Remove(actor); |
||||||
|
} |
||||||
|
|
||||||
|
foreach (Actor actor in incomingMetadata.Actors |
||||||
|
.Filter(a => metadata.Actors.All(a2 => a2.Name != a.Name)) |
||||||
|
.ToList()) |
||||||
|
{ |
||||||
|
metadata.Actors.Add(actor); |
||||||
|
} |
||||||
|
|
||||||
|
// directors
|
||||||
|
foreach (Director director in metadata.Directors |
||||||
|
.Filter(d => incomingMetadata.Directors.All(d2 => d2.Name != d.Name)) |
||||||
|
.ToList()) |
||||||
|
{ |
||||||
|
metadata.Directors.Remove(director); |
||||||
|
} |
||||||
|
|
||||||
|
foreach (Director director in incomingMetadata.Directors |
||||||
|
.Filter(d => metadata.Directors.All(d2 => d2.Name != d.Name)) |
||||||
|
.ToList()) |
||||||
|
{ |
||||||
|
metadata.Directors.Add(director); |
||||||
|
} |
||||||
|
|
||||||
|
// writers
|
||||||
|
foreach (Writer writer in metadata.Writers |
||||||
|
.Filter(w => incomingMetadata.Writers.All(w2 => w2.Name != w.Name)) |
||||||
|
.ToList()) |
||||||
|
{ |
||||||
|
metadata.Writers.Remove(writer); |
||||||
|
} |
||||||
|
|
||||||
|
foreach (Writer writer in incomingMetadata.Writers |
||||||
|
.Filter(w => metadata.Writers.All(w2 => w2.Name != w.Name)) |
||||||
|
.ToList()) |
||||||
|
{ |
||||||
|
metadata.Writers.Add(writer); |
||||||
|
} |
||||||
|
|
||||||
|
// guids
|
||||||
|
foreach (MetadataGuid guid in metadata.Guids |
||||||
|
.Filter(g => incomingMetadata.Guids.All(g2 => g2.Guid != g.Guid)) |
||||||
|
.ToList()) |
||||||
|
{ |
||||||
|
metadata.Guids.Remove(guid); |
||||||
|
} |
||||||
|
|
||||||
|
foreach (MetadataGuid guid in incomingMetadata.Guids |
||||||
|
.Filter(g => metadata.Guids.All(g2 => g2.Guid != g.Guid)) |
||||||
|
.ToList()) |
||||||
|
{ |
||||||
|
metadata.Guids.Add(guid); |
||||||
|
} |
||||||
|
|
||||||
|
metadata.ReleaseDate = incomingMetadata.ReleaseDate; |
||||||
|
|
||||||
|
// poster
|
||||||
|
Artwork incomingPoster = |
||||||
|
incomingMetadata.Artwork.FirstOrDefault(a => a.ArtworkKind == ArtworkKind.Poster); |
||||||
|
if (incomingPoster != null) |
||||||
|
{ |
||||||
|
Artwork poster = metadata.Artwork.FirstOrDefault(a => a.ArtworkKind == ArtworkKind.Poster); |
||||||
|
if (poster == null) |
||||||
|
{ |
||||||
|
poster = new Artwork { ArtworkKind = ArtworkKind.Poster }; |
||||||
|
metadata.Artwork.Add(poster); |
||||||
|
} |
||||||
|
|
||||||
|
poster.Path = incomingPoster.Path; |
||||||
|
poster.DateAdded = incomingPoster.DateAdded; |
||||||
|
poster.DateUpdated = incomingPoster.DateUpdated; |
||||||
|
} |
||||||
|
|
||||||
|
// fan art
|
||||||
|
Artwork incomingFanArt = |
||||||
|
incomingMetadata.Artwork.FirstOrDefault(a => a.ArtworkKind == ArtworkKind.FanArt); |
||||||
|
if (incomingFanArt != null) |
||||||
|
{ |
||||||
|
Artwork fanArt = metadata.Artwork.FirstOrDefault(a => a.ArtworkKind == ArtworkKind.FanArt); |
||||||
|
if (fanArt == null) |
||||||
|
{ |
||||||
|
fanArt = new Artwork { ArtworkKind = ArtworkKind.FanArt }; |
||||||
|
metadata.Artwork.Add(fanArt); |
||||||
|
} |
||||||
|
|
||||||
|
fanArt.Path = incomingFanArt.Path; |
||||||
|
fanArt.DateAdded = incomingFanArt.DateAdded; |
||||||
|
fanArt.DateUpdated = incomingFanArt.DateUpdated; |
||||||
|
} |
||||||
|
|
||||||
|
// version
|
||||||
|
MediaVersion version = existing.MediaVersions.Head(); |
||||||
|
MediaVersion incomingVersion = incoming.MediaVersions.Head(); |
||||||
|
version.Name = incomingVersion.Name; |
||||||
|
version.DateAdded = incomingVersion.DateAdded; |
||||||
|
|
||||||
|
// media file
|
||||||
|
MediaFile file = version.MediaFiles.Head(); |
||||||
|
MediaFile incomingFile = incomingVersion.MediaFiles.Head(); |
||||||
|
file.Path = incomingFile.Path; |
||||||
|
|
||||||
|
await dbContext.SaveChangesAsync(); |
||||||
|
} |
||||||
|
|
||||||
|
private async Task<Either<BaseError, MediaItemScanResult<JellyfinMovie>>> AddMovie( |
||||||
|
TvContext dbContext, |
||||||
|
JellyfinLibrary library, |
||||||
|
JellyfinMovie movie) |
||||||
|
{ |
||||||
|
try |
||||||
|
{ |
||||||
|
// blank out etag for initial save in case other updates fail
|
||||||
|
movie.Etag = string.Empty; |
||||||
|
|
||||||
|
movie.LibraryPathId = library.Paths.Head().Id; |
||||||
|
|
||||||
|
await dbContext.AddAsync(movie); |
||||||
|
await dbContext.SaveChangesAsync(); |
||||||
|
|
||||||
|
await dbContext.Entry(movie).Reference(m => m.LibraryPath).LoadAsync(); |
||||||
|
await dbContext.Entry(movie.LibraryPath).Reference(lp => lp.Library).LoadAsync(); |
||||||
|
return new MediaItemScanResult<JellyfinMovie>(movie) { IsAdded = true }; |
||||||
|
} |
||||||
|
catch (Exception ex) |
||||||
|
{ |
||||||
|
return BaseError.New(ex.ToString()); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
Loading…
Reference in new issue