Module:TA4TK:myTest

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

-- Required modules for SPARQL queries and HTML table generation
local sparql = require('SPARQL')
local mwHtml = require('mw.html')
local json = require("mw.text") 

local p = {}

function p.getImageWithLegend(frame)
    local pidLocalImage  = "P1088"    -- Property ID for the local image
    local pidMediaLegend = "P648"   -- Property ID for the (qualifier) media legend
	local qid = "Q4610"           -- The target entity
    local width = "480px"  
    local defaultLegend = "No legend available."
    local imageLegend1 = "No legend available."
    local imageLegend2 = "No legend available."
    
    
    -- Attempt to retrieve entity data
    local entity = mw.wikibase.getEntity(qid)
    if not entity or not entity.claims[pidLocalImage] then
        return "No image found"
    end
    
    
    local imageClaims = entity.claims[pidLocalImage]
    -- Extract the first image filename from P1088
    local imageFilename1 = imageClaims[1].mainsnak.datavalue.value
    if not imageFilename1 then
        return "No valid image found"
    end
	
	imageLegend1 = imageClaims[1].qualifiers[pidMediaLegend][1].datavalue.value.text

	-- Extract the second image filename from P1088
	local imageFilename2 = imageClaims[2].mainsnak.datavalue.value
    if not imageFilename2 then
    	--return one image
	    return string.format("[[File:%s|thumb|center|%s|%s]]", imageFilename1,width, imageLegend1)
    else     	
		imageLegend2 = imageClaims[2].qualifiers[pidMediaLegend][1].datavalue.value.text
    end

    --return two images
	return string.format("[[File:%s|thumb|left|%s|%s]]  [[File:%s|thumb|center|%s|%s]]", imageFilename1, width, imageLegend1, imageFilename2, width, imageLegend2)
   
end

function p.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


function p.hello()
	local str = "Hello World!"
	return str
end

function p.renderFormula()
    return mw.getCurrentFrame():extensionTag{
		name="math",
		content= "\\sin x", 
	}
end

-- Function to convert JSON results into a comma-separated string
function p.convertJsonToCommaSeparatedList(jsonResults)
	local resultsString = ""
	
	if jsonResults and jsonResults.results and jsonResults.results.bindings then
        local bindings = jsonResults.results.bindings
        for i = 0, #bindings do
            local binding = bindings[i]
            if binding.valueLabel and binding.valueLabel.value then
                if resultsString ~= "" then
                    resultsString = resultsString .. ", "
                end
                
                local name = binding.valueLabel.value
                if string.find(name, "https://") then
                	name = "Unnamed task"
                end
                
                local link = binding.value.value
                link = link:gsub("entity/Q", "wiki/Task:")
                
                local nameAndLink = "[" .. link .. " " .. name .. "]"

                resultsString = resultsString .. nameAndLink
            end
        end
    end

    return resultsString
end


-- Function to build the list
function p.buildList(frame)
	
    -- Retrieve target1 from frame arguments or return error message if not set
	local target1 = frame.args[1]
    if not target1 or target1 == '' then
        return "No ID given"
    end    
    
    -- Constructing the SPARQL query with dynamic entity target1
    local sparqlQuery = [[
PREFIX target1: <https://staging.mardi4nfdi.org/entity/]] .. target1 .. [[>
SELECT ?value ?valueLabel WHERE {
  ?value wdt:P715 target1:.
  target1: rdfs:label ?valueLabel
}
    ]]
    
	-- Executing the SPARQL query and retrieving results in JSON format
	local jsonResults = sparql.runQuery(sparqlQuery)
	
	-- Handle error in SPARQL query execution
	if jsonResults and jsonResults.error then
    	mw.log("Error in SPARQL query: " .. tostring(jsonResults.error))
    	return nil
	end

	if not jsonResults then
        return "Could not fetch data."
	end
	
	local resultString = p.convertJsonToCommaSeparatedList(jsonResults)
    return resultString
    
end

