emall-api/FutureMailAPI/Controllers/AIController.cs

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;
        }
    }
}