!! options
parsoid-compatible
version=2
!! end


!! article
Module:MySPARQL
!! text
local sparql = require('SPARQL')
local mwHtml = require('mw.html')

local p = {}


local function trimAndLower(str)
    if str == nil then return nil end
    str = str:gsub("^%s*(.-)%s*$", "%1")  -- Trim spaces from both ends
    return str:lower()  -- Convert to lowercase
end

local function convertJsonToTable(jsonResults)
    local resultsTable = {}
    if jsonResults and jsonResults.results and jsonResults.results.bindings then
        local bindings = jsonResults.results.bindings
        for j=0, #bindings do
            local row = {}
            for key, value in pairs(bindings[j]) do
                table.insert(row, value.value)
            end
            table.insert(resultsTable, row)
        end
    end
    return resultsTable
end

local function createHtmlTable(dataTable, headers, combineFirstTwoColumns)
    local htmlTable = mwHtml.create('table')
    htmlTable:addClass('wikitable'):attr('border', '1')

    if combineFirstTwoColumns and #headers > 1 then
        local headerRow = htmlTable:tag('tr')
        headerRow:tag('th'):wikitext(headers[0] .. " + " .. headers[1])
        for j = 2, #headers do
            headerRow:tag('th'):wikitext(headers[j])
        end
        for _, row in ipairs(dataTable) do
            local dataRow = htmlTable:tag('tr')
            local combinedData = '[' .. row[1] .. ' ' .. row[2] .. ']'
            dataRow:tag('td'):wikitext(combinedData)
            for j = 3, #row do
                dataRow:tag('td'):wikitext(row[j])
            end
        end
    else
        if #headers > 1 then
            local headerRow = htmlTable:tag('tr')
            for j = 0, #headers do
                headerRow:tag('th'):wikitext(headers[j])
            end
        end
        for _, row in ipairs(dataTable) do
            local dataRow = htmlTable:tag('tr')
            for _, data in ipairs(row) do
                dataRow:tag('td'):wikitext(data)
            end
        end
    end

    return tostring(htmlTable)
end

function p.buildTableFromSparql(frame)
    local sparqlQuery = frame.args[1]
    local combineFirstTwoColumns = trimAndLower(frame.args[2]) == "true"
    -- PHP function sparql.runQuery(query) is called
    local jsonResults = sparql.runQuery(sparqlQuery)
    local headers = {}
    if jsonResults and jsonResults.head and jsonResults.head.vars then
        headers = jsonResults.head.vars
    end
    local dataTable = convertJsonToTable(jsonResults)
    return createHtmlTable(dataTable, headers, combineFirstTwoColumns)
end

return p

!! endarticle


!! test
must have 3 rows and 3 columns with headers
!! config
!! wikitext
{{#invoke:MySPARQL|buildTableFromSparql|
PREFIX wdt: <https://portal.mardi4nfdi.de/prop/direct/>
PREFIX wd: <https://portal.mardi4nfdi.de/entity/>

SELECT
  ?work
  ?title
  ?publication_date
WHERE {
  ?work wdt:P16 wd:Q195254 .
  ?work wdt:P159 ?title .
  OPTIONAL {
    ?work wdt:P28 ?publication_datetime .
    BIND(xsd:date(?publication_datetime) AS ?publication_date)
  }
}
ORDER BY DESC(?publication_date)
}}
!! html
<table class="wikitable" border="1"><tbody><tr><th>work</th><th>title</th><th>publication_date</th></tr><tr><td><a rel="nofollow" class="external free" href="https://portal.mardi4nfdi.de/entity/Q2020028">https://portal.mardi4nfdi.de/entity/Q2020028</a></td><td>Regularity lemmas for clustering graphs</td><td>2021-04-23</td></tr><tr><td><a rel="nofollow" class="external free" href="https://portal.mardi4nfdi.de/entity/Q2291703">https://portal.mardi4nfdi.de/entity/Q2291703</a></td><td>Slow Fibonacci walks</td><td>2020-01-31</td></tr><tr><td><a rel="nofollow" class="external free" href="https://portal.mardi4nfdi.de/entity/Q2404825">https://portal.mardi4nfdi.de/entity/Q2404825</a></td><td>Juggling card sequences</td><td>2017-09-20</td></tr></tbody></table>
!! end

!! test
must have 3 rows and 2 columns with headers
!! config
!! wikitext
{{#invoke:MySPARQL|buildTableFromSparql|
PREFIX wdt: <https://portal.mardi4nfdi.de/prop/direct/>
PREFIX wd: <https://portal.mardi4nfdi.de/entity/>

SELECT
  ?work
  ?title
  ?publication_date
WHERE {
  ?work wdt:P16 wd:Q195254 .
  ?work wdt:P159 ?title .
  OPTIONAL {
    ?work wdt:P28 ?publication_datetime .
    BIND(xsd:date(?publication_datetime) AS ?publication_date)
  }
}
ORDER BY DESC(?publication_date)
|
true
}}
!! html
<table class="wikitable" border="1"><tbody><tr><th>work + title</th><th>publication_date</th></tr><tr><td><a rel="nofollow" class="external text" href="https://portal.mardi4nfdi.de/entity/Q2020028">Regularity lemmas for clustering graphs</a></td><td>2021-04-23</td></tr><tr><td><a rel="nofollow" class="external text" href="https://portal.mardi4nfdi.de/entity/Q2291703">Slow Fibonacci walks</a></td><td>2020-01-31</td></tr><tr><td><a rel="nofollow" class="external text" href="https://portal.mardi4nfdi.de/entity/Q2404825">Juggling card sequences</a></td><td>2017-09-20</td></tr></tbody></table>
!! end