using DD.Persistence.Database.Entity;
using DD.Persistence.Models;
using DD.Persistence.Repositories;
using Mapster;
using Microsoft.EntityFrameworkCore;

namespace DD.Persistence.Repository.Repositories;
public class DataSchemeRepository : IDataSchemeRepository
{
    protected DbContext db;
    public DataSchemeRepository(DbContext db)
    {
        this.db = db;
    }
    protected virtual IQueryable<DataScheme> GetQueryReadOnly() => db.Set<DataScheme>();

    public virtual async Task Add(DataSchemeDto dataSourceSystemDto, CancellationToken token)
    {
        var entity = dataSourceSystemDto.Adapt<DataScheme>();

        await db.Set<DataScheme>().AddAsync(entity, token);
        await db.SaveChangesAsync(token);
    }

    public virtual async Task<DataSchemeDto?> Get(Guid dataSchemeId, CancellationToken token)
    {
        var query = GetQueryReadOnly()
            .Where(e => e.DiscriminatorId == dataSchemeId);
        var entity = await query.ToArrayAsync();
        var dto = entity.Select(e => e.Adapt<DataSchemeDto>()).FirstOrDefault();

        return dto;
    }
}