147 lines
5.2 KiB
Lua
147 lines
5.2 KiB
Lua
-- This file can be loaded as a telescope extension
|
|
local M = {}
|
|
|
|
-- Custom theme picker
|
|
-- Most of the code is copied from telescope colorscheme plugin, mostly for preview creation
|
|
M.theme_switcher = function(opts)
|
|
local pickers, finders, previewers, actions, action_state, utils, conf
|
|
if pcall(require, "telescope") then
|
|
pickers = require "telescope.pickers"
|
|
finders = require "telescope.finders"
|
|
previewers = require "telescope.previewers"
|
|
|
|
actions = require "telescope.actions"
|
|
action_state = require "telescope.actions.state"
|
|
utils = require "telescope.utils"
|
|
conf = require("telescope.config").values
|
|
else
|
|
error "Cannot find telescope!"
|
|
end
|
|
|
|
local local_utils = require "utils"
|
|
local reload_theme = local_utils.reload_theme
|
|
|
|
-- get a table of available themes
|
|
local themes = local_utils.list_themes()
|
|
if next(themes) ~= nil then
|
|
-- save this to use it for later to restore if theme not changed
|
|
local current_theme = vim.g.nvchad_theme
|
|
local new_theme = ""
|
|
local change = false
|
|
|
|
-- buffer number and name
|
|
local bufnr = vim.api.nvim_get_current_buf()
|
|
local bufname = vim.api.nvim_buf_get_name(bufnr)
|
|
|
|
local previewer
|
|
|
|
-- in case its not a normal buffer
|
|
if vim.fn.buflisted(bufnr) ~= 1 then
|
|
local deleted = false
|
|
local function del_win(win_id)
|
|
if win_id and vim.api.nvim_win_is_valid(win_id) then
|
|
utils.buf_delete(vim.api.nvim_win_get_buf(win_id))
|
|
pcall(vim.api.nvim_win_close, win_id, true)
|
|
end
|
|
end
|
|
|
|
previewer =
|
|
previewers.new {
|
|
preview_fn = function(_, entry, status)
|
|
if not deleted then
|
|
deleted = true
|
|
del_win(status.preview_win)
|
|
del_win(status.preview_border_win)
|
|
end
|
|
reload_theme(entry.value)
|
|
end
|
|
}
|
|
else
|
|
-- show current buffer content in previewer
|
|
previewer =
|
|
previewers.new_buffer_previewer {
|
|
get_buffer_by_name = function()
|
|
return bufname
|
|
end,
|
|
define_preview = function(self, entry)
|
|
if vim.loop.fs_stat(bufname) then
|
|
conf.buffer_previewer_maker(bufname, self.state.bufnr, {bufname = self.state.bufname})
|
|
else
|
|
local lines = vim.api.nvim_buf_get_lines(bufnr, 0, -1, false)
|
|
vim.api.nvim_buf_set_lines(self.state.bufnr, 0, -1, false, lines)
|
|
end
|
|
reload_theme(entry.value)
|
|
end
|
|
}
|
|
end
|
|
|
|
local picker =
|
|
pickers.new(
|
|
{
|
|
prompt_title = "Set NvChad color",
|
|
finder = finders.new_table(themes),
|
|
previewer = previewer,
|
|
sorter = conf.generic_sorter(opts),
|
|
attach_mappings = function()
|
|
actions.select_default:replace(
|
|
-- if a entry is selected, change current_theme to that
|
|
function(prompt_bufnr)
|
|
local selection = action_state.get_selected_entry()
|
|
new_theme = selection.value
|
|
change = true
|
|
actions.close(prompt_bufnr)
|
|
end
|
|
)
|
|
return true
|
|
end
|
|
}
|
|
)
|
|
|
|
-- rewrite picker.close_windows
|
|
local close_windows = picker.close_windows
|
|
picker.close_windows = function(status)
|
|
close_windows(status)
|
|
-- now apply the theme, if success, then ask for default theme change
|
|
local final_theme
|
|
if change then
|
|
final_theme = new_theme
|
|
else
|
|
final_theme = current_theme
|
|
end
|
|
|
|
if reload_theme(final_theme) then
|
|
if change then
|
|
-- ask for confirmation to set as default theme
|
|
local ans = string.lower(vim.fn.input("Set " .. new_theme .. " as default theme ? [y/N] ")) == "y"
|
|
local_utils.clear_cmdline()
|
|
if ans then
|
|
local_utils.change_theme(current_theme, final_theme)
|
|
else
|
|
-- will be used in restoring nvchad theme var
|
|
final_theme = current_theme
|
|
end
|
|
end
|
|
else
|
|
final_theme = current_theme
|
|
end
|
|
-- set nvchad_theme global var
|
|
vim.g.nvchad_theme = final_theme
|
|
end
|
|
-- launch the telescope picker
|
|
picker:find()
|
|
else
|
|
print("No themes found in " .. themes_folder)
|
|
end
|
|
end
|
|
|
|
-- register theme swticher as themes to telescope
|
|
local present, telescope = pcall(require, "telescope")
|
|
if present then
|
|
return telescope.register_extension {
|
|
exports = {
|
|
themes = M.theme_switcher
|
|
}
|
|
}
|
|
else
|
|
error "Cannot find telescope!"
|
|
end
|