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
131
132
133
|
local files = require 'files'
local guide = require 'parser.guide'
local vm = require 'vm'
local config = require 'config'
local function typeHint(uri, edits, start, finish)
local ast = files.getAst(uri)
if not ast then
return
end
guide.eachSourceBetween(ast.ast, start, finish, function (source)
if source.type ~= 'local'
and source.type ~= 'setglobal'
and source.type ~= 'tablefield'
and source.type ~= 'tableindex'
and source.type ~= 'setfield'
and source.type ~= 'setindex' then
return
end
if source[1] == '_' then
return
end
-- 排除掉 xx = function 与 xx = {}
if source.value and (source.value.type == 'function' or source.value.type == 'table') then
return
end
if source.parent.type == 'funcargs' then
if not config.config.hint.paramType then
return
end
else
if not config.config.hint.setType then
return
end
end
local infer = vm.getInferType(source, 0)
if infer == 'any'
or infer == 'nil' then
return
end
local src = source
if source.type == 'tablefield' then
src = source.field
elseif source.type == 'tableindex' then
src = source.index
end
edits[#edits+1] = {
newText = (':%s'):format(infer),
start = src.finish,
finish = src.finish,
}
end)
end
local function getArgNames(func)
if not func.args or #func.args == 0 then
return nil
end
local names = {}
for _, arg in ipairs(func.args) do
if arg.type == '...' then
break
end
names[#names+1] = arg[1] or ''
end
if #names == 0 then
return nil
end
return names
end
local function paramName(uri, edits, start, finish)
if not config.config.hint.paramName then
return
end
local ast = files.getAst(uri)
if not ast then
return
end
guide.eachSourceBetween(ast.ast, start, finish, function (source)
if source.type ~= 'call' then
return
end
if not source.args then
return
end
local defs = vm.getDefs(source.node, 0)
if not defs then
return
end
local args
for _, def in ipairs(defs) do
if def.value then
def = def.value
end
if def.type == 'function' then
args = getArgNames(def)
if args then
break
end
end
end
if not args then
return
end
if source.node and source.node.type == 'getmethod' then
table.remove(args, 1)
end
for i, arg in ipairs(source.args) do
if arg.type == 'nil'
or arg.type == 'number'
or arg.type == 'string'
or arg.type == 'boolean'
or arg.type == 'table'
or arg.type == 'function' then
if args[i] and args[i] ~= '' then
edits[#edits+1] = {
newText = ('%s:'):format(args[i]),
start = arg.start,
finish = arg.start - 1,
}
end
end
end
end)
end
return function (uri, start, finish)
local edits = {}
typeHint(uri, edits, start, finish)
paramName(uri, edits, start, finish)
return edits
end
|
