mirror of
https://github.com/notwa/mm
synced 2024-11-05 02:39:02 -08:00
62 lines
1.4 KiB
Lua
62 lines
1.4 KiB
Lua
local function strpad(num, count, pad)
|
|
num = tostring(num)
|
|
return (pad:rep(count)..num):sub(#num)
|
|
end
|
|
|
|
local function add_zeros(num, count)
|
|
return strpad(num, count - 1, '0')
|
|
end
|
|
|
|
local function mixed_sorter(a, b)
|
|
a = type(a) == 'number' and add_zeros(a, 16) or tostring(a)
|
|
b = type(b) == 'number' and add_zeros(b, 16) or tostring(b)
|
|
return a < b
|
|
end
|
|
|
|
-- loosely based on http://lua-users.org/wiki/SortedIteration
|
|
-- the original didn't make use of closures for who knows why
|
|
local function order_keys(t)
|
|
local oi = {}
|
|
for key in pairs(t) do
|
|
table.insert(oi, key)
|
|
end
|
|
table.sort(oi, mixed_sorter)
|
|
return oi
|
|
end
|
|
|
|
local function opairs(t, cache)
|
|
local oi = cache and cache[t] or order_keys(t)
|
|
if cache then
|
|
cache[t] = oi
|
|
end
|
|
local i = 0
|
|
return function()
|
|
i = i + 1
|
|
local key = oi[i]
|
|
if key then return key, t[key] end
|
|
end
|
|
end
|
|
|
|
local function traverse(path)
|
|
if not path then return end
|
|
local parent = _G
|
|
local key
|
|
for w in path:gfind("[%w_]+") do
|
|
if key then
|
|
parent = rawget(parent, key)
|
|
if type(parent) ~= 'table' then return end
|
|
end
|
|
key = w
|
|
end
|
|
if not key then return end
|
|
return {parent=parent, key=key}
|
|
end
|
|
|
|
return {
|
|
strpad = strpad,
|
|
add_zeros = add_zeros,
|
|
mixed_sorter = mixed_sorter,
|
|
order_keys = order_keys,
|
|
opairs = opairs,
|
|
traverse = traverse,
|
|
}
|