-
Notifications
You must be signed in to change notification settings - Fork 13
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
2 changed files
with
3 additions
and
188 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,192 +1,7 @@ | ||
| local utils = require("coq_3p.utils") | ||
|
|
||
| return function(spec) | ||
| vim.g.tabby_trigger_mode = "manual" | ||
| vim.g.tabby_keybinding_accept = "<NOP>" | ||
|
|
||
| local request_id = 0 | ||
|
|
||
| local ctx = function(row, col, line) | ||
| local before_cursor = utils.split_line(line, col) | ||
| local ls = utils.linesep() | ||
| local buf = vim.api.nvim_get_current_buf() | ||
| local ft = vim.api.nvim_buf_get_option(buf, "filetype") | ||
| local language = (vim.g.tabby_filetype_dict or {})[ft] or ft | ||
| local lines = vim.api.nvim_buf_get_text(buf, 0, 0, row, col, {}) | ||
| local before = table.concat(lines, ls) | ||
| local pos = vim.fn.strchars(before) | ||
| local around = vim.api.nvim_buf_get_lines(buf, 0, -1, false) | ||
| local text = table.concat(around, ls) | ||
| local offset = pos - vim.fn.strchars(before_cursor) | ||
| local context = { | ||
| filepath = vim.api.nvim_buf_get_name(buf), | ||
| language = language, | ||
| manually = false, | ||
| position = pos, | ||
| text = text | ||
| } | ||
| return context, buf, offset | ||
| end | ||
|
|
||
| local acc = {} | ||
|
|
||
| local resp_cb = function(buf, row, offset) | ||
| vim.validate { | ||
| buf = {buf, "number"}, | ||
| row = {row, "number"}, | ||
| offset = {offset, "number"} | ||
| } | ||
|
|
||
| return function(resp) | ||
| if type(resp) == "userdata" then | ||
| return | ||
| end | ||
| vim.validate {id = {resp.id, "string"}, choices = {resp.choices, "table"}} | ||
| acc = | ||
| (function() | ||
| local aacc = {} | ||
| for _, val in pairs(resp.choices) do | ||
| vim.validate {val = {val, "table"}} | ||
| vim.validate { | ||
| index = {val.index, "number"}, | ||
| text = {val.text, "string"}, | ||
| replaceRange = {val.replaceRange, "table"} | ||
| } | ||
| vim.validate { | ||
| start = {val.replaceRange.start, "number"}, | ||
| ["end"] = {val.replaceRange["end"], "number"} | ||
| } | ||
| local arguments = { | ||
| type = "view", | ||
| choice_index = val.index, | ||
| completion_id = resp.id | ||
| } | ||
| local start = (function() | ||
| local start = val.replaceRange.start | ||
| return start == 0 and start or start - offset | ||
| end)() | ||
| local v = { | ||
| buf = buf, | ||
| row = row, | ||
| text = val.text, | ||
| start = start, | ||
| fin = val.replaceRange["end"] - offset, | ||
| arguments = arguments | ||
| } | ||
| table.insert(aacc, v) | ||
| vim.fn["tabby#agent#PostEvent"](arguments) | ||
| end | ||
| return aacc | ||
| end)() | ||
| end | ||
| end | ||
|
|
||
| local items = function(row, col, line) | ||
| local buf = vim.api.nvim_get_current_buf() | ||
| local items = {} | ||
| for _, val in pairs(acc) do | ||
| vim.validate {val = {val, "table"}} | ||
| vim.validate { | ||
| buf = {val.buf, "number"}, | ||
| row = {val.row, "number"}, | ||
| text = {val.text, "string"}, | ||
| start = {val.start, "number"}, | ||
| fin = {val.fin, "number"}, | ||
| arguments = {val.arguments, "table"} | ||
| } | ||
| if val.buf == buf and val.row == row then | ||
| local col_diff = col - val.start | ||
| local almost_same_col = math.abs(col_diff) <= utils.MAX_COL_DIFF | ||
| local fin = (function() | ||
| if val.fin >= col then | ||
| return val.fin | ||
| else | ||
| return val.fin + col_diff | ||
| end | ||
| end)() | ||
| local range = { | ||
| start = {line = row, character = val.start}, | ||
| ["end"] = {line = row, character = fin} | ||
| } | ||
|
|
||
| if almost_same_col then | ||
| local item = { | ||
| preselect = true, | ||
| label = val.text, | ||
| documentation = val.text, | ||
| textEdit = { | ||
| newText = val.text, | ||
| range = range | ||
| }, | ||
| command = { | ||
| title = "TAB", | ||
| command = "#TAB", | ||
| arguments = val.arguments | ||
| } | ||
| } | ||
| table.insert(items, item) | ||
| end | ||
| end | ||
| end | ||
| return items | ||
| end | ||
|
|
||
| local pull = function(row, col, line) | ||
| if vim.g.tabby_filetype_dict == nil then | ||
| return | ||
| end | ||
| if request_id ~= 0 then | ||
| vim.fn["tabby#agent#CancelRequest"](request_id) | ||
| end | ||
|
|
||
| local request_context, buf, offset = ctx(row, col, line) | ||
| request_id = | ||
| vim.fn["tabby#agent#ProvideCompletions"]( | ||
| request_context, | ||
| resp_cb(buf, row, offset) | ||
| ) | ||
| end | ||
|
|
||
| local fn = function(args, callback) | ||
| local row, col = unpack(args.pos) | ||
| callback( | ||
| { | ||
| isIncomplete = true, | ||
| items = items(row, col, args.line) | ||
| } | ||
| ) | ||
| end | ||
|
|
||
| local f = "tabby#agent#PostEvent" | ||
| local ff = "*" .. f | ||
|
|
||
| local exec = function(val) | ||
| local arguments = val.arguments | ||
| vim.validate {arguments = {arguments, "table"}} | ||
| arguments.type = "select" | ||
| if vim.fn.exists(ff) then | ||
| vim.fn[f](arguments) | ||
| end | ||
| end | ||
|
|
||
| local events = { | ||
| "CursorHoldI", | ||
| "InsertEnter", | ||
| "TextChangedI", | ||
| "TextChangedP" | ||
| } | ||
| vim.api.nvim_create_autocmd( | ||
| events, | ||
| { | ||
| pattern = {"*"}, | ||
| callback = function() | ||
| local row, col = unpack(vim.api.nvim_win_get_cursor(0)) | ||
| row = row - 1 | ||
| local line = vim.api.nvim_get_current_line() | ||
| pull(row, col, line) | ||
| end | ||
| } | ||
| vim.api.nvim_err_writeln( | ||
| [[Please update to latest version of coq.nvim. This functionality has been pulled into main. (requires tabby_ml LSP)]] | ||
| ) | ||
|
|
||
| return fn, {exec = exec} | ||
| end |