Module:Task

Documentation for this module may be created at Module:Task/doc

-- Required modules for SPARQL queries and HTML table generation
-- Module for executing SPARQL queries
local sparql = require('SPARQL')
-- MediaWiki library for logging and utilities
local mw = require('mw')
local json = require("mw.text") 

-- Main table to hold all functions
local p = {}


-- Function to get Formulations with Quantities
function p.getFormulationsWithQuantities(frame)
	
	local entityId = frame.args[1]
    
    -- Validate input parameter
    if not entityId or entityId == '' then
        return "Error: No entity ID provided"
    end

    -- Constructing the SPARQL query with dynamic entity entityId
    -- P1560: contains property id
    -- P989: defining formula property id
    local sparqlQuery = [[
PREFIX entityId: <https://portal.mardi4nfdi.de/entity/]] .. entityId .. [[>
	SELECT ?Formula ?defining_formulation ?IDFormula 
	WHERE {
      entityId: wdt:P1560 ?IDFormula.
      ?IDFormula  rdfs:label ?Formula.
      ?IDFormula wdt:P989 ?defining_formulation.
	}
    ]]
    
	-- Executing the SPARQL query and retrieving results in JSON format
	local jsonResults = sparql.runQuery(sparqlQuery)
	
		-- Validate results
	if not jsonResults or not jsonResults.results or not jsonResults.results.bindings then
       	return "\"No mathematical expressions found.\""
    end
	
	--local resultTable = p.convertJsonToTable(jsonResults)
	--return "<pre>" .. mw.text.nowiki(json.jsonEncode(resultTable)) .. "</pre>"
    local jsonString = mw.text.jsonEncode(jsonResults)
    -- return "<pre>" .. mw.text.nowiki(jsonString) .. "</pre>"
    return p.extractDefiningFormulationsWithQuantities(jsonResults)

end

-- Function to extract Formulations with Quantities 
function p.extractDefiningFormulationsWithQuantities(jsonResults)
	local frame = mw.getCurrentFrame()
	-- Table to store formulations
    local formulations = {}
    -- Table to store quantity symbol and name id 
    local quantitySymbolNameIdPairs = {}
    
     -- Get the number of mathematical formulations
	local totalFormulations = #jsonResults.results.bindings
    
    if totalFormulations > 0 then
    	-- Loop through the bindings
		for index = 0, totalFormulations  do
    		local item = jsonResults.results.bindings[index]
    		local equationLabel = item.Formula.value
        	local equationIDURL = item.IDFormula.value
        	local equationWithUrl = string.format('[%s %s]', tostring(equationIDURL), tostring(equationLabel))
        	local equationID = equationIDURL:match(".*/(.-)$") or equationIDURL -- Extracts the part after the last '/'
        	
        	local equation = mw.wikibase.getEntity(equationID)
		    -- P989 refers to property defining formula 
		    local definingFormula = equation:getBestStatements('P989')
		    if definingFormula and #definingFormula > 0 then
    			local value = definingFormula[1].mainsnak.datavalue.value
    			mathTag = frame:extensionTag{
	        		name = "math",
	         		content = value
        		}
	        	--table.insert(formulations, "| " .. equationWithUrl  .. " || " .. mathTag)
	        	table.insert(formulations, 
    string.format('| style="width:30%%;" | %s || style="width:70%%;" | %s', equationWithUrl, mathTag)
)

			else
				table.insert(formulations, "| " .. equationWithUrl)
		    end
		
        	quantitySymbolNameIdPairs = p.extractQuantities(equationID)
        	
        	if type(quantitySymbolNameIdPairs) == "table" then
        		-- Accessing the stored pairs
				for i, pair in ipairs(quantitySymbolNameIdPairs) do
					local pairFirstValue = pair[1]
					local pairFirstValue = mw.text.decode(pairFirstValue or "")
					local quantitySymbolMathTag = frame:extensionTag{
						name = "math",
            			content = pairFirstValue
        			}
        			-- Construct the Portal URL
        			local url = "https://portal.mardi4nfdi.de/wiki/Item:"
        			local fullUrl = url .. pair[3]
        			local labelWithUrl = string.format('[%s %s]', tostring(fullUrl), tostring(pair[2]))
        	
            		table.insert(formulations, "| " .. quantitySymbolMathTag   ..  " symbol represents "  .. labelWithUrl)
				end
			table.insert(formulations, "| " .. " " )
			else
				-- Error: extractQuantities did not return a table"
			end
		end

		 -- Construct the Wikitext table
    	local wikitextTable = [[
{| class="wikitable" style="table-layout: fixed; width: 1200px;"
]] .. table.concat(formulations, "\n|-\n") .. "\n|}"

    	return wikitextTable
    else 
    	return "No mathematical expressions found."
    end
end


function p.extractQuantities(qid)
	-- Property ID for in defining formula 
    local pidInDefiningFormula  = "P983"
    -- Property ID for the (qualifier) symbol represents
    local pidSymbolRepresents = "P984"   

    -- Attempt to retrieve entity data
    local entity = mw.wikibase.getEntity(qid)
    if not entity or not entity.claims[pidInDefiningFormula] then
        return "No formulation found"
    end
    
    
    local inDefiningFormulaClaims = entity.claims[pidInDefiningFormula]
    local count = 0
    -- Table to store pairs of quantity symbol and quantity name
    local quantitySymbolNameIdPairs = {}
	for _ in pairs(inDefiningFormulaClaims or {}) do
    	count = count + 1
    	-- Get the quantity symbol
    	local quantitySymbol = inDefiningFormulaClaims[count].mainsnak.datavalue.value
    	if not quantitySymbol then
        	return "No valid symbol found"
    	end
    	quantity = inDefiningFormulaClaims[count].qualifiers[pidSymbolRepresents][1].datavalue.value.text
        quantityId = inDefiningFormulaClaims[count].qualifiers[pidSymbolRepresents][1].datavalue.value.id
	    local quantityName = mw.wikibase.label(quantityId)
	    -- Insert pair into the table
	    table.insert(quantitySymbolNameIdPairs, {quantitySymbol, quantityName, quantityId})
    end
    return quantitySymbolNameIdPairs
end


return p