Module:Dice

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

-- Module:Dice
-- Processes dice input strings and outputs HTML spans with roll links, templates, and data attributes

local p = {}
local html = mw.html

-- Trim whitespace from both ends
local function trim(s)
  return (s:gsub("^%s*(.-)%s*$", "%1"))
end

-- Split the full roll string into parts and operators (+, -)
local function splitParts(s)
  local parts = {}
  local i = 1
  while i <= #s do
    local c = s:sub(i,i)
    if c == "+" or c == "-" then
      table.insert(parts, c)
      i = i + 1
    elseif c == " " then
      i = i + 1
    else
      local j = s:find("[%+%-]", i)
      if j then
        table.insert(parts, trim(s:sub(i, j-1)))
        i = j
      else
        table.insert(parts, trim(s:sub(i)))
        break
      end
    end
  end
  return parts
end

-- Capitalize first letter
local function capitalize(s)
  return (s:gsub("^%l", string.upper))
end

-- Parse an individual part into prefix, dice, and qualifiers
local function parsePart(part)
  local s = trim(part)
  local prefixCount, prefixFlag, diceCount, diceFaces, qualifiers = nil, nil, nil, nil, {}

  -- Extract qualifiers in [brackets]
  local main, qualStr = s:match("^(.-)%[([^%]]+)%]$")
  if qualStr then
    s = trim(main)
    for q in qualStr:gmatch("([^ ]+)") do
      table.insert(qualifiers, q)
    end
    -- Sort qualifiers alphabetically
    table.sort(qualifiers)
  end

  -- Extract parenthesized prefix, e.g. (2@p), (@f), or (2 * @p)
  local rawPref = s:match("^%(([^)]+)%)")
  if rawPref then
    -- Try to match "<number>*@<flag>"
    local num, flag = rawPref:match("^(%d+)%s*%*%s*@(%a+)$")
    if not num then
      -- Try "<number>@<flag>" or just "@<flag>"
      num, flag = rawPref:match("^(%d*)@(%a+)$")
    end
    prefixCount = tonumber(num) or 1
    prefixFlag = flag
    s = s:gsub("^%b()", "", 1)
  else
    -- Pure prefix without dice
    local pure = s:match("^@(%a+)$")
    if pure then
      prefixCount = 1
      prefixFlag = pure
      s = ""
    end
  end

  -- Extract dice, e.g. 2d6 or d8
  if s:match("^%d*d%d+") then
    local cnt, faces = s:match("^(%d*)d(%d+)")
    diceCount = tonumber(cnt) or 1
    diceFaces = faces
  end

  return prefixCount, prefixFlag, diceCount, diceFaces, qualifiers
end

-- Main entry for #invoke:Dice|d|<roll>|<type>
function p.d(frame)
  local fullRoll = frame.args[1] or ""
  local typ = frame.args[2] or "none"
  local parts = splitParts(fullRoll)
  local quickParts, content = {}, {}

  for _, part in ipairs(parts) do
    if part == "+" or part == "-" then
      table.insert(quickParts, part)
      table.insert(content, " " .. part .. " ")
    else
      local prefixCount, prefixFlag, diceCount, diceFaces, qualifiers = parsePart(part)
      -- Build quick-roll (ignore flags)
      if diceCount then
        table.insert(quickParts, diceCount .. "d" .. diceFaces)
      end

      -- Prefix template for non-'none' types
      if prefixFlag and typ ~= "none" and diceCount then
        local tpl = "{{" .. string.upper(prefixFlag)
        if prefixCount and prefixCount ~= 1 then tpl = tpl .. "|" .. prefixCount end
        tpl = tpl .. "}}"
        table.insert(content, frame:preprocess(tpl))
      end

      -- Build dice link
      if diceCount then
        -- Always show count
        local displayDice = diceCount .. "d" .. diceFaces
        local linkLabel = displayDice
        -- Embed prefix in label when type='none' and no qualifiers
        if prefixFlag and typ == "none" and #qualifiers == 0 then
          local prefixLabel = string.upper(prefixFlag)
          if prefixCount and prefixCount > 1 then
            prefixLabel = prefixCount .. prefixLabel
          end
          linkLabel = prefixLabel .. displayDice
        end
        table.insert(content, string.format("[https://dice.run/#/d/%dd%s %s]", diceCount, diceFaces, linkLabel))
      elseif prefixFlag and not diceCount then
        -- Pure prefix outputs template always
        table.insert(content, frame:preprocess("{{" .. string.upper(prefixFlag) .. "}}"))
      end

      -- Qualifier labels
      for _, q in ipairs(qualifiers) do
        table.insert(content, frame:preprocess(" {{L|" .. capitalize(typ) .. "|" .. capitalize(q) .. "}}"))
      end
    end
  end

  -- Assemble quick-roll and content
  local quickRoll = table.concat(quickParts, " ")
  local inner = table.concat(content)

  -- Generate span
  local span = html.create('span')
    :addClass('dice')
    :attr('data-full-roll', fullRoll)
    :attr('data-quick-roll', quickRoll)
    :attr('data-type', typ)
    :wikitext(inner)

  return span:allDone()
end

return p