Editing Module:Test

DefInfo = {}
local wiki = {}

local function runTime()
  local clock = string.format("%i", os.clock() * 1000)
  return clock
end

------------------------------------------------------------------
-- deal with differences between MediaWiki and dev environments --
------------------------------------------------------------------

if mw then
  log = mw.log
  logObject = mw.logObject

  local timeDataEnd = runTime()
  log(string.format('@%ims, data loaded in %ims', timeDataEnd, timeDataEnd - timeDataStart))

  Data   = mw.loadData('Module:Test/data')
  
  local timeDataEnd = runTime()
  log(string.format('@%ims, data loaded in %ims', timeDataEnd, timeDataEnd - timeDataStart))
  
  Util   = require("Module:Test/lib/util")
  Search = require("Module:Test/lib/search")
  VF     = require("Module:Test/data/virtual")

  log(string.format('@%ims, modules loaded', runTime()))

  frame = mw and mw.getCurrentFrame()
else
  logStore = {}

  log = function(str)
    table.insert(logStore, str)
  end

  logObject = function(obj, prefix)
    if prefix then
      assert(type(prefix) == "string")
      table.insert(logStore, prefix .. " = " .. Inspect(obj))
    else
      table.insert(logStore, Inspect(obj))
    end
  end

  function pp(tbl, title) -- pretty print tables
    Util.hl(title)
    print(Inspect(tbl))
  end

  local timeDataStart = runTime()

  Data    = require "data/data"

  local timeDataEnd = runTime()
  log(string.format('@%ims, data loaded in %ims', timeDataEnd, timeDataEnd - timeDataStart))

  Util    = require "lib/util"
  Search  = require "lib/search"
  Inspect = require "lib/inspect"
  VF      = require "data/virtual"

  log(string.format('@%ims, modules loaded', runTime()))
end

-----------------------
-- private functions --
-----------------------

function DefInfo.vardefine(name, value)
  assert(name, "vardefine: missing argument #1 (variable to definePrefix)")
  assert(type(name) == "string", string.format("vardefine: bad argument #1 (string expected, got %s)", type(name)))
  assert(value, "vardefine: missing argument #2 (value to assign)")
  assert(type(value) == "string" or type(value) == "number" or type(value) =="boolean", string.format("vardefine: bad argument #2 (string, number or boolean expected, got %s)", type(value)))
  frame:callParserFunction('#vardefine', name, value)
end

function DefInfo.expandDef(def, runMe)
  if not runMe then return nil end
  local vFuncs = VF
  for fName,func in pairs(vFuncs) do
    if func(def) then
      log(string.format('@%ims, expandDef: %s expanded with %s', runTime(), def.defName, fName))
    end
  end
end

function DefInfo.mergeParents(baseDef, ignoreKeys)
  local ancestorIDs = {}
  local mergedDef = {}
  local def = baseDef

  while def._.ParentName do
    local parentID = def._.DefCategory .. ":" .. def._.ParentName
    table.insert(ancestorIDs, parentID)
    def = Data[parentID]
  end

  ancestorIDs = Util.table.reverse(ancestorIDs)
  table.insert(ancestorIDs, baseDef._.DefCategory .. ":" .. baseDef.defName)

  for _,parentID in ipairs(ancestorIDs) do
    Util.table.overwrite(mergedDef, Data[parentID], ignoreKeys)
  end

  return mergedDef
end

function DefInfo.getDef(defID, expandVF)
  if expandVF ~= false then expandVF = true end
  local ignoreKeys = {"Abstract", "Name", "ParentName"}
  local baseDef
  local def

  if not defID then return nil end

  for _,def in pairs(Data) do
    if def.defName == defID then
      baseDef = def
      break
    elseif string.upper(def.label or '') == string.upper(defID) then
      baseDef = def
      break
    end
  end

  if not baseDef then return nil end

  def = DefInfo.mergeParents(baseDef, ignoreKeys)

  DefInfo.expandDef(def, expandVF)

  return def
end

local function setPrefix(tbl, parentKey)
  local mt = getmetatable(tbl) or {}

  for k,v in pairs(tbl) do
    local prefix = parentKey .. "_" .. k
    if type(v) == 'table' then
      setPrefix(v, prefix)
    else
      mt[k] = prefix
    end
  end

  setmetatable(tbl, mt)
end

local function definePrefixed(tbl)
  for k,v in pairs(tbl) do
    if type(v) ~= 'table' then
      local mt = getmetatable(tbl)
      log(string.format('%s = %s', mt[k], tostring(v)))
      if mw then DefInfo.vardefine(mt[k], v) end
    else
      definePrefixed(v)
    end
  end
end

----------------------
-- public interface --
----------------------

