「モジュール:Pie chart」を編集中

local p = {}

function p.render(frame)
    local args = frame.args
    local caption = args["caption"] or ""
    local values = {}
    local labels = {}
    
    for i = 1, 10 do
        if args["label" .. i] and args["value" .. i] then
            table.insert(labels, args["label" .. i])
            table.insert(values, tonumber(args["value" .. i]) or 0)
        end
    end
    
    local total = 0
    for _, v in ipairs(values) do
        total = total + v
    end
    
    if total == 0 then
        return "<div style='color: red;'>Error: No valid values provided.</div>"
    end

    local data = {}
    for i, v in ipairs(values) do
        table.insert(data, string.format('{ "category": "%s", "value": %d }', labels[i], v))
    end
    
    local graph = string.format([[
    <graph>
    {
      "type": "pie",
      "data": {
        "values": [ %s ]
      },
      "encode": {
        "tooltip": { "field": "category", "type": "nominal" }
      }
    }
    </graph>
    ]], table.concat(data, ", "))

    return "<div style='text-align: center;'>" ..
        (caption ~= "" and "<b>" .. caption .. "</b><br>" or "") ..
        graph .. "</div>"
end

return p