Gaze/SpaceVim
1
0
Fork 0
mirror of https://github.com/SpaceVim/SpaceVim.git synced 2025-02-09 08:30:06 +08:00
Code Issues Projects Releases Wiki Activity

perf(SourceCounter): import sourcecounter

Browse Source
This commit is contained in:
Eric Wong 2024-07-24 19:12:58 +08:00
parent a974fb5646
commit 40edbe8d6c
2 changed files with 159 additions and 120 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

273
bundle/SourceCounter.vim/autoload/SourceCounter.vim
View File

@ -4,147 +4,186 @@ let s:NOTI = SpaceVim#api#import('notify')
scriptencoding utf-8 scriptencoding utf-8
let s:support_ft = ['vim', 'java', 'c', 'py', 'md', 'txt'] let s:support_ft = ['vim', 'java', 'c', 'py', 'md', 'txt']
function! SourceCounter#View(bang, ...) abort function! SourceCounter#View(bang, ...) abort
let result = [] call s:NOTI.notify(string(a:000))
if a:0 let fts = []
let fts = a:000 let dirs = []
else let result = {}
let fts = s:support_ft let argv_type = ''
for argv in a:000
if argv == '-d'
let argv_type = 'dir'
continue
elseif argv == '-ft'
let argv_type = 'filetype'
continue
endif endif
call s:NOTI.notify('counting for: ' . join(fts, ', ')) if argv_type == 'dir'
" return call add(dirs, argv)
elseif argv_type == 'filetype'
call add(fts, argv)
endif
endfor
call s:NOTI.notify('counting for: ' . join(fts, ', '))
" return
for dir in dirs
for ft in fts for ft in fts
let _rs = s:counter(ft) let _rs = s:counter(ft, dir)
if !empty(_rs) if !empty(_rs)
call add(result, _rs) if has_key(result, ft)
let result[ft].files = result[ft].files + _rs.files
let result[ft].lines = result[ft].lines + _rs.lines
else
let result[ft] = {
\ 'files' : _rs.files,
\ 'lines' : _rs.lines,
\ }
endif endif
endif
endfor endfor
let result = sort(deepcopy(result), function('s:compare')) endfor
let table = s:draw_table(result) let result = sort(s:build(result), function('s:compare'))
if a:bang let table = s:draw_table(result)
tabnew if a:bang
for line in table tabnew
call append(line('$'), line) setlocal buftype=nofile bufhidden=wipe nobuflisted nolist noswapfile nowrap cursorline nospell nonumber norelativenumber
endfor setlocal modifiable
else noautocmd normal! gg"_dG
call s:NOTI.notify(join(table, "\n")) for line in table
endif call append(line('$'), line)
endfor
normal! G
setlocal nomodifiable
nnoremap <buffer><silent> q :bd<CR>
else
call s:NOTI.notify(join(table, "\n"))
endif
endfunction
function! s:build(rst) abort
let rst = []
for k in keys(a:rst)
call add(rst, [k, a:rst[k].files, a:rst[k].lines])
endfor
return rst
endfunction endfunction
function! s:compare(a, b) abort function! s:compare(a, b) abort
let m = get(g:, 'source_counter_sort', 'files') let m = get(g:, 'source_counter_sort', 'files')
if m ==# 'lines' if m ==# 'lines'
return a:a[2] == a:b[2] ? 0 : a:a[2] > a:b[2] ? -1 : 1 return a:a[2] == a:b[2] ? 0 : a:a[2] > a:b[2] ? -1 : 1
else else
return a:a[1] == a:b[1] ? 0 : a:a[1] > a:b[1] ? -1 : 1 return a:a[1] == a:b[1] ? 0 : a:a[1] > a:b[1] ? -1 : 1
endif endif
endfunction endfunction
" https://en.wikipedia.org/wiki/Box-drawing_character " https://en.wikipedia.org/wiki/Box-drawing_character
function! s:draw_table(rst) abort function! s:draw_table(rst) abort
if &encoding ==# 'utf-8' if &encoding ==# 'utf-8'
let top_left_corner = '╭' let top_left_corner = '╭'
let top_right_corner = '╮' let top_right_corner = '╮'
let bottom_left_corner = '╰' let bottom_left_corner = '╰'
let bottom_right_corner = '╯' let bottom_right_corner = '╯'
let side = '│' let side = '│'
let top_bottom_side = '─' let top_bottom_side = '─'
let middle = '┼' let middle = '┼'
let top_middle = '┬' let top_middle = '┬'
let left_middle = '├' let left_middle = '├'
let right_middle = '┤' let right_middle = '┤'
let bottom_middle = '┴' let bottom_middle = '┴'
else else
let top_left_corner = '*' let top_left_corner = '*'
let top_right_corner = '*' let top_right_corner = '*'
let bottom_left_corner = '*' let bottom_left_corner = '*'
let bottom_right_corner = '*' let bottom_right_corner = '*'
let side = '|' let side = '|'
let top_bottom_side = '-' let top_bottom_side = '-'
let middle = '*' let middle = '*'
let top_middle = '*' let top_middle = '*'
let left_middle = '*' let left_middle = '*'
let right_middle = '*' let right_middle = '*'
let bottom_middle = '*' let bottom_middle = '*'
endif endif
let table = [] let table = []
let top_line = top_left_corner let top_line = top_left_corner
\ . repeat(top_bottom_side, 15) \ . repeat(top_bottom_side, 15)
\ . top_middle \ . top_middle
\ . repeat(top_bottom_side, 15) \ . repeat(top_bottom_side, 15)
\ . top_middle \ . top_middle
\ . repeat(top_bottom_side, 15) \ . repeat(top_bottom_side, 15)
\ . top_right_corner \ . top_right_corner
let middle_line = left_middle let middle_line = left_middle
\ . repeat(top_bottom_side, 15) \ . repeat(top_bottom_side, 15)
\ . middle \ . middle
\ . repeat(top_bottom_side, 15) \ . repeat(top_bottom_side, 15)
\ . middle \ . middle
\ . repeat(top_bottom_side, 15) \ . repeat(top_bottom_side, 15)
\ . right_middle \ . right_middle
let bottom_line = bottom_left_corner let bottom_line = bottom_left_corner
\ . repeat(top_bottom_side, 15) \ . repeat(top_bottom_side, 15)
\ . bottom_middle \ . bottom_middle
\ . repeat(top_bottom_side, 15) \ . repeat(top_bottom_side, 15)
\ . bottom_middle \ . bottom_middle
\ . repeat(top_bottom_side, 15) \ . repeat(top_bottom_side, 15)
\ . bottom_right_corner \ . bottom_right_corner
call add(table, top_line) call add(table, top_line)
let result = [['filetype', 'files', 'lines']] + a:rst let result = [['filetype', 'files', 'lines']] + a:rst
for rsl in result for rsl in result
let ft_line = side let ft_line = side
\ . rsl[0] . repeat(' ', 15 - strwidth(rsl[0])) \ . rsl[0] . repeat(' ', 15 - strwidth(rsl[0]))
\ . side \ . side
\ . rsl[1] . repeat(' ', 15 - strwidth(rsl[1])) \ . rsl[1] . repeat(' ', 15 - strwidth(rsl[1]))
\ . side \ . side
\ . rsl[2] . repeat(' ', 15 - strwidth(rsl[2])) \ . rsl[2] . repeat(' ', 15 - strwidth(rsl[2]))
\ . side \ . side
call add(table, ft_line) call add(table, ft_line)
call add(table, middle_line) call add(table, middle_line)
endfor endfor
let table[-1] = bottom_line let table[-1] = bottom_line
return table return table
endfunction endfunction
function! s:list_files_stdout(id, data, event) abort function! s:list_files_stdout(id, data, event) abort
endfunction endfunction
function! s:list_files_exit(id, data, event) abort function! s:list_files_exit(id, data, event) abort
endfunction endfunction
function! s:counter(ft) abort function! s:counter(ft, dir) abort
let path = getcwd() if executable('ag')
let partten = '**/*.' . a:ft
if executable('ag')
if has('nvim')
let files = systemlist(['ag','-g', '.' . a:ft . '$'])
else
let files = split(system('ag -g .'.a:ft.'$'),nr2char(10))
endif
else
let files = globpath(l:path, l:partten, 0, 1)
endif
if len(files) == 0
return []
endif
let lines = 0
if has('nvim') if has('nvim')
if len(files) > 380 let files = systemlist(['ag','-g', '.' . a:ft . '$'])
while !empty(files)
let lines += matchstr(systemlist(['wc', '-l'] + remove(files, 0, min([380, len(files) - 1])))[-1], '\d\+')
endwhile
else
let lines = matchstr(systemlist(['wc', '-l'] + files)[-1], '\d\+')
endif
else else
for fl in files let files = split(system('ag -g .'.a:ft.'$'),nr2char(10))
let lines += str2nr(matchstr(system('wc -l '. fl), '\d\+'))
endfor
endif endif
return [a:ft, len(files), lines] else
let partten = '**/*.' . a:ft
let files = globpath(a:dir, l:partten, 0, 1)
endif
if len(files) == 0
return []
endif
let lines = 0
let _file_count = len(files)
if has('nvim')
if len(files) > 380
while !empty(files)
let _fs = remove(files, 0, min([380, len(files) - 1]))
let _rst = systemlist(['wc', '-l'] + _fs)[-1]
let lines += matchstr(_rst, '\d\+')
endwhile
else
let lines = matchstr(systemlist(['wc', '-l'] + files)[-1], '\d\+')
endif
else
for fl in files
let lines += str2nr(matchstr(system('wc -l '. fl), '\d\+'))
endfor
endif
return {'files' : _file_count, 'lines' : lines}
endfunction endfunction

6
bundle/SourceCounter.vim/doc/SourceCounter.vim.txt
View File

@ -3,9 +3,9 @@ wsdjeg *SourceCounter.vim*
============================================================================== ==============================================================================
CONTENTS *SourceCounter.vim-contents* CONTENTS *SourceCounter.vim-contents*
1. Introduction....................................|SourceCounter.vim-intro| 1. Introduction..................................... |SourceCounter.vim-intro|
2. Configuration..................................|SourceCounter.vim-config| 2. Configuration................................... |SourceCounter.vim-config|
3. Commands.....................................|SourceCounter.vim-commands| 3. Commands...................................... |SourceCounter.vim-commands|
============================================================================== ==============================================================================
INTRODUCTION *SourceCounter.vim-intro* INTRODUCTION *SourceCounter.vim-intro*