using SqlSugar;
using Volo.Abp.Domain.Entities.Auditing;
using Yi.Framework.AiHub.Domain.Shared.Enums;

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

[SugarTable("Ai_ImageStoreTask")]
public class ImageStoreTaskAggregateRoot : FullAuditedAggregateRoot<Guid>
{
    /// <summary>
    /// 提示词
    /// </summary>
    [SugarColumn(ColumnDataType = StaticConfig.CodeFirst_BigString)]
    public string Prompt { get; set; }

    /// <summary>
    /// 参考图Url
    /// </summary>
    [SugarColumn(IsJson = true)]
    public List<string> ReferenceImageUrls { get; set; }

    /// <summary>
    /// 图片绝对路径
    /// </summary>
    public string? StoreUrl { get; set; }

    /// <summary>
    /// 任务状态
    /// </summary>
    public TaskStatusEnum TaskStatus { get; set; } = TaskStatusEnum.Processing;

    /// <summary>
    /// 用户id
    /// </summary>
    public Guid UserId { get; set; }

    /// <summary>
    /// 设置成功
    /// </summary>
    /// <param name="storeUrl"></param>
    public void SetSuccess(string storeUrl)
    {
        TaskStatus = TaskStatusEnum.Success;
        StoreUrl = storeUrl;
    }
}