local Date = require('Module:Date')._Date

local yesno = require('Module:Yesno')



local p = {}



-- Local variables.

local reviewer = nil;

local df = "mdy";

local entity = nil;

local genRefs = true;

local showSystem = true;

local showUpdateLink = true;

local system = nil;

local systemId = nil;

local systemFormat = "colon";

local updateLinkStyle = nil;

local entities = {};



-- Translation table for converting numeric-IDs to shorthand aliases.

local systemAliases = {

	10677 = 'PS1',

	1323662 = 'PS1', -- Placeholder, this is actually the series but could be mistakenly used for PS1.

	10680 = 'PS2',

	10683 = 'PS3',

	5014725 = 'PS4',

	16338 = 'PC',

	8079 = 'Wii',

	56942 = 'WiiU',

	132020 = 'XBOX',

	48263 = 'X360',

	13361286 = 'XONE',

	203597 = '3DS',

	188808 = 'PSV',

	170323 = 'DS', -- Sometimes has been NDS

	170325 = 'PSP',

	48493 = 'IOS', -- iOS, iPhone, iPad

	94 = 'AND', -- Android

	186437 = 'GB',

	188642 = 'GBA',

	203992 = 'GBC',

	184198 = 'DC',

	200912 = 'SAT',

	172742 = 'NES',

	183259 = 'SNES',

	184839 = 'N64',

	182172 = 'GC', -- Sometimes has been NGC

	19610114 = 'NS', -- Nintendo Switch

	98973368 = 'XSX/S', -- Xbox Series X and Series S (Output label differs from the input parameter, XSXS, per RFC)

	63184502 = 'PS5'

}



-- Translation table for converting system aliases to QIDs

local systemIDs = {

	'PS1' = 10677,

	'PS2' = 10680,

	'PS3' = 10683,

	'PS4' = 5014725,

	'PC' = 16338,

	'WII' = 8079,

	'WIIU' = 56942,

	'XBOX' = 132020,

	'X360' = 48263,

	'XONE' = 13361286,

	'3DS' = 203597,

	'PSV' = 188808,

	'DS' = 170323,

	'NDS' = 170323,

	'PSP' = 170325,

	'IOS' = 48493,

	'AND' = 94,

	'GB' = 186437,

	'GBA' = 188642,

	'GBC' = 203992,

	'DC' = 184198,

	'SAT' = 200912,

	'NES' = 172742,

	'SNES' = 183259,

	'N64' = 184839,

	'GC' = 182172,

	'NGC' = 182172,

	'NS' = 19610114,

	'XSXS' = 98973368, -- This is the template parameter, which cannot contain a / or | character.

	'PS5' = 63184502

}



-- List of accepted aggregator arguments and their related QID.

local aggregatorAliases = {

    150248 = 'MC',

    40160 = 'GR',

    21039459 = 'OC'

}   



-- List of accepted aggregator arguments and their related QID.

local aggregatorIDs = {

    'MC' = 150248,

    'GR' = 40160,

    'OC' = 21039459

}   



-- List of accepted reviewer arguments and their related QID.

local reviewerAliases = {

	591573 = 'FAM',

	207708 = 'IGN'

}   



-- List of accepted reviewer arguments and their related QID.

local reviewerIDs = {

	'FAM' = 591573,

	'IGN' = 207708

} 



local function sortByPlatform(a,b) 

	local platformA = "";

	local platformB = "";

	if(a'qualifiers']['P400' ~= nil and a'qualifiers']['P400'][1 ~= nil) then

		platformA = p.getSystemAlias(a'qualifiers']['P400'][1]['datavalue']['value']['numeric-id']);

		if(platformA == nil) then

			platformA = mw.wikibase.label('Q'..a'qualifiers']['P400'][1]['datavalue']['value']['numeric-id']);

		end;

	end;

	if(b'qualifiers']['P400' ~= nil and b'qualifiers']['P400'][1 ~= nil) then

		platformB = p.getSystemAlias(b'qualifiers']['P400'][1]['datavalue']['value']['numeric-id']);

		if(platformB == nil) then

			platformB = mw.wikibase.label('Q'..b'qualifiers']['P400'][1]['datavalue']['value']['numeric-id']);

		end;

	end;		

		

	return platformA < platformB

