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
184
|
local guide = require 'parser.guide'
local files = require 'files'
local encoder = require 'encoder'
local offsetEncoding = 'utf16'
local m = {}
---@alias position {line: integer, character: integer}
local function rawPackPosition(uri, pos)
local row, col = guide.rowColOf(pos)
if col > 0 then
local state = files.getState(uri)
local text = files.getText(uri)
if text then
local lineOffset = state.lines[row]
if lineOffset then
local start = lineOffset
local finish = lineOffset + col - 1
if start <= #text and finish <= #text then
col = encoder.len(offsetEncoding, text, lineOffset, lineOffset + col - 1)
end
else
col = 0
end
end
end
return {
line = row,
character = col,
}
end
local function diffedPackPosition(uri, pos)
local state = files.getState(uri)
local offset = guide.positionToOffset(state, pos)
local originOffset = files.diffedOffsetBack(uri, offset)
local originLines = files.getOriginLines(uri)
local originPos = guide.offsetToPositionByLines(originLines, originOffset)
local row, col = guide.rowColOf(originPos)
if col > 0 then
local text = files.getOriginText(uri)
if text then
local lineOffset = originLines[row]
local finalOffset = math.min(lineOffset + col - 1, #text + 1)
col = encoder.len(offsetEncoding, text, lineOffset, finalOffset)
end
end
return {
line = row,
character = col,
}
end
---@param uri uri
---@param pos integer
---@return position
function m.packPosition(uri, pos)
if files.hasDiffed(uri) then
return diffedPackPosition(uri, pos)
else
return rawPackPosition(uri, pos)
end
end
local function rawUnpackPosition(uri, position)
local row, col = position.line, position.character
if col > 0 then
local state = files.getState(uri)
local text = files.getText(uri)
if state and text then
local lineOffset = state.lines[row]
local textOffset = encoder.offset(offsetEncoding, text, col + 1, lineOffset)
if textOffset and lineOffset then
col = textOffset - lineOffset
end
end
end
local pos = guide.positionOf(row, col)
return pos
end
local function diffedUnpackPosition(uri, position)
local row, col = position.line, position.character
local originLines = files.getOriginLines(uri)
if col > 0 then
local text = files.getOriginText(uri)
if text then
local lineOffset = originLines[row]
local textOffset = encoder.offset(offsetEncoding, text, col + 1, lineOffset)
if textOffset and lineOffset then
col = textOffset - lineOffset
end
end
end
local state = files.getState(uri)
local originPos = guide.positionOf(row, col)
local originOffset = guide.positionToOffsetByLines(originLines, originPos)
local offset = files.diffedOffset(uri, originOffset)
local pos = guide.offsetToPosition(state, offset)
return pos
end
---@param uri uri
---@param position position
---@return integer
function m.unpackPosition(uri, position)
if files.hasDiffed(uri) then
return diffedUnpackPosition(uri, position)
else
return rawUnpackPosition(uri, position)
end
end
---@alias range {start: position, end: position}
---@param uri uri
---@param start integer
---@param finish integer
---@return range
function m.packRange(uri, start, finish)
local range = {
start = m.packPosition(uri, start),
['end'] = m.packPosition(uri, finish),
}
return range
end
---@param uri uri
---@param range range
---@return integer start
---@return integer finish
function m.unpackRange(uri, range)
local start = m.unpackPosition(uri, range.start)
local finish = m.unpackPosition(uri, range['end'])
return start, finish
end
---@alias location {uri: uri, range: range}
---@param uri string
---@param range range
---@return location
function m.location(uri, range)
return {
uri = uri,
range = range,
}
end
---@alias locationLink {targetUri:uri, targetRange: range, targetSelectionRange: range, originSelectionRange: range}
---@param uri string
---@param range range
---@param selection range
---@param origin range
---@return locationLink
function m.locationLink(uri, range, selection, origin)
return {
targetUri = uri,
targetRange = range,
targetSelectionRange = selection,
originSelectionRange = origin,
}
end
---@alias textEdit {range: range, newText: string}
---@param range range
---@param newtext string
---@return textEdit
function m.textEdit(range, newtext)
return {
range = range,
newText = newtext,
}
end
function m.setOffsetEncoding(encoding)
offsetEncoding = encoding:lower():gsub('%-', '')
end
return m
|