-- Author: Achuan-2
-- https://github.com/Achuan-2/pandoc_word_template/blob/main/image-title-to-caption.lua
-- 作用：pandoc默认把图片alt作为图片caption，改为将图片title设置为caption，并在只有alt文本时确保无caption

-- 创建caption的函数
function create_caption(text)
    if text:find("^%s*$") then
        return {}
    else
        return {pandoc.Str(text)}
    end
end

-- 处理Para块
function Para(para)
    -- 检查段落是否只包含一个图片
    if #para.content == 1 and para.content[1].t == "Image" then
        local img = para.content[1]
        -- 如果图片有标题，将其包装为Figure并将标题设为caption
        if img.title and img.title ~= "" then
            local content = {pandoc.Plain({img})}
            local caption = create_caption(img.title)
            return pandoc.Figure(content, caption)
        else
            -- 如果只有alt文本（无标题），也将其包装为Figure但不生成caption
            local content = {pandoc.Plain({img})}
            local caption = {}
            return pandoc.Figure(content, caption)
        end
    end
    -- 如果不是单独的图片，返回未更改的段落
    return para
end

-- 处理Figure块
function Figure(fig)
    -- 查找Figure中的图片
    for i, block in ipairs(fig.content) do
        if block.t == "Plain" then
            for j, inline in ipairs(block.content) do
                if inline.t == "Image" then
                    local img = inline
                    -- 如果图片有标题，将其设为Figure的caption
                    if img.title and img.title ~= "" then
                        fig.caption = create_caption(img.title)
                    else
                        -- 如果无标题，确保Figure无caption
                        fig.caption = {}
                    end
                end
            end
        end
    end
    return fig
end

-- 返回过滤器
return {{Para = Para}, {Figure = Figure}}