RobDeliveryAPI/Controllers/RobotController.cs

using Application.Abstractions.Interfaces;
using Application.DTOs.RobotDTOs;
using Application.DTOs.MapDTOs;
using Microsoft.AspNetCore.Authorization;
using Microsoft.AspNetCore.Mvc;
using Microsoft.AspNetCore.SignalR;
using RobDeliveryAPI.Hubs;
using System.Security.Claims;

namespace RobDeliveryAPI.Controllers
{
    [ApiController]
    [Route("api/[controller]")]
    [Authorize]
    public class RobotController : ControllerBase
    {
        private readonly IRobotService _robotService;
        private readonly IHubContext<MapHub> _mapHubContext;

        public RobotController(IRobotService robotService, IHubContext<MapHub> mapHubContext)
        {
            _robotService = robotService;
            _mapHubContext = mapHubContext;
        }

        [HttpPost]
        [Authorize(Roles = "Admin")]
        public async Task<IActionResult> CreateRobot([FromBody] CreateRobotDTO robotDto)
        {
            try
            {
                var result = await _robotService.CreateRobotAsync(robotDto);
                return CreatedAtAction(nameof(GetRobotById), new { id = result.Id }, result);
            }
            catch (ArgumentException ex)
            {
                return BadRequest(new { error = ex.Message });
            }
            catch (Exception ex)
            {
                return StatusCode(500, new { error = "An error occurred", details = ex.Message });
            }
        }

        [HttpGet("{id}")]
        [Authorize(Roles = "Admin")]
        public async Task<IActionResult> GetRobotById(int id)
        {
            var robot = await _robotService.GetRobotByIdAsync(id);
            return robot == null ? NotFound(new { error = "Robot not found" }) : Ok(robot);
        }

        [HttpGet]
        [Authorize(Roles = "Admin")]
        public async Task<IActionResult> GetAllRobots()
        {
            var robots = await _robotService.GetAllRobotsAsync();
            return Ok(robots);
        }

        [HttpGet("status/{status}")]
        [Authorize(Roles = "Admin")]
        public async Task<IActionResult> GetByStatus(RobotStatus status)
        {
            var robots = await _robotService.GetByStatusAsync(status);
            return Ok(robots);
        }

        [HttpGet("type/{type}")]
        [Authorize(Roles = "Admin")]
        public async Task<IActionResult> GetByType(RobotType type)
        {
            var robots = await _robotService.GetByTypeAsync(type);
            return Ok(robots);
        }

        [HttpGet("available")]
        public async Task<IActionResult> GetAvailableRobots()
        {
            var robots = await _robotService.GetAvailableRobotsAsync();
            return Ok(robots);
        }

        [HttpPatch]
        [Authorize(Roles = "Admin")]
        public async Task<IActionResult> UpdateRobot([FromBody] UpdateRobotDTO robotDto)
        {
            try
            {
                if (robotDto.Id != robotDto.Id) return BadRequest(new { error = "ID mismatch" });
                var result = await _robotService.UpdateRobotAsync(robotDto);
                return result ? Ok(new { message = "Robot updated" }) : NotFound(new { error = "Robot not found" });
            }
            catch (ArgumentException ex)
            {
                return BadRequest(new { error = ex.Message });
            }
        }

        [HttpDelete("{id}")]
        [Authorize(Roles = "Admin")]
        public async Task<IActionResult> DeleteRobot(int id)
        {
            var result = await _robotService.DeleteRobotAsync(id);
            return result ? Ok(new { message = "Robot deleted" }) : NotFound(new { error = "Robot not found" });
        }

        // IoT Device Endpoint - Robot Status Update
        [HttpPost("status")]
        [Authorize]
        public async Task<IActionResult> UpdateRobotStatus([FromBody] RobotStatusUpdateDTO statusUpdate)
        {
            try
            {
                // Get robot ID from token
                var robotIdClaim = User.FindFirst("RobotId")?.Value;
                var userRole = User.FindFirst(ClaimTypes.Role)?.Value;

                int robotId;

                // If Admin, they must provide robot ID in the request or use their own ID claim
                if (userRole == "Admin")
                {
                    var idClaim = User.FindFirst("Id")?.Value;
                    if (string.IsNullOrEmpty(idClaim) || !int.TryParse(idClaim, out robotId))
                    {
                        return Unauthorized(new { error = "Invalid token" });
                    }
                }
                // If IoT device, use RobotId from token
                else if (!string.IsNullOrEmpty(robotIdClaim) && int.TryParse(robotIdClaim, out robotId))
                {
                    // Robot can only update its own status
                }
                else
                {
                    return Unauthorized(new { error = "Invalid robot token" });
                }

                // Update robot status through service
                var robot = await _robotService.UpdateRobotStatusAsync(robotId, statusUpdate);
                if (robot == null)
                {
                    return NotFound(new { error = "Robot not found" });
                }

                // Broadcast robot position update to all connected clients via SignalR
                var robotPositionUpdate = new RobotMapPositionDTO
                {
                    Id = robot.Id,
                    Name = robot.Name,
                    Model = robot.Model,
                    Type = (RobotType)Enum.Parse(typeof(RobotType), robot.TypeName),
                    TypeName = robot.TypeName,
                    Status = (RobotStatus)Enum.Parse(typeof(RobotStatus), robot.StatusName),
                    StatusName = robot.StatusName,
                    BatteryLevel = robot.BatteryLevel,
                    Latitude = statusUpdate.CurrentLatitude,
                    Longitude = statusUpdate.CurrentLongitude,
                    CurrentNodeId = robot.CurrentNodeId,
                    CurrentNodeName = robot.CurrentNodeName,
                    TargetNodeId = statusUpdate.TargetNodeId,
                    ActiveOrdersCount = robot.ActiveOrdersCount
                };

                await _mapHubContext.Clients.All.SendAsync("ReceiveRobotUpdate", robotPositionUpdate);

                return Ok(new
                {
                    message = "Robot status updated successfully",
                    robotId = robot.Id,
                    status = robot.StatusName,
                    batteryLevel = robot.BatteryLevel,
                    currentNodeId = robot.CurrentNodeId,
                    targetNodeId = statusUpdate.TargetNodeId,
                    coordinates = statusUpdate.CurrentLatitude != null && statusUpdate.CurrentLongitude != null
                        ? new { latitude = statusUpdate.CurrentLatitude, longitude = statusUpdate.CurrentLongitude }
                        : null
                });
            }
            catch (ArgumentException ex)
            {
                return BadRequest(new { error = ex.Message });
            }
            catch (Exception ex)
            {
                return StatusCode(500, new { error = "An error occurred", details = ex.Message });
            }
        }

