mroot/lazyvim
1
0
Fork 0
mirror of https://github.com/LazyVim/LazyVim.git synced 2026-07-25 14:01:06 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor: tabnine

Browse source
This commit is contained in:
Folke Lemaitre 2023-10-14 18:23:34 +02:00
parent be92041084
commit c810074a48
No known key found for this signature in database
GPG key ID: 41F8B1FBACAE2040
3 changed files with 9 additions and 56 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
lua/lazyvim/config/init.lua
View file

@ -84,6 +84,7 @@ local defaults = {
Snippet = "",
String = "",
Struct = "󰆼 ",
TabNine = "󰏚 ",
Text = "",
TypeParameter = "",
Unit = "",

1
lua/lazyvim/plugins/coding.lua
View file

@ -83,6 +83,7 @@ return {
local icons = require("lazyvim.config").icons.kinds
if icons[item.kind] then
item.kind = icons[item.kind] .. item.kind
item.menu = ""
end
return item
end,

63
lua/lazyvim/plugins/extras/coding/tabnine.lua
View file

@ -1,5 +1,4 @@
-- TODO: add TabNine icon if is available
local tabnine_icon = ""
local Util = require("lazyvim.util")
return {
-- Tabnine cmp source
@ -30,25 +29,12 @@ return {
priority = 100,
})
-- Format the completion menu, remove the percentage and add icons
opts.formatting.format = function(entry, vim_item)
local icons = require("lazyvim.config").icons.kinds
if icons[vim_item.kind] then
vim_item.kind = icons[vim_item.kind] .. vim_item.kind
end
-- Add tabnine icon and hide percentage in the menu
opts.formatting.format = Util.inject.args(opts.formatting.format, function(entry, item)
-- Hide percentage in the menu
if entry.source.name == "cmp_tabnine" then
vim_item.kind = tabnine_icon .. " TabNine"
vim_item.menu = ""
if (entry.completion_item.data or {}).multiline then
vim_item.kind = vim_item.kind .. " " .. "[ML]"
end
item.menu = ""
end
return vim_item
end
end)
end,
},
-- Show TabNine status in lualine
@ -57,43 +43,8 @@ return {
optional = true,
event = "VeryLazy",
opts = function(_, opts)
local started = false
local function status()
if not package.loaded["cmp"] then
return
end
for _, s in ipairs(require("cmp").core.sources) do
if s.name == "cmp_tabnine" then
if s.source:is_available() then
started = true
else
return started and "error" or nil
end
if s.status == s.SourceStatus.FETCHING then
return "pending"
end
return "ok"
end
end
end
local Util = require("lazyvim.util")
local colors = {
ok = Util.ui.fg("Special"),
error = Util.ui.fg("DiagnosticError"),
pending = Util.ui.fg("DiagnosticWarn"),
}
table.insert(opts.sections.lualine_x, 3, {
function()
return tabnine_icon
end,
cond = function()
return status() ~= nil
end,
color = function()
return colors[status()] or colors.ok
end,
})
local icon = require("lazyvim.config").icons.kinds.TabNine
table.insert(opts.sections.lualine_x, 2, require("lazyvim.util").lualine.cmp_source("cmp_tabnine", icon))
end,
},
}