namespace Admin.NET.Core.Ai.Services.DataBase;

using Admin.NET.Core.Ai.Entitys;
using Admin.NET.Core.Ai.Enums;
using Admin.NET.Core.Ai.Services.DataBase.Dto;

/// <summary>
/// 异步对数据库进行操作
/// </summary>
public class ChatChannelActionService : IScoped
{
    private readonly ILogger<ChatChannelActionService> _logger;
    private readonly SqlSugarRepository<LLMChatSummaryHistory> _chatSummaryHistoryService;
    private readonly SqlSugarRepository<LLMChatHistory> _chatHistoryService;

    /// <summary>
    /// 构造函数
    /// </summary>
    /// <param name="logger"></param>
    /// <param name="chatSummaryHistoryService"></param>
    /// <param name="chatHistoryService"></param>
    public ChatChannelActionService(ILogger<ChatChannelActionService> logger,
        SqlSugarRepository<LLMChatSummaryHistory> chatSummaryHistoryService,
        SqlSugarRepository<LLMChatHistory> chatHistoryService)
    {
        _logger = logger;
        _chatSummaryHistoryService = chatSummaryHistoryService;
        _chatHistoryService = chatHistoryService;
    }

    /// <summary>
    /// 追加聊天记录
    /// </summary>
    public async Task AppendAsync(LLMChatSummaryHistory chatSummaryHistory)
    {
        await _chatSummaryHistoryService.Context.CopyNew()
            .InsertNav(chatSummaryHistory)
            .Include(z => z.Histories)
            .ExecuteCommandAsync();
    }

    /// <summary>
    /// 追加聊天明细
    /// </summary>
    public async Task AppendItemAsync(LLMChatSummaryHistory chatSummaryHistory)
    {
        chatSummaryHistory.Histories.ForEach(item => item.SummaryId = chatSummaryHistory.Id);
        await _chatHistoryService.Context.CopyNew()
            .Insertable(chatSummaryHistory.Histories)
            .ExecuteCommandAsync();
    }

    /// <summary>
    /// 删除所有聊天记录
    /// </summary>
    public async Task DeleteAllAsync(LLMChatSummaryHistory chatSummaryHistory)
    {
        await _chatSummaryHistoryService.Context.CopyNew()
            .DeleteNav<LLMChatSummaryHistory>(u => u.Id == chatSummaryHistory.Id)
            .Include(u => u.Histories)
            .ExecuteCommandAsync();
    }

    /// <summary>
    /// 重命名摘要
    /// </summary>
    public async Task RenameSummaryAsync(LLMChatSummaryHistory chatSummaryHistory)
    {
        var oldChatSummaryHistory = await _chatSummaryHistoryService.CopyNew()
            .AsQueryable()
            .Where(u => u.Id == chatSummaryHistory.Id)
            .FirstAsync();
        oldChatSummaryHistory.Summary = chatSummaryHistory.Summary;
        await _chatSummaryHistoryService.Context.CopyNew()
            .Updateable(oldChatSummaryHistory)
            .ExecuteCommandAsync();
    }
}