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
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
|
local files = require 'files'
local guide = require 'parser.guide'
local proto = require 'proto.proto'
local lookBackward = require 'core.look-backward'
local util = require 'utility'
local client = require 'client'
---@param uri uri
---@param change table
local function removeSpacesAfterEnter(uri, change)
if not change.text:match '^\r?\n[\t ]+\r?\n$' then
return false
end
local state = files.getState(uri)
if not state then
return false
end
local lines = state.originLines or state.lines
local text = state.originText or state.lua
---@cast text -?
local edits = {}
-- 清除前置空格
local startPos = guide.positionOf(change.range.start.line, change.range.start.character)
local startOffset = guide.positionToOffsetByLines(lines, startPos)
local leftOffset
for offset = startOffset, lines[change.range.start.line], -1 do
leftOffset = offset
local char = text:sub(offset, offset)
if char ~= ' ' and char ~= '\t' then
break
end
end
if leftOffset and leftOffset < startOffset then
edits[#edits+1] = {
start = leftOffset,
finish = startOffset,
text = '',
}
end
-- 清除后置空格
local endOffset = startOffset + #change.text
local _, rightOffset = text:find('^[\t ]+', endOffset + 1)
if rightOffset then
edits[#edits+1] = {
start = endOffset,
finish = rightOffset,
text = '',
}
end
if #edits == 0 then
return nil
end
return edits
end
local function getIndent(state, row)
local offset = state.lines[row]
local indent = state.lua:match('^[\t ]*', offset)
return indent
end
---@param state parser.state
---@param pos integer
---@return parser.object
local function getBlock(state, pos)
local block
guide.eachSourceContain(state.ast, pos, function (src)
if not src.bstart then
return
end
if not block or block.bstart < src.bstart then
block = src
end
end)
return block
end
---@param uri uri
local function fixWrongIndent(uri, change)
if not change.text:match '^\r?\n[\t ]+$' then
return false
end
local state = files.getState(uri)
if not state then
return false
end
local position = guide.positionOf(change.range.start.line, change.range.start.character)
local row = guide.rowColOf(position)
local myIndent = getIndent(state, row + 1)
local lastOffset = lookBackward.findAnyOffset(state.lua, guide.positionToOffset(state, position))
if not lastOffset then
return
end
local lastPosition = guide.offsetToPosition(state, lastOffset)
local lastRow = guide.rowColOf(lastPosition)
local lastIndent = getIndent(state, lastRow)
if #myIndent <= #lastIndent then
return
end
if not util.stringStartWith(myIndent, lastIndent) then
return
end
local myBlock = getBlock(state, lastPosition)
if myBlock.bstart >= lastPosition then
return
end
local endPosition = guide.positionOf(change.range.start.line + 1, #myIndent)
local endOffset = guide.positionToOffset(state, endPosition)
local edits = {}
edits[#edits+1] = {
start = endOffset - #myIndent + #lastIndent,
finish = endOffset,
text = '',
}
return edits
end
---@param uri uri
local function applyEdits(uri, edits)
if #edits == 0 then
return
end
local state = files.getState(uri)
if not state then
return
end
local lines = state.originLines or state.lines
local results = {}
for i, edit in ipairs(edits) do
local startPos = guide.offsetToPositionByLines(lines, edit.start)
local endPos = guide.offsetToPositionByLines(lines, edit.finish)
local startRow, startCol = guide.rowColOf(startPos)
local endRow, endCol = guide.rowColOf(endPos)
results[i] = {
range = {
start = {
line = startRow,
character = startCol,
},
['end'] = {
line = endRow,
character = endCol,
}
},
newText = edit.text,
}
end
proto.request('workspace/applyEdit', {
label = 'Fix Indent',
edit = {
changes = {
[state.uri] = results
}
},
})
end
return function (uri, changes)
if not client.getOption('fixIndents') then
return
end
local firstChange = changes[1]
if firstChange.range then
local edits = removeSpacesAfterEnter(uri, firstChange)
or fixWrongIndent(uri, firstChange)
if edits then
applyEdits(uri, edits)
end
end
end
|
