mirror of
https://github.com/SpaceVim/SpaceVim.git
synced 2025-04-14 15:19:12 +08:00
163 lines
3.8 KiB
VimL
163 lines
3.8 KiB
VimL
"=============================================================================
|
|
" dict.vim --- SpaceVim dict API
|
|
" Copyright (c) 2016-2020 Wang Shidong & Contributors
|
|
" Author: Wang Shidong < wsdjeg at 163.com >
|
|
" URL: https://spacevim.org
|
|
" License: GPLv3
|
|
"=============================================================================
|
|
|
|
""
|
|
" @section data#dict, api-data-dict
|
|
" @parentsection api
|
|
" provides some functions to manipulate a dict.
|
|
"
|
|
" make({keys}, {values}[, {fill}])
|
|
"
|
|
" make a dictionary from two list, the {keys} and {values}.
|
|
"
|
|
" swap({dict})
|
|
"
|
|
" swap the keys and values in a dictionary.
|
|
"
|
|
" make_index
|
|
"
|
|
" make a dictionary from a list, use
|
|
|
|
|
|
|
|
|
|
|
|
function! SpaceVim#api#data#dict#get() abort
|
|
return map({
|
|
\ 'make' : '',
|
|
\ 'swap' : '',
|
|
\ 'make_index' : '',
|
|
\ 'pick' : '',
|
|
\ 'omit' : '',
|
|
\ 'clear' : '',
|
|
\ 'max_by' : '',
|
|
\ 'min_by' : '',
|
|
\ 'foldl' : '',
|
|
\ 'foldr' : '',
|
|
\ 'entrys' : '',
|
|
\ },
|
|
\ "function('s:' . v:key)"
|
|
\ )
|
|
endfunction
|
|
|
|
|
|
if get(g:, '_spacevim_if_lua', 0) || has('nvim')
|
|
function! s:entrys(dict) abort
|
|
lua require("spacevim.api.data.dict").entrys(
|
|
\ require("spacevim").eval("a:dict")
|
|
\ )
|
|
endfunction
|
|
function! s:pick(dict, keys) abort
|
|
lua require("spacevim.api.data.dict").pick(
|
|
\ require("spacevim").eval("a:dict"),
|
|
\ require("spacevim").eval("a:keys")
|
|
\ )
|
|
endfunction
|
|
else
|
|
function! s:entrys(dict) abort
|
|
let entrys = []
|
|
for key in keys(a:dict)
|
|
call add(entrys, {key : a:dict[key]})
|
|
endfor
|
|
return entrys
|
|
endfunction
|
|
function! s:pick(dict, keys) abort
|
|
let new_dict = {}
|
|
for key in a:keys
|
|
if has_key(a:dict, key)
|
|
let new_dict[key] = a:dict[key]
|
|
endif
|
|
endfor
|
|
return new_dict
|
|
endfunction
|
|
endif
|
|
|
|
function! s:make(keys, values, ...) abort
|
|
let dict = {}
|
|
let fill = a:0 ? a:1 : 0
|
|
for i in range(len(a:keys))
|
|
let key = type(a:keys[i]) == type('') ? a:keys[i] : string(a:keys[i])
|
|
if key ==# ''
|
|
throw "SpaceVim API: data#dict: Can't use an empty string for key."
|
|
endif
|
|
let dict[key] = get(a:values, i, fill)
|
|
endfor
|
|
return dict
|
|
endfunction
|
|
|
|
" Swaps keys and values
|
|
function! s:swap(dict) abort
|
|
return s:make(values(a:dict), keys(a:dict))
|
|
endfunction
|
|
|
|
" Makes a index dict from a list
|
|
function! s:make_index(list, ...) abort
|
|
let value = a:0 ? a:1 : 1
|
|
return s:make(a:list, [], value)
|
|
endfunction
|
|
|
|
|
|
function! s:omit(dict, keys) abort
|
|
let new_dict = copy(a:dict)
|
|
for key in a:keys
|
|
if has_key(a:dict, key)
|
|
call remove(new_dict, key)
|
|
endif
|
|
endfor
|
|
return new_dict
|
|
endfunction
|
|
|
|
function! s:clear(dict) abort
|
|
for key in keys(a:dict)
|
|
call remove(a:dict, key)
|
|
endfor
|
|
return a:dict
|
|
endfunction
|
|
|
|
function! s:_max_by(dict, expr) abort
|
|
let dict = s:swap(map(copy(a:dict), a:expr))
|
|
let key = dict[max(keys(dict))]
|
|
return [key, a:dict[key]]
|
|
endfunction
|
|
|
|
function! s:max_by(dict, expr) abort
|
|
if empty(a:dict)
|
|
throw 'SpaceVim API: data#dict: Empty dictionary'
|
|
endif
|
|
return s:_max_by(a:dict, a:expr)
|
|
endfunction
|
|
|
|
function! s:min_by(dict, expr) abort
|
|
if empty(a:dict)
|
|
throw 'SpaceVim API: data#dict: Empty dictionary'
|
|
endif
|
|
return s:_max_by(a:dict, '-(' . a:expr . ')')
|
|
endfunction
|
|
|
|
function! s:_foldl(f, init, xs) abort
|
|
let memo = a:init
|
|
for [k, v] in a:xs
|
|
let expr = substitute(a:f, 'v:key', string(k), 'g')
|
|
let expr = substitute(expr, 'v:val', string(v), 'g')
|
|
let expr = substitute(expr, 'v:memo', string(memo), 'g')
|
|
unlet memo
|
|
let memo = eval(expr)
|
|
endfor
|
|
return memo
|
|
endfunction
|
|
|
|
function! s:foldl(f, init, dict) abort
|
|
return s:_foldl(a:f, a:init, items(a:dict))
|
|
endfunction
|
|
|
|
function! s:foldr(f, init, dict) abort
|
|
return s:_foldl(a:f, a:init, reverse(items(a:dict)))
|
|
endfunction
|
|
|
|
" vim:set et sw=2 cc=80:
|