Module:EncounterTable

---

local p = {}

--a lot taken from lol gamepedia
function p.gsplit( text, pattern, plain )
	if not pattern then pattern = '%s*,%s*' end
	local s, l = 1, text:len()
	return function ()
		if s then
			local e, n = text:find( pattern, s, plain )
			local ret
			if not e then
				ret = text:sub( s )
				s = nil
			elseif n < e then
				-- Empty separator!
				ret = text:sub( s, e )
				if e < l then
					s = e + 1
				else
					s = nil
				end
			else
				ret = e > s and text:sub( s, e - 1 ) or ''
				s = n + 1
			end
			return ret
		end
	end, nil, nil
end

function p.split(text, pattern, plain)
	if not text then
		return {}
	end
	local ret = {}
	for m in gmatch(text, pattern, plain) do
		ret[#ret+1] = m
	end
	return ret
end

function p.splitArgs(input, fieldlist, sep)
    if not input or input == '' then return end
    sep = sep or '%s*;;;%s*'
    local result = {}
    local inputTbl = p.split(input,sep)
    for i, v in ipairs(fieldlist) do
        if not inputTbl[i] then
            error(('Missing parameter %s - maybe wrong child template?'):format(v))
        end
        if inputTbl[i] ~= '' then 
            result[v] = inputTbl[i]
        end
    end
    return result
end

function p.splitEncounterTableArgs(row)
    local TABLE_ARGS = { 'table', 'item', 'quantity', 'weight' }
    return p.splitArgs(row, TABLE_ARGS)
end


function p.start(frame)
 local tblName = frame.args['name'] or ''
 local result = {}
 local totalWeight = 0
 for i, row in ipairs(frame.args) do
  local newrow = p.splitEncounterTableArgs(row)
  result[i] = newrow
  totalWeight = totalWeight + tonumber(newrow['weight'])
 end
 
  --second pass: first row is either an item of a table (where the output is Item from (table)
  local finalTable = {}
  for i, row in ipairs(table) do
    if row['table'] == 'none' then
      if row['item'] == 'none' then
        error('A table or item must be specified!')
      else
        finalTable['item'] = '[[' .. row['item'] .. ']]'
      end
    elseif row['item'] ~= 'none' then
      error('A table and item cannot be specified at the same time. Choose one ya silly')
    else 
      finalTable['item'] = 'Item from [[EncounterTable:' .. row['table'] .. '|' .. row['table'] .. ']]'
    end
   -- dice tooltip quantity
    finalTable['quantity'] = frame:expandTemplate{title='Dice tooltip', args={row['quantity']}}
    finalTable['weight'] = row['weight']
   -- calculate final chance
    finalTable['chance'] = row['weight']/totalWeight
  end
--format table. 
  return p.formatTable(finalTable)
end

function p.formatTable(final)
  local tblheader = '<tr><th>Item</th><th>Quantity</th><th>Weight</th><th>Chance</th></tr>'
  local tr = ''
  for _, v in ipairs(final) do
        tr = tr ..  ' <tr><td>' .. v['item'] .. '</td><td>' .. v['quantity'] .. '</td><td>' .. v['weight'] .. '</td><td>' .. v['chance'] .. '</td></tr>'
  end
    return '<table class="wikitable">' .. tblheader  .. tr .. '</table>'
end

return p