using Dapper; using ErsatzTV.Core; using ErsatzTV.Core.Domain; using ErsatzTV.Core.Errors; using ErsatzTV.Core.Interfaces.Repositories; using ErsatzTV.Core.Metadata; using Microsoft.EntityFrameworkCore; using Microsoft.Extensions.Logging; namespace ErsatzTV.Infrastructure.Data.Repositories; public class OtherVideoRepository : IOtherVideoRepository { private readonly IDbContextFactory _dbContextFactory; private readonly ILogger _logger; public OtherVideoRepository(IDbContextFactory dbContextFactory, ILogger logger) { _dbContextFactory = dbContextFactory; _logger = logger; } public async Task>> GetOrAdd( LibraryPath libraryPath, LibraryFolder libraryFolder, string path) { await using TvContext dbContext = await _dbContextFactory.CreateDbContextAsync(); Option maybeExisting = await dbContext.OtherVideos .AsNoTracking() .Include(i => i.OtherVideoMetadata) .ThenInclude(ovm => ovm.Genres) .Include(i => i.OtherVideoMetadata) .ThenInclude(ovm => ovm.Tags) .Include(i => i.OtherVideoMetadata) .ThenInclude(ovm => ovm.Studios) .Include(i => i.OtherVideoMetadata) .ThenInclude(ovm => ovm.Guids) .Include(i => i.OtherVideoMetadata) .ThenInclude(ovm => ovm.Actors) .Include(i => i.OtherVideoMetadata) .ThenInclude(ovm => ovm.Actors) .ThenInclude(a => a.Artwork) .Include(i => i.OtherVideoMetadata) .ThenInclude(ovm => ovm.Directors) .Include(i => i.OtherVideoMetadata) .ThenInclude(ovm => ovm.Writers) .Include(ov => ov.LibraryPath) .ThenInclude(lp => lp.Library) .Include(ov => ov.MediaVersions) .ThenInclude(ov => ov.MediaFiles) .Include(ov => ov.MediaVersions) .ThenInclude(ov => ov.Streams) .Include(ov => ov.TraktListItems) .ThenInclude(tli => tli.TraktList) .OrderBy(i => i.MediaVersions.First().MediaFiles.First().Path) .SingleOrDefaultAsync(i => i.MediaVersions.First().MediaFiles.First().Path == path); return await maybeExisting.Match( mediaItem => Right>( new MediaItemScanResult(mediaItem) { IsAdded = false }).AsTask(), async () => await AddOtherVideo(dbContext, libraryPath.Id, libraryFolder.Id, path)); } public async Task> FindOtherVideoPaths(LibraryPath libraryPath) { await using TvContext dbContext = await _dbContextFactory.CreateDbContextAsync(); return await dbContext.Connection.QueryAsync( @"SELECT MF.Path FROM MediaFile MF INNER JOIN MediaVersion MV on MF.MediaVersionId = MV.Id INNER JOIN OtherVideo O on MV.OtherVideoId = O.Id INNER JOIN MediaItem MI on O.Id = MI.Id WHERE MI.LibraryPathId = @LibraryPathId", new { LibraryPathId = libraryPath.Id }); } public async Task> DeleteByPath(LibraryPath libraryPath, string path) { await using TvContext dbContext = await _dbContextFactory.CreateDbContextAsync(); List ids = await dbContext.Connection.QueryAsync( @"SELECT O.Id FROM OtherVideo O INNER JOIN MediaItem MI on O.Id = MI.Id INNER JOIN MediaVersion MV on O.Id = MV.OtherVideoId INNER JOIN MediaFile MF on MV.Id = MF.MediaVersionId WHERE MI.LibraryPathId = @LibraryPathId AND MF.Path = @Path", new { LibraryPathId = libraryPath.Id, Path = path }).Map(result => result.ToList()); foreach (int otherVideoId in ids) { OtherVideo otherVideo = await dbContext.OtherVideos.FindAsync(otherVideoId); if (otherVideo != null) { dbContext.OtherVideos.Remove(otherVideo); } } await dbContext.SaveChangesAsync(); return ids; } public async Task AddGenre(OtherVideoMetadata metadata, Genre genre) { await using TvContext dbContext = await _dbContextFactory.CreateDbContextAsync(); return await dbContext.Connection.ExecuteAsync( "INSERT INTO Genre (Name, OtherVideoMetadataId) VALUES (@Name, @MetadataId)", new { genre.Name, MetadataId = metadata.Id }).Map(result => result > 0); } public async Task AddTag(OtherVideoMetadata metadata, Tag tag) { await using TvContext dbContext = await _dbContextFactory.CreateDbContextAsync(); return await dbContext.Connection.ExecuteAsync( "INSERT INTO Tag (Name, OtherVideoMetadataId, ExternalCollectionId) VALUES (@Name, @MetadataId, @ExternalCollectionId)", new { tag.Name, MetadataId = metadata.Id, tag.ExternalCollectionId }).Map(result => result > 0); } public async Task AddStudio(OtherVideoMetadata metadata, Studio studio) { await using TvContext dbContext = await _dbContextFactory.CreateDbContextAsync(); return await dbContext.Connection.ExecuteAsync( "INSERT INTO Studio (Name, OtherVideoMetadataId) VALUES (@Name, @MetadataId)", new { studio.Name, MetadataId = metadata.Id }).Map(result => result > 0); } public async Task AddActor(OtherVideoMetadata metadata, Actor actor) { await using TvContext dbContext = await _dbContextFactory.CreateDbContextAsync(); int? artworkId = null; if (actor.Artwork != null) { artworkId = await dbContext.Connection.QuerySingleAsync( $""" INSERT INTO Artwork (ArtworkKind, DateAdded, DateUpdated, Path) VALUES (@ArtworkKind, @DateAdded, @DateUpdated, @Path); SELECT {TvContext.LastInsertedRowId} """, new { ArtworkKind = (int)actor.Artwork.ArtworkKind, actor.Artwork.DateAdded, actor.Artwork.DateUpdated, actor.Artwork.Path }); } return await dbContext.Connection.ExecuteAsync( "INSERT INTO Actor (Name, Role, `Order`, OtherVideoMetadataId, ArtworkId) VALUES (@Name, @Role, @Order, @MetadataId, @ArtworkId)", new { actor.Name, actor.Role, actor.Order, MetadataId = metadata.Id, ArtworkId = artworkId }) .Map(result => result > 0); } public async Task AddDirector(OtherVideoMetadata metadata, Director director) { await using TvContext dbContext = await _dbContextFactory.CreateDbContextAsync(); return await dbContext.Connection.ExecuteAsync( "INSERT INTO Director (Name, OtherVideoMetadataId) VALUES (@Name, @MetadataId)", new { director.Name, MetadataId = metadata.Id }).Map(result => result > 0); } public async Task AddWriter(OtherVideoMetadata metadata, Writer writer) { await using TvContext dbContext = await _dbContextFactory.CreateDbContextAsync(); return await dbContext.Connection.ExecuteAsync( "INSERT INTO Writer (Name, OtherVideoMetadataId) VALUES (@Name, @MetadataId)", new { writer.Name, MetadataId = metadata.Id }).Map(result => result > 0); } public async Task> GetOtherVideosForCards(List ids) { await using TvContext dbContext = await _dbContextFactory.CreateDbContextAsync(); return await dbContext.OtherVideoMetadata .AsNoTracking() .Filter(ovm => ids.Contains(ovm.OtherVideoId)) .Include(ovm => ovm.OtherVideo) .Include(ovm => ovm.Artwork) .Include(ovm => ovm.OtherVideo) .ThenInclude(ov => ov.MediaVersions) .ThenInclude(mv => mv.MediaFiles) .OrderBy(ovm => ovm.SortTitle) .ToListAsync(); } private async Task>> AddOtherVideo( TvContext dbContext, int libraryPathId, int libraryFolderId, string path) { try { if (await MediaItemRepository.MediaFileAlreadyExists(path, libraryPathId, dbContext, _logger)) { return new MediaFileAlreadyExists(); } var otherVideo = new OtherVideo { LibraryPathId = libraryPathId, MediaVersions = [ new MediaVersion { MediaFiles = [new MediaFile { Path = path, LibraryFolderId = libraryFolderId }], Streams = [] } ], TraktListItems = [] }; await dbContext.OtherVideos.AddAsync(otherVideo); await dbContext.SaveChangesAsync(); await dbContext.Entry(otherVideo).Reference(m => m.LibraryPath).LoadAsync(); await dbContext.Entry(otherVideo.LibraryPath).Reference(lp => lp.Library).LoadAsync(); return new MediaItemScanResult(otherVideo) { IsAdded = true }; } catch (Exception ex) { return BaseError.New(ex.Message); } } }