-- Function to build the list
function p.getFormulations(frame)
	
    -- Constructing the SPARQL query with dynamic entity target1
    local sparqlQuery = [[
    SELECT ?Formula ?defining_formulation
	# Q4610 refers to Electron Shuttling Model
	# P715 refers to contains 
	# P29 refers to defining formulation
	WHERE {
      wd:Q4610 wdt:P715 ?IDFormula.
      ?IDFormula  rdfs:label ?Formula.
      ?IDFormula wdt:P29 ?defining_formulation
	}
    ]]
    
	-- Executing the SPARQL query and retrieving results in JSON format
	local jsonResults = sparql.runQuery(sparqlQuery)
	
	-- Handle error in SPARQL query execution
	if jsonResults and jsonResults.error then
    	mw.log("Error in SPARQL query: " .. tostring(jsonResults.error))
    	return nil
	end

	if not jsonResults then
        return "Could not fetch data."
	end
	
	--local testResults = p.convertJsonToCommaSeparatedList(jsonResults)
	--return testResults
	local resultTable = p.convertJsonToTable(jsonResults)
    -- return resultTable
    
    --return "<pre>" .. mw.text.nowiki(json.jsonEncode(resultTable)) .. "</pre>"
    local jsonString = mw.text.jsonEncode(jsonResults)
    -- return "<pre>" .. mw.text.nowiki(jsonString) .. "</pre>"
    return p.extractDefiningFormulations(jsonString)

end

local json = require("mw.text") -- Load the JSON library

function p.extractDefiningFormulations(jsonString)
	local frame = mw.getCurrentFrame()
    local data = mw.text.jsonDecode(jsonString) -- Convert JSON string to Lua table
    
    if not data or not data.results or not data.results.bindings then
        return "Error: Invalid JSON structure"
    end

    local formulations = {} -- Table to store extracted values

    -- Loop through the bindings and collect "defining_formulation" values
    for _, item in ipairs(data.results.bindings) do
        if not item or not item.defining_formulation or not item.defining_formulation.value then
        	return "Error: Missing item.defining_formulation.value"
        else
        	local mathML = item.defining_formulation.value -- Get the MathML string
        	local equationLabel = item.Formula.value
        	 -- Use pattern matching to find the alttext attribute
    		local alttext = mathML:match('alttext="([^"]+)"')
    		-- Pattern to match content after "\displaystyle " and before the last "}"
		    local match = alttext:match("\\displaystyle%s*(.*)}%s*$")
            --table.insert(formulations, item.defining_formulation.value)
              local mathTag = frame:extensionTag{
            	name = "math",
            	content = match
        	  }
              -- table.insert(formulations, "| " .. mathTag) -- Add row to the table
              table.insert(formulations, "| " .. equationLabel .. " || " .. mathTag)
       --     table.insert(formulations, match)
        end
        
    end
  
    -- Construct the Wikitext table
    local wikitextTable = "{| class='wikitable'\n" .. table.concat(formulations, "\n|-\n") .. "\n|}"
	return wikitextTable
end

function p.extractDefiningFormulationsWithQuantities(jsonString)
	local frame = mw.getCurrentFrame()
    local data = mw.text.jsonDecode(jsonString) -- Convert JSON string to Lua table
    
    if not data or not data.results or not data.results.bindings then
        return "Error: Invalid JSON structure"
    end

    local formulations = {} -- Table to store extracted values

    -- Loop through the bindings and collect "defining_formulation" values
    for _, item in ipairs(data.results.bindings) do
        if not item or not item.defining_formulation or not item.defining_formulation.value then
        	return "Error: Missing item.defining_formulation.value"
        else
        	local mathML = item.defining_formulation.value -- Get the MathML string
        	local equationLabel = item.Formula.value
        	 -- Use pattern matching to find the alttext attribute
    		local alttext = mathML:match('alttext="([^"]+)"')
    		-- Pattern to match content after "\displaystyle " and before the last "}"
		    local match = alttext:match("\\displaystyle%s*(.*)}%s*$")
            --table.insert(formulations, item.defining_formulation.value)
              local mathTag = frame:extensionTag{
            	name = "math",
            	content = match
        	  }
        	  local quantityMathML = item.in_defining_formula.value
    		  -- Match the content inside alttext="{\displaystyle ... }"
    		  local alttext = string.match(quantityMathML, 'alttext="{\\displaystyle%s*(.-)%s*}"')
              local mathQuantity = frame:extensionTag{
            	name = "math",
            	content = alttext
        	  }
              -- table.insert(formulations, "| " .. mathTag) -- Add row to the table
              table.insert(formulations, "| " .. equationLabel .. " || " .. mathTag)
              table.insert(formulations, "| " .. mathQuantity ..  " || " .. "symbol represents:")
   
        end
        
    end
        
    -- Construct the Wikitext table
    local wikitextTable = "{| class='wikitable'\n" .. table.concat(formulations, "\n|-\n") .. "\n|}"
	return wikitextTable
