ErsatzTV/ErsatzTV.Infrastructure/Data/Repositories/MediaSourceRepository.cs

using System.Collections.Generic;
using System.Data;
using System.Linq;
using System.Threading.Tasks;
using Dapper;
using ErsatzTV.Core.Domain;
using ErsatzTV.Core.Interfaces.Repositories;
using LanguageExt;
using Microsoft.EntityFrameworkCore;
using static LanguageExt.Prelude;

namespace ErsatzTV.Infrastructure.Data.Repositories
{
    public class MediaSourceRepository : IMediaSourceRepository
    {
        private readonly IDbConnection _dbConnection;
        private readonly IDbContextFactory<TvContext> _dbContextFactory;

        public MediaSourceRepository(
            IDbContextFactory<TvContext> dbContextFactory,
            IDbConnection dbConnection)
        {
            _dbContextFactory = dbContextFactory;
            _dbConnection = dbConnection;
        }

        public async Task<LocalMediaSource> Add(LocalMediaSource localMediaSource)
        {
            await using TvContext context = _dbContextFactory.CreateDbContext();
            await context.LocalMediaSources.AddAsync(localMediaSource);
            await context.SaveChangesAsync();
            return localMediaSource;
        }

        public async Task<PlexMediaSource> Add(PlexMediaSource plexMediaSource)
        {
            await using TvContext context = _dbContextFactory.CreateDbContext();
            await context.PlexMediaSources.AddAsync(plexMediaSource);
            await context.SaveChangesAsync();
            return plexMediaSource;
        }

        public async Task<List<MediaSource>> GetAll()
        {
            await using TvContext context = _dbContextFactory.CreateDbContext();
            List<MediaSource> all = await context.MediaSources.ToListAsync();
            foreach (PlexMediaSource plex in all.OfType<PlexMediaSource>())
            {
                await context.Entry(plex).Collection(p => p.Connections).LoadAsync();
            }

            return all;
        }

        public Task<List<PlexMediaSource>> GetAllPlex()
        {
            using TvContext context = _dbContextFactory.CreateDbContext();
            return context.PlexMediaSources
                .Include(p => p.Connections)
                .ToListAsync();
        }

        public Task<List<PlexLibrary>> GetPlexLibraries(int plexMediaSourceId)
        {
            using TvContext context = _dbContextFactory.CreateDbContext();
            return context.PlexLibraries
                .Filter(l => l.MediaSourceId == plexMediaSourceId)
                .ToListAsync();
        }

        public Task<List<PlexPathReplacement>> GetPlexPathReplacements(int plexMediaSourceId)
        {
            using TvContext context = _dbContextFactory.CreateDbContext();
            return context.PlexPathReplacements
                .Filter(r => r.PlexMediaSourceId == plexMediaSourceId)
                .ToListAsync();
        }

        public Task<Option<PlexLibrary>> GetPlexLibrary(int plexLibraryId)
        {
            using TvContext context = _dbContextFactory.CreateDbContext();
            return context.PlexLibraries
                .Include(l => l.Paths)
                .OrderBy(l => l.Id) // https://github.com/dotnet/efcore/issues/22579
                .SingleOrDefaultAsync(l => l.Id == plexLibraryId)
                .Map(Optional);
        }

        public Task<Option<MediaSource>> Get(int id)
        {
            using TvContext context = _dbContextFactory.CreateDbContext();
            return context.MediaSources
                .OrderBy(s => s.Id) // https://github.com/dotnet/efcore/issues/22579
                .SingleOrDefaultAsync(s => s.Id == id)
                .Map(Optional);
        }

        public Task<Option<PlexMediaSource>> GetPlex(int id)
        {
            using TvContext context = _dbContextFactory.CreateDbContext();
            return context.PlexMediaSources
                .Include(p => p.Connections)
                .Include(p => p.Libraries)
                .Include(p => p.PathReplacements)
                .OrderBy(s => s.Id) // https://github.com/dotnet/efcore/issues/22579
                .SingleOrDefaultAsync(p => p.Id == id)
                .Map(Optional);
        }

