using Company.Domain.InstitutionContractExtensionTempAgg;
using MongoDB.Driver;

namespace CompanyManagement.Infrastructure.Mongo.InstitutionContractInsertTempRepo;

public class InstitutionContractExtenstionTempRepository:IInstitutionContractExtenstionTempRepository
{
    private readonly IMongoCollection<InstitutionContractExtensionTemp> _institutionExtenstionTemp;

    public InstitutionContractExtenstionTempRepository(IMongoDatabase database)
    {
        _institutionExtenstionTemp = database.GetCollection<InstitutionContractExtensionTemp>("InstitutionContractExtenstionTemp");
    }

    public async Task Create(InstitutionContractExtensionTemp institutionContract)
    {
        await _institutionExtenstionTemp.InsertOneAsync(institutionContract);
    }

    public async Task<InstitutionContractExtensionTemp> GetPreviousExtenstionData(long contractingPartyId)
    {
        var entity = await _institutionExtenstionTemp
            .Find(x => x.PreviousId == contractingPartyId)
            .FirstOrDefaultAsync();
        return entity;
    }

    public async Task Remove(Guid id)
    {
        await _institutionExtenstionTemp.DeleteOneAsync(x=>x.Id == id);
    }
}