123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339 |
- let s:plugin_name = expand('<sfile>:t:r')
- let s:vital_base_dir = expand('<sfile>:h')
- let s:project_root = expand('<sfile>:h:h:h')
- let s:is_vital_vim = s:plugin_name is# 'vital'
- let s:loaded = {}
- let s:cache_sid = {}
- " function() wrapper
- if v:version > 703 || v:version == 703 && has('patch1170')
- function! s:_function(fstr) abort
- return function(a:fstr)
- endfunction
- else
- function! s:_SID() abort
- return matchstr(expand('<sfile>'), '<SNR>\zs\d\+\ze__SID$')
- endfunction
- let s:_s = '<SNR>' . s:_SID() . '_'
- function! s:_function(fstr) abort
- return function(substitute(a:fstr, 's:', s:_s, 'g'))
- endfunction
- endif
- function! vital#{s:plugin_name}#new() abort
- return s:new(s:plugin_name)
- endfunction
- function! vital#{s:plugin_name}#import(...) abort
- if !exists('s:V')
- let s:V = s:new(s:plugin_name)
- endif
- return call(s:V.import, a:000, s:V)
- endfunction
- let s:Vital = {}
- function! s:new(plugin_name) abort
- let base = deepcopy(s:Vital)
- let base._plugin_name = a:plugin_name
- return base
- endfunction
- function! s:vital_files() abort
- if !exists('s:vital_files')
- let s:vital_files = map(
- \ s:is_vital_vim ? s:_global_vital_files() : s:_self_vital_files(),
- \ 'fnamemodify(v:val, ":p:gs?[\\\\/]?/?")')
- endif
- return copy(s:vital_files)
- endfunction
- let s:Vital.vital_files = s:_function('s:vital_files')
- function! s:import(name, ...) abort dict
- let target = {}
- let functions = []
- for a in a:000
- if type(a) == type({})
- let target = a
- elseif type(a) == type([])
- let functions = a
- endif
- unlet a
- endfor
- let module = self._import(a:name)
- if empty(functions)
- call extend(target, module, 'keep')
- else
- for f in functions
- if has_key(module, f) && !has_key(target, f)
- let target[f] = module[f]
- endif
- endfor
- endif
- return target
- endfunction
- let s:Vital.import = s:_function('s:import')
- function! s:load(...) abort dict
- for arg in a:000
- let [name; as] = type(arg) == type([]) ? arg[: 1] : [arg, arg]
- let target = split(join(as, ''), '\W\+')
- let dict = self
- let dict_type = type({})
- while !empty(target)
- let ns = remove(target, 0)
- if !has_key(dict, ns)
- let dict[ns] = {}
- endif
- if type(dict[ns]) == dict_type
- let dict = dict[ns]
- else
- unlet dict
- break
- endif
- endwhile
- if exists('dict')
- call extend(dict, self._import(name))
- endif
- unlet arg
- endfor
- return self
- endfunction
- let s:Vital.load = s:_function('s:load')
- function! s:unload() abort dict
- let s:loaded = {}
- let s:cache_sid = {}
- unlet! s:vital_files
- endfunction
- let s:Vital.unload = s:_function('s:unload')
- function! s:exists(name) abort dict
- if a:name !~# '\v^\u\w*%(\.\u\w*)*$'
- throw 'vital: Invalid module name: ' . a:name
- endif
- return s:_module_path(a:name) isnot# ''
- endfunction
- let s:Vital.exists = s:_function('s:exists')
- function! s:search(pattern) abort dict
- let paths = s:_extract_files(a:pattern, self.vital_files())
- let modules = sort(map(paths, 's:_file2module(v:val)'))
- return s:_uniq(modules)
- endfunction
- let s:Vital.search = s:_function('s:search')
- function! s:plugin_name() abort dict
- return self._plugin_name
- endfunction
- let s:Vital.plugin_name = s:_function('s:plugin_name')
- function! s:_self_vital_files() abort
- let builtin = printf('%s/__%s__/', s:vital_base_dir, s:plugin_name)
- let installed = printf('%s/_%s/', s:vital_base_dir, s:plugin_name)
- let base = builtin . ',' . installed
- return split(globpath(base, '**/*.vim', 1), "\n")
- endfunction
- function! s:_global_vital_files() abort
- let pattern = 'autoload/vital/__*__/**/*.vim'
- return split(globpath(&runtimepath, pattern, 1), "\n")
- endfunction
- function! s:_extract_files(pattern, files) abort
- let tr = {'.': '/', '*': '[^/]*', '**': '.*'}
- let target = substitute(a:pattern, '\.\|\*\*\?', '\=tr[submatch(0)]', 'g')
- let regexp = printf('autoload/vital/[^/]\+/%s.vim$', target)
- return filter(a:files, 'v:val =~# regexp')
- endfunction
- function! s:_file2module(file) abort
- let filename = fnamemodify(a:file, ':p:gs?[\\/]?/?')
- let tail = matchstr(filename, 'autoload/vital/_\w\+/\zs.*\ze\.vim$')
- return join(split(tail, '[\\/]\+'), '.')
- endfunction
- " @param {string} name e.g. Data.List
- function! s:_import(name) abort dict
- if has_key(s:loaded, a:name)
- return copy(s:loaded[a:name])
- endif
- let module = self._get_module(a:name)
- if has_key(module, '_vital_created')
- call module._vital_created(module)
- endif
- let export_module = filter(copy(module), 'v:key =~# "^\\a"')
- " Cache module before calling module.vital_loaded() to avoid cyclic
- " dependences but remove the cache if module._vital_loaded() fails.
- " let s:loaded[a:name] = export_module
- let s:loaded[a:name] = export_module
- if has_key(module, '_vital_loaded')
- try
- call module._vital_loaded(vital#{s:plugin_name}#new())
- catch
- unlet s:loaded[a:name]
- throw 'vital: fail to call ._vital_loaded(): ' . v:exception
- endtry
- endif
- return copy(s:loaded[a:name])
- endfunction
- let s:Vital._import = s:_function('s:_import')
- " s:_get_module() returns module object wihch has all script local functions.
- function! s:_get_module(name) abort dict
- let funcname = s:_import_func_name(self.plugin_name(), a:name)
- if s:_exists_autoload_func_with_source(funcname)
- return call(funcname, [])
- else
- return s:_get_builtin_module(a:name)
- endif
- endfunction
- function! s:_get_builtin_module(name) abort
- return s:sid2sfuncs(s:_module_sid(a:name))
- endfunction
- if s:is_vital_vim
- " For vital.vim, we can use s:_get_builtin_module directly
- let s:Vital._get_module = s:_function('s:_get_builtin_module')
- else
- let s:Vital._get_module = s:_function('s:_get_module')
- endif
- function! s:_import_func_name(plugin_name, module_name) abort
- return printf('vital#_%s#%s#import', a:plugin_name, s:_dot_to_sharp(a:module_name))
- endfunction
- function! s:_module_sid(name) abort
- let path = s:_module_path(a:name)
- if !filereadable(path)
- throw 'vital: module not found: ' . a:name
- endif
- let vital_dir = s:is_vital_vim ? '__\w\+__' : printf('_\{1,2}%s\%%(__\)\?', s:plugin_name)
- let base = join([vital_dir, ''], '[/\\]\+')
- let p = base . substitute('' . a:name, '\.', '[/\\\\]\\+', 'g')
- let sid = s:_sid(path, p)
- if !sid
- call s:_source(path)
- let sid = s:_sid(path, p)
- if !sid
- throw printf('vital: cannot get <SID> from path: %s', path)
- endif
- endif
- return sid
- endfunction
- function! s:_module_path(name) abort
- return get(s:_extract_files(a:name, s:vital_files()), 0, '')
- endfunction
- function! s:_module_sid_base_dir() abort
- return s:is_vital_vim ? &rtp : s:project_root
- endfunction
- function! s:_dot_to_sharp(name) abort
- return substitute(a:name, '\.', '#', 'g')
- endfunction
- " It will sources autoload file if a given func is not already defined.
- function! s:_exists_autoload_func_with_source(funcname) abort
- if exists('*' . a:funcname)
- " Return true if a given func is already defined
- return 1
- endif
- " source a file which may include a given func definition and try again.
- let path = 'autoload/' . substitute(substitute(a:funcname, '#[^#]*$', '.vim', ''), '#', '/', 'g')
- call s:_runtime(path)
- return exists('*' . a:funcname)
- endfunction
- function! s:_runtime(path) abort
- execute 'runtime' fnameescape(a:path)
- endfunction
- function! s:_source(path) abort
- execute 'source' fnameescape(a:path)
- endfunction
- " @vimlint(EVL102, 1, l:_)
- " @vimlint(EVL102, 1, l:__)
- function! s:_sid(path, filter_pattern) abort
- let unified_path = s:_unify_path(a:path)
- if has_key(s:cache_sid, unified_path)
- return s:cache_sid[unified_path]
- endif
- for line in filter(split(s:_redir(':scriptnames'), "\n"), 'v:val =~# a:filter_pattern')
- let [_, sid, path; __] = matchlist(line, '^\s*\(\d\+\):\s\+\(.\+\)\s*$')
- if s:_unify_path(path) is# unified_path
- let s:cache_sid[unified_path] = sid
- return s:cache_sid[unified_path]
- endif
- endfor
- return 0
- endfunction
- function! s:_redir(cmd) abort
- let [save_verbose, save_verbosefile] = [&verbose, &verbosefile]
- set verbose=0 verbosefile=
- redir => res
- silent! execute a:cmd
- redir END
- let [&verbose, &verbosefile] = [save_verbose, save_verbosefile]
- return res
- endfunction
- if filereadable(expand('<sfile>:r') . '.VIM') " is case-insensitive or not
- let s:_unify_path_cache = {}
- " resolve() is slow, so we cache results.
- " Note: On windows, vim can't expand path names from 8.3 formats.
- " So if getting full path via <sfile> and $HOME was set as 8.3 format,
- " vital load duplicated scripts. Below's :~ avoid this issue.
- function! s:_unify_path(path) abort
- if has_key(s:_unify_path_cache, a:path)
- return s:_unify_path_cache[a:path]
- endif
- let value = tolower(fnamemodify(resolve(fnamemodify(
- \ a:path, ':p')), ':~:gs?[\\/]?/?'))
- let s:_unify_path_cache[a:path] = value
- return value
- endfunction
- else
- function! s:_unify_path(path) abort
- return resolve(fnamemodify(a:path, ':p:gs?[\\/]?/?'))
- endfunction
- endif
- " copied and modified from Vim.ScriptLocal
- let s:SNR = join(map(range(len("\<SNR>")), '"[\\x" . printf("%0x", char2nr("\<SNR>"[v:val])) . "]"'), '')
- function! s:sid2sfuncs(sid) abort
- let fs = split(s:_redir(printf(':function /^%s%s_', s:SNR, a:sid)), "\n")
- let r = {}
- let pattern = printf('\m^function\s<SNR>%d_\zs\w\{-}\ze(', a:sid)
- for fname in map(fs, 'matchstr(v:val, pattern)')
- let r[fname] = function(s:_sfuncname(a:sid, fname))
- endfor
- return r
- endfunction
- "" Return funcname of script local functions with SID
- function! s:_sfuncname(sid, funcname) abort
- return printf('<SNR>%s_%s', a:sid, a:funcname)
- endfunction
- if exists('*uniq')
- function! s:_uniq(list) abort
- return uniq(a:list)
- endfunction
- else
- function! s:_uniq(list) abort
- let i = len(a:list) - 1
- while 0 < i
- if a:list[i] ==# a:list[i - 1]
- call remove(a:list, i)
- endif
- let i -= 1
- endwhile
- return a:list
- endfunction
- endif
|