end;



local function buildCite(reference) 

	local referenceUrl = nil;

	local cite = nil;

	

	if(reference'snaks']['P854' ~= nil and reference'snaks']['P854'][1 ~= nil) then

		referenceUrl = reference'snaks']['P854'][1]['datavalue']['value'];	

	end;



	if(referenceUrl ~= nil and referenceUrl ~= "") then

		cite = "{{cite web|url="..referenceUrl;

			

		local pubdate = nil;

		local accessdate = nil;

		local publisher = nil;

		local work = nil;

		local title = nil;

		local archiveUrl = nil;

		local archiveDate = nil;

		local authors = {};



		if(reference'snaks']['P577' ~= nil and reference'snaks']['P577'][1 ~= nil) then

			pubdate = reference'snaks']['P577'][1]['datavalue']['value']['time'];

		end;

		if(reference'snaks']['P813' ~= nil and reference'snaks']['P813'][1 ~= nil) then

			accessdate = reference'snaks']['P813'][1]['datavalue']['value']['time'];

		end;

		if(reference'snaks']['P123' ~= nil and reference'snaks']['P123'][1 ~= nil) then

			publisher = mw.wikibase.label('Q'..reference'snaks']['P123'][1]['datavalue']['value']['numeric-id']);

		end;

		if(reference'snaks']['P1433' ~= nil and reference'snaks']['P1433'][1 ~= nil) then

			work = mw.wikibase.label('Q'..reference'snaks']['P1433'][1]['datavalue']['value']['numeric-id']);

		end;			

		if(reference'snaks']['P1476' ~= nil and reference'snaks']['P1476'][1 ~= nil) then

			title = reference'snaks']['P1476'][1]['datavalue']['value']['text'];

		end;

		if(reference'snaks']['P1065' ~= nil and reference'snaks']['P1065'][1 ~= nil) then

			archiveUrl = reference'snaks']['P1065'][1]['datavalue']['value'];

		end;

		if(reference'snaks']['P2960' ~= nil and reference'snaks']['P2960'][1 ~= nil) then

			archiveDate = reference'snaks']['P2960'][1]['datavalue']['value']['time'];

		end;	

		if(reference'snaks']['P50' ~= nil and #reference'snaks']['P50' > 0) then

			for i,authorDat in pairs(reference'snaks']['P50']) do

				local authorQid = 'Q'..authorDat'datavalue']['value']['numeric-id'];

				if(entitiesauthorQid == nil) then

					entitiesauthorQid = mw.wikibase.getEntity(authorQid);

				end;

				

				local author = {};

				author'fullname' = mw.wikibase.label(authorQid); -- Default to label

				author'first' = nil;

				author'last' = nil;

				

				if(entitiesauthorQid]['claims']['P735' ~= nil and entitiesauthorQid]['claims']['P735'][1 ~= nil) then

					author'first' = mw.wikibase.label('Q'..entitiesauthorQid]['claims']['P735'][1]['mainsnak']['datavalue']['value']['numeric-id']);

				end;

				if(entitiesauthorQid]['claims']['P734' ~= nil and entitiesauthorQid]['claims']['P734'][1 ~= nil) then

					author'last' = mw.wikibase.label('Q'..entitiesauthorQid]['claims']['P734'][1]['mainsnak']['datavalue']['value']['numeric-id']);

				end;

				

				table.insert(authors, author);

			end;

		end;			

		

		if(title ~= nil and title ~= "") then

			cite = cite .. "|title="..title;

		end;

		if(publisher ~= nil and publisher ~= "") then

			cite = cite .. "|publisher="..publisher;

		end;

		if(work ~= nil and work ~= "") then

			cite = cite .. "|work="..work;

		end;		

		if(pubdate ~= nil and pubdate ~= "") then

			local pubdateText = Date(pubdate):text(df);



			cite = cite .. "|date="..pubdateText;

		end;		

		if(accessdate ~= nil and accessdate ~= "") then

			local accessdateText = Date(accessdate):text(df);



			cite = cite .. "|accessdate="..accessdateText;

		end;

		if(archiveUrl ~= nil and archiveUrl ~= "" and archiveDate ~= nil and archiveDate ~= "") then

		    local archivedateText = Date(archiveDate):text(df);

			cite = cite .. "|archiveurl="..archiveUrl;

			cite = cite .. "|archivedate="..archivedateText;

		end;		

		if(#authors > 0) then

			for i,author in pairs(authors) do

				if(author'first' ~= nil and author'last' ~= nil and author'first' ~= "" and author'last' ~= "") then

					if(#authors == 1) then

						cite = cite .."|last="..author'last'.."|first="..author'first'];

					else

						cite = cite .."|last"..i.."="..author'last'.."|first"..i.."="..author'first'];

					end;

				else

					if(#authors == 1) then

						cite = cite .."|author="..author'fullname'];

					else

						cite = cite .."|author"..i.."="..author'fullname'];

					end;					

				end;

			end;			

		end;

		

		

		cite = cite..'}}';	

	end;

	

	return cite;