        // Get robot's own info (for IoT devices)
        [HttpGet("me")]
        [Authorize(Roles = "Iot")]
        public async Task<IActionResult> GetMyRobotInfo()
        {
            var robotIdClaim = User.FindFirst("RobotId")?.Value;
            if (string.IsNullOrEmpty(robotIdClaim) || !int.TryParse(robotIdClaim, out int robotId))
            {
                return Unauthorized(new { error = "Invalid robot token" });
            }

            var robot = await _robotService.GetRobotByIdAsync(robotId);
            if (robot == null)
            {
                return NotFound(new { error = "Robot not found" });
            }

            return Ok(robot);
        }

        // Get orders assigned to this robot (for IoT devices)
        [HttpGet("my-orders")]
        [Authorize(Roles = "Iot")]
        public async Task<IActionResult> GetMyOrders()
        {
            try
            {
                var robotIdClaim = User.FindFirst("RobotId")?.Value;
                if (string.IsNullOrEmpty(robotIdClaim) || !int.TryParse(robotIdClaim, out int robotId))
                {
                    return Unauthorized(new { error = "Invalid robot token" });
                }

                var orders = await _robotService.GetMyOrdersAsync(robotId);
                return Ok(orders);
            }
            catch (ArgumentException ex)
            {
                return NotFound(new { error = ex.Message });
            }
            catch (Exception ex)
            {
                return StatusCode(500, new { error = "An error occurred while retrieving orders", details = ex.Message });
            }
        }

        // Accept an assigned order (for IoT devices)
        [HttpPost("order/{orderId}/accept")]
        [Authorize(Roles = "Iot")]
        public async Task<IActionResult> AcceptOrder(int orderId)
        {
            try
            {
                var robotIdClaim = User.FindFirst("RobotId")?.Value;
                if (string.IsNullOrEmpty(robotIdClaim) || !int.TryParse(robotIdClaim, out int robotId))
                {
                    return Unauthorized(new { error = "Invalid robot token" });
                }

                var result = await _robotService.AcceptOrderAsync(robotId, orderId);
                return Ok(result);
            }
            catch (ArgumentException ex)
            {
                return NotFound(new { error = ex.Message });
            }
            catch (InvalidOperationException ex)
            {
                return BadRequest(new { error = ex.Message });
            }
            catch (Exception ex)
            {
                return StatusCode(500, new { error = "An error occurred while accepting the order", details = ex.Message });
            }
        }

        // Update order delivery phase (for IoT devices)
        [HttpPost("order/{orderId}/phase")]
        [Authorize(Roles = "Iot")]
        public async Task<IActionResult> UpdateOrderPhase(int orderId, [FromBody] OrderPhaseUpdateDTO phaseUpdate)
        {
            try
            {
                var robotIdClaim = User.FindFirst("RobotId")?.Value;
                if (string.IsNullOrEmpty(robotIdClaim) || !int.TryParse(robotIdClaim, out int robotId))
                {
                    return Unauthorized(new { error = "Invalid robot token" });
                }

                var result = await _robotService.UpdateOrderPhaseAsync(robotId, orderId, phaseUpdate);
                return Ok(new
                {
                    message = "Order phase updated successfully",
                    orderId = orderId,
                    phase = phaseUpdate.Phase,
                    timestamp = phaseUpdate.Timestamp
                });
            }
            catch (ArgumentException ex)
            {
                return BadRequest(new { error = ex.Message });
            }
            catch (InvalidOperationException ex)
            {
                return BadRequest(new { error = ex.Message });
            }
            catch (Exception ex)
            {
                return StatusCode(500, new { error = "An error occurred while updating order phase", details = ex.Message });
            }
        }

        [HttpPost("test-routing")]
        [Authorize(Roles = "Admin")]
        public async Task<IActionResult> TestRouting([FromBody] RoutingTestDTO testDto)
        {
            try
            {
                var result = await _robotService.RoutingTestAsync(testDto);
                return Ok(result);
            }
            catch (ArgumentException ex)
            {
                return BadRequest(new { error = ex.Message });
            }
            catch (Exception ex)
            {
                return StatusCode(500, new { error = "An error occurred during routing test", details = ex.Message });
            }
        }
    }
}