        public async Task<Option<PlexMediaSource>> GetPlexByLibraryId(int plexLibraryId)
        {
            int? id = await _dbConnection.QuerySingleAsync<int?>(
                @"SELECT L.MediaSourceId FROM Library L
                INNER JOIN PlexLibrary PL on L.Id = PL.Id
                WHERE L.Id = @PlexLibraryId",
                new { PlexLibraryId = plexLibraryId });

            await using TvContext context = _dbContextFactory.CreateDbContext();
            return await context.PlexMediaSources
                .Include(p => p.Connections)
                .Include(p => p.Libraries)
                .OrderBy(p => p.Id)
                .SingleOrDefaultAsync(p => p.Id == id)
                .Map(Optional);
        }

        public Task<List<PlexPathReplacement>> GetPlexPathReplacementsByLibraryId(int plexLibraryPathId)
        {
            using TvContext context = _dbContextFactory.CreateDbContext();
            return context.PlexPathReplacements
                .FromSqlRaw(
                    @"select ppr.* from LibraryPath lp
                    inner join PlexLibrary pl ON pl.Id = lp.LibraryId
                    inner join Library l ON l.Id = pl.Id
                    inner join PlexPathReplacement ppr on ppr.PlexMediaSourceId = l.MediaSourceId
                    where lp.Id = {0}",
                    plexLibraryPathId)
                .Include(ppr => ppr.PlexMediaSource)
                .ToListAsync();
        }

        public Task<int> CountMediaItems(int id)
        {
            using TvContext context = _dbContextFactory.CreateDbContext();
            return context.MediaItems
                .CountAsync(i => i.LibraryPath.Library.MediaSourceId == id);
        }

        public async Task Update(LocalMediaSource localMediaSource)
        {
            await using TvContext context = _dbContextFactory.CreateDbContext();
            context.LocalMediaSources.Update(localMediaSource);
            await context.SaveChangesAsync();
        }

        public async Task Update(
            PlexMediaSource plexMediaSource,
            List<PlexConnection> sortedConnections,
            List<PlexConnection> toAdd,
            List<PlexConnection> toDelete)
        {
            await _dbConnection.ExecuteAsync(
                @"UPDATE PlexMediaSource SET
                  ProductVersion = @ProductVersion,
                  Platform = @Platform,
                  PlatformVersion = @PlatformVersion,
                  ServerName = @ServerName
                  WHERE Id = @Id",
                new
                {
                    plexMediaSource.ProductVersion,
                    plexMediaSource.Platform,
                    plexMediaSource.PlatformVersion,
                    plexMediaSource.ServerName,
                    plexMediaSource.Id
                });

            foreach (PlexConnection add in toAdd)
            {
                await _dbConnection.ExecuteAsync(
                    @"INSERT INTO PlexConnection (IsActive, Uri, PlexMediaSourceId)
                    VALUES (0, @Uri, @PlexMediaSourceId)",
                    new { add.Uri, PlexMediaSourceId = plexMediaSource.Id });
            }

            foreach (PlexConnection delete in toDelete)
            {
                await _dbConnection.ExecuteAsync(
                    @"DELETE FROM PlexConnection WHERE Id = @Id",
                    new { delete.Id });
            }

            int activeCount = await _dbConnection.QuerySingleAsync<int>(
                @"SELECT COUNT(*) FROM PlexConnection WHERE IsActive = 1 AND PlexMediaSourceId = @PlexMediaSourceId",
                new { PlexMediaSourceId = plexMediaSource.Id });
            if (activeCount == 0)
            {
                Option<PlexConnection> toActivate =
                    sortedConnections.FirstOrDefault(c => toDelete.All(d => d.Id != c.Id));

                // update on uri because connections from Plex API don't have our local ids
                await toActivate.IfSomeAsync(
                    async c => await _dbConnection.ExecuteAsync(
                        @"UPDATE PlexConnection SET IsActive = 1 WHERE Uri = @Uri",
                        new { c.Uri }));
            }
        }

