diff options
Diffstat (limited to 'script')
| -rw-r--r-- | script/files.lua | 30 | ||||
| -rw-r--r-- | script/provider/provider.lua | 15 | ||||
| -rw-r--r-- | script/text-merger.lua | 99 | ||||
| -rw-r--r-- | script/utility.lua | 25 |
4 files changed, 143 insertions, 26 deletions
diff --git a/script/files.lua b/script/files.lua index b085e7fd..3f3d633e 100644 --- a/script/files.lua +++ b/script/files.lua @@ -19,6 +19,7 @@ if platform.OS == 'Windows' then unicode = require 'bee.unicode' end +---@class files local m = {} m.openMap = {} @@ -206,6 +207,24 @@ function m.setRawText(uri, text) m.astMap[uri] = nil end +function m.getCachedRows(uri) + uri = getUriKey(uri) + local file = m.fileMap[uri] + if not file then + return nil + end + return file.rows +end + +function m.setCachedRows(uri, rows) + uri = getUriKey(uri) + local file = m.fileMap[uri] + if not file then + return + end + file.rows = rows +end + --- 获取文件版本 function m.getVersion(uri) uri = getUriKey(uri) @@ -633,17 +652,6 @@ function m.diffedOffsetBack(uri, offset) return smerger.getOffsetBack(file._diffInfo, offset) end -function m.clearDiff(uri) - uri = m.getUri(uri) - local file = m.fileMap[uri] - if not file then - return - end - file._diffInfo = nil - file.text = file.originText - m.linesMap[uri] = m.originLinesMap[uri] -end - --- 将光标位置转化为 position ---@param uri uri ---@param offset integer diff --git a/script/provider/provider.lua b/script/provider/provider.lua index 917f66cc..bfd51416 100644 --- a/script/provider/provider.lua +++ b/script/provider/provider.lua @@ -17,6 +17,7 @@ local fs = require 'bee.filesystem' local lang = require 'language' local plugin = require 'plugin' local progress = require 'progress' +local tm = require 'text-merger' local function updateConfig() local diagnostics = require 'provider.diagnostic' @@ -270,18 +271,8 @@ proto.on('textDocument/didChange', function (params) if not files.isLua(uri) and not files.isOpen(uri) then return end - files.clearDiff(uri) - local text = files.getText(uri) or '' - for _, change in ipairs(changes) do - if change.range then - local start, finish = files.unrange(uri, change.range) - text = text:sub(1, start - 1) .. change.text .. text:sub(finish) - else - text = change.text - end - files.setRawText(uri, text) - end - files.setRawText(uri, '') + log.debug('changes', util.dump(changes)) + local text = tm(uri, changes) files.setText(uri, text, true) end) diff --git a/script/text-merger.lua b/script/text-merger.lua new file mode 100644 index 00000000..9e01ba1c --- /dev/null +++ b/script/text-merger.lua @@ -0,0 +1,99 @@ +local files = require 'files' +local util = require 'utility' + +local function splitRows(text) + local rows = {} + for line in util.eachLine(text, true) do + rows[#rows+1] = line + end + return rows +end + +local function getLeft(text, char) + local left + local length = util.utf8Len(text) + + if char == 0 then + left = '' + elseif char >= length then + left = text + else + left = text:sub(1, utf8.offset(text, char + 1) - 1) + end + + return left +end + +local function getRight(text, char) + local right + local length = util.utf8Len(text) + + if char == 0 then + right = text + elseif char >= length then + right = '' + else + right = text:sub(utf8.offset(text, char + 1)) + end + + return right +end + +local function mergeRows(rows, change) + local startLine = change.range['start'].line + 1 + local startChar = change.range['start'].character + local endLine = change.range['end'].line + 1 + local endChar = change.range['end'].character + + local insertRows = splitRows(change.text) + -- 先把双方的行数调整成一致 + local delta = #insertRows - (endLine - startLine + 1) + if delta ~= 0 then + table.move(rows, endLine, #rows, endLine + delta) + -- 如果行数变少了，要清除多余的行 + if delta < 0 then + for i = #rows, #rows + delta + 1, -1 do + rows[i] = nil + end + end + end + -- 先处理第一行和最后一行 + local newEndLine = startLine + #insertRows - 1 + local left = getLeft(rows[startLine], startChar) + local right = getRight(rows[newEndLine], endChar) + if startLine == newEndLine then + rows[startLine] = left .. insertRows[1] .. right + else + rows[startLine] = left .. insertRows[1] + rows[newEndLine] = insertRows[#insertRows] .. right + end + -- 修改中间的每一行 + for i = 2, #insertRows - 1 do + local currentLine = startLine + i - 1 + local insertText = insertRows[i] + rows[currentLine] = insertText + end +end + +return function (uri, changes) + local text + for _, change in ipairs(changes) do + if change.range then + local rows = files.getCachedRows(uri) + if not rows then + text = text or files.getOriginText(uri) + rows = splitRows(text) + end + mergeRows(rows, change) + files.setCachedRows(uri, rows) + else + files.setCachedRows(uri, nil) + text = change.text + end + end + local rows = files.getCachedRows(uri) + if rows then + text = table.concat(rows) + end + return text +end diff --git a/script/utility.lua b/script/utility.lua index a1eec7eb..a98bef92 100644 --- a/script/utility.lua +++ b/script/utility.lua @@ -547,24 +547,43 @@ function m.tableMultiRemove(t, index) end end -function m.eachLine(text) +---遍历文本的每一行 +---@param text string +---@param keepNL boolean # 保留换行符 +---@return fun(text:string):string +function m.eachLine(text, keepNL) local offset = 1 local lineCount = 0 + local lastLine return function () if offset > #text then + if not lastLine then + lastLine = '' + return '' + end return nil end lineCount = lineCount + 1 local nl = text:find('[\r\n]', offset) if not nl then - local lastLine = text:sub(offset) + lastLine = text:sub(offset) offset = #text + 1 return lastLine end - local line = text:sub(offset, nl - 1) + local line if text:sub(nl, nl + 1) == '\r\n' then + if keepNL then + line = text:sub(offset, nl + 1) + else + line = text:sub(offset, nl - 1) + end offset = nl + 2 else + if keepNL then + line = text:sub(offset, nl) + else + line = text:sub(offset, nl - 1) + end offset = nl + 1 end return line |