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
|
local files = require 'files'
local guide = require 'parser.guide'
local proto = require 'proto'
local lang = require 'language'
local converter = require 'proto.converter'
local function isInString(ast, offset)
return guide.eachSourceContain(ast.ast, offset, function (source)
if source.type == 'string' then
return true
end
end) or false
end
---@async
return function (data)
local uri = data.uri
local text = files.getText(uri)
local state = files.getState(uri)
if not state or not text then
return
end
local lines = state.lines
local textEdit = {}
for i = 0, #lines do
local startOffset = lines[i]
local finishOffset = text:find('[\r\n]', startOffset) or (#text + 1)
local lastOffset = finishOffset - 1
local lastChar = text:sub(lastOffset, lastOffset)
if lastChar ~= ' ' and lastChar ~= '\t' then
goto NEXT_LINE
end
local lastPos = guide.offsetToPosition(state, lastOffset)
if isInString(state.ast, lastPos) then
goto NEXT_LINE
end
local firstOffset = startOffset
for n = lastOffset - 1, startOffset, -1 do
local char = text:sub(n, n)
if char ~= ' ' and char ~= '\t' then
firstOffset = n + 1
break
end
end
local firstPos = guide.offsetToPosition(state, firstOffset) - 1
textEdit[#textEdit+1] = {
range = converter.packRange(uri, firstPos, lastPos),
newText = '',
}
::NEXT_LINE::
end
if #textEdit == 0 then
return
end
proto.awaitRequest('workspace/applyEdit', {
label = lang.script.COMMAND_REMOVE_SPACE,
edit = {
changes = {
[uri] = textEdit,
}
},
})
end
|
