using FreeTubeSync.Database;
using Microsoft.EntityFrameworkCore;

namespace FreeTubeSync;

public class Repository<TEntity>(DataContext dbContext) : IRepository<TEntity> where TEntity : class
{
    public async Task AddAsync(TEntity entity, CancellationToken ct, bool sync = true)
    {
        dbContext.Set<TEntity>().Add(entity);
        if(sync)
            await dbContext.SaveChangesAsync(ct);
    }

    public async Task UpdateAsync(TEntity entity, CancellationToken ct, bool sync = true)
    {
        dbContext.Set<TEntity>().Update(entity);
        if (sync)
            await dbContext.SaveChangesAsync(ct);
    }

    public async Task DeleteAsync(TEntity entity, CancellationToken ct, bool sync = true)
    {
        dbContext.Set<TEntity>().Remove(entity);
        if (sync)
            await dbContext.SaveChangesAsync(ct);
    }

    public async Task<TEntity?> GetByIdAsync(string id, CancellationToken ct)
    {
        return await dbContext.Set<TEntity>().FindAsync(id, ct);
    }

    public async Task<IEnumerable<TEntity>> GetAllAsync(CancellationToken ct)
    {
        return await dbContext.Set<TEntity>().ToListAsync<TEntity>(ct);
    }
}