using Managing.Application.Abstractions;
using Managing.Domain.Backtests;
using Managing.Infrastructure.Databases.MongoDb;
using Managing.Infrastructure.Databases.MongoDb.Abstractions;
using Managing.Infrastructure.Databases.MongoDb.Collections;

namespace Managing.Infrastructure.Databases;

public class BacktestRepository : IBacktestRepository
{
    private readonly IMongoRepository<BacktestDto> _backtestRepository;

    public BacktestRepository(IMongoRepository<BacktestDto> backtestRepository)
    {
        _backtestRepository = backtestRepository;
    }

    public void DeleteAllBacktests()
    {
        _backtestRepository.DropCollection();
    }

    public void DeleteBacktestById(string id)
    {
        _backtestRepository.DeleteById(id);
    }

    public IEnumerable<Backtest> GetBacktests()
    {
        var backtests = _backtestRepository.FindAll();
        return backtests.Select(b => MongoMappers.Map(b));
    }

    public void InsertBacktest(Backtest backtest)
    {
        _backtestRepository.InsertOne(MongoMappers.Map(backtest));
    }
}