Module:Plan

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

local p = {}

function p.wrap( s, length )
	s = mw.text.trim( s )
	local oldindex = 1
	local index = string.find( s, ' ' )
	while ( index ) do
		if ( index - oldindex >= length ) then
			local temp = string.sub( s, 1, index - 1 ) .. '<BR/>' .. string.sub( s, index + 1 )
			s = temp
			oldindex = index + 4
		end
		index = string.find( s, ' ', index + 1)
	end
	return s
end

function p.wrapArray( a, length )
	s = ''
	for k,v in pairs( a ) do
		if #s > 0 then
			s = s .. '<BR/>'
		end
		s = s .. p.wrap( v, length )
	end
	return s
end

function p.label( a )
	return '"' .. p.wrapArray( a, 30 ) .. '"'
end

function p.cleanseNodeName( s )
	s = string.gsub( s, ':', '-')
	s = string.gsub( s, '/', '-')
	return s
end

function p.buildGoalGraph( plan )
	local plans = mw.smw.getQueryResult( '[[' .. plan .. ']]|?Goal' )
	if plans == nil then
		return ''
	end
	local plan_name = p.cleanseNodeName( plan )
	local s = 'graph TB\n'
	s = s .. 'classDef mermaid-goal fill:#104e8b,stroke:black;\n'
	s = s .. 'classDef mermaid-outcome fill:#458b74,stroke:black;\n'
	local node = table.remove( plans.results )
	s = s .. plan_name .. '[' ..
		p.label( { table.concat( node.printouts.Goal ) } ) .. ']\n'
	s = s .. 'class ' .. plan_name .. ' mermaid-goal\n'
	local outcomes = mw.smw.getQueryResult( '[[Category:Outcomes]][[Plan::' .. plan .. '|?Description' )
	for k,v in pairs( outcomes.results ) do
		local outcome_name = p.cleanseNodeName( v.fulltext )
		s = s .. outcome_name .. '[' ..
			 p.label( { table.concat( v.printouts.Description ) } ) .. ']\n'
		s = s .. 'class ' .. outcome_name .. ' mermaid-outcome\n'
		s = s .. 'click ' .. outcome_name .. ' "' .. v.fullurl .. '"\n'
		s = s .. plan_name .. ' -->' .. outcome_name .. '\n'
	end
	return s
end


function p.buildOutcomeGraph( outcome )
	local outcomes = mw.smw.getQueryResult( '[[' .. outcome .. ']]|?Description' )
    if outcomes == nil then
    	return ''
    end
	local outcome_name = p.cleanseNodeName( outcome )
	local s = 'graph TB\n'
	s = s .. 'classDef mermaid-outcome fill:#458b74,stroke:black;\n'
	s = s .. 'classDef mermaid-output fill:#888888,stroke:black;\n'
	local node = table.remove( outcomes.results )
	s = s .. outcome_name .. '[' ..
		p.label( { table.concat( node.printouts.Description ) } ) .. ']\n'
	s = s .. 'class ' .. outcome_name .. ' mermaid-outcome\n'
	local outputs = mw.smw.getQueryResult(
		'[[Category:Outputs]][[Parent Outcome::' .. outcome .. ']]|?Description|?Primary Team|?Collaborating Team' )
	for k,v in pairs( outputs.results ) do
		local output_name = p.cleanseNodeName( v.fulltext )
		local primary = table.concat( v.printouts['Primary Team'] )
		if #primary > 0 then
			primary = '<BR/>Primary Team: ' .. primary
		end
		local collaborating = table.concat( v.printouts['Collaborating Team'], ', ' )
		if #collaborating > 0 then
			collaborating = 'Collaborating Teams: ' .. collaborating
		end
		s = s .. output_name .. '[' ..
			p.label( { table.concat( v.printouts.Description ), primary, collaborating } )
		s = s .. ']\n'
		s = s .. 'class ' .. output_name .. ' mermaid-output\n'
		s = s .. 'click ' .. output_name.. ' "' .. v.fullurl .. '"\n'
		s = s .. outcome_name .. '-->' .. output_name .. '\n'
	end
	return s
end

function p.showGoal(frame)
	local s = p.buildGoalGraph( frame.args[1] )
	return '<div style="width:100%;text-align:center;font-size:14pt;">' ..
		frame:callParserFunction{ name = '#mermaid', args = { s } } .. '</div>'
end

function p.showOutcome(frame)
	local s = p.buildOutcomeGraph( frame.args[1] )
	return '<div style="width:100%;text-align:center;font-size:14pt;">' ..
		frame:callParserFunction{ name = '#mermaid', args = { s } } .. '</div>'
end

return p