wordcount.vim 1.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940
  1. " MIT License. Copyright (c) 2013-2015 Bailey Ling.
  2. " vim: et ts=2 sts=2 sw=2
  3. let s:filetypes = get(g:, 'airline#extensions#wordcount#filetypes', '\vhelp|markdown|rst|org')
  4. " adapted from http://stackoverflow.com/questions/114431/fast-word-count-function-in-vim
  5. function! s:update()
  6. if &ft !~ s:filetypes
  7. unlet! b:airline_wordcount
  8. return
  9. endif
  10. let old_status = v:statusmsg
  11. let position = getpos(".")
  12. exe "silent normal! g\<c-g>"
  13. let stat = v:statusmsg
  14. call setpos('.', position)
  15. let v:statusmsg = old_status
  16. let parts = split(stat)
  17. if len(parts) > 11
  18. let cnt = str2nr(split(stat)[11])
  19. let spc = g:airline_symbols.space
  20. let b:airline_wordcount = cnt . spc . 'words' . spc . g:airline_right_alt_sep . spc
  21. else
  22. unlet! b:airline_wordcount
  23. endif
  24. endfunction
  25. function! airline#extensions#wordcount#apply(...)
  26. if &ft =~ s:filetypes
  27. call airline#extensions#prepend_to_section('z', '%{get(b:, "airline_wordcount", "")}')
  28. endif
  29. endfunction
  30. function! airline#extensions#wordcount#init(ext)
  31. call a:ext.add_statusline_func('airline#extensions#wordcount#apply')
  32. autocmd BufReadPost,CursorMoved,CursorMovedI * call s:update()
  33. endfunction