end;



local function printReviewRow(frame, reviewscore)

	local score = nil;



	if(reviewscore'mainsnak']['datavalue' ~= nil and reviewscore'mainsnak']['datavalue']['value' ~= nil) then

		score = reviewscore'mainsnak']['datavalue']['value'];

	else

		return "";

	end;

	

	local ret = ""

	local system = nil;

	local reference = nil;	



	if(reviewscore'qualifiers']['P400' ~= nil and reviewscore'qualifiers']['P400'][1 ~= nil) then

		system = p.getSystemAlias(reviewscore'qualifiers']['P400'][1]['datavalue']['value']['numeric-id']);

	end	

	if(system ~= nil and system ~= "" and showSystem) then

		if(systemFormat == "para") then

			ret = ret.."("..system..") ";

		else

			ret = ret..system..": ";

		end;

	end;



	ret = ret..score;



	if(reviewscore'references' ~= nil and reviewscore'references'][1 ~= nil and genRefs) then

		local cite = buildCite(reviewscore'references'][1]);

		

		if(cite ~= nil) then

			local scoreBy = p.getAggregatorAlias(reviewscore'qualifiers']['P447'][1]['datavalue']['value']['numeric-id']);

			if(scoreBy == nil) then

				scoreBy = p.getReviewerAlias(reviewscore'qualifiers']['P447'][1]['datavalue']['value']['numeric-id']);

			end;



			local name = entity:getLabel()..'-'..scoreBy;

			if(system ~= nil and system ~= "") then

				name = name..system;

			end;



			cite = frame:extensionTag{ name = "ref", args = {name=name}, content=cite };

			ret = ret..cite;

		end;

	end;



	return ret.."<br />";

end



function p.getSystemAlias(numericId)

	return systemAliasesnumericId];

end



function p.getSystemID(system)

	return systemIDssystem];

end



function p.getAggregatorAlias(numericId)

	return aggregatorAliasesnumericId];

end



function p.getAggregatorID(system)

	return aggregatorIDssystem];

end



function p.getReviewerAlias(numericId)

	return reviewerAliasesnumericId];

end



function p.getReviewerID(system)

	return reviewerIDssystem];

end



