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
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
|
local files = require 'files'
local searcher = require 'core.searcher'
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
local mark = {}
searcher.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
if source.value and searcher.isLiteral(source.value) 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
if not src then
return
end
if mark[src] then
return
end
mark[src] = true
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 = {}
if func.parent.type == 'setmethod' then
names[#names+1] = 'self'
end
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 hasLiteralArgInCall(call)
if not call.args then
return false
end
for _, arg in ipairs(call.args) do
if searcher.isLiteral(arg) then
return true
end
end
return false
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
local mark = {}
searcher.eachSourceBetween(ast.ast, start, finish, function (source)
if source.type ~= 'call' then
return
end
if not hasLiteralArgInCall(source) 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 not mark[arg] and searcher.isLiteral(arg) then
mark[arg] = true
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
|
