using Dpz.Core.Service.Network.Models;
using Dpz.Core.Web.Pager;

namespace Dpz.Core.Web.Controllers;

[CheckAuthorize]
public class AiChatController(IAiChatService aiChatService, ILogger<AiChatController> logger)
    : Controller
{
    public async Task<IActionResult> Index()
    {
        // 获取用户的会话列表
        var sessions = await aiChatService.GetSessionListAsync(User.RequiredUserId);
        IPagedList<VmAiChatRecord> records = PagedList<VmAiChatRecord>.Empty();
        string? defaultSessionId = null;
        AiModel? aiModel = null;

        // 第一个会话作为默认会话
        if (sessions.Count > 0)
        {
            defaultSessionId = sessions[0].SessionId;
            records = await aiChatService.GetSessionMessagesAsync(defaultSessionId, 1, 500);
            if (Enum.TryParse(sessions[0].ModelType, out AiModel modelType))
            {
                aiModel = modelType;
            }
        }
        var model = new AiChatModel(
            sessions,
            defaultSessionId,
            aiModel ?? AiModel.Gpt5Mini,
            records
        );
        return View(model);
    }

    /// <summary>
    /// 获取会话列表
    /// </summary>
    [HttpGet]
    public async Task<IActionResult> Sessions()
    {
        var sessions = await aiChatService.GetSessionListAsync(User.RequiredUserId);
        return Json(new ResultInfo(true, sessions));
    }

    /// <summary>
    /// 删除会话
    /// </summary>
    [HttpDelete]
    public async Task<IActionResult> DeleteSession(string sessionId)
    {
        var result = await aiChatService.DeleteSessionAsync(User.RequiredUserId, sessionId);
        return Json(new ResultInfo(result));
    }

    [HttpGet]
    public async Task<IActionResult> ChatMessages(
        string? sessionId,
        int pageIndex = 1,
        int pageSize = 500
    )
    {
        if (string.IsNullOrEmpty(sessionId))
        {
            return PartialView(
                "_ChatMessages",
                new AiChatMessageModel(null, AiModel.Gpt5Mini, PagedList<VmAiChatRecord>.Empty())
            );
        }

        var session = await aiChatService.GetSessionAsync(User.RequiredUserId, sessionId);
        if (session == null)
        {
            return NotFound();
        }
        var messages = await aiChatService.GetSessionMessagesAsync(sessionId, pageIndex, pageSize);

        AiModel? aiModel = null;
        if (Enum.TryParse(session.ModelType, out AiModel modelType))
        {
            aiModel = modelType;
        }
        var model = new AiChatMessageModel(sessionId, aiModel ?? AiModel.Gpt5Mini, messages);

        return PartialView("_ChatMessages", model);
    }

    /// <summary>
    /// 更新会话名称
    /// </summary>
    [HttpPut]
    public async Task<IActionResult> UpdateSessionName(string sessionId, string newName)
    {
        var result = await aiChatService.UpdateSessionNameAsync(
            User.RequiredUserId,
            sessionId,
            newName
        );
        return Json(new ResultInfo(result));
    }

    /// <summary>
    /// 获取会话详情
    /// </summary>
    [HttpGet]
    public async Task<IActionResult> GetSessionModelType(string sessionId)
    {
        var session = await aiChatService.GetSessionAsync(User.RequiredUserId, sessionId);
        if (session == null)
        {
            return Json(ResultInfo.ToFail("会话不存在"));
        }

        if (!Enum.TryParse(session.ModelType, out AiModel modelType))
        {
            modelType = AiModel.Gpt5Mini;
        }

        return Json(new ResultInfo(true, (int)modelType));
    }

    /// <summary>
    /// 更新会话模型
    /// </summary>
    [HttpPut]
    public async Task<IActionResult> UpdateSessionModelType(string sessionId, AiModel modelType)
    {
        // 更新会话的模型类型
        var updated = await aiChatService.UpdateSessionModelTypeAsync(
            User.RequiredUserId,
            sessionId,
            modelType.ToString()
        );
        logger.LogInformation(
            "Updated model type to {ModelValue} for session {SessionId}",
            modelType,
            sessionId
        );
        return Json(new ResultInfo(updated));
    }

    [HttpPost]
    public async Task<IActionResult> CancelMessage(
        [FromQuery] string? sessionId = null,
        [FromQuery] string? connectionId = null
    )
    {
        if (string.IsNullOrWhiteSpace(sessionId))
        {
            logger.LogWarning("Session id is null or empty");
            return BadRequest("Session id is null or empty");
        }
        if (string.IsNullOrWhiteSpace(connectionId))
        {
            logger.LogWarning("connection id null or empty");
            return BadRequest("connection id null or empty");
        }
        // 精确取消指定会话的流
        var tokenKey = $"{sessionId}_{connectionId}";
        if (WebToolsExtensions.CancellationTokens.TryRemove(tokenKey, out var cts))
        {
            try
            {
                await cts.CancelAsync();
                logger.LogInformation(
                    "Stream message cancellation requested and executed, Token Key: {TokenKey}",
                    tokenKey
                );
                return Ok();
            }
            catch (Exception ex)
            {
                logger.LogError(ex, "Error cancelling stream, Token Key: {TokenKey}", tokenKey);
                return BadRequest(ex.Message);
            }
            finally
            {
                cts.Dispose();
            }
        }

        logger.LogWarning(
            "No active stream found for cancellation, Token Key: {TokenKey}",
            tokenKey
        );
        return BadRequest($"No active stream found for cancellation, Token Key: {tokenKey}");
    }

    /// <summary>
    /// 删除消息
    /// </summary>
    /// <param name="messageId"></param>
    /// <returns></returns>
    [HttpDelete]
    public async Task<IActionResult> DeleteMessage(string messageId)
    {
        var result = await aiChatService.DeleteMessageAsync(messageId, User.RequiredUserId);
        if (result.Success)
        {
            return Json(new ResultInfo(true));
        }
        return Json(new ResultInfo(false, result.Message));
    }
}