44 lines
2.2 KiB
VimL
44 lines
2.2 KiB
VimL
" Based on <https://github.com/vim-airline/vim-airline/blob/70b06be4b067fec44756e843e2445cce5c97082f/autoload/airline/extensions/example.vim>
|
|
|
|
function! airline#extensions#dotfiles_filesize#init(ext) abort
|
|
call airline#parts#define_function('dotfiles_filesize', 'airline#extensions#dotfiles_filesize#get')
|
|
call a:ext.add_statusline_func('airline#extensions#dotfiles_filesize#apply')
|
|
endfunction
|
|
|
|
function! airline#extensions#dotfiles_filesize#apply(...) abort
|
|
call airline#extensions#append_to_section('y', airline#section#create_right(['', '', 'dotfiles_filesize']))
|
|
endfunction
|
|
|
|
" Finally, this function will be invoked from the statusline.
|
|
function! airline#extensions#dotfiles_filesize#get() abort
|
|
" Use several preliminary checks to prevent frequent updates. You see,
|
|
" wordcount() has to iterate the entire file to calculate its byte size.
|
|
" <https://github.com/vim-airline/vim-airline/blob/ecac148e19fe28d0f13af5f99d9500c2eadefe4c/autoload/airline/extensions/wordcount.vim#L78-L82>
|
|
" <https://github.com/vim-airline/vim-airline/blob/2e9df43962539e3a05e5571c63ccf6356451d46c/autoload/airline/extensions/lsp.vim#L95-L100>
|
|
" Implementation of wordcount: <https://github.com/neovim/neovim/blob/dab6b08a1e5571d0a0c4cb1f2f1af7000870c652/src/nvim/ops.c#L5568-L5808>
|
|
if empty(get(b:, 'dotfiles_filesize_str', '')) ||
|
|
\ (get(b:, 'dotfiles_filesize_changedtick', 0) !=# b:changedtick &&
|
|
\ reltimefloat(reltime()) - get(b:, 'dotfiles_filesize_timer') >=# get(g:, 'airline#extensions#dotfiles_filesize#update_delay', 0))
|
|
let bytes = wordcount().bytes
|
|
let b:dotfiles_filesize = bytes
|
|
|
|
let factor = 1
|
|
for unit in ['B', 'K', 'M', 'G']
|
|
let next_factor = factor * 1024
|
|
if bytes <# next_factor
|
|
let number_str = printf('%.2f', (bytes * 1.0) / factor)
|
|
" remove trailing zeros
|
|
let number_str = substitute(number_str, '\v(\.0*)=$', '', '')
|
|
let b:dotfiles_filesize_str = number_str . unit
|
|
break
|
|
endif
|
|
let factor = next_factor
|
|
endfor
|
|
|
|
let b:dotfiles_filesize_changedtick = b:changedtick
|
|
let b:dotfiles_filesize_timer = reltimefloat(reltime())
|
|
endif
|
|
|
|
return b:dotfiles_filesize_str
|
|
endfunction
|