end

function p.extractQuantities(frame)
    local pidInDefiningFormula  = "P597"    -- Property ID for in defining formula 
    local pidSymbolRepresents = "P598"   -- Property ID for the (qualifier) symbol represents
	local qid = "Q4577"           -- The target entity
    
    -- 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 imageClaims = entity.claims[pidInDefiningFormula]
    -- Extract the first image filename from P1088
    local imageFilename1 = imageClaims[1].mainsnak.datavalue.value
    if not imageFilename1 then
        return "No valid formulation found"
    end
	quantity = imageClaims[1].qualifiers[pidSymbolRepresents][1].datavalue.value.text
	--return quantity	
	-- return imageClaims[1].qualifiers
	
	-- return "<pre>" .. mw.text.nowiki(json.jsonEncode(imageClaims[1].qualifiers)) .. "</pre>"
--	return  "<pre>" .. mw.text.nowiki(json.jsonEncode( imageClaims[1].qualifiers.P598[1].datavalue.value.id)) .. "</pre>" 
	-- Attempt to retrieve entity data
    quantityId = imageClaims[1].qualifiers.P598[1].datavalue.value.id
    local label = mw.wikibase.label(quantityId)
    return label

end


-- Function to build the list
function p.getFormulationsWithQuantities(frame)
	
    -- Constructing the SPARQL query with dynamic entity target1
    local sparqlQuery = [[
    SELECT ?Formula ?defining_formulation ?in_defining_formula
	# Q4610 refers to Electron Shuttling Model
	# P715 refers to contains 
	# P29 refers to defining formula
	# P597 refers to in defining formula
	WHERE {
      wd:Q4610 wdt:P715 ?IDFormula.
      ?IDFormula  rdfs:label ?Formula.
      ?IDFormula wdt:P29 ?defining_formulation.
      ?IDFormula wdt:P597 ?in_defining_formula
	}
    ]]
    
	-- Executing the SPARQL query and retrieving results in JSON format
	local jsonResults = sparql.runQuery(sparqlQuery)
	
	-- Handle error in SPARQL query execution
	if jsonResults and jsonResults.error then
    	mw.log("Error in SPARQL query: " .. tostring(jsonResults.error))
    	return nil
	end

	if not jsonResults then
        return "Could not fetch data."
	end
	
	--local testResults = p.convertJsonToCommaSeparatedList(jsonResults)
	--return testResults
	local resultTable = p.convertJsonToTable(jsonResults)
    --return resultTable
    
    --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(jsonString)

end

-- Function to build the description
function p.buildDescription(frame)
	
    -- Retrieve target1 from frame arguments or return error message if not set
	local target1 = frame.args[1]
    if not target1 or target1 == '' then
        return "No ID given"
    end    
    
    -- Constructing the SPARQL query with dynamic entity target1
    local sparqlQuery = [[
SELECT ?Description
WHERE {
      wd:Q4610 wdt:P896 ?Description.
}
    ]]

	-- Executing the SPARQL query and retrieving results in JSON format
	local jsonResults = sparql.runQuery(sparqlQuery)
    return jsonResults['results']['bindings'][0]['Description']['value']
end



function p.getImage(frame)
    local entityId = "Q4610"  -- The target entity
    local propertyId = "P1088" -- Property assumed to store the image
    local width = "480px"      -- Desired image width

    -- Attempt to retrieve entity data
    local entity = mw.wikibase.getEntity(entityId)
    if not entity or not entity.claims[propertyId] then
        return "No image found"
    end

    -- Extract the first image filename from P1088
    local imageFilename1 = entity.claims[propertyId][1].mainsnak.datavalue.value
    if not imageFilename1 then
        return "No valid image found"
    end
	-- Extract the second image filename from P1088
	local imageFilename2 = entity.claims[propertyId][2].mainsnak.datavalue.value
    if not imageFilename2 then
    	--return one image
	    return string.format("[[File:%s|thumb|center|%s]]", imageFilename1, width)
    else     	
    	--return two images
	     return string.format("[[File:%s|left|%s]]  [[File:%s|center|%s]] \n \n  \n ", imageFilename1, width, imageFilename2, width)
    end
end



return p