--helpers
local module_frame = mw.getCurrentFrame()
tier_text = {module_frame:expandTemplate{title="Low"},
module_frame:expandTemplate{title="Common"},
module_frame:expandTemplate{title="Moderate"},
module_frame:expandTemplate{title="High"},
module_frame:expandTemplate{title="Epic"},
module_frame:expandTemplate{title="Legendary"},
module_frame:expandTemplate{title="Artifact"},
}
metal_text = {"Bronze", "Iron", "Steel", "Cobalt", "Bloodsteel", "Orichalcum"}
wood_text = {"Wooden", "Bamboo", "Hardwood", "Ironwood", "Frostwood", "Eldwood"}
upper_first = function(str)
return str:sub(1, 1):upper() .. str:sub(2)
end
str_split_iter = function(str, sep, escape)
if escape then
sep = "%"..sep
end
return string.gmatch(str, "[^"..sep.."]+")
end
format_probability = function(p_flt)
-- format probability string
local p_perc = p_flt * 100
local p_str = tostring(p_perc)
if p_str ~= tostring(math.floor(p_perc)) then
p_str = string.format("%.2f", p_perc)
end
return p_str
end
-- worker function
process = function(frame, txt, entries, depth, p_prev, reuse_p)
local p_order = {}
local hidep
if not reuse_p then
--sort indices by drop chance
for idx, entry in ipairs(entries) do
p_order[idx] = {idx, entry[1]}
end
table.sort(p_order, function(a, b) return a[2] > b[2] end)
hidep = 0
else
-- use entries directly (re-using same probability)
p_order = entries
hidep = 1
end
-- process entries
for _, val in ipairs(p_order) do
-- unpack data
local entry
if not reuse_p then
entry = entries[val[1]]
else
entry = val
end
local p_flt, box
if not reuse_p then
p_flt = entry[1] * p_prev
box = entry[2]
else
p_flt = p_prev
box = entry
end
local box_type = box.t
local box_content = box.c
-- format probability string
local p_str = format_probability(p_flt)
-- handle box type
if box_type == "Item" then
-- single item row
local item = box_content[2]
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootRowVariable',
args={depth=depth, item=item, n=1, p=p_str, hidep=hidep} }
elseif box_type == "Nothing" then
-- single row
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootRowVariable',
args={depth=depth, item="Nothing", p=p_str, hidep=hidep, noimg=1} }
elseif box_type == "MultiDrop" then
-- multidrop
local nested_box = box_content[1]
local nested_type = nested_box.t
local low = box_content[2]
local high = box_content[3]
local n_str
if low == high then
n_str = tostring(low)
else
n_str = string.format("%s - %s", low, high)
end
if nested_type == "Item" then
-- multiple of one item
local item = nested_box.c[2]
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootRowVariable',
args={depth=depth, item=item, n=n_str, p=p_str, hidep=hidep} }
elseif nested_type == "SubTable" then
-- multidrop from nested table
local sub_depth = depth + 1
local sub_box = nested_box.c[1]
local sub_path = nested_box.c[2]
local path_parts = {}
for part in str_split_iter(sub_path, '.', true) do
path_parts[#path_parts+1] = part
end
local item_category = path_parts[3]
local last_part = path_parts[#path_parts]
local name
local collapsed = 1
--material sub tables (hard-code matching for now)
if item_category == "food" then
if last_part == "wild_ingredients" then
name = "Wild Ingredients"
elseif last_part == "prepared" then
name = "Prepared Food"
elseif last_part == "farm_ingredients" then
name = "Farm Ingredients"
else
name = "Food"
end
elseif item_category == "materials" then
local name
if last_part == "gems" then
name = "Gems"
elseif last_part == "underground" then
name = "Underground Materials"
else
name = "Materials"
end
else
name = "Misc"
collapsed = 0
end
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootHeaderVariable',
args={depth=sub_depth, collapsed=collapsed, title=name, n=n_str, p=p_str, hidep=hidep} }
txt = process(frame, txt, sub_box, sub_depth, p_flt, false)
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootEndNested'}
elseif nested_type == "All" then
local sub_depth = depth + 1
local sub_box_content = nested_box.c
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootHeaderVariable',
args={depth=sub_depth, title="All", collapsed=0, p=p_str, hidep=hidep} }
txt = process(frame, txt, sub_box_content, sub_depth, p_flt, true)
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootEndNested'}
-- spacer row approach
-- txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootRowSpacer'}
-- txt = process(frame, txt, box_content, sub_depth, p_flt/100, true)
-- txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootRowSpacer'}
else
-- other case (unhandled for now)
end
elseif box_type == "SubTable" then
local sub_depth = depth + 1
local sub_box = box_content[1]
local sub_path = box_content[2]
local override = false
-- sub table header
local path_sections = {}
for section in str_split_iter(sub_path, '.', true) do
-- for section in string.gmatch(sub_path, "[^%.]+") do
path_sections[#path_sections+1] = section
end
local last_section = path_sections[#path_sections]
local second_last_section = path_sections[#path_sections-1]
local is_modular = false
if last_section ~= nil then
is_modular = "tier" == string.sub(last_section, 1, 4)
end
local idx_tier
if is_modular then
idx_tier = 1 + tonumber(string.sub(last_section, -1, -1))
end
local item_category = path_sections[3]
if item_category == "weapons" then
if not is_modular then
-- non-modular weapon
local name = upper_first(last_section).." Weapon"
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootHeaderVariable',
args={depth=sub_depth, title=name, p=p_str, hidep=hidep} }
else
if second_last_section == "components" then
-- primary component
local name = tier_text[idx_tier].." Primary Component"
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootHeaderVariable',
args={depth=sub_depth, title=name, n=1, p=p_str, hidep=hidep} }
local metal = metal_text[idx_tier]
local wood = wood_text[idx_tier]
local p_wep = string.format("%.2f", 100 * p_flt / 6)
local page = "Modular Crafting" -- page link override doesn't seem to work
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootRowVariable',
args={depth=sub_depth, item=metal.." Sword Blade", p=p_wep, noimg=1, page=page} }
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootRowVariable',
args={depth=sub_depth, item=metal.." Axe Head", p=p_wep, noimg=1, page=page} }
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootRowVariable',
args={depth=sub_depth, item=metal.." Hammer Head", p=p_wep, noimg=1, page=page} }
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootRowVariable',
args={depth=sub_depth, item=wood.." Bow Limbs", p=p_wep, noimg=1, page=page} }
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootRowVariable',
args={depth=sub_depth, item=wood.." Sceptre Shaft", p=p_wep, noimg=1, page=page} }
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootRowVariable',
args={depth=sub_depth, item=wood.." Staff Shaft", p=p_wep, noimg=1, page=page} }
-- **TODO**
-- MediaWiki template for primary components of parameterised tier (sans outer header and end)
override = true
elseif second_last_section == "secondary" then
-- secondary component
-- **TODO**
-- MediaWiki template for secondary components (sans outer header and end)
override = true
else
-- modular weapon
local name = tier_text[idx_tier].." Weapon"
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootHeaderVariable',
args={depth=sub_depth, title=name, n=1, p=p_str, hidep=hidep} }
local metal = metal_text[idx_tier]
local wood = wood_text[idx_tier]
local p_wep = string.format("%.2f", 100 * p_flt / 6)
local page = "Weapons" -- page link override doesn't seem to work
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootRowVariable',
args={depth=sub_depth, item=metal.." Sword", p=p_wep, noimg=1, page=page} }
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootRowVariable',
args={depth=sub_depth, item=metal.." Axe", p=p_wep, noimg=1, page=page} }
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootRowVariable',
args={depth=sub_depth, item=metal.." Hammer", p=p_wep, noimg=1, page=page} }
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootRowVariable',
args={depth=sub_depth, item=wood.." Bow", p=p_wep, noimg=1, page=page} }
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootRowVariable',
args={depth=sub_depth, item=wood.." Sceptre", p=p_wep, noimg=1, page=page} }
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootRowVariable',
args={depth=sub_depth, item=wood.." Staff", p=p_wep, noimg=1, page=page} }
-- **TODO**
-- MediaWiki template for complete modular weapon of parameterised tier (sans outer header and end)
override = true
end
end
elseif item_category == "armor" then
if not is_modular then
-- specific armor
local name = upper_first(last_section)
if name == "Cloth" then
name = tier_text[1].." Armor"
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootHeaderVariable',
args={depth=sub_depth, title=name, n=1, p=p_str, hidep=hidep} }
else
if name == "Wool" then
name = "Woolen"
elseif name == "Silk" then
name = "Silken"
elseif name == "Lifecloth" then
name = "Druid"
elseif name == "Plate" then
name = "Primal"
end
local image = name.." Armor Set.png"
name = name.." Armor"
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootHeaderVariable',
args={depth=sub_depth, title=name, image=image, p=p_str, hidep=hidep} }
end
else
-- modular armor tier
local name = tier_text[idx_tier].." Armor"
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootHeaderVariable',
args={depth=sub_depth, title=name, n=1, p=p_str, hidep=hidep} }
end
elseif item_category == "food" then
--food sub tables (hard-code matching for now)
local name
if last_section == "wild_ingredients" then
name = "Wild Ingredients"
elseif last_section == "prepared" then
name = "Prepared Food"
elseif last_section == "farm_ingredients" then
name = "Farm Ingredients"
else
name = "Food"
end
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootHeaderVariable',
args={depth=sub_depth, collapsed=1, title=name, p=p_str, hidep=hidep} }
elseif item_category == "materials" then
local name
if last_section == "gems" then
name = "Gems"
elseif last_section == "underground" then
name = "Underground Materials"
else
name = "Materials"
end
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootHeaderVariable',
args={depth=sub_depth, collapsed=1, title=name, p=p_str, hidep=hidep} }
else
-- non weeapon/armor sub table
-- local name = last_section..second_last_section
local name = "Misc"
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootHeaderVariable',
args={depth=sub_depth, collapsed=0, title=name, p=p_str, hidep=hidep} }
end
-- sub table rows (unless direct template was used)
if not override then
txt = process(frame, txt, sub_box, sub_depth, p_flt, false)
end
-- sub table end
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootEndNested'}
elseif box_type == "All" then
local sub_depth = depth + 1
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootHeaderVariable',
args={depth=sub_depth, title="All", collapsed=0, p=p_str, hidep=hidep} }
txt = process(frame, txt, box_content, sub_depth, p_flt, true)
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootEndNested'}
-- spacer row approach
-- txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootRowSpacer'}
-- txt = process(frame, txt, box_content, sub_depth, p_flt/100, true)
-- txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootRowSpacer'}
else
-- fallback (shouldn't reach)
local name = "Error"
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootRowVariable',
args={depth=depth, item=name, p=0, hidep=1, noimg=1} }
end
end
return txt
end
-- mainline code
local module = {}
local drops_json = mw.text.jsonDecode(module_frame:expandTemplate{title="User:Horblegorble/Sandbox/Template/LootData"})
local sources_json = mw.text.jsonDecode(module_frame:expandTemplate{title="User:Horblegorble/Sandbox/Template/SourceData"})
module.drops = function(frame)
-- generates drops table for an entity
-- handle args
-- local entity = tostring(frame.args[1])
-- local loot_page = "User:"..entity.."Loot"
-- local json = mw.text.jsonDecode(frame:expandTemplate{title=loot_page})
-- get entity name
-- local page_sections = {}
-- for idx, section in ipairs(str_split_iter(frame.args[1])) do
-- page_sections[idx] = section
-- end
-- local entity = page_sections[#page_sections]
local entity = frame.args[1]
local export_date = drops_json[1]
-- find data entry for entity
local data = false
for _, entry in ipairs(drops_json[2]) do
if entry[1] == entity then
data = entry[2]
break
end
end
if not data then
return "entity not found in drops data"
end
-- table start
local txt = frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootHeaderVariable'}
-- do the work
txt = process(frame, txt, data, 0, 1.0, false)
-- table end
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/LootEnd',
args={date=export_date} }
return txt
end
module.sources = function(frame)
-- generates table of entity drop sources for an item
local item = frame.args[1]
local export_date = sources_json[1]
local data = false
for _, entry in ipairs(sources_json[2]) do
if entry[1] == item then
data = entry[2]
break
end
end
if not data then
return frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/NoSources',
args={date=export_date}}
end
-- sort entries by average
local p_order = {}
for idx, entry in ipairs(data) do
p_order[idx] = {idx, entry[8]}
end
table.sort(p_order, function(a, b) return a[2] > b[2] end)
-- table start
local txt = frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/SourceHeader',
args={sortable=1}}
-- do the work
for _, val in ipairs(p_order) do
entry = data[val[1]]
local entity = entry[1]
if entity ~= "" then
local variant = entry[2]
local amount = entry[3]
local n_sort = entry[4]
local chance = entry[5]
local rolls = entry[6]
local p_sort = entry[7] * 100
-- local avg = entry[8]
if amount[1] == amount[2] then
amount = tostring(amount[1])
else
amount = string.format("%i-%i", amount[1], amount[2])
end
if rolls[1] == rolls[2] then
if rolls[1] == 1 then
rolls = ''
else
rolls = tostring(rolls[1])
end
else
rolls = string.format("%i-%i", rolls[1], rolls[2])
end
local p_str = format_probability(chance)
local page = ""
if string.sub(entity, 1, 1) == "#" then
page = "Containers"
entity = string.sub(entity, 2, -1)
end
txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/SourceRow',
args={entity=entity, variant=variant, n=amount, n_sort=n_sort,
p=p_str, rolls=rolls, p_sort=p_sort, page=page}}
end
end
-- table end
local txt = txt .. frame:expandTemplate{title='User:Horblegorble/Sandbox/Template/SourceEnd',
args={date=export_date} }
return txt
end
return module
