using Distribution.Models;
using Microsoft.AspNetCore.Mvc;

namespace Distribution.Features.DeliveryRide;

[ApiController]
[Route("api")]
public class DeliveryTaskController(DeliveryTaskService service) : ControllerBase
{
    [HttpPost("rides/{rideId:guid}/tasks/generate")]
    public async Task<IActionResult> Generate(Guid rideId, CancellationToken ct)
    {
        try
        {
            var result = await service.GenerateAsync(rideId, ct);
            return Ok(result);
        }
        catch (KeyNotFoundException) { return NotFound(); }
        catch (InvalidOperationException ex) { return BadRequest(new { error = ex.Message }); }
    }

    [HttpGet("rides/{rideId:guid}/tasks")]
    [HttpGet("rides/{rideId:guid}/delivery-tasks")]
    public async Task<IActionResult> GetByRide(Guid rideId)
    {
        var result = await service.GetByRideIdAsync(rideId);
        return Ok(result);
    }

    [HttpGet("delivery-tasks")]
    public async Task<IActionResult> GetAll(
        [FromQuery] DeliveryTaskStatus? status,
        [FromQuery] int page = 1,
        [FromQuery] int pageSize = 20)
    {
        var result = await service.GetAllAsync(status, page, pageSize);
        return Ok(result);
    }

    [HttpGet("delivery-tasks/{id:guid}")]
    public async Task<IActionResult> GetById(Guid id)
    {
        var result = await service.GetByIdAsync(id);
        return result is null ? NotFound() : Ok(result);
    }

    [HttpPut("delivery-tasks/{id:guid}/quantities")]
    public async Task<IActionResult> SetQuantities(Guid id, SetQuantitiesRequest request, CancellationToken ct)
    {
        try
        {
            var result = await service.SetQuantitiesAsync(id, request, ct);
            return Ok(result);
        }
        catch (KeyNotFoundException) { return NotFound(); }
        catch (InvalidOperationException ex) { return BadRequest(new { error = ex.Message }); }
    }

    [HttpPost("delivery-tasks/{id:guid}/confirm")]
    public async Task<IActionResult> Confirm(Guid id, CancellationToken ct)
    {
        try
        {
            var result = await service.ConfirmAsync(id, ct);
            return Ok(result);
        }
        catch (KeyNotFoundException) { return NotFound(); }
        catch (InvalidOperationException ex) { return BadRequest(new { error = ex.Message }); }
    }

    [HttpPost("delivery-tasks/{id:guid}/fail")]
    public async Task<IActionResult> Fail(Guid id, FailTaskRequest request, CancellationToken ct)
    {
        try
        {
            var result = await service.FailAsync(id, request.Note, ct);
            return Ok(result);
        }
        catch (KeyNotFoundException) { return NotFound(); }
        catch (InvalidOperationException ex) { return BadRequest(new { error = ex.Message }); }
    }
}