100 lines
2.6 KiB
Lua
100 lines
2.6 KiB
Lua
|
local files = require 'files'
|
||
|
|
local guide = require 'parser.guide'
|
||
|
|
local proto = require 'proto'
|
||
|
|
local lang = require 'language'
|
||
|
|
local converter = require 'proto.converter'
|
||
|
|
|
||
|
|
local opMap = {
|
||
|
|
['+'] = true,
|
||
|
|
['-'] = true,
|
||
|
|
['*'] = true,
|
||
|
|
['/'] = true,
|
||
|
|
['//'] = true,
|
||
|
|
['^'] = true,
|
||
|
|
['<<'] = true,
|
||
|
|
['>>'] = true,
|
||
|
|
['&'] = true,
|
||
|
|
['|'] = true,
|
||
|
|
['~'] = true,
|
||
|
|
['..'] = true,
|
||
|
|
}
|
||
|
|
|
||
|
|
local literalMap = {
|
||
|
|
['number'] = true,
|
||
|
|
['integer'] = true,
|
||
|
|
['boolean'] = true,
|
||
|
|
['string'] = true,
|
||
|
|
['table'] = true,
|
||
|
|
}
|
||
|
|
|
||
|
|
---@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 start, finish = converter.unpackRange(state, data.range)
|
||
|
|
|
||
|
|
local result = guide.eachSourceContain(state.ast, start, function (source)
|
||
|
|
if source.start ~= start
|
||
|
|
or source.finish ~= finish then
|
||
|
|
return
|
||
|
|
end
|
||
|
|
if not source.op or source.op.type ~= 'or' then
|
||
|
|
return
|
||
|
|
end
|
||
|
|
local first = source[1]
|
||
|
|
local second = source[2]
|
||
|
|
-- a + b or 0 --> a + (b or 0)
|
||
|
|
do
|
||
|
|
if first.op
|
||
|
|
and opMap[first.op.type]
|
||
|
|
and first.type ~= 'unary'
|
||
|
|
and not second.op
|
||
|
|
and literalMap[second.type] then
|
||
|
|
return {
|
||
|
|
start = source[1][2].start,
|
||
|
|
finish = source[2].finish,
|
||
|
|
}
|
||
|
|
end
|
||
|
|
end
|
||
|
|
-- a or b + c --> (a or b) + c
|
||
|
|
do
|
||
|
|
if second.op
|
||
|
|
and opMap[second.op.type]
|
||
|
|
and second.type ~= 'unary'
|
||
|
|
and not first.op
|
||
|
|
and literalMap[second[1].type] then
|
||
|
|
return {
|
||
|
|
start = source[1].start,
|
||
|
|
finish = source[2][1].finish,
|
||
|
|
}
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end)
|
||
|
|
|
||
|
|
if not result then
|
||
|
|
return
|
||
|
|
end
|
||
|
|
|
||
|
|
proto.awaitRequest('workspace/applyEdit', {
|
||
|
|
label = lang.script.COMMAND_REMOVE_SPACE,
|
||
|
|
edit = {
|
||
|
|
changes = {
|
||
|
|
[uri] = {
|
||
|
|
{
|
||
|
|
range = converter.packRange(state, result.start, result.finish),
|
||
|
|
newText = ('(%s)'):format(text:sub(
|
||
|
|
guide.positionToOffset(state, result.start + 1),
|
||
|
|
guide.positionToOffset(state, result.finish)
|
||
|
|
)),
|
||
|
|
}
|
||
|
|
},
|
||
|
|
}
|
||
|
|
},
|
||
|
|
})
|
||
|
|
end
|