using Mapster;
using SqlSugar;
using Volo.Abp.Domain.Entities.Auditing;
using Yi.Framework.AiHub.Domain.Entities.ValueObjects;
using Yi.Framework.AiHub.Domain.Shared.Dtos;
using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi;
using Yi.Framework.AiHub.Domain.Shared.Enums;

namespace Yi.Framework.AiHub.Domain.Entities.Chat;

[SugarTable("Ai_Message")]
[SugarIndex($"index_{{table}}_{nameof(UserId)}_{nameof(SessionId)}",
    $"{nameof(UserId)}", OrderByType.Desc,
    $"{nameof(SessionId)}", OrderByType.Desc
)]
public class MessageAggregateRoot : FullAuditedAggregateRoot<Guid>
{
    public MessageAggregateRoot()
    {
    }

    public MessageAggregateRoot(Guid? userId, Guid? sessionId, string content, string role, string modelId,
        ThorUsageResponse? tokenUsage, Guid? tokenId = null)
    {
        UserId = userId;
        SessionId = sessionId;
        TokenId = tokenId ?? Guid.Empty;
        //如果没有会话，不存储对话内容
        Content = sessionId is null ? null : content;
        Role = role;
        ModelId = modelId;
        if (tokenUsage is not null)
        {
            long inputTokenCount =
                (tokenUsage.PromptTokens.HasValue && tokenUsage.PromptTokens.Value != 0)
                    ? tokenUsage.PromptTokens.Value
                    : (tokenUsage.InputTokens.HasValue && tokenUsage.InputTokens.Value != 0)
                        ? tokenUsage.InputTokens.Value
                        : 0;

            long outputTokenCount =
                (tokenUsage.CompletionTokens.HasValue && tokenUsage.CompletionTokens.Value != 0)
                    ? tokenUsage.CompletionTokens.Value
                    : (tokenUsage.OutputTokens.HasValue && tokenUsage.OutputTokens.Value != 0)
                        ? tokenUsage.OutputTokens.Value
                        : 0;


            this.TokenUsage = new TokenUsageValueObject
            {
                OutputTokenCount = outputTokenCount,
                InputTokenCount = inputTokenCount,
                TotalTokenCount = tokenUsage.TotalTokens ?? 0
            };
        }

        this.MessageType = sessionId is null ? MessageTypeEnum.Api : MessageTypeEnum.Web;
    }

    public Guid? UserId { get; set; }
    public Guid? SessionId { get; set; }

    /// <summary>
    /// Token密钥Id（通过API调用时记录，Web调用为Guid.Empty）
    /// </summary>
    public Guid TokenId { get; set; }

    [SugarColumn(ColumnDataType = StaticConfig.CodeFirst_BigString)]
    public string? Content { get; set; }

    public string Role { get; set; }
    public string ModelId { get; set; }
    public string? Remark { get; set; }

    [SugarColumn(IsOwnsOne = true)] public TokenUsageValueObject TokenUsage { get; set; } = new TokenUsageValueObject();

    public MessageTypeEnum MessageType { get; set; }
}