Module:Database/data

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

local species = mw.text.jsonDecode(mw.title.new('Data:Species'):getContent())
local moves = mw.text.jsonDecode(mw.title.new('Data:Moves'):getContent())

local db = {
	species = {
		by_name = {},
		by_index = {},
		unnumbered = {},
		by_tag = {},
		by_type = {},
		min_index = 1,
		max_index = 120
	},
	moves = {
		by_name = {},
		by_tag = {},
		by_type = {}
	}
}

db.species.unknown = {
	name = "Unknown",
	description = "an unknown species",
	elemental_type = "Typeless",
	bestiary_bios = {""},
	evolves_from = {},
	evolves_to = {},
	habitats = {},
	moves = { initial = {}, tags = {}, upgrades = {} },
	stats = {
		max_ap = 0,
		max_hp = 0,
		melee_attack = 0,
		melee_defense = 0,
		move_slots = 0,
		ranged_attack = 0,
		ranged_defense = 0,
		record_rate = 0,
		speed = 0
	}
}

db.moves.unknown = {
	accuracy = 100,
	ap_cost = 0,
	can_be_copied = true,
	category = "Miscellaneous",
	crit_damage_percent = 150,
	crit_rate = 0.0625,
	default_target = "DEFAULT_TARGET_ENEMY",
	description = "Unknown move.",
	elemental_type = "Typeless",
	is_passive_only = false,
	name = "Unknown",
	physicality = "MELEE",
	power = 0,
	priority = 0,
	tags = {},
	target_type = "TARGET_ONE"
}

function species_sort_index(species)
	local i = species.bestiary_index
	if i == nil or i < 0 then
		return 99999
	else
		return i
	end
end

function cmp_species(a, b)
	local ai = species_sort_index(a)
	local bi = species_sort_index(b)
	return ai < bi
end

function cmp_moves(a, b)
	return a.name < b.name
end

function normalize_species(species)
	species.trimmed_description = species.description
	if string.sub(species.trimmed_description, 1, 3) == "an " then
		species.trimmed_description = string.sub(species.trimmed_description, 4)
	elseif string.sub(species.trimmed_description, 1, 2) == "a " then
		species.trimmed_description = string.sub(species.trimmed_description, 3)
	end
	
	if species.bestiary_index ~= nil and species.bestiary_index < 0 then
		species.bestiary_index = nil
	end
	
	if species.bestiary_index ~= nil then
		species.index = string.format("%03d", species.bestiary_index)
	else
		species.index = "???"
	end
	
	for i,name in ipairs(species.evolves_from) do
		if name == "" then
			species.evolves_from[i] = "Unknown"
		end
	end
	for i,evo in ipairs(species.evolves_to) do
		if evo.evolved_form == "" then
			evo.evolved_form = "Unknown"
		end
		evo.unevolved_form = species.name
	end
	for i,habitat in ipairs(species.habitats) do
		habitat = string.gsub(habitat, "[\"“”]", "")
		local pos = string.find(habitat, ",")
		if pos ~= nil then
			habitat = string.sub(habitat, 1, pos - 1)
		end
		species.habitats[i] = habitat
	end
	
	table.insert(species.moves.tags, "any")
	table.insert(species.moves.tags, string.lower(species.elemental_type))
end

function find_evo(evo_from, evo_to)
	for _,evo in pairs(evo_from.evolves_to) do
		if evo.evolved_form == evo_to.name then
			return evo
		end
	end
	return nil
end

function late_fix_species(species)
	local new_evo_from = {}
	for _,evo_from in ipairs(species.evolves_from) do
		local evo_from_species = db.species.by_name[evo_from]
		if evo_from_species ~= nil then
			local evo = find_evo(evo_from_species, species)
			if evo ~= nil then
				table.insert(new_evo_from, evo)
			end
		end
	end
	species.evolves_from = new_evo_from
end

normalize_species(db.species.unknown)

for _,s in ipairs(species) do
	normalize_species(s)
	db.species.by_name[s.name] = s
	
	if s.bestiary_index ~= nil then
		db.species.by_index[s.bestiary_index] = s
		db.species.min_index = math.min(db.species.min_index, s.bestiary_index)
		db.species.max_index = math.max(db.species.max_index, s.bestiary_index)
	else
		table.insert(db.species.unnumbered, s)
	end
	
	if db.species.by_type[s.elemental_type] == nil then
		db.species.by_type[s.elemental_type] = {}
	end
	table.insert(db.species.by_type[s.elemental_type], s)
	
	for _,tag in ipairs(s.moves.tags) do
		if db.species.by_tag[tag] == nil then
			db.species.by_tag[tag] = {}
		end
		table.insert(db.species.by_tag[tag], s)
	end
end
table.sort(db.species.unnumbered, cmp_species)
for _,species in pairs(db.species.by_tag) do
	table.sort(species, cmp_species)
end
for _,species in pairs(db.species.by_type) do
	table.sort(species, cmp_species)
end
for _,species in pairs(db.species.by_name) do
	late_fix_species(species)
end

for _,m in ipairs(moves) do
	db.moves.by_name[m.name] = m
	
	if db.moves.by_type[m.elemental_type] == nil then
		db.moves.by_type[m.elemental_type] = {}
	end
	table.insert(db.moves.by_type[m.elemental_type], m)
	
	for _,tag in ipairs(m.tags) do
		if db.moves.by_tag[tag] == nil then
			db.moves.by_tag[tag] = {}
		end
		table.insert(db.moves.by_tag[tag], m)
	end
end
for _,moves in pairs(db.moves.by_type) do
	table.sort(moves, cmp_moves)
end
for _,moves in pairs(db.moves.by_tag) do
	table.sort(moves, cmp_moves)
end

return db