function p.setReviewer(iReviewer)

	-- No reviewer, stop. Must have reviewer at least.

	if(iReviewer == nil or iReviewer == "") then

		return "Missing reviewer";

	end;

	

	-- See if supplied reviewer is in the aggregator table.

	iReviewer = string.upper(iReviewer)

	reviewer = p.getAggregatorID(iReviewer);

	if(reviewer == nil or reviewer == "") then

		-- No? Maybe in the reviewer table.

		reviewer = p.getReviewerID(iReviewer);

		if(reviewer == nil or reviewer == "") then

			return "Invalid reviewer";

		end;

	end;	



	return nil;

end;



function p.setDateFormat(iDf)

	-- Check for a date format parameter. Default to mdy if missing.

	if(iDf ~= nil and iDf ~= "") then

		df = string.lower(iDf);

	end;

end;



function p.setSystemFormat(iSf)

	if(iSf ~= nil and iSf ~= "") then

		systemFormat = string.lower(iSf);

	end;

end;



function p.setUpdateLinkStyle(iStyle)

	if(iStyle ~= nil and iStyle ~= "") then

		updateLinkStyle = string.lower(iStyle);

	end;

end;



function p.setGame(iGame)

	-- Check for a game parameter. If missing, default to current article.

	if(iGame ~= nil and iGame ~= "") then

		if(entitiesiGame == nil and mw.wikibase ~= nil) then

 			entitiesiGame = mw.wikibase.getEntity(iGame);

		end;

		

		entity = entitiesiGame

	else

		-- Need to research if we can determine the entity's ID before retrieving it.

		if(mw.wikibase ~= nil) then

			entity = mw.wikibase.getEntity();

			if(entity ~= nil) then

				entitiesentity'id']] = entity;

			end;

		end;

	end;

	

	if(entity == nil) then

		return "No matching wikidata entity found";

	end;

	

	return nil;

end;



function p.setSystem(iSystem)

	-- Check for system parameter, and resolve it's QID if possible.

	if(iSystem ~= nil and iSystem ~= "") then

		system = string.upper(iSystem);

		systemId = p.getSystemID(system);

	elseif(not showSystem) then

		-- If no system was specified, force showSystem on.

		showSystem = true;

	end;

end;



function p.setGenerateReferences(iGenRefs)

	-- Reference suppression.

	if(iGenRefs ~= nil and iGenRefs ~= "") then

		genRefs = yesno(iGenRefs, true);

	end;

end;



function p.setShowSystem(iShowSystem)

	-- Suppression of system aliases in front of score, i.e. (XBOX) xx/100.

	if(iShowSystem ~= nil and iShowSystem ~= "") then

		showSystem = yesno(iShowSystem, false);

	end;

	if(system == nil or system == '') then

		-- If no system was specified, force showSystem on.

		showSystem = true;

	end;

end;



function p.setShowUpdateLink(iShowUpdateLink)

	-- Suppression of update link to Wikidata at the end of the score, i.e. (XBOX) xx/100[+].

	if(iShowUpdateLink ~= nil and iShowUpdateLink ~= "") then

		showUpdateLink = yesno(iShowUpdateLink, false);

	end;

end;



function p.getUpdateLink()

	if(updateLinkStyle == "pen") then

		return "[[File:Blue pencil.svg|frameless|text-top|10px|alt=Edit this on Wikidata|link=https://www.wikidata.org/wiki/"..entity'id'.."?uselang="..mw.language.getContentLanguage().code.."#P444|Edit this on Wikidata]]";

	elseif(updateLinkStyle == "noSub") then

		return '[[d:'..entity'id'..'#P444|&#91;±&#93;]]';

	elseif(updateLinkStyle == "text and pen") then

		return '<span style="position: relative;"><span style="position: absolute; right: 0;">[[File:Blue pencil.svg|10px|baseline|link=|alt=]]</span>[[d:'..entity'id'..'#P444|<span style="position: relative; padding-right: 14px;">Edit on Wikidata</span>]]</span>'

	end;

	

	return '<sub>[[d:'..entity'id'..'#P444|&#91;±&#93;]]</sub>';

