Module:Img

---

local p = {}
local findId = require("Module:FindId")

local dataModuleNames = {
	item = "Module:Items/data",
	enchantment = "Module:Enchantment/data",
	location = "Module:Location/data",
	monster = "Module:Monsters/data",
	ability = "Module:Abilities/data",
	other = "Module:OtherImages/data"
}

local idListType = {
	[1] = {
		[1] = "item",
		[2] = "itemImage",
		[3] = "itemIcon"
	},
	[2] = {
		[1] = "monster",
		[2] = "image"
	},
	[3] = {
		[1] = "enchantment",
		[2] = "buffIcon"
	},
	[4] = {
		[1] = "location",
		[2] = "locationImage"
	},
	[5] = {
		[1] = "ability",
		[2] = "abilityImage"
	}
}

local loadedDataModules = {}

local alternativeNames = {
	fishing_enchantment = 23,
	cooking_enchatment = 24,
	crafting_enchantment = 25,
	runecrafting_enchantment = 27,
	farming_enchantment = 39,
	fishing = 0,
	cooking = 0,
	crafting = 0,
	farming = 0,
	runecrafting = 0
}

--
-- Loads data modules
--
-- @param dataType {string}
-- @return {data table}
--
function p.loadData (dataType)
	local moduleName = dataModuleNames[dataType]
	if loadedDataModules[moduleName] == nil then
		loadedDataModules[moduleName] = mw.loadData(moduleName)
	end
	return loadedDataModules[moduleName]
end

--
-- Helps capitalize the first letter of each word
--
-- @param first {string}
-- @param rest {string}
-- @return {strings}
--
local function tchelper(first, rest)
    return first:upper()..rest:lower()
end

--
-- Capitalize the first letter of a word in a string
--
-- @param s {string}
-- @return {string}
--
local function capitalize(s)
	s = s:gsub("(%a)([%w_']*)", tchelper):gsub(" Of "," of "):gsub(" The "," the "):gsub("Ii","II")
	return s
end

--
-- Hyphenates name
--
-- @param name {string}
-- @return {string}
--
local function hyphenateName(name)
    local lName = name:lower()
    return lName:gsub('^%s*(.-)%s*$', '%1'):gsub("%s+", "_")
end

--
-- Generates a full url
--
-- @param url {string}
-- @return {string}
--
local function fullUrl(url)
	local newUrl = url
	if url:sub(1,5) == "https" then
		return newUrl
	end

	if url:sub(1,1) ~= "/" then
		newUrl = "/" .. newUrl
	end

	newUrl = "https://www.play.idlescape.com" .. newUrl
	return newUrl
end

-- 
-- Finds input's id and input's type
--
-- @param name {string}
-- @return {number}
-- @return {string}
local function findInputId(name)
	local id
	local inputType
	for index, value in ipairs(idListType) do
		id = findId._findId({name, value[1]})
		if type(id) == "number" then
			inputType = value[1]
			return id , inputType
		end
	end
	return 0
end

--
-- Finds arg's image url
--
-- @param id {number}
-- @param inputType {string}
-- @return {string}
--
local function findImageUrl(id, inputType)
	local imageUrl
	for index, value in ipairs(idListType) do
		if value[1] == inputType then
			if inputType == "item" then
				if p.loadData(inputType)[tostring(id)][value[3]] then
					return fullUrl(p.loadData(inputType)[tostring(id)][value[3]])
				end
			end
			imageUrl = fullUrl(p.loadData(inputType)[tostring(id)][value[2]])
			return imageUrl
		end
	end
	return 0
end

--
-- Generates string with page link and <img> tag
--
-- @param name {string}
-- @param url {string}
-- @param width {number}
-- @param height {number}
-- @param word {number}
-- @return {string}
--
local function image(name, url, width, height, word)
	local s = url
	s = "src=\"".. s .. "\""
	s = s .. " alt=\"" .. capitalize(name) .. "\""
	s = s .. " width=\"" .. width .. "px\""
	s = s .. " height=\"" .. height .. "px\""
	s = '<img ' .. s .. '>'
	if tonumber(word) == 1 then
		s = s .. "&nbsp;" .. capitalize(name)
	end
	s = '[[' .. capitalize(name) .. '|' .. s .. ']]'
	return s
end

--
-- Finds otherimage
--
-- @param name {string}
-- @return {string}
--
local function findOtherImage(name)
	local hName = hyphenateName(name)
	for key, value in pairs(p.loadData("other")) do
		if hName == key then
			return value
		end
	end
	return 0
end

--
-- Generates string with page link and <img> tag
--
-- @param name {string}
-- @param url {string}
-- @param width {number}
-- @param height {number}
-- @param word {number}
-- @return {string}
--
local function otherImage(name, url, width, height, word)
	local s = url
	s = s .. "\"" .. width .. "px\""
	s = s .. " height=\"" .. height .. "px\">"
	if tonumber(word) == 1 then
		s = s .. "&nbsp;".. capitalize(name)
	end
	s = s .. ']]'
	return s
end

--
-- Main img method accessed through #invoke
--
-- @param frame {table}
-- @return {string}
--
function p.img(frame)
	return p._img(frame:getParent().args)
end

--
-- img method to allow it to be called by other modules
--
-- @param _args {table}
-- @return {string}
--
function p._img(_args)
	local width = _args[2]
	local height = _args[3]
	local id
	local inputType
	local url
	local word = _args["word"]
	local otherimage
	local s
	local error
	if not width then
		width = 20
	end
	if not height then
		height = width
	end
	if not _args[1] then
		return "Couldn't get name"
	end
	if alternativeNames[hyphenateName(_args[1])] then
		id = alternativeNames[hyphenateName(_args[1])]
		inputType = "enchantment"
	else
		id, inputType = findInputId(_args[1])
	end
	if id == 0 then
		otherimage = findOtherImage(_args[1])
	end
	if otherimage == 0 then
		error = "<span class=\"rt-commentedText tooltip tooltip-dotted\" title=\"Given name doesn't exist in any of the data modules, Check spelling or update data modules.\">".. _args[1] .."</span>"
		return error
	elseif otherimage then
		s = otherImage(_args[1], otherimage, width, height, word)
		return s
	end
	url = findImageUrl(id, inputType)
	if url == 0 then
		return "couldn't find image url."
	end
	s = image(_args[1], url, width, height, word)
	return s
end

return p