nixos/lua-lsp/script/proto/converter.lua

local guide   = require 'parser.guide'
local files   = require 'files'
local encoder = require 'encoder'

local offsetEncoding = 'utf16'

local m = {}

---@alias position {line: integer, character: integer}

---@param row integer
---@param col integer
---@return position
function m.position(row, col)
    return {
        line      = row,
        character = col,
    }
end

---@param state parser.state
---@param pos integer
---@return position
local function rawPackPosition(state, pos)
    local row, col = guide.rowColOf(pos)
    if col > 0 then
        local text  = state.lua
        if state and 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

---@param state parser.state
---@param pos integer
---@return position
local function diffedPackPosition(state, pos)
    local offset       = guide.positionToOffset(state, pos)
    local originOffset = files.diffedOffsetBack(state, offset)
    local originPos    = guide.offsetToPositionByLines(state.originLines, originOffset)
    local row, col     = guide.rowColOf(originPos)
    if col > 0 then
        local text = state.originText
        if text then
            local lineOffset  = state.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 state parser.state
---@param pos integer
---@return position
function m.packPosition(state, pos)
    if files.hasDiffed(state) then
        return diffedPackPosition(state, pos)
    else
        return rawPackPosition(state, pos)
    end
end

---@param state parser.state
---@param position position
---@return integer
local function rawUnpackPosition(state, position)
    local row, col = position.line, position.character
    if col > 0 then
        local text  = state.lua
        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

---@param state parser.state
---@param position position
---@return integer
local function diffedUnpackPosition(state, position)
    local row, col = position.line, position.character
    if col > 0 then
        local lineOffset = state.originLines[row]
        if lineOffset then
            local textOffset = encoder.offset(offsetEncoding, state.originText, col + 1, lineOffset)
            if textOffset and lineOffset then
                col = textOffset - lineOffset
            end
        end
    end
    local originPos    = guide.positionOf(row, col)
    local originOffset = guide.positionToOffsetByLines(state.originLines, originPos)
    local offset       = files.diffedOffset(state, originOffset)
    local pos          = guide.offsetToPosition(state, offset)
    return pos
end

---@param state    parser.state
---@param position position
---@return integer
function m.unpackPosition(state, position)
    if files.hasDiffed(state) then
        return diffedUnpackPosition(state, position)
    else
        return rawUnpackPosition(state, position)
    end
end

---@alias range {start: position, end: position}

---@param state  parser.state
---@param start  integer
---@param finish integer
---@return range
function m.packRange(state, start, finish)
    local range = {
        start   = m.packPosition(state, start),
        ['end'] = m.packPosition(state, finish),
    }
    return range
end

---@param start position
---@param finish position
---@return range
function m.range(start, finish)
    return {
        start   = start,
        ['end'] = finish,
    }
end

---@param state parser.state
---@param range range
---@return integer start
---@return integer finish
function m.unpackRange(state, range)
    local start  = m.unpackPosition(state, range.start)
    local finish = m.unpackPosition(state, 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