local p = {}



local default_color = '#F8F9FA'



local categories = {

	party_not_in_list = '[[Category:Pages using Political party with unknown party]]',

	shortname_not_in_list = '[[Category:Pages using Political party with missing shortname]]',

	color_not_in_list = '[[Category:Pages using Political party with missing color]]',

}



local function create_error(error_message)

	return string.format('<strong class="error">%s</strong>', error_message)

end



local function getFirstLetter(party)

	local index = mw.ustring.sub(party, 1, 1)

	-- Set index for non-A-Z starts

	if string.match(index, '%A') then

		return '1'

	end

	return string.upper(index)

end



local function stripToNil(text)

	-- If text is a string, return its trimmed content, or nil if empty.

	-- Otherwise return text (which may, for example, be nil).

	if type(text) == 'string' then

		text = text:match('(%S.-)%s*$')

		local delink = require('Module:Delink')._delink

		text = delink({text, wikilinks = "target"})

	end

	return text

end



-- Example of having all the data - color and names - in one table. Requires one page to be edited instead of two when adding a new party.

function p._fetch(args)

	if not args1 then

		return create_error("parameter 1 should be a party name.")

	end



	if not args2 then

		return create_error("parameter 2 should be the output type.")

	end



 	local party = stripToNil(args1])

	local out_type = stripToNil(args2])

	if out_type == 'colour' then

		out_type = 'color'

	end

	local index = getFirstLetter(party)

	

	-- Load data from submodule

	local data = mw.loadData('Module:Political party/' .. index)

	local data_all = data.full



	local party_alt = data.alternateparty

	local party_info

	if party_alt then

		party = party_alt

		if data_allparty_alt then

			party_info = data_allparty_alt

		else

			index = getFirstLetter(party_alt)

			data = mw.loadData('Module:Political party/' .. index)

			party_info = data.fullparty_alt

		end

	else

		party_info = data_allparty

	end



	-- Check if database value exists

	-- * Not even in database - return given error or input

	-- * No color - return error

	-- * No shortname/abbrev - return first non-blank of abbrev->shortname->input

	if not party_info then

		if out_type == 'color' then

			return args.error or default_color

		else

			return args.error or party

		end

	end

	local return_value = party_infoout_type

	if return_value == "" then

		if out_type == 'color' then

			return args.error or create_error("Value not in template. Please request that it be added.")

		elseif out_type == 'abbrev' then

			if party_info.shortname ~= "" then

				return party_info.shortname

			else

				return party

			end

		elseif out_type == 'shortname' then

			if party_info.abbrev ~= "" then

				return party_info.abbrev 

			else

				return party

			end

		elseif out_type == 'link' then

			return nil

		else

			return party

		end

	elseif return_value == nil then

		if out_type == 'link' or out_type == 'altname' then

			return party

		end

	end



	if out_type == 'color' and string.find(return_value, '#') then

		return_value = string.gsub(return_value, '#', '&#35;')

	end

	return return_value	

end



function p.fetch(frame)

	-- Initialise and populate variables

	local getArgs = require("Module:Arguments").getArgs

	local args = getArgs(frame)

	

	return p._fetch(args)

end



return p