Creating the genre repository

This commit is contained in:
Zoe Roux 2020-06-03 02:08:17 +02:00
parent 4414d94c7e
commit bd1ef0e15b
2 changed files with 122 additions and 27 deletions

View File

@ -0,0 +1,87 @@
using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using Kyoo.Models;
using Kyoo.Models.Exceptions;
using Microsoft.EntityFrameworkCore;
namespace Kyoo.Controllers
{
public class GenreRepository : IGenreRepository
{
private readonly DatabaseContext _database;
public GenreRepository(DatabaseContext database)
{
_database = database;
}
public async Task<Genre> Get(long id)
{
return await _database.Genres.FirstOrDefaultAsync(x => x.ID == id);
}
public async Task<Genre> Get(string slug)
{
return await _database.Genres.FirstOrDefaultAsync(x => x.Slug == slug);
}
public async Task<IEnumerable<Genre>> Search(string query)
{
return await _database.Genres
.Where(genre => EF.Functions.Like(genre.Name, $"%{query}%"))
.Take(20)
.ToListAsync();
}
public async Task<IEnumerable<Genre>> GetAll()
{
return await _database.Genres.ToListAsync();
}
public async Task<long> Create(Genre obj)
{
if (obj == null)
throw new ArgumentNullException(nameof(obj));
await _database.Genres.AddAsync(obj);
await _database.SaveChangesAsync();
return obj.ID;
}
public async Task<long> CreateIfNotExists(Genre obj)
{
if (obj == null)
throw new ArgumentNullException(nameof(obj));
Genre old = await Get(obj.Slug);
if (old != null)
return obj.ID;
return await Create(obj);
}
public async Task Edit(Genre edited, bool resetOld)
{
if (edited == null)
throw new ArgumentNullException(nameof(edited));
Genre old = await Get(edited.Slug);
if (old == null)
throw new ItemNotFound($"No genre found with the slug {edited.Slug}.");
if (resetOld)
Utility.Nullify(old);
Utility.Merge(old, edited);
await _database.SaveChangesAsync();
}
public async Task Delete(Genre obj)
{
_database.Genres.Remove(obj);
await _database.SaveChangesAsync();
}
}
}

View File

@ -29,49 +29,36 @@ namespace Kyoo.Controllers
_providers = providers;
}
public Task<Show> Get(long id)
public async Task<Show> Get(long id)
{
return Task.FromResult(_database.Shows.FirstOrDefault(x => x.ID == id));
return await _database.Shows.FirstOrDefaultAsync(x => x.ID == id);
}
public Task<Show> Get(string slug)
public async Task<Show> Get(string slug)
{
return Task.FromResult(_database.Shows.FirstOrDefault(x => x.Slug == slug));
return await _database.Shows.FirstOrDefaultAsync(x => x.Slug == slug);
}
public Task<IEnumerable<Show>> Search(string query)
public async Task<IEnumerable<Show>> Search(string query)
{
return Task.FromResult<IEnumerable<Show>>(
_database.Shows.FromSqlInterpolated($@"SELECT * FROM Shows WHERE Shows.Title LIKE {$"%{query}%"}
OR Shows.Aliases LIKE {$"%{query}%"}").Take(20).ToList());
return await _database.Shows
.FromSqlInterpolated($@"SELECT * FROM Shows WHERE Shows.Title LIKE {$"%{query}%"}
OR Shows.Aliases LIKE {$"%{query}%"}")
.Take(20)
.ToListAsync();
}
public Task<IEnumerable<Show>> GetAll()
public async Task<IEnumerable<Show>> GetAll()
{
return Task.FromResult<IEnumerable<Show>>(_database.Shows.ToList());
return await _database.Shows.ToListAsync();
}
public async Task<long> Create(Show obj)
{
if (obj == null)
throw new ArgumentNullException(nameof(obj));
obj.StudioID = await _studio.CreateIfNotExists(obj.Studio);
obj.GenreLinks = (await Task.WhenAll(obj.GenreLinks.Select(async x =>
{
x.GenreID = await _genres.CreateIfNotExists(x.Genre);
return x;
}))).ToList();
obj.People = (await Task.WhenAll(obj.People.Select(async x =>
{
x.PeopleID = await _people.CreateIfNotExists(x.People);
return x;
}))).ToList();
obj.ExternalIDs = (await Task.WhenAll(obj.ExternalIDs.Select(async x =>
{
x.ProviderID = await _providers.CreateIfNotExists(x.Provider);
return x;
}))).ToList();
await Validate(obj);
obj.Seasons = null;
obj.Episodes = null;
@ -105,9 +92,30 @@ namespace Kyoo.Controllers
if (resetOld)
Utility.Nullify(old);
Utility.Merge(old, edited);
await Validate(old);
await _database.SaveChangesAsync();
}
private async Task Validate(Show obj)
{
obj.StudioID = await _studio.CreateIfNotExists(obj.Studio);
obj.GenreLinks = (await Task.WhenAll(obj.GenreLinks.Select(async x =>
{
x.GenreID = await _genres.CreateIfNotExists(x.Genre);
return x;
}))).ToList();
obj.People = (await Task.WhenAll(obj.People.Select(async x =>
{
x.PeopleID = await _people.CreateIfNotExists(x.People);
return x;
}))).ToList();
obj.ExternalIDs = (await Task.WhenAll(obj.ExternalIDs.Select(async x =>
{
x.ProviderID = await _providers.CreateIfNotExists(x.Provider);
return x;
}))).ToList();
}
public async Task Delete(Show show)
{
_database.Shows.Remove(show);