I150/srcs/RobotNet10/FleetManager/RobotNet10.FleetManager/Controllers/RobotModelImagesController.cs

using Microsoft.AspNetCore.Mvc;
using RobotNet10.FleetManager.Services;
using RobotNet10.FleetManager.Shared.DTOs.Responses;

namespace RobotNet10.FleetManager.Controllers;

/// <summary>
/// API controller for managing robot model images
/// </summary>
[ApiController]
[Route("api/robot-models/{robotModelId}/image")]
public class RobotModelImagesController(
    IRobotModelImageStorageService imageStorageService,
    IRobotModelService robotModelService,
    Services.Logger<RobotModelImagesController> logger) : ControllerBase
{
    private readonly IRobotModelImageStorageService _imageStorageService = imageStorageService;
    private readonly IRobotModelService _robotModelService = robotModelService;
    private readonly Services.Logger<RobotModelImagesController> _logger = logger;

    private const long MaxFileSize = 10 * 1024 * 1024; // 10MB

    /// <summary>
    /// Get robot model image
    /// </summary>
    [HttpGet]
    [ProducesResponseType(typeof(FileResult), StatusCodes.Status200OK)]
    [ProducesResponseType(StatusCodes.Status404NotFound)]
    public async Task<IActionResult> GetImage(Guid robotModelId)
    {
        try
        {
            // Verify robot model exists
            var model = await _robotModelService.GetByIdAsync(robotModelId);
            if (model == null)
            {
                return NotFound(new ErrorResponseDto { Error = $"Robot model with ID {robotModelId} not found." });
            }

            var imageStream = await _imageStorageService.GetImageAsync(robotModelId);
            if (imageStream == null)
            {
                return NotFound(new ErrorResponseDto { Error = $"Image for robot model {robotModelId} not found." });
            }

            return File(imageStream, "image/png", $"{robotModelId}.png");
        }
        catch (Exception ex)
        {
            _logger.Error($"Error getting image for robot model {robotModelId}: {ex.Message}");
            return StatusCode(500, new ErrorResponseDto { Error = "An error occurred while retrieving the image." });
        }
    }

    /// <summary>
    /// Upload robot model image
    /// </summary>
    [HttpPost]
    [RequestSizeLimit(MaxFileSize)]
    [ProducesResponseType(StatusCodes.Status200OK)]
    [ProducesResponseType(StatusCodes.Status400BadRequest)]
    [ProducesResponseType(StatusCodes.Status404NotFound)]
    public async Task<IActionResult> UploadImage(Guid robotModelId, IFormFile file)
    {
        try
        {
            // Verify robot model exists
            var model = await _robotModelService.GetByIdAsync(robotModelId);
            if (model == null)
            {
                return NotFound(new ErrorResponseDto { Error = $"Robot model with ID {robotModelId} not found." });
            }

            // Validate file
            if (file == null || file.Length == 0)
            {
                return BadRequest(new ErrorResponseDto { Error = "No file provided." });
            }

            if (file.Length > MaxFileSize)
            {
                return BadRequest(new ErrorResponseDto { Error = $"File size exceeds maximum allowed size of {MaxFileSize / (1024 * 1024)}MB." });
            }

            // Validate file extension
            var extension = Path.GetExtension(file.FileName).ToLowerInvariant();
            if (extension != ".png")
            {
                return BadRequest(new ErrorResponseDto { Error = "Only PNG files are allowed." });
            }

            // Read image dimensions
            using (var stream = file.OpenReadStream())
            {
                var (width, height) = await _imageStorageService.GetImageDimensionsAsync(stream);

                // Save image
                stream.Position = 0;
                await _imageStorageService.SaveImageAsync(robotModelId, stream);

                // Update robot model with image dimensions
                await _robotModelService.UpdateAsync(robotModelId, new Shared.DTOs.RobotModel.UpdateRobotModelRequest
                {
                    ImageWidth = width,
                    ImageHeight = height
                });
            }

            return Ok(new { Error = "Image uploaded successfully." });
        }
        catch (InvalidOperationException ex)
        {
            return BadRequest(new ErrorResponseDto { Error = ex.Message });
        }
        catch (KeyNotFoundException ex)
        {
            return NotFound(new ErrorResponseDto { Error = ex.Message });
        }
        catch (Exception ex)
        {
            _logger.Error($"Error uploading image for robot model {robotModelId}: {ex.Message}");
            return StatusCode(500, new ErrorResponseDto { Error = "An error occurred while uploading the image." });
        }
    }

    /// <summary>
    /// Delete robot model image
    /// </summary>
    [HttpDelete]
    [ProducesResponseType(StatusCodes.Status204NoContent)]
    [ProducesResponseType(StatusCodes.Status404NotFound)]
    public async Task<IActionResult> DeleteImage(Guid robotModelId)
    {
        try
        {
            // Verify robot model exists
            var model = await _robotModelService.GetByIdAsync(robotModelId);
            if (model == null)
            {
                return NotFound(new ErrorResponseDto { Error = $"Robot model with ID {robotModelId} not found." });
            }

            var deleted = await _imageStorageService.DeleteImageAsync(robotModelId);
            if (!deleted)
            {
                return NotFound(new ErrorResponseDto { Error = $"Image for robot model {robotModelId} not found." });
            }

            return NoContent();
        }
        catch (Exception ex)
        {
            _logger.Error($"Error deleting image for robot model {robotModelId}: {ex.Message}");
            return StatusCode(500, new ErrorResponseDto { Error = "An error occurred while deleting the image." });
        }
    }
}