function wiki.getDefName(frame)
  local defName
  local label = frame.args[1]

  if not label then
    logObject(frame.args, string.format("getDefName @ %ims: missing argument #1 (label)\nframe.args", runTime()))
    return nil
  end

  for defID,def in pairs(Data) do
    if string.upper(def.label or "") == string.upper(label) then
      defName = def.defName
    end
  end

  if not defName then
    logObject(frame.args, string.format("getDefName @ %ims: '%s' not found\nframe.args", runTime(), label))
    return nil
  end

  log(string.format("@%ims, getDefName: retrieved defName", runTime()))
  return defName
end

function wiki.count(frame)
  local query = wiki.query(frame)
  if queried then
    return #queried
  end
end

function wiki.query(frame)

  local argLen = Util.table.count(frame.args, "number") -- #frame.args won't work as expected, check the doc

  if not frame.args['defName'] and not frame.args['label'] then
    logObject(frame.args, string.format('query @ %ims: missing an identifying argument (defName or label)\nframe.args', runTime()))
    return nil
  end

  local def = DefInfo.getDef(frame.args['defName']) or DefInfo.getDef(frame.args['label'])

  if not def then
    logObject(frame.args, string.format("query @ %ims: Def not found\nframe.args", runTime()))
    return nil
  end

  if def and argLen == 0 then
    logObject(def, string.format("['%s:%s'] @ %ims", def._.DefCategory, def.defName, runTime()))
    return nil
  end

  local prune = def

  for i,arg in ipairs(frame.args) do -- arguments
    arg = tonumber(arg) or arg -- frame.args are always strings on MediaWiki so convert back the numbers

    if i == argLen and frame.args["sibling"] then
      prune = Search.find({nil, frame.args["sibling"]} , prune)
      if not prune then
        logObject(frame.args, string.format("query @ %ims: bad argument 'sibling' ('%s' not found')\nframe.args", runTime(), frame.args["sibling"]))
        return nil
      else
        prune = Search.meta.parent.table[arg]
        if not prune then
          logObject(frame.args, string.format("query @ %ims: bad argument #%i ('%s' is not a sibling of '%s')", runTime(), i, arg, frame.args["sibling"]))
          return nil
        end
      end
    end

    if i < argLen or i == argLen and not frame.args["sibling"] then
      prune = Search.find(arg, prune)
      if not prune then
        logObject(frame.args, string.format("query @ %ims: bad argument #%i ('%s' not found)\nframe.args", runTime(), i, frame.args[i]))
        return nil
      else
        if type(prune) ~= 'table' and i < argLen then
          log(string.format("query @ %ims: warning Def ['%s'] argument #%i ('%s' returns a value, all extra arguments ignored)", runTime(), def['label'], i, frame.args[i]))
          return prune
        end
      end
    end

  end -- for arguments

  if type(prune) == "table" then
    log(string.format("@%ims, query: table vardefined", runTime()))
    setPrefix(prune, frame.args[argLen])
    definePrefixed(prune)
    queried = prune
    return nil
  end

  log(string.format("@%ims, query: %s printed", runTime(), type(prune)))
  return prune

end
-- {{User:Dr. Strangelove/Template:Infobox/sandbox|{{{movespeed|}}}|statBases|MoveSpeed|title=Some movin|SMWP=Move Speed Base}}
-- {{#invoke:Test|print|{{SUBPAGENAME}}}}
function wiki.print(frame)
  local subpagename = frame.args[1]
  local pFrame = frame:getParent()

  if not pFrame.args then return "no arguments passed" end
  if not pFrame.args[1] then return "missing argument #1" end
  if not pFrame.args.title then return "missing named argument 'title'" end

  local ibPropValue = pFrame.args[1]

  local qFrame = {
    args = {}
  }

  for i,v in ipairs(pFrame.args) do
    if i > 1 then
      table.insert(qFrame.args, v)
    end
  end

  qFrame.args.label = 'hare'

  mw.logObject(qFrame, 'qFrame')

  local q = wiki.query(qFrame)

  mw.logObject(q, 'q')
  mw.logObject(pFrame.args.SMWP, 'pFrame.args.SMWP')
  mw.logObject(ibPropValue, 'ibPropValue')

  if pFrame.args.SMWP and ibPropValue then
    frame:callParserFunction('#set', pFrame.args.SMWP .. '=' .. ibPropValue)
  end

  local sOutputText

  if ibPropValue == '' then
    sOutputText = string.format(';%s\n:%s', pFrame.args.title, q)
  else
    sOutputText = string.format(';%s\n:%s', pFrame.args.title, ibPropValue)
  end

  return sOutputText
end

---------------------------------
-- simulate module invocation  --
---------------------------------

if not mw then
  local simframe = { ["args"] = {} }
  simframe.args['label'] = 'ancient cryptosleep casket'
--~   simframe.args[1] = 'verbs'
--~   simframe.args[2] = 'label'
  wiki.query(simframe)
end

----------------------------------------
-- simulate wiki log while developing --
----------------------------------------

if not mw then
  Util.hl("DefInfo log")
  for _,v in ipairs(logStore) do
    print(v)
  end
end

if mw then
  return wiki
else
  return DefInfo
end