初始化

2025-10-16 09:56:36 +08:00
commit de704db577
272 changed files with 37331 additions and 0 deletions
--- a/FutureMailAPI/Controllers/AIController.cs
+++ b/FutureMailAPI/Controllers/AIController.cs
@@ -0,0 +1,130 @@
+using Microsoft.AspNetCore.Authorization;
+using Microsoft.AspNetCore.Mvc;
+using FutureMailAPI.Services;
+using FutureMailAPI.DTOs;
+using System.Security.Claims;
+
+namespace FutureMailAPI.Controllers
+{
+    [ApiController]
+    [Route("api/v1/ai")]
+    [Authorize]
+    public class AIController : ControllerBase
+    {
+        private readonly IAIAssistantService _aiAssistantService;
+        private readonly ILogger<AIController> _logger;
+        
+        public AIController(IAIAssistantService aiAssistantService, ILogger<AIController> logger)
+        {
+            _aiAssistantService = aiAssistantService;
+            _logger = logger;
+        }
+        
+        /// <summary>
+        /// AI写作辅助
+        /// </summary>
+        /// <param name="request">写作辅助请求</param>
+        /// <returns>AI生成的内容和建议</returns>
+        [HttpPost("writing-assistant")]
+        public async Task<ActionResult<ApiResponse<WritingAssistantResponseDto>>> WritingAssistant([FromBody] WritingAssistantRequestDto request)
+        {
+            if (!ModelState.IsValid)
+            {
+                return BadRequest(ApiResponse<WritingAssistantResponseDto>.ErrorResult("输入数据无效"));
+            }
+            
+            try
+            {
+                var result = await _aiAssistantService.GetWritingAssistanceAsync(request);
+                
+                if (!result.Success)
+                {
+                    return BadRequest(result);
+                }
+                
+                return Ok(result);
+            }
+            catch (Exception ex)
+            {
+                _logger.LogError(ex, "获取写作辅助时发生错误");
+                return StatusCode(500, ApiResponse<WritingAssistantResponseDto>.ErrorResult("服务器内部错误"));
+            }
+        }
+        
+        /// <summary>
+        /// 情感分析
+        /// </summary>
+        /// <param name="request">情感分析请求</param>
+        /// <returns>情感分析结果</returns>
+        [HttpPost("sentiment-analysis")]
+        public async Task<ActionResult<ApiResponse<SentimentAnalysisResponseDto>>> SentimentAnalysis([FromBody] SentimentAnalysisRequestDto request)
+        {
+            if (!ModelState.IsValid)
+            {
+                return BadRequest(ApiResponse<SentimentAnalysisResponseDto>.ErrorResult("输入数据无效"));
+            }
+            
+            try
+            {
+                var result = await _aiAssistantService.AnalyzeSentimentAsync(request);
+                
+                if (!result.Success)
+                {
+                    return BadRequest(result);
+                }
+                
+                return Ok(result);
+            }
+            catch (Exception ex)
+            {
+                _logger.LogError(ex, "进行情感分析时发生错误");
+                return StatusCode(500, ApiResponse<SentimentAnalysisResponseDto>.ErrorResult("服务器内部错误"));
+            }
+        }
+        
+        /// <summary>
+        /// 未来预测
+        /// </summary>
+        /// <param name="request">未来预测请求</param>
+        /// <returns>未来预测结果</returns>
+        [HttpPost("future-prediction")]
+        public async Task<ActionResult<ApiResponse<FuturePredictionResponseDto>>> FuturePrediction([FromBody] FuturePredictionRequestDto request)
+        {
+            if (!ModelState.IsValid)
+            {
+                return BadRequest(ApiResponse<FuturePredictionResponseDto>.ErrorResult("输入数据无效"));
+            }
+            
+            try
+            {
+                var result = await _aiAssistantService.PredictFutureAsync(request);
+                
+                if (!result.Success)
+                {
+                    return BadRequest(result);
+                }
+                
+                return Ok(result);
+            }
+            catch (Exception ex)
+            {
+                _logger.LogError(ex, "进行未来预测时发生错误");
+                return StatusCode(500, ApiResponse<FuturePredictionResponseDto>.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 null;
+        }
+    }
+}