using Dpz.Core.Service.Network;
using Dpz.Core.Service.Network.Models;
using Medallion.Threading;

namespace Dpz.Core.Web.Library.Api.Service;

public class GenerateSessionTitleService(
    IAiChatService aiChatService,
    ChatCompletionService chatCompletionService,
    IDistributedLockProvider distributedLockProvider,
    ILogger<GenerateSessionTitleService> logger
)
{
    public async Task<string?> GenerateAsync(
        string userId,
        string sessionId,
        string questionContent,
        string answerContent,
        string modelType
    )
    {
        string? sessionName = null;
        var lockKey = $"AI_Analyze.Chat.Lock:{userId}_{sessionId}";
        try
        {
            logger.LogInformation(
                "start generate session title. UserId: {UserId}, SessionId: {SessionId}",
                userId,
                sessionId
            );
            await using (await distributedLockProvider.AcquireLockAsync(lockKey))
            {
                var prompt = new StringBuilder("问：")
                    .AppendLine()
                    .AppendLine(questionContent)
                    .AppendLine()
                    .AppendLine("------")
                    .AppendLine()
                    .AppendLine("答：")
                    .AppendLine()
                    .AppendLine(answerContent)
                    .AppendLine()
                    .AppendLine("------")
                    .AppendLine()
                    .AppendLine("请给以上对话简单总结出一个标题")
                    .ToString();
                var message = new ChatMessage { Role = "user", Message = prompt };

                var result = await chatCompletionService.SendChatAsync(
                    [message],
                    x => x.Stream = false
                );

                if (result is { Success: true, Data: not null, Data.Choices.Count: > 0 })
                {
                    sessionName = result.Data.Choices[0].Message.Content;
                    await aiChatService.CreateSessionAsync(
                        userId,
                        sessionId,
                        sessionName,
                        modelType
                    );
                }
            }
            logger.LogInformation(
                "generate session title completed. UserId: {UserId}, SessionId: {SessionId},SessionName:{SessionName}",
                userId,
                sessionId,
                sessionName
            );
        }
        catch (Exception e)
        {
            logger.LogError(
                e,
                "generate session title failed. UserId: {UserId}, SessionId: {SessionId}",
                userId,
                sessionId
            );
        }

        return sessionName;
    }
}