306 lines
15 KiB
Fennel
306 lines
15 KiB
Fennel
(import-macros {: deferred_cmd} :../macros/cmd)
|
|
|
|
(let [mod {}
|
|
api vim.api
|
|
heirline-utils (require :heirline.utils)
|
|
heirline-conditions (require :heirline.conditions)
|
|
web-devicons (require :nvim-web-devicons)
|
|
colors (. (require :nifoc.theme) :colors)
|
|
formatting (require :nifoc.formatting)
|
|
nifoc-treesitter (require :nifoc.treesitter)]
|
|
(fn buffer-variable-exists? [key]
|
|
(not= (. vim :b key) nil))
|
|
|
|
(fn get-current-line []
|
|
(let [cl (api.nvim_win_get_cursor 0)]
|
|
(. cl 1)))
|
|
|
|
(fn get-total-lines []
|
|
(api.nvim_buf_line_count 0))
|
|
|
|
(fn max-number [nums]
|
|
(math.max (unpack nums)))
|
|
|
|
;; Utils
|
|
(set mod.default-hl (fn []
|
|
{:bg colors.black}))
|
|
(set mod.space {:provider " "})
|
|
(set mod.spacer {:provider " " :hl {:fg colors.bg :bg colors.bg}})
|
|
(set mod.push-right {:provider "%="})
|
|
(set mod.space-if-count {:condition #(> $1.check-count 0) :provider " "})
|
|
(set mod.space-if-length {:condition #(> $1.check-length 0) :provider " "})
|
|
(set mod.space-if-count-or-length
|
|
{:condition #(or (> $1.check-count 0) (> $1.check-length 0))
|
|
:provider " "})
|
|
;; Mode
|
|
(set mod.vi-mode {:init (fn [self]
|
|
(let [mode (. (api.nvim_get_mode) :mode)]
|
|
(set self.mode mode)))
|
|
:update :ModeChanged
|
|
:static {:mode-names {:n :NORMAL
|
|
:no :O-PENDING
|
|
:nov :O-PENDING
|
|
:noV :O-PENDING
|
|
"no\022" :O-PENDING
|
|
:niI :NORMAL
|
|
:niR :NORMAL
|
|
:niV :NORMAL
|
|
:nt :NORMAL
|
|
:v :VISUAL
|
|
:vs :VISUAL
|
|
:V :V-LINE
|
|
:Vs :V-LINE
|
|
"\022" :V-BLOCK
|
|
"\022s" :V-BLOCK
|
|
:s :SELECT
|
|
:S :S-LINE
|
|
"\019" :S-BLOCK
|
|
:i :INSERT
|
|
:ic :INSERT
|
|
:ix :INSERT
|
|
:R :REPLACE
|
|
:Rc :REPLACE
|
|
:Rx :REPLACE
|
|
:Rv :V-REPLACE
|
|
:Rvc :V-REPLACE
|
|
:Rvx :V-REPLACE
|
|
:c :COMMAND
|
|
:cv :EX
|
|
:ce :EX
|
|
:r :REPLACE
|
|
:rm :MORE
|
|
:r? :CONFIRM
|
|
:! :SHELL
|
|
:t :TERMINAL}
|
|
:mode-hl {:n {:fg colors.black
|
|
:bg colors.purple
|
|
:bold true}
|
|
:i {:fg colors.black
|
|
:bg colors.green
|
|
:bold true}
|
|
:v {:fg colors.black
|
|
:bg colors.pink
|
|
:bold true}
|
|
:V {:fg colors.black
|
|
:bg colors.pink
|
|
:bold true}
|
|
"\022" {:fg colors.black
|
|
:bg colors.pink
|
|
:bold true}
|
|
:c {:fg colors.black
|
|
:bg colors.yellow
|
|
:bold true}
|
|
:s {:fg colors.black
|
|
:bg colors.orange
|
|
:bold true}
|
|
:S {:fg colors.black
|
|
:bg colors.orange
|
|
:bold true}
|
|
"\019" {:fg colors.black
|
|
:bg colors.orange
|
|
:bold true}
|
|
:R {:fg colors.black
|
|
:bg colors.yellow
|
|
:bold true}
|
|
:r {:fg colors.black
|
|
:bg colors.yellow
|
|
:bold true}
|
|
:! {:fg colors.black
|
|
:bg colors.purple
|
|
:bold true}
|
|
:t {:fg colors.black
|
|
:bg colors.purple
|
|
:bold true}}}
|
|
:provider #(.. " " (. $1 :mode-names $1.mode) " ")
|
|
:hl (fn [self]
|
|
(let [short-mode (self.mode:sub 1 1)]
|
|
(. self :mode-hl short-mode)))})
|
|
;; Filetype
|
|
(set mod.filetype-block
|
|
{:init (fn [self]
|
|
(set self.filename (api.nvim_buf_get_name 0)))})
|
|
(set mod.file-icon {:init (fn [self]
|
|
(let [filename self.filename
|
|
ext (vim.fn.fnamemodify filename ":e")
|
|
(icon color) (web-devicons.get_icon_color filename
|
|
ext
|
|
{:default true})]
|
|
(set self.icon icon)
|
|
(set self.icon-color color)))
|
|
:provider (fn [self]
|
|
(when self.icon
|
|
(.. self.icon " ")))
|
|
:hl (fn [self]
|
|
{:fg self.icon-color})})
|
|
(set mod.filetype {:provider (fn []
|
|
(let [ft vim.bo.filetype]
|
|
(if (> (ft:len) 0) ft "no ft")))
|
|
:hl {:fg colors.white}})
|
|
(set mod.encoding {:provider (fn []
|
|
(let [vim-enc (if (not= vim.bo.fenc nil)
|
|
vim.bo.fenc
|
|
vim.o.enc)
|
|
enc (if (> (vim-enc:len) 0) vim-enc
|
|
"no enc")]
|
|
(.. enc " ")))
|
|
:hl {:fg colors.white}})
|
|
(set mod.filetype-block
|
|
(heirline-utils.insert mod.filetype-block mod.file-icon mod.filetype
|
|
{:provider " | " :hl {:fg colors.white}}
|
|
mod.encoding))
|
|
;; git
|
|
(set mod.git {:condition heirline-conditions.is_git_repo
|
|
:init (fn [self]
|
|
(let [git-status vim.b.gitsigns_status_dict]
|
|
(set self.git-head git-status.head)
|
|
(set self.git-added (or git-status.added 0))
|
|
(set self.git-removed (or git-status.removed 0))
|
|
(set self.git-changed (or git-status.changed 0))
|
|
(set self.check-count
|
|
(max-number [self.git-added
|
|
self.git-removed
|
|
self.git-changed]))
|
|
(set self.check-length (length self.git-head))))
|
|
1 mod.space-if-count-or-length
|
|
2 {:provider #(.. " " $1.git-head " ")
|
|
:hl {:fg colors.black :bg colors.orange :bold true}}
|
|
3 mod.space
|
|
4 {:provider (fn [self]
|
|
(.. " " self.git-added " "))
|
|
:hl {:fg colors.bright_green}}
|
|
5 {:provider (fn [self]
|
|
(.. " " self.git-removed " "))
|
|
:hl {:fg colors.bright_red}}
|
|
6 {:provider (fn [self]
|
|
(.. " " self.git-changed))
|
|
:hl {:fg colors.cyan}}})
|
|
;; Diagnostics
|
|
(set mod.diagnostics {:condition heirline-conditions.has_diagnostics
|
|
:init (fn [self]
|
|
(let [d vim.diagnostic]
|
|
(set self.errors
|
|
(length (d.get 0
|
|
{:severity d.severity.ERROR})))
|
|
(set self.warnings
|
|
(length (d.get 0
|
|
{:severity d.severity.WARN})))
|
|
(set self.info
|
|
(length (d.get 0
|
|
{:severity d.severity.INFO})))
|
|
(set self.hints
|
|
(length (d.get 0
|
|
{:severity d.severity.HINT})))
|
|
(set self.check-count
|
|
(max-number [self.errors
|
|
self.warnings
|
|
self.info
|
|
self.hints]))))
|
|
:update [:DiagnosticChanged :BufEnter]
|
|
1 mod.space-if-count
|
|
2 {:provider (fn [self]
|
|
(let [spacer (if (or (> self.warnings 0)
|
|
(> self.info 0)
|
|
(> self.hints 0))
|
|
" " "")]
|
|
(when (> self.errors 0)
|
|
(.. " " self.errors spacer))))
|
|
:hl {:fg colors.red}}
|
|
3 {:provider (fn [self]
|
|
(let [spacer (if (or (> self.info 0)
|
|
(> self.hints 0))
|
|
" " "")]
|
|
(when (> self.warnings 0)
|
|
(.. " " self.warnings spacer))))
|
|
:hl {:fg colors.yellow}}
|
|
4 {:provider (fn [self]
|
|
(let [spacer (if (> self.hints 0) " " "")]
|
|
(when (> self.info 0)
|
|
(.. " " self.info spacer))))
|
|
:hl {:fg colors.cyan}}
|
|
5 {:provider (fn [self]
|
|
(when (> self.hints 0)
|
|
(.. " " self.hints)))
|
|
:hl {:fg colors.cyan}}})
|
|
;; Current Function
|
|
(set mod.current-function
|
|
{:condition heirline-conditions.lsp_attached
|
|
:init (fn [self]
|
|
(let [ctx vim.b.nifoc_lsp_current_context]
|
|
(if (and (not= ctx nil) (> (ctx:len) 0))
|
|
(do
|
|
(set self.current-ctx ctx)
|
|
(set self.check-length 1))
|
|
(do
|
|
(set self.current-ctx nil)
|
|
(set self.check-length 0)))))
|
|
1 mod.space-if-length
|
|
2 {:provider (fn [self]
|
|
self.current-ctx)
|
|
:hl {:fg colors.white}}})
|
|
;; Buffer Options
|
|
(set mod.buffer-options
|
|
{:static {:format {:dos "" :unix "" :mac ""}}
|
|
:hl {:fg colors.black :bg colors.orange}
|
|
1 mod.space
|
|
2 {:condition #(buffer-variable-exists? :nifoc_lsp_enabled)
|
|
:provider " "
|
|
:on_click {:name :heirline_buffer_options_lsp
|
|
:callback #(deferred_cmd {:cmd :LspInfo} 200)}}
|
|
3 {:condition #(formatting.active?) :provider " "}
|
|
4 {:condition #(nifoc-treesitter.active?) :provider " "}
|
|
5 {:condition #vim.wo.spell :provider "暈"}
|
|
6 {:provider (fn [self]
|
|
(let [f vim.bo.fileformat]
|
|
(.. (. self :format f) " ")))}})
|
|
;; Position
|
|
(set mod.position
|
|
{:init (fn [self]
|
|
(let [pos (api.nvim_win_get_cursor 0)]
|
|
(set self.position-line (tostring (. pos 1)))
|
|
(set self.position-column (tostring (. pos 2)))))
|
|
:provider (fn [self]
|
|
(string.format " %3s:%-3s " self.position-line
|
|
self.position-column))
|
|
:hl {:fg colors.black :bg colors.purple :bold true}})
|
|
;; Scrollbar
|
|
(set mod.scrollbar {:init (fn [self]
|
|
(set self.current-line (get-current-line))
|
|
(set self.total-lines (get-total-lines)))
|
|
:static {:scrollbar-icons ["🭶"
|
|
"🭷"
|
|
"🭸"
|
|
"🭹"
|
|
"🭺"
|
|
"🭻"]}
|
|
:provider (fn [self]
|
|
(let [i (+ (math.floor (* (/ (- self.current-line
|
|
1)
|
|
self.total-lines)
|
|
(length self.scrollbar-icons)))
|
|
1)
|
|
new-scrollbar (. self :scrollbar-icons
|
|
i)]
|
|
(string.rep new-scrollbar 2)))
|
|
:hl {:fg colors.purple}})
|
|
;; Search count
|
|
(set mod.search-count
|
|
{:condition #(> vim.v.hlsearch 0)
|
|
:init (fn [self]
|
|
(set self.count (vim.fn.searchcount {:timeout 5})))
|
|
:provider (fn [self]
|
|
(string.format "[%s/%s]" self.count.current
|
|
self.count.total))
|
|
:hl {:fg colors.black :bg colors.purple}})
|
|
;; Custom Mode
|
|
|
|
(fn mod.custom-mode [str fg bg]
|
|
{:provider (.. " " str " ")
|
|
:hl {:fg (. colors fg) :bg (. colors bg) :bold true}})
|
|
|
|
(fn mod.shell-mode [fg bg]
|
|
{:provider (fn []
|
|
(.. " " vim.b.nifoc_shell_mode " "))
|
|
:hl {:fg (. colors fg) :bg (. colors bg) :bold true}})
|
|
|
|
mod)
|
|
|