58 lines
2.2 KiB
Fennel
58 lines
2.2 KiB
Fennel
(let [mod {}
|
|
b vim.b
|
|
cmd vim.cmd
|
|
api vim.api
|
|
set-bufvar vim.api.nvim_buf_set_var]
|
|
(fn mod.setup []
|
|
(let [usercmd vim.api.nvim_create_user_command
|
|
augroup (vim.api.nvim_create_augroup :NifocFormatting {:clear true})
|
|
aucmd vim.api.nvim_create_autocmd]
|
|
(usercmd :FormatDisableBuffer mod.disable-for-buffer
|
|
{:desc "Disable Formatting for the current buffer"})
|
|
(usercmd :FormatEnableBuffer mod.enable-for-buffer
|
|
{:desc "Enable Formatting for the current buffer"})
|
|
(aucmd :BufWritePre {:callback mod.maybe-format-buffer
|
|
:group augroup
|
|
:desc "Run Formatter"})))
|
|
|
|
(fn run-neoformat [formatprg]
|
|
(let [neoformat (.. "Neoformat " formatprg)]
|
|
(cmd (.. "try | undojoin | " neoformat " | catch /E790/ | " neoformat
|
|
" | endtry"))))
|
|
|
|
(fn run-lsp-format []
|
|
(if (not= b.nifoc_formatter_filter_lsp_client nil)
|
|
(vim.lsp.buf.format {:filter #(= $1.name
|
|
b.nifoc_formatter_filter_lsp_client)
|
|
:timeout_ms 1000})
|
|
(vim.lsp.buf.format {:timeout_ms 1000})))
|
|
|
|
(fn mod.enable-for-buffer []
|
|
(set-bufvar 0 :nifoc_formatter_disabled 0))
|
|
|
|
(fn mod.disable-for-buffer []
|
|
(set-bufvar 0 :nifoc_formatter_disabled 1))
|
|
|
|
(fn mod.active? []
|
|
(if (= b.nifoc_formatter_disabled 1) false
|
|
(= b.nifoc_lsp_formatter_enabled 1) true
|
|
(not= (vim.opt_local.formatprg:get) "") true
|
|
false))
|
|
|
|
(fn mod.maybe-enable-lsp [client bufnr]
|
|
(when (client.supports_method :textDocument/rangeFormatting)
|
|
(api.nvim_buf_set_option bufnr :formatexpr "v:lua.vim.lsp.formatexpr()"))
|
|
(when (client.supports_method :textDocument/formatting)
|
|
(set-bufvar bufnr :nifoc_lsp_formatter_enabled 1)))
|
|
|
|
(fn mod.maybe-format-buffer []
|
|
(let [formatprg (vim.opt_local.formatprg:get)
|
|
formatprg-exe (-> formatprg (vim.split " " {:trimempty true}) (. 1))]
|
|
(if (= b.nifoc_formatter_disabled 1) nil
|
|
(= b.nifoc_formatter_force_formatprg 1) (run-neoformat formatprg-exe)
|
|
(= b.nifoc_lsp_formatter_enabled 1) (run-lsp-format)
|
|
(not= formatprg-exe nil) (run-neoformat formatprg-exe))))
|
|
|
|
mod)
|
|
|