using Distribution.Data;
using Distribution.Models;
using Microsoft.EntityFrameworkCore;

namespace Distribution.Features.Ride;

public class RideRepository(AppDBContext context) : IRideRepository
{
    public async Task<DeliveryRun?> GetByIdAsync(Guid id) =>
        await context.DeliveryRuns
            .Include(r => r.Vehicle)
            .Include(r => r.DriverProfile).ThenInclude(dp => dp.User)
            .FirstOrDefaultAsync(r => r.Id == id);

    public async Task<List<DeliveryRun>> GetAllAsync(RideStatus? status, Guid? driverProfileId, Guid? vehicleId, int page, int pageSize) =>
        await context.DeliveryRuns
            .Include(r => r.Vehicle)
            .Include(r => r.DriverProfile).ThenInclude(dp => dp.User)
            .Where(r =>
                (status == null || r.Status == status) &&
                (driverProfileId == null || r.FK_DriverProfileId == driverProfileId) &&
                (vehicleId == null || r.FK_VehicleId == vehicleId))
            .Skip((page - 1) * pageSize)
            .Take(pageSize)
            .ToListAsync();

    public async Task<int> GetCountAsync(RideStatus? status, Guid? driverProfileId, Guid? vehicleId) =>
        await context.DeliveryRuns
            .CountAsync(r =>
                (status == null || r.Status == status) &&
                (driverProfileId == null || r.FK_DriverProfileId == driverProfileId) &&
                (vehicleId == null || r.FK_VehicleId == vehicleId));

    public async Task<bool> HasOpenRideForVehicleAsync(Guid vehicleId, Guid? excludeRideId = null) =>
        await context.DeliveryRuns.AnyAsync(r =>
            r.FK_VehicleId == vehicleId &&
            r.Id != excludeRideId &&
            r.Status == RideStatus.Open);

    public async Task AddAsync(DeliveryRun ride) =>
        await context.DeliveryRuns.AddAsync(ride);

    public void Update(DeliveryRun ride) =>
        context.DeliveryRuns.Update(ride);
}