diff options
| -rw-r--r-- | server/src/core/code_action.lua | 13 |
1 files changed, 8 insertions, 5 deletions
diff --git a/server/src/core/code_action.lua b/server/src/core/code_action.lua index 7291e3a7..465d5b6f 100644 --- a/server/src/core/code_action.lua +++ b/server/src/core/code_action.lua @@ -246,7 +246,7 @@ local function solveSyntaxByFix(uri, err, lines, callback) } end -local function findEndPosition(err, lines, row, endrow) +local function findEndPosition(lines, row, endrow) if endrow == row then return { newText = ' end', @@ -279,8 +279,8 @@ local function findEndPosition(err, lines, row, endrow) end end -local function solveSyntaxByAddEnd(uri, err, lines, callback) - local row = lines:rowcol(err.start) +local function solveSyntaxByAddEnd(uri, start, lines, callback) + local row = lines:rowcol(start) local line = lines[row] if not line then return nil @@ -296,7 +296,7 @@ local function solveSyntaxByAddEnd(uri, err, lines, callback) edit = { changes = { [uri] = { - findEndPosition(err, lines, row, i - 1) + findEndPosition(lines, row, i - 1) } } } @@ -331,7 +331,10 @@ local function solveSyntax(lsp, uri, data, callback) solveSyntaxByAddDoEnd(uri, data, callback) end if err.type == 'MISS_END' then - solveSyntaxByAddEnd(uri, err, lines, callback) + solveSyntaxByAddEnd(uri, err.start, lines, callback) + end + if err.type == 'MISS_SYMBOL' and err.info.symbol == 'end' then + solveSyntaxByAddEnd(uri, err.info.related[1], lines, callback) end if err.fix then solveSyntaxByFix(uri, err, lines, callback) |