Kyoo/Kyoo/Controllers/Repositories/SeasonRepository.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using Kyoo.Models;
using Kyoo.Models.Exceptions;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.DependencyInjection;

namespace Kyoo.Controllers
{
	public class SeasonRepository : ISeasonRepository
	{
		private readonly DatabaseContext _database;
		private readonly IServiceProvider _serviceProvider;


		public SeasonRepository(DatabaseContext database, IServiceProvider serviceProvider)
		{
			_database = database;
			_serviceProvider = serviceProvider;
		}
		
		public async Task<Season> Get(int id)
		{
			return await _database.Seasons.FirstOrDefaultAsync(x => x.ID == id);
		}

		public Task<Season> Get(string slug)
		{
			int index = slug.IndexOf("-s", StringComparison.Ordinal);
			if (index == -1)
				throw new InvalidOperationException("Invalid season slug. Format: {showSlug}-s{seasonNumber}");
			string showSlug = slug.Substring(0, index);
			if (!int.TryParse(slug.Substring(index + 2), out int seasonNumber))
				throw new InvalidOperationException("Invalid season slug. Format: {showSlug}-s{seasonNumber}");
			return Get(showSlug, seasonNumber);
		}
		
		public async Task<Season> Get(string showSlug, int seasonNumber)
		{
			return await _database.Seasons.FirstOrDefaultAsync(x => x.Show.Slug == showSlug 
			                                                        && x.SeasonNumber == seasonNumber);
		}

		public async Task<ICollection<Season>> Search(string query)
		{
			return await _database.Seasons
				.Where(x => EF.Functions.Like(x.Title, $"%{query}%"))
				.Take(20)
				.ToListAsync();
		}

		public async Task<ICollection<Season>> GetAll()
		{
			return await _database.Seasons.ToListAsync();
		}

		public async Task<int> Create(Season obj)
		{
			if (obj == null)
				throw new ArgumentNullException(nameof(obj));
			
			await Validate(obj);
			_database.Entry(obj).State = EntityState.Added;
			if (obj.ExternalIDs != null)
				foreach (MetadataID entry in obj.ExternalIDs)
					_database.Entry(entry).State = EntityState.Added;
			await _database.SaveChangesAsync();
			return obj.ID;
		}
		
		public async Task<int> CreateIfNotExists(Season obj)
		{
			if (obj == null)
				throw new ArgumentNullException(nameof(obj));

			Season old = await Get(obj.Slug);
			if (old != null)
				return old.ID;
			return await Create(obj);
		}

		public async Task Edit(Season edited, bool resetOld)
		{
			if (edited == null)
				throw new ArgumentNullException(nameof(edited));
			
			Season old = await Get(edited.Slug);

			if (old == null)
				throw new ItemNotFound($"No season found with the slug {edited.Slug}.");
			
			if (resetOld)
				Utility.Nullify(old);
			Utility.Merge(old, edited);

			await Validate(old);
			await _database.SaveChangesAsync();
		}

		private async Task Validate(Season obj)
		{
			if (obj.ShowID <= 0)
				throw new InvalidOperationException($"Can't store a season not related to any show (showID: {obj.ShowID}).");

			if (obj.ExternalIDs != null)
			{
				obj.ExternalIDs = (await Task.WhenAll(obj.ExternalIDs.Select(async x =>
				{
					using IServiceScope serviceScope = _serviceProvider.CreateScope();
					IProviderRepository providers = serviceScope.ServiceProvider.GetService<IProviderRepository>();
					
					x.ProviderID = await providers.CreateIfNotExists(x.Provider);
					return x;
				}))).ToList();
			}
		}

		public async Task Delete(Season obj)
		{
			_database.Seasons.Remove(obj);
			await _database.SaveChangesAsync();
		}
		
		public async Task<ICollection<Season>> GetSeasons(int showID)
		{
			return await _database.Seasons.Where(x => x.ShowID == showID).ToListAsync();
		}

		public async Task<ICollection<Season>> GetSeasons(string showSlug)
		{
			return await _database.Seasons.Where(x => x.Show.Slug == showSlug).ToListAsync();
		}
	}
}