ersatztv/ErsatzTV.Infrastructure/Data/Repositories/MetadataRepository.cs

using System;
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 MetadataRepository : IMetadataRepository
    {
        private readonly IDbConnection _dbConnection;
        private readonly IDbContextFactory<TvContext> _dbContextFactory;

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

        public Task<bool> RemoveActor(Actor actor) =>
            _dbConnection.ExecuteAsync("DELETE FROM Actor WHERE Id = @ActorId", new { ActorId = actor.Id })
                .Map(result => result > 0);

        public async Task<bool> Update(Metadata metadata)
        {
            await using TvContext dbContext = _dbContextFactory.CreateDbContext();
            dbContext.Entry(metadata).State = EntityState.Modified;
            return await dbContext.SaveChangesAsync() > 0;
        }

        public async Task<bool> Add(Metadata metadata)
        {
            await using TvContext dbContext = _dbContextFactory.CreateDbContext();
            dbContext.Entry(metadata).State = EntityState.Added;

            foreach (Genre genre in Optional(metadata.Genres).Flatten())
            {
                dbContext.Entry(genre).State = EntityState.Added;
            }

            foreach (Tag tag in Optional(metadata.Tags).Flatten())
            {
                dbContext.Entry(tag).State = EntityState.Added;
            }

            foreach (Studio studio in Optional(metadata.Studios).Flatten())
            {
                dbContext.Entry(studio).State = EntityState.Added;
            }

            if (metadata is ArtistMetadata artistMetadata)
            {
                foreach (Style style in Optional(artistMetadata.Styles).Flatten())
                {
                    dbContext.Entry(style).State = EntityState.Added;
                }

                foreach (Mood mood in Optional(artistMetadata.Moods).Flatten())
                {
                    dbContext.Entry(mood).State = EntityState.Added;
                }
            }

            foreach (Actor actor in Optional(metadata.Actors).Flatten())
            {
                dbContext.Entry(actor).State = EntityState.Added;
                if (actor.Artwork != null)
                {
                    dbContext.Entry(actor.Artwork).State = EntityState.Added;
                }
            }

            foreach (MetadataGuid guid in Optional(metadata.Guids).Flatten())
            {
                dbContext.Entry(guid).State = EntityState.Added;
            }

            if (metadata is MovieMetadata movieMetadata)
            {
                foreach (Director director in Optional(movieMetadata.Directors).Flatten())
                {
                    dbContext.Entry(director).State = EntityState.Added;
                }

                foreach (Writer writer in Optional(movieMetadata.Writers).Flatten())
                {
                    dbContext.Entry(writer).State = EntityState.Added;
                }
            }

            if (metadata is EpisodeMetadata episodeMetadata)
            {
                foreach (Director director in Optional(episodeMetadata.Directors).Flatten())
                {
                    dbContext.Entry(director).State = EntityState.Added;
                }

                foreach (Writer writer in Optional(episodeMetadata.Writers).Flatten())
                {
                    dbContext.Entry(writer).State = EntityState.Added;
                }
            }

            return await dbContext.SaveChangesAsync() > 0;
        }

        public async Task<bool> UpdateLocalStatistics(
            int mediaVersionId,
            MediaVersion incoming,
            bool updateVersion = true)
        {
            await using TvContext dbContext = _dbContextFactory.CreateDbContext();
            Option<MediaVersion> maybeVersion = await dbContext.MediaVersions
                .Include(v => v.Streams)
                .OrderBy(v => v.Id)
                .SingleOrDefaultAsync(v => v.Id == mediaVersionId)
                .Map(Optional);

            return await maybeVersion.Match(
                async existing =>
                {
                    if (updateVersion)
                    {
                        existing.DateUpdated = incoming.DateUpdated;
                        existing.Duration = incoming.Duration;
                        existing.SampleAspectRatio = incoming.SampleAspectRatio;
                        existing.DisplayAspectRatio = incoming.DisplayAspectRatio;
                        existing.Width = incoming.Width;
                        existing.Height = incoming.Height;
                        existing.VideoScanKind = incoming.VideoScanKind;
                    }

                    var toAdd = incoming.Streams.Filter(s => existing.Streams.All(es => es.Index != s.Index)).ToList();
                    var toRemove = existing.Streams.Filter(es => incoming.Streams.All(s => s.Index != es.Index))
                        .ToList();
                    var toUpdate = incoming.Streams.Except(toAdd).ToList();

                    // add
                    existing.Streams.AddRange(toAdd);

                    // remove
                    existing.Streams.RemoveAll(s => toRemove.Contains(s));

                    // update
                    foreach (MediaStream incomingStream in toUpdate)
                    {
                        MediaStream existingStream = existing.Streams.First(s => s.Index == incomingStream.Index);

                        existingStream.Codec = incomingStream.Codec;
                        existingStream.Profile = incomingStream.Profile;
                        existingStream.MediaStreamKind = incomingStream.MediaStreamKind;
                        existingStream.Language = incomingStream.Language;
                        existingStream.Channels = incomingStream.Channels;
                        existingStream.Title = incomingStream.Title;
                        existingStream.Default = incomingStream.Default;
                        existingStream.Forced = incomingStream.Forced;
                    }

                    return await dbContext.SaveChangesAsync() > 0;
                },
                () => Task.FromResult(false));
        }

        public Task<bool> UpdatePlexStatistics(int mediaVersionId, MediaVersion incoming) =>
            _dbConnection.ExecuteAsync(
                @"UPDATE MediaVersion SET
                  DateUpdated = @DateUpdated
                  WHERE Id = @MediaVersionId",
                new
                {
                    incoming.DateUpdated,
                    MediaVersionId = mediaVersionId
                }).Map(result => result > 0);

        public Task<Unit> UpdateArtworkPath(Artwork artwork) =>
            _dbConnection.ExecuteAsync(
                "UPDATE Artwork SET Path = @Path, DateUpdated = @DateUpdated WHERE Id = @Id",
                new { artwork.Path, artwork.DateUpdated, artwork.Id }).ToUnit();

        public Task<Unit> AddArtwork(Metadata metadata, Artwork artwork)
        {
            var parameters = new
            {
                artwork.ArtworkKind, metadata.Id, artwork.DateAdded, artwork.DateUpdated, artwork.Path
            };

            return metadata switch
            {
                MovieMetadata => _dbConnection.ExecuteAsync(
                        @"INSERT INTO Artwork (ArtworkKind, MovieMetadataId, DateAdded, DateUpdated, Path)
                            VALUES (@ArtworkKind, @Id, @DateAdded, @DateUpdated, @Path)",
                        parameters)
                    .ToUnit(),
                ShowMetadata => _dbConnection.ExecuteAsync(
                        @"INSERT INTO Artwork (ArtworkKind, ShowMetadataId, DateAdded, DateUpdated, Path)
                            VALUES (@ArtworkKind, @Id, @DateAdded, @DateUpdated, @Path)",
                        parameters)
                    .ToUnit(),
                SeasonMetadata => _dbConnection.ExecuteAsync(
                        @"INSERT INTO Artwork (ArtworkKind, SeasonMetadataId, DateAdded, DateUpdated, Path)
                            VALUES (@ArtworkKind, @Id, @DateAdded, @DateUpdated, @Path)",
                        parameters)
                    .ToUnit(),
                EpisodeMetadata => _dbConnection.ExecuteAsync(
                        @"INSERT INTO Artwork (ArtworkKind, EpisodeMetadataId, DateAdded, DateUpdated, Path)
                            VALUES (@ArtworkKind, @Id, @DateAdded, @DateUpdated, @Path)",
                        parameters)
                    .ToUnit(),
                ArtistMetadata => _dbConnection.ExecuteAsync(
                        @"INSERT INTO Artwork (ArtworkKind, ArtistMetadataId, DateAdded, DateUpdated, Path)
                            Values (@ArtworkKind, @Id, @DateAdded, @DateUpdated, @Path)",
                        parameters)
                    .ToUnit(),
                MusicVideoMetadata => _dbConnection.ExecuteAsync(
                        @"INSERT INTO Artwork (ArtworkKind, MusicVideoMetadataId, DateAdded, DateUpdated, Path)
                            VALUES (@ArtworkKind, @Id, @DateAdded, @DateUpdated, @Path)",
                        parameters)
                    .ToUnit(),
                _ => Task.FromResult(Unit.Default)
            };
        }

        public Task<Unit> RemoveArtwork(Metadata metadata, ArtworkKind artworkKind) =>
            _dbConnection.ExecuteAsync(
                @"DELETE FROM Artwork WHERE ArtworkKind = @ArtworkKind AND (MovieMetadataId = @Id
                OR ShowMetadataId = @Id OR SeasonMetadataId = @Id OR EpisodeMetadataId = @Id)",
                new { ArtworkKind = artworkKind, metadata.Id }).ToUnit();

        public Task<Unit> MarkAsUpdated(ShowMetadata metadata, DateTime dateUpdated) =>
            _dbConnection.ExecuteAsync(
                @"UPDATE ShowMetadata SET DateUpdated = @DateUpdated WHERE Id = @Id",
                new { DateUpdated = dateUpdated, metadata.Id }).ToUnit();

        public Task<Unit> MarkAsUpdated(SeasonMetadata metadata, DateTime dateUpdated) =>
            _dbConnection.ExecuteAsync(
                @"UPDATE SeasonMetadata SET DateUpdated = @DateUpdated WHERE Id = @Id",
                new { DateUpdated = dateUpdated, metadata.Id }).ToUnit();

        public Task<Unit> MarkAsUpdated(MovieMetadata metadata, DateTime dateUpdated) =>
            _dbConnection.ExecuteAsync(
                @"UPDATE MovieMetadata SET DateUpdated = @DateUpdated WHERE Id = @Id",
                new { DateUpdated = dateUpdated, metadata.Id }).ToUnit();

        public Task<Unit> MarkAsUpdated(EpisodeMetadata metadata, DateTime dateUpdated) =>
            _dbConnection.ExecuteAsync(
                @"UPDATE EpisodeMetadata SET DateUpdated = @DateUpdated WHERE Id = @Id",
                new { DateUpdated = dateUpdated, metadata.Id }).ToUnit();

        public Task<Unit> MarkAsExternal(ShowMetadata metadata) =>
            _dbConnection.ExecuteAsync(
                @"UPDATE ShowMetadata SET MetadataKind = @Kind WHERE Id = @Id",
                new { metadata.Id, Kind = (int) MetadataKind.External }).ToUnit();

        public Task<Unit> SetContentRating(ShowMetadata metadata, string contentRating) =>
            _dbConnection.ExecuteAsync(
                @"UPDATE ShowMetadata SET ContentRating = @ContentRating WHERE Id = @Id",
                new { metadata.Id, ContentRating = contentRating }).ToUnit();

        public Task<Unit> MarkAsExternal(MovieMetadata metadata) =>
            _dbConnection.ExecuteAsync(
                @"UPDATE MovieMetadata SET MetadataKind = @Kind WHERE Id = @Id",
                new { metadata.Id, Kind = (int) MetadataKind.External }).ToUnit();

        public Task<Unit> SetContentRating(MovieMetadata metadata, string contentRating) =>
            _dbConnection.ExecuteAsync(
                @"UPDATE MovieMetadata SET ContentRating = @ContentRating WHERE Id = @Id",
                new { metadata.Id, ContentRating = contentRating }).ToUnit();

        public Task<bool> RemoveGuid(MetadataGuid guid) =>
            _dbConnection.ExecuteAsync("DELETE FROM MetadataGuid WHERE Id = @GuidId", new { GuidId = guid.Id })
                .Map(result => result > 0);

        public Task<bool> AddGuid(Metadata metadata, MetadataGuid guid) =>
            metadata switch
            {
                MovieMetadata =>
                    _dbConnection.ExecuteAsync(
                        "INSERT INTO MetadataGuid (Guid, MovieMetadataId) VALUES (@Guid, @MetadataId)",
                        new { guid.Guid, MetadataId = metadata.Id }).Map(result => result > 0),
                ShowMetadata =>
                    _dbConnection.ExecuteAsync(
                        "INSERT INTO MetadataGuid (Guid, ShowMetadataId) VALUES (@Guid, @MetadataId)",
                        new { guid.Guid, MetadataId = metadata.Id }).Map(result => result > 0),
                SeasonMetadata =>
                    _dbConnection.ExecuteAsync(
                        "INSERT INTO MetadataGuid (Guid, SeasonMetadataId) VALUES (@Guid, @MetadataId)",
                        new { guid.Guid, MetadataId = metadata.Id }).Map(result => result > 0),
                EpisodeMetadata =>
                    _dbConnection.ExecuteAsync(
                        "INSERT INTO MetadataGuid (Guid, EpisodeMetadataId) VALUES (@Guid, @MetadataId)",
                        new { guid.Guid, MetadataId = metadata.Id }).Map(result => result > 0),
                ArtistMetadata =>
                    _dbConnection.ExecuteAsync(
                        "INSERT INTO MetadataGuid (Guid, ArtistMetadataId) VALUES (@Guid, @MetadataId)",
                        new { guid.Guid, MetadataId = metadata.Id }).Map(result => result > 0),
                _ => throw new NotSupportedException()
            };

        public Task<bool> RemoveDirector(Director director) =>
            _dbConnection.ExecuteAsync("DELETE FROM Director WHERE Id = @DirectorId", new { DirectorId = director.Id })
                .Map(result => result > 0);

        public Task<bool> RemoveWriter(Writer writer) =>
            _dbConnection.ExecuteAsync("DELETE FROM Writer WHERE Id = @WriterId", new { WriterId = writer.Id })
                .Map(result => result > 0);

        public Task<bool> RemoveGenre(Genre genre) =>
            _dbConnection.ExecuteAsync("DELETE FROM Genre WHERE Id = @GenreId", new { GenreId = genre.Id })
                .Map(result => result > 0);

        public Task<bool> RemoveTag(Tag tag) =>
            _dbConnection.ExecuteAsync("DELETE FROM Tag WHERE Id = @TagId", new { TagId = tag.Id })
                .Map(result => result > 0);

        public Task<bool> RemoveStudio(Studio studio) =>
            _dbConnection.ExecuteAsync("DELETE FROM Studio WHERE Id = @StudioId", new { StudioId = studio.Id })
                .Map(result => result > 0);

        public Task<bool> RemoveStyle(Style style) =>
            _dbConnection.ExecuteAsync("DELETE FROM Style WHERE Id = @StyleId", new { StyleId = style.Id })
                .Map(result => result > 0);

        public Task<bool> RemoveMood(Mood mood) =>
            _dbConnection.ExecuteAsync("DELETE FROM Mood WHERE Id = @MoodId", new { MoodId = mood.Id })
                .Map(result => result > 0);
    }
}