        public async Task<Unit> UpdateLibraries(
            int plexMediaSourceId,
            List<PlexLibrary> toAdd,
            List<PlexLibrary> toDelete)
        {
            await using TvContext dbContext = _dbContextFactory.CreateDbContext();

            foreach (PlexLibrary add in toAdd)
            {
                add.MediaSourceId = plexMediaSourceId;
                dbContext.Entry(add).State = EntityState.Added;
                foreach (LibraryPath path in add.Paths)
                {
                    dbContext.Entry(path).State = EntityState.Added;
                }
            }

            foreach (PlexLibrary delete in toDelete)
            {
                dbContext.Entry(delete).State = EntityState.Deleted;
            }

            await dbContext.SaveChangesAsync();

            return Unit.Default;
        }

        public async Task<Unit> UpdatePathReplacements(
            int plexMediaSourceId,
            List<PlexPathReplacement> toAdd,
            List<PlexPathReplacement> toUpdate,
            List<PlexPathReplacement> toDelete)
        {
            foreach (PlexPathReplacement add in toAdd)
            {
                await _dbConnection.ExecuteAsync(
                    @"INSERT INTO PlexPathReplacement
                    (PlexPath, LocalPath, PlexMediaSourceId)
                    VALUES (@PlexPath, @LocalPath, @PlexMediaSourceId)",
                    new { add.PlexPath, add.LocalPath, PlexMediaSourceId = plexMediaSourceId });
            }

            foreach (PlexPathReplacement update in toUpdate)
            {
                await _dbConnection.ExecuteAsync(
                    @"UPDATE PlexPathReplacement
                    SET PlexPath = @PlexPath, LocalPath = @LocalPath
                    WHERE Id = @Id",
                    new { update.PlexPath, update.LocalPath, update.Id });
            }

            foreach (PlexPathReplacement delete in toDelete)
            {
                await _dbConnection.ExecuteAsync(
                    @"DELETE FROM PlexPathReplacement WHERE Id = @Id",
                    new { delete.Id });
            }

            return Unit.Default;
        }

        public async Task Update(PlexLibrary plexMediaSourceLibrary)
        {
            await using TvContext context = _dbContextFactory.CreateDbContext();
            context.PlexLibraries.Update(plexMediaSourceLibrary);
            await context.SaveChangesAsync();
        }

        public async Task Delete(int mediaSourceId)
        {
            await using TvContext context = _dbContextFactory.CreateDbContext();
            MediaSource mediaSource = await context.MediaSources.FindAsync(mediaSourceId);
            context.MediaSources.Remove(mediaSource);
            await context.SaveChangesAsync();
        }

        public async Task<List<int>> DeleteAllPlex()
        {
            await using TvContext context = _dbContextFactory.CreateDbContext();

            List<PlexMediaSource> allMediaSources = await context.PlexMediaSources.ToListAsync();
            context.PlexMediaSources.RemoveRange(allMediaSources);

            List<PlexLibrary> allPlexLibraries = await context.PlexLibraries.ToListAsync();
            context.PlexLibraries.RemoveRange(allPlexLibraries);

            List<int> movieIds = await context.PlexMovies.Map(pm => pm.Id).ToListAsync();
            List<int> showIds = await context.PlexShows.Map(ps => ps.Id).ToListAsync();

            await context.SaveChangesAsync();

            return movieIds.Append(showIds).ToList();
        }

