初始化

2025-10-16 09:56:36 +08:00
commit de704db577
272 changed files with 37331 additions and 0 deletions
--- a/FutureMailAPI/Controllers/UploadController.cs
+++ b/FutureMailAPI/Controllers/UploadController.cs
@@ -0,0 +1,115 @@
+using Microsoft.AspNetCore.Mvc;
+using Microsoft.AspNetCore.Authorization;
+using FutureMailAPI.Services;
+using FutureMailAPI.DTOs;
+using System.Security.Claims;
+
+namespace FutureMailAPI.Controllers
+{
+    [ApiController]
+    [Route("api/v1/upload")]
+    [Authorize]
+    public class UploadController : ControllerBase
+    {
+        private readonly IFileUploadService _fileUploadService;
+        private readonly ILogger<UploadController> _logger;
+        
+        public UploadController(IFileUploadService fileUploadService, ILogger<UploadController> logger)
+        {
+            _fileUploadService = fileUploadService;
+            _logger = logger;
+        }
+        
+        /// <summary>
+        /// 上传附件
+        /// </summary>
+        /// <param name="request">文件上传请求</param>
+        /// <returns>上传结果</returns>
+        [HttpPost("attachment")]
+        public async Task<IActionResult> UploadAttachment([FromForm] FileUploadWithFileRequestDto request)
+        {
+            try
+            {
+                var userId = GetCurrentUserId();
+                if (userId <= 0)
+                {
+                    return Unauthorized(ApiResponse<object>.ErrorResult("无效的用户令牌"));
+                }
+                
+                if (request.File == null || request.File.Length == 0)
+                {
+                    return BadRequest(ApiResponse<object>.ErrorResult("请选择要上传的文件"));
+                }
+                
+                var result = await _fileUploadService.UploadFileAsync(request.File, userId, request);
+                
+                if (result.Success)
+                {
+                    return Ok(result);
+                }
+                
+                return BadRequest(result);
+            }
+            catch (Exception ex)
+            {
+                _logger.LogError(ex, "上传附件时发生错误");
+                return StatusCode(500, ApiResponse<FileUploadResponseDto>.ErrorResult("服务器内部错误"));
+            }
+        }
+        
+        /// <summary>
+        /// 上传头像
+        /// </summary>
+        /// <param name="request">文件上传请求</param>
+        /// <returns>上传结果</returns>
+        [HttpPost("avatar")]
+        public async Task<IActionResult> UploadAvatar([FromForm] FileUploadWithFileRequestDto request)
+        {
+            try
+            {
+                var userId = GetCurrentUserId();
+                if (userId <= 0)
+                {
+                    return Unauthorized(ApiResponse<object>.ErrorResult("无效的用户令牌"));
+                }
+                
+                if (request.File == null || request.File.Length == 0)
+                {
+                    return BadRequest(ApiResponse<object>.ErrorResult("请选择要上传的头像文件"));
+                }
+                
+                // 设置头像特定的属性
+                request.Type = AttachmentType.IMAGE;
+                request.Category = "avatar";
+                
+                var result = await _fileUploadService.UploadFileAsync(request.File, userId, request);
+                
+                if (result.Success)
+                {
+                    return Ok(result);
+                }
+                
+                return BadRequest(result);
+            }
+            catch (Exception ex)
+            {
+                _logger.LogError(ex, "上传头像时发生错误");
+                return StatusCode(500, ApiResponse<FileUploadResponseDto>.ErrorResult("服务器内部错误"));
+            }
+        }
+        
+        /// <summary>
+        /// 从JWT令牌中获取当前用户ID
+        /// </summary>
+        /// <returns>用户ID</returns>
+        private int GetCurrentUserId()
+        {
+            var userIdClaim = User.FindFirst(ClaimTypes.NameIdentifier);
+            if (userIdClaim != null && int.TryParse(userIdClaim.Value, out int userId))
+            {
+                return userId;
+            }
+            return 0;
+        }
+    }
+}