using AngleSharp.Html.Parser;
using Dpz.Core.Infrastructure;
using Dpz.Core.MessageQueue.Abstractions;
using Dpz.Core.MessageQueue.Models;
using Dpz.Core.Public.ViewModel.Messages;
using Dpz.Core.Service.Mediator;

namespace Dpz.Core.Web.Jobs.MessageHandlers;

/// <summary>
/// 删除 Markdown 内容消息处理器
/// 解析 Markdown 中的图片并继续发布删除图片消息
/// </summary>
public class DeleteMarkdownHandler(
    IMessagePublisher<RemoveImagesMessage> removeImagesPublisher,
    ILogger<DeleteMarkdownHandler> logger
) : IMessageHandler<DeleteMarkdownMessage>
{
    private readonly HtmlParser _htmlParser = new();

    public async Task<MessageHandlerResult> HandleAsync(
        DeleteMarkdownMessage message,
        CancellationToken cancellationToken
    )
    {
        try
        {
            if (message.MarkdownContents.Count == 0)
            {
                logger.LogInformation("没有需要处理的 Markdown 内容");
                return MessageHandlerResult.Ok();
            }

            var allImages = new List<string>();

            foreach (var markdown in message.MarkdownContents)
            {
                if (string.IsNullOrWhiteSpace(markdown))
                {
                    continue;
                }

                var images = await ExtractImagesFromMarkdownAsync(markdown);
                allImages.AddRange(images);
            }

            if (allImages.Count > 0)
            {
                // 批量发布删除图片队列消息
                var messages = allImages.Select(x => new RemoveImagesMessage { ImageUrl = x });
                await removeImagesPublisher.PublishBatchAsync(
                    messages,
                    cancellationToken: cancellationToken
                );

                logger.LogInformation(
                    "处理 DeleteMarkdown 完成，从 {MarkdownCount} 个 Markdown 中提取到 {ImageCount} 张图片并发布到队列",
                    message.MarkdownContents.Count,
                    allImages.Count
                );
            }
            else
            {
                logger.LogInformation(
                    "处理 DeleteMarkdown 完成，从 {MarkdownCount} 个 Markdown 中未提取到图片",
                    message.MarkdownContents.Count
                );
            }

            return MessageHandlerResult.Ok();
        }
        catch (Exception ex)
        {
            logger.LogError(ex, "处理 DeleteMarkdown 消息失败");
            return MessageHandlerResult.Fail($"处理 DeleteMarkdown 消息失败: {ex.Message}");
        }
    }

    /// <summary>
    /// 从 Markdown 内容中提取图片 URL
    /// </summary>
    private async Task<List<string>> ExtractImagesFromMarkdownAsync(string markdown)
    {
        try
        {
            // 将 Markdown 转换为 HTML
            var htmlContent = markdown.MarkdownToHtml();

            // 解析 HTML
            var document = await _htmlParser.ParseDocumentAsync(htmlContent);
            var imageElements = document.GetElementsByTagName("img");

            // 提取图片 URL
            return imageElements.GetElementsImageUrls();
        }
        catch (Exception ex)
        {
            logger.LogError(ex, "从 Markdown 提取图片失败");
            return [];
        }
    }
}