        public async Task<List<int>> DeletePlex(PlexMediaSource plexMediaSource)
        {
            List<int> mediaItemIds = await _dbConnection.QueryAsync<int>(
                    @"SELECT MediaItem.Id FROM MediaItem
                  INNER JOIN LibraryPath LP on MediaItem.LibraryPathId = LP.Id
                  INNER JOIN Library L on LP.LibraryId = L.Id
                  WHERE L.MediaSourceId = @PlexMediaSourceId",
                    new { PlexMediaSourceId = plexMediaSource.Id })
                .Map(result => result.ToList());

            await _dbConnection.ExecuteAsync(
                @"DELETE FROM MediaSource WHERE Id = @PlexMediaSourceId",
                new { PlexMediaSourceId = plexMediaSource.Id });

            return mediaItemIds;
        }

        public async Task<List<int>> DisablePlexLibrarySync(List<int> libraryIds)
        {
            await _dbConnection.ExecuteAsync(
                "UPDATE PlexLibrary SET ShouldSyncItems = 0 WHERE Id IN @ids",
                new { ids = libraryIds });

            await _dbConnection.ExecuteAsync(
                "UPDATE Library SET LastScan = null WHERE Id IN @ids",
                new { ids = libraryIds });

            List<int> movieIds = await _dbConnection.QueryAsync<int>(
                @"SELECT m.Id FROM MediaItem m
                INNER JOIN PlexMovie pm ON pm.Id = m.Id
                INNER JOIN LibraryPath lp ON lp.Id = m.LibraryPathId
                INNER JOIN Library l ON l.Id = lp.LibraryId
                WHERE l.Id IN @ids",
                new { ids = libraryIds }).Map(result => result.ToList());

            await _dbConnection.ExecuteAsync(
                @"DELETE FROM MediaItem WHERE Id IN
                (SELECT m.Id FROM MediaItem m
                INNER JOIN PlexMovie pm ON pm.Id = m.Id
                INNER JOIN LibraryPath lp ON lp.Id = m.LibraryPathId
                INNER JOIN Library l ON l.Id = lp.LibraryId
                WHERE l.Id IN @ids)",
                new { ids = libraryIds });

            await _dbConnection.ExecuteAsync(
                @"DELETE FROM MediaItem WHERE Id IN
                (SELECT m.Id FROM MediaItem m
                INNER JOIN PlexEpisode pe ON pe.Id = m.Id
                INNER JOIN LibraryPath lp ON lp.Id = m.LibraryPathId
                INNER JOIN Library l ON l.Id = lp.LibraryId
                WHERE l.Id IN @ids)",
                new { ids = libraryIds });

            await _dbConnection.ExecuteAsync(
                @"DELETE FROM MediaItem WHERE Id IN
                (SELECT m.Id FROM MediaItem m
                INNER JOIN PlexSeason ps ON ps.Id = m.Id
                INNER JOIN LibraryPath lp ON lp.Id = m.LibraryPathId
                INNER JOIN Library l ON l.Id = lp.LibraryId
                WHERE l.Id IN @ids)",
                new { ids = libraryIds });

            List<int> showIds = await _dbConnection.QueryAsync<int>(
                @"SELECT m.Id FROM MediaItem m
                INNER JOIN PlexShow ps ON ps.Id = m.Id
                INNER JOIN LibraryPath lp ON lp.Id = m.LibraryPathId
                INNER JOIN Library l ON l.Id = lp.LibraryId
                WHERE l.Id IN @ids",
                new { ids = libraryIds }).Map(result => result.ToList());

            await _dbConnection.ExecuteAsync(
                @"DELETE FROM MediaItem WHERE Id IN
                (SELECT m.Id FROM MediaItem m
                INNER JOIN PlexShow ps ON ps.Id = m.Id
                INNER JOIN LibraryPath lp ON lp.Id = m.LibraryPathId
                INNER JOIN Library l ON l.Id = lp.LibraryId
                WHERE l.Id IN @ids)",
                new { ids = libraryIds });

            return movieIds.Append(showIds).ToList();
        }

        public Task EnablePlexLibrarySync(IEnumerable<int> libraryIds) =>
            _dbConnection.ExecuteAsync(
                "UPDATE PlexLibrary SET ShouldSyncItems = 1 WHERE Id IN @ids",
                new { ids = libraryIds });
    }
}