end;



function p.getSitelink()

	return mw.wikibase.sitelink(entity'id']);

end;



function p.getLabel()

	return mw.wikibase.label(entity'id']);

end;



function p.getParts()

	local ret = {};	

	

	-- Loop all of "has Part" for this title

	local parts = entity'claims']['P527'];	

	if(parts) then

		for i,part in pairs(parts) do

			table.insert(ret,"Q"..part'mainsnak']['datavalue']['value']['numeric-id']);

		end;		

	end;



	return ret;	

end;



function p.getEarliestPublicationDate()

	local ret = {};

	

	local pubDates = entity'claims']['P577'];	

	if(pubDates) then

		for i,pubDate in pairs(pubDates) do

			if(pubDate'mainsnak']['datavalue']) then

				local timestamp = pubDate'mainsnak']['datavalue']['value']['time'];

				local accessdate = Date(timestamp);	

				table.insert(ret,accessdate);

			end;

		end;		

	end;



	if(#ret < 1) then

		return nil;

	end;



	table.sort(ret);

	

	return ret1];

end;



function p.printReviewScores(frame)

	local ret = "";	

	

	-- Loop all of "review scores" for this title

	local reviewscores = entity'claims']['P444'];	

	if(reviewscores) then

		-- Find reviews that qualify for printing and insert into array.

		local reviewsToPrint = {}

    	for i,review in pairs(reviewscores) do

    		if(review'qualifiers' ~= nil) then

				local scoreBy = nil 

				if(review'qualifiers']['P447' ~= nil and review'qualifiers']['P447'][1 ~= nil) then

					scoreBy = review'qualifiers']['P447'][1]['datavalue']['value']['numeric-id'];

				end;

				if(scoreBy == reviewer) then

					-- If template specified a system, we need to check for the specific system and only output that one.

					if(system == nil or system == "") then

						-- No system specified, so output each one found.

						table.insert(reviewsToPrint,review);

					else

						-- Get platform if it exists.

						if(review'qualifiers']['P400' ~= nil and review'qualifiers']['P400'][1 ~= nil) then

							-- Try to match based on QID.

							local reviewSysId = review'qualifiers']['P400'][1]['datavalue']['value']['numeric-id'];

							if(systemId == reviewSysId) then

								table.insert(reviewsToPrint,review);

							else 

								-- If that failed, try to match based on label.

								local systemName = mw.wikibase.label('Q'..reviewSysId);

								if(systemName ~= nil and string.upper(systemName) == system) then

									table.insert(reviewsToPrint,review);

								end;

							end;

						end;

					end;

				end;

			end;    		

		end;

	

		-- Sort the array by platform label.

    	table.sort(reviewsToPrint, sortByPlatform);

    	

    	-- If a system was not specified, showSystem has defaulted to true. If this title only has one platform and one review, we will turn it off.

    	-- Note: If the title has zero or more platforms defined, we leave showSystem on. We are unable to determine if this is a single-platform game.

    	--if((system == nil or system == "") and #reviewsToPrint == 1 and entity['claims']['P400'] ~= nil and #entity['claims']['P400'] == 1) then

    	-- Simplifying this based on discussion at [Template:Video game reviews]. If there's only one review, don't display system unless explicitly requested.

    	if((system == nil or system == "") and #reviewsToPrint == 1) then

    		showSystem = false;

    	end;



		-- Print the reviews

    	for i,review in ipairs(reviewsToPrint) do

    		ret = ret .. printReviewRow(frame, review);

		end;

	end;



	if(ret ~= "") then

		ret = string.sub(ret, 1, -7);

	elseif(not showUpdateLink) then

		ret = nil;

	end;

	

	-- Add edit link at end if showUpdateLink is on.

	if(showUpdateLink) then 

		ret = ret .. p.getUpdateLink();

	end;	



	return ret;

end;



return p