debug/dump.lua at main · cfadmin-cn/debug · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
local USAGE = [[

dump [command] [key1] [key1] [keyN] :

  [global] - dump global table (`_G`).

  [registery] - dump lua debug registery table.

  [filename] - dump already loaded package and its return table .

  --

  `keyX` means we can get `deep value` like `table[key1][key2]..[keyN]`

  e.g :
   1. dump cf wait
   2. dump global string
]]

local function DUMPALL(name, root)
  local KSIZE,  T_USERDATA, T_STRING, T_FUNCTION, T_NUMBER, T_THREAD, T_BOOLEAN, T_TABLE = 0, 0, 0, 0, 0, 0, 0, 0
  local counter = {}
  local content = {}
  for k, v in pairs(root) do
    KSIZE = KSIZE + 1
    local value = nil
    if type(v) == 'string' then
      T_STRING = T_STRING + 1
      value = "'" .. v .. "'"
    elseif type(v) =="boolean" then
      T_BOOLEAN = T_BOOLEAN + 1
      value = v and "true" or "false"
    elseif type(v) == 'userdata' then
      T_USERDATA = T_USERDATA + 1
      value = tostring(v)
    elseif type(v) == 'function' then
      T_FUNCTION = T_FUNCTION + 1
      local info = debug.getinfo(v)
      local val = false
      if info.short_src ~= '[C]' then
        val = tostring(v) .. "(".. info.short_src .. ":" .. info.linedefined .. ")"
      end
      value = val or tostring(v)
    elseif type(v) == 'number' then
      T_NUMBER = T_NUMBER + 1
      value = v
    elseif type(v) == 'thread' then
      T_THREAD = T_THREAD + 1
      value = tostring(v)
    elseif type(v) == 'table' then
      T_TABLE = T_TABLE + 1
      value = tostring(v)
    end
    if type(k) == 'number' then
      content[#content+1] = "[" .. k .. "]" .. ' = ' .. value
    else
      content[#content+1] = "['" .. tostring(k) .. "']" .. ' = ' .. value
    end
  end
  counter[#counter+1] = "total keys count: " .. KSIZE
  counter[#counter+1] = T_STRING > 0 and ("string value count: " .. T_STRING) or nil
  counter[#counter+1] = T_NUMBER > 0 and ("number value count: " .. T_NUMBER) or nil
  counter[#counter+1] = T_BOOLEAN > 0 and ("boolean value count: " .. T_BOOLEAN) or nil
  counter[#counter+1] = T_FUNCTION > 0 and ("function value count: " .. T_FUNCTION) or nil
  counter[#counter+1] = T_USERDATA > 0 and ("usedata value count: " .. T_USERDATA) or nil
  counter[#counter+1] = T_THREAD > 0 and ("thread value count: " .. T_THREAD) or nil
  counter[#counter+1] = T_TABLE > 0 and ("table value count: " .. T_TABLE) or nil
  local tab = { '\r\n' .. name .. "{\r\n  " .. table.concat(content, "\r\n  ") .. "\r\n}", "counter: \r\n  " .. table.concat(counter, '\r\n  ')}
  return table.concat(tab, "\r\n\r\n") .. "\r\n\r\nDone."
end

local function DUMPVALUE(k, v)
  if type(v) == 'string' then
    v = "'" .. v .. "'"
  elseif type(v) =="boolean" then
    v = v and "true" or "false"
  elseif type(v) == 'userdata' then
    v = tostring(v)
  elseif type(v) == 'function' then
    local info = debug.getinfo(v)
    local val = false
    if info.short_src ~= '[C]' then
      val = tostring(v) .. "(".. info.short_src .. ":" .. info.linedefined .. ")"
    end
    v = val or tostring(v)
  elseif type(v) == 'number' then
    v = tostring(v)
  elseif type(v) == 'thread' then
    v = tostring(v)
  else
    v = tostring(v)
  end
  return "\r\n" .. "['" .. k .. "']" .. " = " .. v .. "\r\n\r\nDown."
end

return function (name, ...)
  if not name then
    return USAGE
  end
  local ROOT
  if name == 'global' or name == 'g' then
    name = 'global'
    ROOT = _ENV or _G
  elseif name == 'registery' or name == 'r' then
    name = 'registery'
    ROOT = debug.getregistry ()
  else
    local t = package.loaded[name]
    if t == nil then
      return "\r\nNot import this package yet.\r\n"
    end
    ROOT = t
  end
  if type(ROOT) ~= 'table' then
    return DUMPVALUE(name, ROOT)
  end
  local args
  local argv = select("#", ...)
  if argv > 0 then
    args = {...}
  end
  for level = 1, argv do
    local tab = ROOT[args[level]] or ROOT[tonumber(args[level])] or ROOT[math.tointeger(args[level])]
    if type(tab) ~= 'table' then
      return DUMPVALUE(args[level], tab)
    end
    ROOT = tab
  end
  return DUMPALL(args and args[#args] or name, ROOT)
end