Fri, 22 Nov 2019 14:43:37 -0600
improve tig
724 | 1 | " vim-plug: Vim plugin manager |
2 | " ============================ | |
3 | " | |
4 | " Download plug.vim and put it in ~/.vim/autoload | |
5 | " | |
6 | " curl -fLo ~/.vim/autoload/plug.vim --create-dirs \ | |
7 | " https://raw.githubusercontent.com/junegunn/vim-plug/master/plug.vim | |
8 | " | |
9 | " Edit your .vimrc | |
10 | " | |
11 | " call plug#begin('~/.vim/plugged') | |
12 | " | |
13 | " " Make sure you use single quotes | |
14 | " | |
15 | " " Shorthand notation; fetches https://github.com/junegunn/vim-easy-align | |
16 | " Plug 'junegunn/vim-easy-align' | |
17 | " | |
18 | " " Any valid git URL is allowed | |
19 | " Plug 'https://github.com/junegunn/vim-github-dashboard.git' | |
20 | " | |
21 | " " Multiple Plug commands can be written in a single line using | separators | |
22 | " Plug 'SirVer/ultisnips' | Plug 'honza/vim-snippets' | |
23 | " | |
24 | " " On-demand loading | |
25 | " Plug 'scrooloose/nerdtree', { 'on': 'NERDTreeToggle' } | |
26 | " Plug 'tpope/vim-fireplace', { 'for': 'clojure' } | |
27 | " | |
28 | " " Using a non-master branch | |
29 | " Plug 'rdnetto/YCM-Generator', { 'branch': 'stable' } | |
30 | " | |
31 | " " Using a tagged release; wildcard allowed (requires git 1.9.2 or above) | |
32 | " Plug 'fatih/vim-go', { 'tag': '*' } | |
33 | " | |
34 | " " Plugin options | |
35 | " Plug 'nsf/gocode', { 'tag': 'v.20150303', 'rtp': 'vim' } | |
36 | " | |
37 | " " Plugin outside ~/.vim/plugged with post-update hook | |
38 | " Plug 'junegunn/fzf', { 'dir': '~/.fzf', 'do': './install --all' } | |
39 | " | |
40 | " " Unmanaged plugin (manually installed and updated) | |
41 | " Plug '~/my-prototype-plugin' | |
42 | " | |
43 | " " Initialize plugin system | |
44 | " call plug#end() | |
45 | " | |
46 | " Then reload .vimrc and :PlugInstall to install plugins. | |
47 | " | |
48 | " Plug options: | |
49 | " | |
50 | "| Option | Description | | |
51 | "| ----------------------- | ------------------------------------------------ | | |
52 | "| `branch`/`tag`/`commit` | Branch/tag/commit of the repository to use | | |
53 | "| `rtp` | Subdirectory that contains Vim plugin | | |
54 | "| `dir` | Custom directory for the plugin | | |
55 | "| `as` | Use different name for the plugin | | |
56 | "| `do` | Post-update hook (string or funcref) | | |
57 | "| `on` | On-demand loading: Commands or `<Plug>`-mappings | | |
58 | "| `for` | On-demand loading: File types | | |
59 | "| `frozen` | Do not update unless explicitly specified | | |
60 | " | |
61 | " More information: https://github.com/junegunn/vim-plug | |
62 | " | |
63 | " | |
64 | " Copyright (c) 2017 Junegunn Choi | |
65 | " | |
66 | " MIT License | |
67 | " | |
68 | " Permission is hereby granted, free of charge, to any person obtaining | |
69 | " a copy of this software and associated documentation files (the | |
70 | " "Software"), to deal in the Software without restriction, including | |
71 | " without limitation the rights to use, copy, modify, merge, publish, | |
72 | " distribute, sublicense, and/or sell copies of the Software, and to | |
73 | " permit persons to whom the Software is furnished to do so, subject to | |
74 | " the following conditions: | |
75 | " | |
76 | " The above copyright notice and this permission notice shall be | |
77 | " included in all copies or substantial portions of the Software. | |
78 | " | |
79 | " THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, | |
80 | " EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF | |
81 | " MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND | |
82 | " NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE | |
83 | " LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION | |
84 | " OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION | |
85 | " WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. | |
86 | ||
87 | if exists('g:loaded_plug') | |
88 | finish | |
89 | endif | |
90 | let g:loaded_plug = 1 | |
91 | ||
92 | let s:cpo_save = &cpo | |
93 | set cpo&vim | |
94 | ||
95 | let s:plug_src = 'https://github.com/junegunn/vim-plug.git' | |
96 | let s:plug_tab = get(s:, 'plug_tab', -1) | |
97 | let s:plug_buf = get(s:, 'plug_buf', -1) | |
98 | let s:mac_gui = has('gui_macvim') && has('gui_running') | |
99 | let s:is_win = has('win32') | |
100 | let s:nvim = has('nvim-0.2') || (has('nvim') && exists('*jobwait') && !s:is_win) | |
101 | let s:vim8 = has('patch-8.0.0039') && exists('*job_start') | |
102 | let s:me = resolve(expand('<sfile>:p')) | |
103 | let s:base_spec = { 'branch': 'master', 'frozen': 0 } | |
104 | let s:TYPE = { | |
105 | \ 'string': type(''), | |
106 | \ 'list': type([]), | |
107 | \ 'dict': type({}), | |
108 | \ 'funcref': type(function('call')) | |
109 | \ } | |
110 | let s:loaded = get(s:, 'loaded', {}) | |
111 | let s:triggers = get(s:, 'triggers', {}) | |
112 | ||
113 | function! plug#begin(...) | |
114 | if a:0 > 0 | |
115 | let s:plug_home_org = a:1 | |
116 | let home = s:path(fnamemodify(expand(a:1), ':p')) | |
117 | elseif exists('g:plug_home') | |
118 | let home = s:path(g:plug_home) | |
119 | elseif !empty(&rtp) | |
120 | let home = s:path(split(&rtp, ',')[0]) . '/plugged' | |
121 | else | |
122 | return s:err('Unable to determine plug home. Try calling plug#begin() with a path argument.') | |
123 | endif | |
124 | if fnamemodify(home, ':t') ==# 'plugin' && fnamemodify(home, ':h') ==# s:first_rtp | |
125 | return s:err('Invalid plug home. '.home.' is a standard Vim runtime path and is not allowed.') | |
126 | endif | |
127 | ||
128 | let g:plug_home = home | |
129 | let g:plugs = {} | |
130 | let g:plugs_order = [] | |
131 | let s:triggers = {} | |
132 | ||
133 | call s:define_commands() | |
134 | return 1 | |
135 | endfunction | |
136 | ||
137 | function! s:define_commands() | |
138 | command! -nargs=+ -bar Plug call plug#(<args>) | |
139 | if !executable('git') | |
140 | return s:err('`git` executable not found. Most commands will not be available. To suppress this message, prepend `silent!` to `call plug#begin(...)`.') | |
141 | endif | |
142 | command! -nargs=* -bar -bang -complete=customlist,s:names PlugInstall call s:install(<bang>0, [<f-args>]) | |
143 | command! -nargs=* -bar -bang -complete=customlist,s:names PlugUpdate call s:update(<bang>0, [<f-args>]) | |
144 | command! -nargs=0 -bar -bang PlugClean call s:clean(<bang>0) | |
145 | command! -nargs=0 -bar PlugUpgrade if s:upgrade() | execute 'source' s:esc(s:me) | endif | |
146 | command! -nargs=0 -bar PlugStatus call s:status() | |
147 | command! -nargs=0 -bar PlugDiff call s:diff() | |
148 | command! -nargs=? -bar -bang -complete=file PlugSnapshot call s:snapshot(<bang>0, <f-args>) | |
149 | endfunction | |
150 | ||
151 | function! s:to_a(v) | |
152 | return type(a:v) == s:TYPE.list ? a:v : [a:v] | |
153 | endfunction | |
154 | ||
155 | function! s:to_s(v) | |
156 | return type(a:v) == s:TYPE.string ? a:v : join(a:v, "\n") . "\n" | |
157 | endfunction | |
158 | ||
159 | function! s:glob(from, pattern) | |
160 | return s:lines(globpath(a:from, a:pattern)) | |
161 | endfunction | |
162 | ||
163 | function! s:source(from, ...) | |
164 | let found = 0 | |
165 | for pattern in a:000 | |
166 | for vim in s:glob(a:from, pattern) | |
167 | execute 'source' s:esc(vim) | |
168 | let found = 1 | |
169 | endfor | |
170 | endfor | |
171 | return found | |
172 | endfunction | |
173 | ||
174 | function! s:assoc(dict, key, val) | |
175 | let a:dict[a:key] = add(get(a:dict, a:key, []), a:val) | |
176 | endfunction | |
177 | ||
178 | function! s:ask(message, ...) | |
179 | call inputsave() | |
180 | echohl WarningMsg | |
181 | let answer = input(a:message.(a:0 ? ' (y/N/a) ' : ' (y/N) ')) | |
182 | echohl None | |
183 | call inputrestore() | |
184 | echo "\r" | |
185 | return (a:0 && answer =~? '^a') ? 2 : (answer =~? '^y') ? 1 : 0 | |
186 | endfunction | |
187 | ||
188 | function! s:ask_no_interrupt(...) | |
189 | try | |
190 | return call('s:ask', a:000) | |
191 | catch | |
192 | return 0 | |
193 | endtry | |
194 | endfunction | |
195 | ||
196 | function! s:lazy(plug, opt) | |
197 | return has_key(a:plug, a:opt) && | |
198 | \ (empty(s:to_a(a:plug[a:opt])) || | |
199 | \ !isdirectory(a:plug.dir) || | |
200 | \ len(s:glob(s:rtp(a:plug), 'plugin')) || | |
201 | \ len(s:glob(s:rtp(a:plug), 'after/plugin'))) | |
202 | endfunction | |
203 | ||
204 | function! plug#end() | |
205 | if !exists('g:plugs') | |
206 | return s:err('Call plug#begin() first') | |
207 | endif | |
208 | ||
209 | if exists('#PlugLOD') | |
210 | augroup PlugLOD | |
211 | autocmd! | |
212 | augroup END | |
213 | augroup! PlugLOD | |
214 | endif | |
215 | let lod = { 'ft': {}, 'map': {}, 'cmd': {} } | |
216 | ||
217 | if exists('g:did_load_filetypes') | |
218 | filetype off | |
219 | endif | |
220 | for name in g:plugs_order | |
221 | if !has_key(g:plugs, name) | |
222 | continue | |
223 | endif | |
224 | let plug = g:plugs[name] | |
225 | if get(s:loaded, name, 0) || !s:lazy(plug, 'on') && !s:lazy(plug, 'for') | |
226 | let s:loaded[name] = 1 | |
227 | continue | |
228 | endif | |
229 | ||
230 | if has_key(plug, 'on') | |
231 | let s:triggers[name] = { 'map': [], 'cmd': [] } | |
232 | for cmd in s:to_a(plug.on) | |
233 | if cmd =~? '^<Plug>.\+' | |
234 | if empty(mapcheck(cmd)) && empty(mapcheck(cmd, 'i')) | |
235 | call s:assoc(lod.map, cmd, name) | |
236 | endif | |
237 | call add(s:triggers[name].map, cmd) | |
238 | elseif cmd =~# '^[A-Z]' | |
239 | let cmd = substitute(cmd, '!*$', '', '') | |
240 | if exists(':'.cmd) != 2 | |
241 | call s:assoc(lod.cmd, cmd, name) | |
242 | endif | |
243 | call add(s:triggers[name].cmd, cmd) | |
244 | else | |
245 | call s:err('Invalid `on` option: '.cmd. | |
246 | \ '. Should start with an uppercase letter or `<Plug>`.') | |
247 | endif | |
248 | endfor | |
249 | endif | |
250 | ||
251 | if has_key(plug, 'for') | |
252 | let types = s:to_a(plug.for) | |
253 | if !empty(types) | |
254 | augroup filetypedetect | |
255 | call s:source(s:rtp(plug), 'ftdetect/**/*.vim', 'after/ftdetect/**/*.vim') | |
256 | augroup END | |
257 | endif | |
258 | for type in types | |
259 | call s:assoc(lod.ft, type, name) | |
260 | endfor | |
261 | endif | |
262 | endfor | |
263 | ||
264 | for [cmd, names] in items(lod.cmd) | |
265 | execute printf( | |
266 | \ 'command! -nargs=* -range -bang -complete=file %s call s:lod_cmd(%s, "<bang>", <line1>, <line2>, <q-args>, %s)', | |
267 | \ cmd, string(cmd), string(names)) | |
268 | endfor | |
269 | ||
270 | for [map, names] in items(lod.map) | |
271 | for [mode, map_prefix, key_prefix] in | |
272 | \ [['i', '<C-O>', ''], ['n', '', ''], ['v', '', 'gv'], ['o', '', '']] | |
273 | execute printf( | |
274 | \ '%snoremap <silent> %s %s:<C-U>call <SID>lod_map(%s, %s, %s, "%s")<CR>', | |
275 | \ mode, map, map_prefix, string(map), string(names), mode != 'i', key_prefix) | |
276 | endfor | |
277 | endfor | |
278 | ||
279 | for [ft, names] in items(lod.ft) | |
280 | augroup PlugLOD | |
281 | execute printf('autocmd FileType %s call <SID>lod_ft(%s, %s)', | |
282 | \ ft, string(ft), string(names)) | |
283 | augroup END | |
284 | endfor | |
285 | ||
286 | call s:reorg_rtp() | |
287 | filetype plugin indent on | |
288 | if has('vim_starting') | |
289 | if has('syntax') && !exists('g:syntax_on') | |
290 | syntax enable | |
291 | end | |
292 | else | |
293 | call s:reload_plugins() | |
294 | endif | |
295 | endfunction | |
296 | ||
297 | function! s:loaded_names() | |
298 | return filter(copy(g:plugs_order), 'get(s:loaded, v:val, 0)') | |
299 | endfunction | |
300 | ||
301 | function! s:load_plugin(spec) | |
302 | call s:source(s:rtp(a:spec), 'plugin/**/*.vim', 'after/plugin/**/*.vim') | |
303 | endfunction | |
304 | ||
305 | function! s:reload_plugins() | |
306 | for name in s:loaded_names() | |
307 | call s:load_plugin(g:plugs[name]) | |
308 | endfor | |
309 | endfunction | |
310 | ||
311 | function! s:trim(str) | |
312 | return substitute(a:str, '[\/]\+$', '', '') | |
313 | endfunction | |
314 | ||
315 | function! s:version_requirement(val, min) | |
316 | for idx in range(0, len(a:min) - 1) | |
317 | let v = get(a:val, idx, 0) | |
318 | if v < a:min[idx] | return 0 | |
319 | elseif v > a:min[idx] | return 1 | |
320 | endif | |
321 | endfor | |
322 | return 1 | |
323 | endfunction | |
324 | ||
325 | function! s:git_version_requirement(...) | |
326 | if !exists('s:git_version') | |
327 | let s:git_version = map(split(split(s:system('git --version'))[2], '\.'), 'str2nr(v:val)') | |
328 | endif | |
329 | return s:version_requirement(s:git_version, a:000) | |
330 | endfunction | |
331 | ||
332 | function! s:progress_opt(base) | |
333 | return a:base && !s:is_win && | |
334 | \ s:git_version_requirement(1, 7, 1) ? '--progress' : '' | |
335 | endfunction | |
336 | ||
817 | 337 | function! s:rtp(spec) |
338 | return s:path(a:spec.dir . get(a:spec, 'rtp', '')) | |
339 | endfunction | |
340 | ||
724 | 341 | if s:is_win |
342 | function! s:path(path) | |
343 | return s:trim(substitute(a:path, '/', '\', 'g')) | |
344 | endfunction | |
345 | ||
346 | function! s:dirpath(path) | |
347 | return s:path(a:path) . '\' | |
348 | endfunction | |
349 | ||
350 | function! s:is_local_plug(repo) | |
351 | return a:repo =~? '^[a-z]:\|^[%~]' | |
352 | endfunction | |
795 | 353 | |
354 | " Copied from fzf | |
355 | function! s:wrap_cmds(cmds) | |
806 | 356 | let use_chcp = executable('sed') |
357 | return map([ | |
358 | \ '@echo off', | |
359 | \ 'setlocal enabledelayedexpansion'] | |
360 | \ + (use_chcp ? [ | |
361 | \ 'for /f "usebackq" %%a in (`chcp ^| sed "s/[^0-9]//gp"`) do set origchcp=%%a', | |
362 | \ 'chcp 65001 > nul'] : []) | |
363 | \ + (type(a:cmds) == type([]) ? a:cmds : [a:cmds]) | |
364 | \ + (use_chcp ? ['chcp !origchcp! > nul'] : []) | |
365 | \ + ['endlocal'], | |
366 | \ 'v:val."\r"') | |
795 | 367 | endfunction |
368 | ||
369 | function! s:batchfile(cmd) | |
370 | let batchfile = tempname().'.bat' | |
371 | call writefile(s:wrap_cmds(a:cmd), batchfile) | |
372 | let cmd = plug#shellescape(batchfile, {'shell': &shell, 'script': 1}) | |
373 | if &shell =~# 'powershell\.exe$' | |
374 | let cmd = '& ' . cmd | |
375 | endif | |
376 | return [batchfile, cmd] | |
377 | endfunction | |
724 | 378 | else |
379 | function! s:path(path) | |
380 | return s:trim(a:path) | |
381 | endfunction | |
382 | ||
383 | function! s:dirpath(path) | |
384 | return substitute(a:path, '[/\\]*$', '/', '') | |
385 | endfunction | |
386 | ||
387 | function! s:is_local_plug(repo) | |
388 | return a:repo[0] =~ '[/$~]' | |
389 | endfunction | |
390 | endif | |
391 | ||
392 | function! s:err(msg) | |
393 | echohl ErrorMsg | |
394 | echom '[vim-plug] '.a:msg | |
395 | echohl None | |
396 | endfunction | |
397 | ||
398 | function! s:warn(cmd, msg) | |
399 | echohl WarningMsg | |
400 | execute a:cmd 'a:msg' | |
401 | echohl None | |
402 | endfunction | |
403 | ||
404 | function! s:esc(path) | |
405 | return escape(a:path, ' ') | |
406 | endfunction | |
407 | ||
408 | function! s:escrtp(path) | |
409 | return escape(a:path, ' ,') | |
410 | endfunction | |
411 | ||
412 | function! s:remove_rtp() | |
413 | for name in s:loaded_names() | |
414 | let rtp = s:rtp(g:plugs[name]) | |
415 | execute 'set rtp-='.s:escrtp(rtp) | |
416 | let after = globpath(rtp, 'after') | |
417 | if isdirectory(after) | |
418 | execute 'set rtp-='.s:escrtp(after) | |
419 | endif | |
420 | endfor | |
421 | endfunction | |
422 | ||
423 | function! s:reorg_rtp() | |
424 | if !empty(s:first_rtp) | |
425 | execute 'set rtp-='.s:first_rtp | |
426 | execute 'set rtp-='.s:last_rtp | |
427 | endif | |
428 | ||
429 | " &rtp is modified from outside | |
430 | if exists('s:prtp') && s:prtp !=# &rtp | |
431 | call s:remove_rtp() | |
432 | unlet! s:middle | |
433 | endif | |
434 | ||
435 | let s:middle = get(s:, 'middle', &rtp) | |
436 | let rtps = map(s:loaded_names(), 's:rtp(g:plugs[v:val])') | |
437 | let afters = filter(map(copy(rtps), 'globpath(v:val, "after")'), '!empty(v:val)') | |
438 | let rtp = join(map(rtps, 'escape(v:val, ",")'), ',') | |
439 | \ . ','.s:middle.',' | |
440 | \ . join(map(afters, 'escape(v:val, ",")'), ',') | |
441 | let &rtp = substitute(substitute(rtp, ',,*', ',', 'g'), '^,\|,$', '', 'g') | |
442 | let s:prtp = &rtp | |
443 | ||
444 | if !empty(s:first_rtp) | |
445 | execute 'set rtp^='.s:first_rtp | |
446 | execute 'set rtp+='.s:last_rtp | |
447 | endif | |
448 | endfunction | |
449 | ||
450 | function! s:doautocmd(...) | |
451 | if exists('#'.join(a:000, '#')) | |
452 | execute 'doautocmd' ((v:version > 703 || has('patch442')) ? '<nomodeline>' : '') join(a:000) | |
453 | endif | |
454 | endfunction | |
455 | ||
456 | function! s:dobufread(names) | |
457 | for name in a:names | |
766 | 458 | let path = s:rtp(g:plugs[name]) |
459 | for dir in ['ftdetect', 'ftplugin', 'after/ftdetect', 'after/ftplugin'] | |
724 | 460 | if len(finddir(dir, path)) |
461 | if exists('#BufRead') | |
462 | doautocmd BufRead | |
463 | endif | |
464 | return | |
465 | endif | |
466 | endfor | |
467 | endfor | |
468 | endfunction | |
469 | ||
470 | function! plug#load(...) | |
471 | if a:0 == 0 | |
472 | return s:err('Argument missing: plugin name(s) required') | |
473 | endif | |
474 | if !exists('g:plugs') | |
475 | return s:err('plug#begin was not called') | |
476 | endif | |
477 | let names = a:0 == 1 && type(a:1) == s:TYPE.list ? a:1 : a:000 | |
478 | let unknowns = filter(copy(names), '!has_key(g:plugs, v:val)') | |
479 | if !empty(unknowns) | |
480 | let s = len(unknowns) > 1 ? 's' : '' | |
481 | return s:err(printf('Unknown plugin%s: %s', s, join(unknowns, ', '))) | |
482 | end | |
483 | let unloaded = filter(copy(names), '!get(s:loaded, v:val, 0)') | |
484 | if !empty(unloaded) | |
485 | for name in unloaded | |
486 | call s:lod([name], ['ftdetect', 'after/ftdetect', 'plugin', 'after/plugin']) | |
487 | endfor | |
488 | call s:dobufread(unloaded) | |
489 | return 1 | |
490 | end | |
491 | return 0 | |
492 | endfunction | |
493 | ||
494 | function! s:remove_triggers(name) | |
495 | if !has_key(s:triggers, a:name) | |
496 | return | |
497 | endif | |
498 | for cmd in s:triggers[a:name].cmd | |
499 | execute 'silent! delc' cmd | |
500 | endfor | |
501 | for map in s:triggers[a:name].map | |
502 | execute 'silent! unmap' map | |
503 | execute 'silent! iunmap' map | |
504 | endfor | |
505 | call remove(s:triggers, a:name) | |
506 | endfunction | |
507 | ||
508 | function! s:lod(names, types, ...) | |
509 | for name in a:names | |
510 | call s:remove_triggers(name) | |
511 | let s:loaded[name] = 1 | |
512 | endfor | |
513 | call s:reorg_rtp() | |
514 | ||
515 | for name in a:names | |
516 | let rtp = s:rtp(g:plugs[name]) | |
517 | for dir in a:types | |
518 | call s:source(rtp, dir.'/**/*.vim') | |
519 | endfor | |
520 | if a:0 | |
521 | if !s:source(rtp, a:1) && !empty(s:glob(rtp, a:2)) | |
522 | execute 'runtime' a:1 | |
523 | endif | |
524 | call s:source(rtp, a:2) | |
525 | endif | |
526 | call s:doautocmd('User', name) | |
527 | endfor | |
528 | endfunction | |
529 | ||
530 | function! s:lod_ft(pat, names) | |
531 | let syn = 'syntax/'.a:pat.'.vim' | |
532 | call s:lod(a:names, ['plugin', 'after/plugin'], syn, 'after/'.syn) | |
533 | execute 'autocmd! PlugLOD FileType' a:pat | |
534 | call s:doautocmd('filetypeplugin', 'FileType') | |
535 | call s:doautocmd('filetypeindent', 'FileType') | |
536 | endfunction | |
537 | ||
538 | function! s:lod_cmd(cmd, bang, l1, l2, args, names) | |
539 | call s:lod(a:names, ['ftdetect', 'after/ftdetect', 'plugin', 'after/plugin']) | |
540 | call s:dobufread(a:names) | |
541 | execute printf('%s%s%s %s', (a:l1 == a:l2 ? '' : (a:l1.','.a:l2)), a:cmd, a:bang, a:args) | |
542 | endfunction | |
543 | ||
544 | function! s:lod_map(map, names, with_prefix, prefix) | |
545 | call s:lod(a:names, ['ftdetect', 'after/ftdetect', 'plugin', 'after/plugin']) | |
546 | call s:dobufread(a:names) | |
547 | let extra = '' | |
548 | while 1 | |
549 | let c = getchar(0) | |
550 | if c == 0 | |
551 | break | |
552 | endif | |
553 | let extra .= nr2char(c) | |
554 | endwhile | |
555 | ||
556 | if a:with_prefix | |
557 | let prefix = v:count ? v:count : '' | |
558 | let prefix .= '"'.v:register.a:prefix | |
559 | if mode(1) == 'no' | |
560 | if v:operator == 'c' | |
561 | let prefix = "\<esc>" . prefix | |
562 | endif | |
563 | let prefix .= v:operator | |
564 | endif | |
565 | call feedkeys(prefix, 'n') | |
566 | endif | |
567 | call feedkeys(substitute(a:map, '^<Plug>', "\<Plug>", '') . extra) | |
568 | endfunction | |
569 | ||
570 | function! plug#(repo, ...) | |
571 | if a:0 > 1 | |
572 | return s:err('Invalid number of arguments (1..2)') | |
573 | endif | |
574 | ||
575 | try | |
576 | let repo = s:trim(a:repo) | |
577 | let opts = a:0 == 1 ? s:parse_options(a:1) : s:base_spec | |
578 | let name = get(opts, 'as', fnamemodify(repo, ':t:s?\.git$??')) | |
579 | let spec = extend(s:infer_properties(name, repo), opts) | |
580 | if !has_key(g:plugs, name) | |
581 | call add(g:plugs_order, name) | |
582 | endif | |
583 | let g:plugs[name] = spec | |
584 | let s:loaded[name] = get(s:loaded, name, 0) | |
585 | catch | |
586 | return s:err(v:exception) | |
587 | endtry | |
588 | endfunction | |
589 | ||
590 | function! s:parse_options(arg) | |
591 | let opts = copy(s:base_spec) | |
592 | let type = type(a:arg) | |
593 | if type == s:TYPE.string | |
594 | let opts.tag = a:arg | |
595 | elseif type == s:TYPE.dict | |
596 | call extend(opts, a:arg) | |
597 | if has_key(opts, 'dir') | |
598 | let opts.dir = s:dirpath(expand(opts.dir)) | |
599 | endif | |
600 | else | |
601 | throw 'Invalid argument type (expected: string or dictionary)' | |
602 | endif | |
603 | return opts | |
604 | endfunction | |
605 | ||
606 | function! s:infer_properties(name, repo) | |
607 | let repo = a:repo | |
608 | if s:is_local_plug(repo) | |
609 | return { 'dir': s:dirpath(expand(repo)) } | |
610 | else | |
611 | if repo =~ ':' | |
612 | let uri = repo | |
613 | else | |
614 | if repo !~ '/' | |
615 | throw printf('Invalid argument: %s (implicit `vim-scripts'' expansion is deprecated)', repo) | |
616 | endif | |
617 | let fmt = get(g:, 'plug_url_format', 'https://git::@github.com/%s.git') | |
618 | let uri = printf(fmt, repo) | |
619 | endif | |
620 | return { 'dir': s:dirpath(g:plug_home.'/'.a:name), 'uri': uri } | |
621 | endif | |
622 | endfunction | |
623 | ||
624 | function! s:install(force, names) | |
625 | call s:update_impl(0, a:force, a:names) | |
626 | endfunction | |
627 | ||
628 | function! s:update(force, names) | |
629 | call s:update_impl(1, a:force, a:names) | |
630 | endfunction | |
631 | ||
632 | function! plug#helptags() | |
633 | if !exists('g:plugs') | |
634 | return s:err('plug#begin was not called') | |
635 | endif | |
636 | for spec in values(g:plugs) | |
637 | let docd = join([s:rtp(spec), 'doc'], '/') | |
638 | if isdirectory(docd) | |
639 | silent! execute 'helptags' s:esc(docd) | |
640 | endif | |
641 | endfor | |
642 | return 1 | |
643 | endfunction | |
644 | ||
645 | function! s:syntax() | |
646 | syntax clear | |
647 | syntax region plug1 start=/\%1l/ end=/\%2l/ contains=plugNumber | |
648 | syntax region plug2 start=/\%2l/ end=/\%3l/ contains=plugBracket,plugX | |
649 | syn match plugNumber /[0-9]\+[0-9.]*/ contained | |
650 | syn match plugBracket /[[\]]/ contained | |
651 | syn match plugX /x/ contained | |
652 | syn match plugDash /^-/ | |
653 | syn match plugPlus /^+/ | |
654 | syn match plugStar /^*/ | |
655 | syn match plugMessage /\(^- \)\@<=.*/ | |
656 | syn match plugName /\(^- \)\@<=[^ ]*:/ | |
657 | syn match plugSha /\%(: \)\@<=[0-9a-f]\{4,}$/ | |
658 | syn match plugTag /(tag: [^)]\+)/ | |
659 | syn match plugInstall /\(^+ \)\@<=[^:]*/ | |
660 | syn match plugUpdate /\(^* \)\@<=[^:]*/ | |
661 | syn match plugCommit /^ \X*[0-9a-f]\{7,9} .*/ contains=plugRelDate,plugEdge,plugTag | |
662 | syn match plugEdge /^ \X\+$/ | |
663 | syn match plugEdge /^ \X*/ contained nextgroup=plugSha | |
664 | syn match plugSha /[0-9a-f]\{7,9}/ contained | |
665 | syn match plugRelDate /([^)]*)$/ contained | |
666 | syn match plugNotLoaded /(not loaded)$/ | |
667 | syn match plugError /^x.*/ | |
668 | syn region plugDeleted start=/^\~ .*/ end=/^\ze\S/ | |
669 | syn match plugH2 /^.*:\n-\+$/ | |
670 | syn keyword Function PlugInstall PlugStatus PlugUpdate PlugClean | |
671 | hi def link plug1 Title | |
672 | hi def link plug2 Repeat | |
673 | hi def link plugH2 Type | |
674 | hi def link plugX Exception | |
675 | hi def link plugBracket Structure | |
676 | hi def link plugNumber Number | |
677 | ||
678 | hi def link plugDash Special | |
679 | hi def link plugPlus Constant | |
680 | hi def link plugStar Boolean | |
681 | ||
682 | hi def link plugMessage Function | |
683 | hi def link plugName Label | |
684 | hi def link plugInstall Function | |
685 | hi def link plugUpdate Type | |
686 | ||
687 | hi def link plugError Error | |
688 | hi def link plugDeleted Ignore | |
689 | hi def link plugRelDate Comment | |
690 | hi def link plugEdge PreProc | |
691 | hi def link plugSha Identifier | |
692 | hi def link plugTag Constant | |
693 | ||
694 | hi def link plugNotLoaded Comment | |
695 | endfunction | |
696 | ||
697 | function! s:lpad(str, len) | |
698 | return a:str . repeat(' ', a:len - len(a:str)) | |
699 | endfunction | |
700 | ||
701 | function! s:lines(msg) | |
702 | return split(a:msg, "[\r\n]") | |
703 | endfunction | |
704 | ||
705 | function! s:lastline(msg) | |
706 | return get(s:lines(a:msg), -1, '') | |
707 | endfunction | |
708 | ||
709 | function! s:new_window() | |
710 | execute get(g:, 'plug_window', 'vertical topleft new') | |
711 | endfunction | |
712 | ||
713 | function! s:plug_window_exists() | |
714 | let buflist = tabpagebuflist(s:plug_tab) | |
715 | return !empty(buflist) && index(buflist, s:plug_buf) >= 0 | |
716 | endfunction | |
717 | ||
718 | function! s:switch_in() | |
719 | if !s:plug_window_exists() | |
720 | return 0 | |
721 | endif | |
722 | ||
723 | if winbufnr(0) != s:plug_buf | |
724 | let s:pos = [tabpagenr(), winnr(), winsaveview()] | |
725 | execute 'normal!' s:plug_tab.'gt' | |
726 | let winnr = bufwinnr(s:plug_buf) | |
727 | execute winnr.'wincmd w' | |
728 | call add(s:pos, winsaveview()) | |
729 | else | |
730 | let s:pos = [winsaveview()] | |
731 | endif | |
732 | ||
733 | setlocal modifiable | |
734 | return 1 | |
735 | endfunction | |
736 | ||
737 | function! s:switch_out(...) | |
738 | call winrestview(s:pos[-1]) | |
739 | setlocal nomodifiable | |
740 | if a:0 > 0 | |
741 | execute a:1 | |
742 | endif | |
743 | ||
744 | if len(s:pos) > 1 | |
745 | execute 'normal!' s:pos[0].'gt' | |
746 | execute s:pos[1] 'wincmd w' | |
747 | call winrestview(s:pos[2]) | |
748 | endif | |
749 | endfunction | |
750 | ||
751 | function! s:finish_bindings() | |
752 | nnoremap <silent> <buffer> R :call <SID>retry()<cr> | |
753 | nnoremap <silent> <buffer> D :PlugDiff<cr> | |
754 | nnoremap <silent> <buffer> S :PlugStatus<cr> | |
755 | nnoremap <silent> <buffer> U :call <SID>status_update()<cr> | |
756 | xnoremap <silent> <buffer> U :call <SID>status_update()<cr> | |
757 | nnoremap <silent> <buffer> ]] :silent! call <SID>section('')<cr> | |
758 | nnoremap <silent> <buffer> [[ :silent! call <SID>section('b')<cr> | |
759 | endfunction | |
760 | ||
761 | function! s:prepare(...) | |
762 | if empty(getcwd()) | |
763 | throw 'Invalid current working directory. Cannot proceed.' | |
764 | endif | |
765 | ||
766 | for evar in ['$GIT_DIR', '$GIT_WORK_TREE'] | |
767 | if exists(evar) | |
768 | throw evar.' detected. Cannot proceed.' | |
769 | endif | |
770 | endfor | |
771 | ||
772 | call s:job_abort() | |
773 | if s:switch_in() | |
774 | if b:plug_preview == 1 | |
775 | pc | |
776 | endif | |
777 | enew | |
778 | else | |
779 | call s:new_window() | |
780 | endif | |
781 | ||
782 | nnoremap <silent> <buffer> q :if b:plug_preview==1<bar>pc<bar>endif<bar>bd<cr> | |
783 | if a:0 == 0 | |
784 | call s:finish_bindings() | |
785 | endif | |
786 | let b:plug_preview = -1 | |
787 | let s:plug_tab = tabpagenr() | |
788 | let s:plug_buf = winbufnr(0) | |
789 | call s:assign_name() | |
790 | ||
791 | for k in ['<cr>', 'L', 'o', 'X', 'd', 'dd'] | |
792 | execute 'silent! unmap <buffer>' k | |
793 | endfor | |
794 | setlocal buftype=nofile bufhidden=wipe nobuflisted nolist noswapfile nowrap cursorline modifiable nospell | |
795 | if exists('+colorcolumn') | |
796 | setlocal colorcolumn= | |
797 | endif | |
798 | setf vim-plug | |
799 | if exists('g:syntax_on') | |
800 | call s:syntax() | |
801 | endif | |
802 | endfunction | |
803 | ||
804 | function! s:assign_name() | |
805 | " Assign buffer name | |
806 | let prefix = '[Plugins]' | |
807 | let name = prefix | |
808 | let idx = 2 | |
809 | while bufexists(name) | |
810 | let name = printf('%s (%s)', prefix, idx) | |
811 | let idx = idx + 1 | |
812 | endwhile | |
813 | silent! execute 'f' fnameescape(name) | |
814 | endfunction | |
815 | ||
816 | function! s:chsh(swap) | |
817 | let prev = [&shell, &shellcmdflag, &shellredir] | |
781 | 818 | if !s:is_win && a:swap |
724 | 819 | set shell=sh shellredir=>%s\ 2>&1 |
820 | endif | |
821 | return prev | |
822 | endfunction | |
823 | ||
824 | function! s:bang(cmd, ...) | |
817 | 825 | let batchfile = '' |
724 | 826 | try |
827 | let [sh, shellcmdflag, shrd] = s:chsh(a:0) | |
828 | " FIXME: Escaping is incomplete. We could use shellescape with eval, | |
829 | " but it won't work on Windows. | |
830 | let cmd = a:0 ? s:with_cd(a:cmd, a:1) : a:cmd | |
831 | if s:is_win | |
795 | 832 | let [batchfile, cmd] = s:batchfile(cmd) |
724 | 833 | endif |
834 | let g:_plug_bang = (s:is_win && has('gui_running') ? 'silent ' : '').'!'.escape(cmd, '#!%') | |
835 | execute "normal! :execute g:_plug_bang\<cr>\<cr>" | |
836 | finally | |
837 | unlet g:_plug_bang | |
838 | let [&shell, &shellcmdflag, &shellredir] = [sh, shellcmdflag, shrd] | |
817 | 839 | if s:is_win && filereadable(batchfile) |
724 | 840 | call delete(batchfile) |
841 | endif | |
842 | endtry | |
843 | return v:shell_error ? 'Exit status: ' . v:shell_error : '' | |
844 | endfunction | |
845 | ||
846 | function! s:regress_bar() | |
847 | let bar = substitute(getline(2)[1:-2], '.*\zs=', 'x', '') | |
848 | call s:progress_bar(2, bar, len(bar)) | |
849 | endfunction | |
850 | ||
851 | function! s:is_updated(dir) | |
852 | return !empty(s:system_chomp('git log --pretty=format:"%h" "HEAD...HEAD@{1}"', a:dir)) | |
853 | endfunction | |
854 | ||
855 | function! s:do(pull, force, todo) | |
856 | for [name, spec] in items(a:todo) | |
857 | if !isdirectory(spec.dir) | |
858 | continue | |
859 | endif | |
860 | let installed = has_key(s:update.new, name) | |
861 | let updated = installed ? 0 : | |
862 | \ (a:pull && index(s:update.errors, name) < 0 && s:is_updated(spec.dir)) | |
863 | if a:force || installed || updated | |
864 | execute 'cd' s:esc(spec.dir) | |
865 | call append(3, '- Post-update hook for '. name .' ... ') | |
866 | let error = '' | |
867 | let type = type(spec.do) | |
868 | if type == s:TYPE.string | |
869 | if spec.do[0] == ':' | |
870 | if !get(s:loaded, name, 0) | |
871 | let s:loaded[name] = 1 | |
872 | call s:reorg_rtp() | |
873 | endif | |
874 | call s:load_plugin(spec) | |
875 | try | |
876 | execute spec.do[1:] | |
877 | catch | |
878 | let error = v:exception | |
879 | endtry | |
880 | if !s:plug_window_exists() | |
881 | cd - | |
882 | throw 'Warning: vim-plug was terminated by the post-update hook of '.name | |
883 | endif | |
884 | else | |
885 | let error = s:bang(spec.do) | |
886 | endif | |
887 | elseif type == s:TYPE.funcref | |
888 | try | |
889 | let status = installed ? 'installed' : (updated ? 'updated' : 'unchanged') | |
890 | call spec.do({ 'name': name, 'status': status, 'force': a:force }) | |
891 | catch | |
892 | let error = v:exception | |
893 | endtry | |
894 | else | |
895 | let error = 'Invalid hook type' | |
896 | endif | |
897 | call s:switch_in() | |
898 | call setline(4, empty(error) ? (getline(4) . 'OK') | |
899 | \ : ('x' . getline(4)[1:] . error)) | |
900 | if !empty(error) | |
901 | call add(s:update.errors, name) | |
902 | call s:regress_bar() | |
903 | endif | |
904 | cd - | |
905 | endif | |
906 | endfor | |
907 | endfunction | |
908 | ||
909 | function! s:hash_match(a, b) | |
910 | return stridx(a:a, a:b) == 0 || stridx(a:b, a:a) == 0 | |
911 | endfunction | |
912 | ||
913 | function! s:checkout(spec) | |
914 | let sha = a:spec.commit | |
915 | let output = s:system('git rev-parse HEAD', a:spec.dir) | |
916 | if !v:shell_error && !s:hash_match(sha, s:lines(output)[0]) | |
917 | let output = s:system( | |
918 | \ 'git fetch --depth 999999 && git checkout '.s:esc(sha).' --', a:spec.dir) | |
919 | endif | |
920 | return output | |
921 | endfunction | |
922 | ||
923 | function! s:finish(pull) | |
924 | let new_frozen = len(filter(keys(s:update.new), 'g:plugs[v:val].frozen')) | |
925 | if new_frozen | |
926 | let s = new_frozen > 1 ? 's' : '' | |
927 | call append(3, printf('- Installed %d frozen plugin%s', new_frozen, s)) | |
928 | endif | |
929 | call append(3, '- Finishing ... ') | 4 | |
930 | redraw | |
931 | call plug#helptags() | |
932 | call plug#end() | |
933 | call setline(4, getline(4) . 'Done!') | |
934 | redraw | |
935 | let msgs = [] | |
936 | if !empty(s:update.errors) | |
937 | call add(msgs, "Press 'R' to retry.") | |
938 | endif | |
939 | if a:pull && len(s:update.new) < len(filter(getline(5, '$'), | |
940 | \ "v:val =~ '^- ' && v:val !~# 'Already up.to.date'")) | |
941 | call add(msgs, "Press 'D' to see the updated changes.") | |
942 | endif | |
943 | echo join(msgs, ' ') | |
944 | call s:finish_bindings() | |
945 | endfunction | |
946 | ||
947 | function! s:retry() | |
948 | if empty(s:update.errors) | |
949 | return | |
950 | endif | |
951 | echo | |
952 | call s:update_impl(s:update.pull, s:update.force, | |
953 | \ extend(copy(s:update.errors), [s:update.threads])) | |
954 | endfunction | |
955 | ||
956 | function! s:is_managed(name) | |
957 | return has_key(g:plugs[a:name], 'uri') | |
958 | endfunction | |
959 | ||
960 | function! s:names(...) | |
961 | return sort(filter(keys(g:plugs), 'stridx(v:val, a:1) == 0 && s:is_managed(v:val)')) | |
962 | endfunction | |
963 | ||
964 | function! s:check_ruby() | |
965 | silent! ruby require 'thread'; VIM::command("let g:plug_ruby = '#{RUBY_VERSION}'") | |
966 | if !exists('g:plug_ruby') | |
967 | redraw! | |
968 | return s:warn('echom', 'Warning: Ruby interface is broken') | |
969 | endif | |
970 | let ruby_version = split(g:plug_ruby, '\.') | |
971 | unlet g:plug_ruby | |
972 | return s:version_requirement(ruby_version, [1, 8, 7]) | |
973 | endfunction | |
974 | ||
975 | function! s:update_impl(pull, force, args) abort | |
976 | let sync = index(a:args, '--sync') >= 0 || has('vim_starting') | |
977 | let args = filter(copy(a:args), 'v:val != "--sync"') | |
978 | let threads = (len(args) > 0 && args[-1] =~ '^[1-9][0-9]*$') ? | |
979 | \ remove(args, -1) : get(g:, 'plug_threads', 16) | |
980 | ||
981 | let managed = filter(copy(g:plugs), 's:is_managed(v:key)') | |
982 | let todo = empty(args) ? filter(managed, '!v:val.frozen || !isdirectory(v:val.dir)') : | |
983 | \ filter(managed, 'index(args, v:key) >= 0') | |
984 | ||
985 | if empty(todo) | |
986 | return s:warn('echo', 'No plugin to '. (a:pull ? 'update' : 'install')) | |
987 | endif | |
988 | ||
989 | if !s:is_win && s:git_version_requirement(2, 3) | |
990 | let s:git_terminal_prompt = exists('$GIT_TERMINAL_PROMPT') ? $GIT_TERMINAL_PROMPT : '' | |
991 | let $GIT_TERMINAL_PROMPT = 0 | |
992 | for plug in values(todo) | |
993 | let plug.uri = substitute(plug.uri, | |
994 | \ '^https://git::@github\.com', 'https://github.com', '') | |
995 | endfor | |
996 | endif | |
997 | ||
998 | if !isdirectory(g:plug_home) | |
999 | try | |
1000 | call mkdir(g:plug_home, 'p') | |
1001 | catch | |
1002 | return s:err(printf('Invalid plug directory: %s. '. | |
1003 | \ 'Try to call plug#begin with a valid directory', g:plug_home)) | |
1004 | endtry | |
1005 | endif | |
1006 | ||
1007 | if has('nvim') && !exists('*jobwait') && threads > 1 | |
1008 | call s:warn('echom', '[vim-plug] Update Neovim for parallel installer') | |
1009 | endif | |
1010 | ||
1011 | let use_job = s:nvim || s:vim8 | |
1012 | let python = (has('python') || has('python3')) && !use_job | |
1013 | let ruby = has('ruby') && !use_job && (v:version >= 703 || v:version == 702 && has('patch374')) && !(s:is_win && has('gui_running')) && threads > 1 && s:check_ruby() | |
1014 | ||
1015 | let s:update = { | |
1016 | \ 'start': reltime(), | |
1017 | \ 'all': todo, | |
1018 | \ 'todo': copy(todo), | |
1019 | \ 'errors': [], | |
1020 | \ 'pull': a:pull, | |
1021 | \ 'force': a:force, | |
1022 | \ 'new': {}, | |
1023 | \ 'threads': (python || ruby || use_job) ? min([len(todo), threads]) : 1, | |
1024 | \ 'bar': '', | |
1025 | \ 'fin': 0 | |
1026 | \ } | |
1027 | ||
1028 | call s:prepare(1) | |
1029 | call append(0, ['', '']) | |
1030 | normal! 2G | |
1031 | silent! redraw | |
1032 | ||
1033 | let s:clone_opt = get(g:, 'plug_shallow', 1) ? | |
1034 | \ '--depth 1' . (s:git_version_requirement(1, 7, 10) ? ' --no-single-branch' : '') : '' | |
1035 | ||
806 | 1036 | if has('win32unix') || has('wsl') |
724 | 1037 | let s:clone_opt .= ' -c core.eol=lf -c core.autocrlf=input' |
1038 | endif | |
1039 | ||
1040 | let s:submodule_opt = s:git_version_requirement(2, 8) ? ' --jobs='.threads : '' | |
1041 | ||
1042 | " Python version requirement (>= 2.7) | |
1043 | if python && !has('python3') && !ruby && !use_job && s:update.threads > 1 | |
1044 | redir => pyv | |
1045 | silent python import platform; print platform.python_version() | |
1046 | redir END | |
1047 | let python = s:version_requirement( | |
1048 | \ map(split(split(pyv)[0], '\.'), 'str2nr(v:val)'), [2, 6]) | |
1049 | endif | |
1050 | ||
1051 | if (python || ruby) && s:update.threads > 1 | |
1052 | try | |
1053 | let imd = &imd | |
1054 | if s:mac_gui | |
1055 | set noimd | |
1056 | endif | |
1057 | if ruby | |
1058 | call s:update_ruby() | |
1059 | else | |
1060 | call s:update_python() | |
1061 | endif | |
1062 | catch | |
1063 | let lines = getline(4, '$') | |
1064 | let printed = {} | |
1065 | silent! 4,$d _ | |
1066 | for line in lines | |
1067 | let name = s:extract_name(line, '.', '') | |
1068 | if empty(name) || !has_key(printed, name) | |
1069 | call append('$', line) | |
1070 | if !empty(name) | |
1071 | let printed[name] = 1 | |
1072 | if line[0] == 'x' && index(s:update.errors, name) < 0 | |
1073 | call add(s:update.errors, name) | |
1074 | end | |
1075 | endif | |
1076 | endif | |
1077 | endfor | |
1078 | finally | |
1079 | let &imd = imd | |
1080 | call s:update_finish() | |
1081 | endtry | |
1082 | else | |
1083 | call s:update_vim() | |
1084 | while use_job && sync | |
1085 | sleep 100m | |
1086 | if s:update.fin | |
1087 | break | |
1088 | endif | |
1089 | endwhile | |
1090 | endif | |
1091 | endfunction | |
1092 | ||
1093 | function! s:log4(name, msg) | |
1094 | call setline(4, printf('- %s (%s)', a:msg, a:name)) | |
1095 | redraw | |
1096 | endfunction | |
1097 | ||
1098 | function! s:update_finish() | |
1099 | if exists('s:git_terminal_prompt') | |
1100 | let $GIT_TERMINAL_PROMPT = s:git_terminal_prompt | |
1101 | endif | |
1102 | if s:switch_in() | |
1103 | call append(3, '- Updating ...') | 4 | |
1104 | for [name, spec] in items(filter(copy(s:update.all), 'index(s:update.errors, v:key) < 0 && (s:update.force || s:update.pull || has_key(s:update.new, v:key))')) | |
1105 | let [pos, _] = s:logpos(name) | |
1106 | if !pos | |
1107 | continue | |
1108 | endif | |
1109 | if has_key(spec, 'commit') | |
1110 | call s:log4(name, 'Checking out '.spec.commit) | |
1111 | let out = s:checkout(spec) | |
1112 | elseif has_key(spec, 'tag') | |
1113 | let tag = spec.tag | |
1114 | if tag =~ '\*' | |
795 | 1115 | let tags = s:lines(s:system('git tag --list '.plug#shellescape(tag).' --sort -version:refname 2>&1', spec.dir)) |
724 | 1116 | if !v:shell_error && !empty(tags) |
1117 | let tag = tags[0] | |
1118 | call s:log4(name, printf('Latest tag for %s -> %s', spec.tag, tag)) | |
1119 | call append(3, '') | |
1120 | endif | |
1121 | endif | |
1122 | call s:log4(name, 'Checking out '.tag) | |
1123 | let out = s:system('git checkout -q '.s:esc(tag).' -- 2>&1', spec.dir) | |
1124 | else | |
1125 | let branch = s:esc(get(spec, 'branch', 'master')) | |
1126 | call s:log4(name, 'Merging origin/'.branch) | |
1127 | let out = s:system('git checkout -q '.branch.' -- 2>&1' | |
1128 | \. (has_key(s:update.new, name) ? '' : ('&& git merge --ff-only origin/'.branch.' 2>&1')), spec.dir) | |
1129 | endif | |
1130 | if !v:shell_error && filereadable(spec.dir.'/.gitmodules') && | |
1131 | \ (s:update.force || has_key(s:update.new, name) || s:is_updated(spec.dir)) | |
1132 | call s:log4(name, 'Updating submodules. This may take a while.') | |
1133 | let out .= s:bang('git submodule update --init --recursive'.s:submodule_opt.' 2>&1', spec.dir) | |
1134 | endif | |
1135 | let msg = s:format_message(v:shell_error ? 'x': '-', name, out) | |
1136 | if v:shell_error | |
1137 | call add(s:update.errors, name) | |
1138 | call s:regress_bar() | |
1139 | silent execute pos 'd _' | |
1140 | call append(4, msg) | 4 | |
1141 | elseif !empty(out) | |
1142 | call setline(pos, msg[0]) | |
1143 | endif | |
1144 | redraw | |
1145 | endfor | |
1146 | silent 4 d _ | |
1147 | try | |
1148 | call s:do(s:update.pull, s:update.force, filter(copy(s:update.all), 'index(s:update.errors, v:key) < 0 && has_key(v:val, "do")')) | |
1149 | catch | |
1150 | call s:warn('echom', v:exception) | |
1151 | call s:warn('echo', '') | |
1152 | return | |
1153 | endtry | |
1154 | call s:finish(s:update.pull) | |
1155 | call setline(1, 'Updated. Elapsed time: ' . split(reltimestr(reltime(s:update.start)))[0] . ' sec.') | |
1156 | call s:switch_out('normal! gg') | |
1157 | endif | |
1158 | endfunction | |
1159 | ||
1160 | function! s:job_abort() | |
1161 | if (!s:nvim && !s:vim8) || !exists('s:jobs') | |
1162 | return | |
1163 | endif | |
1164 | ||
1165 | for [name, j] in items(s:jobs) | |
1166 | if s:nvim | |
1167 | silent! call jobstop(j.jobid) | |
1168 | elseif s:vim8 | |
1169 | silent! call job_stop(j.jobid) | |
1170 | endif | |
1171 | if j.new | |
795 | 1172 | call s:system('rm -rf ' . plug#shellescape(g:plugs[name].dir)) |
724 | 1173 | endif |
1174 | endfor | |
1175 | let s:jobs = {} | |
1176 | endfunction | |
1177 | ||
1178 | function! s:last_non_empty_line(lines) | |
1179 | let len = len(a:lines) | |
1180 | for idx in range(len) | |
1181 | let line = a:lines[len-idx-1] | |
1182 | if !empty(line) | |
1183 | return line | |
1184 | endif | |
1185 | endfor | |
1186 | return '' | |
1187 | endfunction | |
1188 | ||
1189 | function! s:job_out_cb(self, data) abort | |
1190 | let self = a:self | |
1191 | let data = remove(self.lines, -1) . a:data | |
1192 | let lines = map(split(data, "\n", 1), 'split(v:val, "\r", 1)[-1]') | |
1193 | call extend(self.lines, lines) | |
1194 | " To reduce the number of buffer updates | |
1195 | let self.tick = get(self, 'tick', -1) + 1 | |
1196 | if !self.running || self.tick % len(s:jobs) == 0 | |
1197 | let bullet = self.running ? (self.new ? '+' : '*') : (self.error ? 'x' : '-') | |
1198 | let result = self.error ? join(self.lines, "\n") : s:last_non_empty_line(self.lines) | |
1199 | call s:log(bullet, self.name, result) | |
1200 | endif | |
1201 | endfunction | |
1202 | ||
1203 | function! s:job_exit_cb(self, data) abort | |
1204 | let a:self.running = 0 | |
1205 | let a:self.error = a:data != 0 | |
1206 | call s:reap(a:self.name) | |
1207 | call s:tick() | |
1208 | endfunction | |
1209 | ||
1210 | function! s:job_cb(fn, job, ch, data) | |
1211 | if !s:plug_window_exists() " plug window closed | |
1212 | return s:job_abort() | |
1213 | endif | |
1214 | call call(a:fn, [a:job, a:data]) | |
1215 | endfunction | |
1216 | ||
1217 | function! s:nvim_cb(job_id, data, event) dict abort | |
1218 | return a:event == 'stdout' ? | |
1219 | \ s:job_cb('s:job_out_cb', self, 0, join(a:data, "\n")) : | |
1220 | \ s:job_cb('s:job_exit_cb', self, 0, a:data) | |
1221 | endfunction | |
1222 | ||
1223 | function! s:spawn(name, cmd, opts) | |
1224 | let job = { 'name': a:name, 'running': 1, 'error': 0, 'lines': [''], | |
1225 | \ 'new': get(a:opts, 'new', 0) } | |
1226 | let s:jobs[a:name] = job | |
795 | 1227 | let cmd = has_key(a:opts, 'dir') ? s:with_cd(a:cmd, a:opts.dir, 0) : a:cmd |
1228 | let argv = s:is_win ? ['cmd', '/s', '/c', '"'.cmd.'"'] : ['sh', '-c', cmd] | |
724 | 1229 | |
1230 | if s:nvim | |
1231 | call extend(job, { | |
1232 | \ 'on_stdout': function('s:nvim_cb'), | |
1233 | \ 'on_exit': function('s:nvim_cb'), | |
1234 | \ }) | |
1235 | let jid = jobstart(argv, job) | |
1236 | if jid > 0 | |
1237 | let job.jobid = jid | |
1238 | else | |
1239 | let job.running = 0 | |
1240 | let job.error = 1 | |
1241 | let job.lines = [jid < 0 ? argv[0].' is not executable' : | |
1242 | \ 'Invalid arguments (or job table is full)'] | |
1243 | endif | |
1244 | elseif s:vim8 | |
1245 | let jid = job_start(s:is_win ? join(argv, ' ') : argv, { | |
1246 | \ 'out_cb': function('s:job_cb', ['s:job_out_cb', job]), | |
1247 | \ 'exit_cb': function('s:job_cb', ['s:job_exit_cb', job]), | |
1248 | \ 'out_mode': 'raw' | |
1249 | \}) | |
1250 | if job_status(jid) == 'run' | |
1251 | let job.jobid = jid | |
1252 | else | |
1253 | let job.running = 0 | |
1254 | let job.error = 1 | |
1255 | let job.lines = ['Failed to start job'] | |
1256 | endif | |
1257 | else | |
1258 | let job.lines = s:lines(call('s:system', [cmd])) | |
1259 | let job.error = v:shell_error != 0 | |
1260 | let job.running = 0 | |
1261 | endif | |
1262 | endfunction | |
1263 | ||
1264 | function! s:reap(name) | |
1265 | let job = s:jobs[a:name] | |
1266 | if job.error | |
1267 | call add(s:update.errors, a:name) | |
1268 | elseif get(job, 'new', 0) | |
1269 | let s:update.new[a:name] = 1 | |
1270 | endif | |
1271 | let s:update.bar .= job.error ? 'x' : '=' | |
1272 | ||
1273 | let bullet = job.error ? 'x' : '-' | |
1274 | let result = job.error ? join(job.lines, "\n") : s:last_non_empty_line(job.lines) | |
1275 | call s:log(bullet, a:name, empty(result) ? 'OK' : result) | |
1276 | call s:bar() | |
1277 | ||
1278 | call remove(s:jobs, a:name) | |
1279 | endfunction | |
1280 | ||
1281 | function! s:bar() | |
1282 | if s:switch_in() | |
1283 | let total = len(s:update.all) | |
1284 | call setline(1, (s:update.pull ? 'Updating' : 'Installing'). | |
1285 | \ ' plugins ('.len(s:update.bar).'/'.total.')') | |
1286 | call s:progress_bar(2, s:update.bar, total) | |
1287 | call s:switch_out() | |
1288 | endif | |
1289 | endfunction | |
1290 | ||
1291 | function! s:logpos(name) | |
1292 | for i in range(4, line('$')) | |
1293 | if getline(i) =~# '^[-+x*] '.a:name.':' | |
1294 | for j in range(i + 1, line('$')) | |
1295 | if getline(j) !~ '^ ' | |
1296 | return [i, j - 1] | |
1297 | endif | |
1298 | endfor | |
1299 | return [i, i] | |
1300 | endif | |
1301 | endfor | |
1302 | return [0, 0] | |
1303 | endfunction | |
1304 | ||
1305 | function! s:log(bullet, name, lines) | |
1306 | if s:switch_in() | |
1307 | let [b, e] = s:logpos(a:name) | |
1308 | if b > 0 | |
1309 | silent execute printf('%d,%d d _', b, e) | |
1310 | if b > winheight('.') | |
1311 | let b = 4 | |
1312 | endif | |
1313 | else | |
1314 | let b = 4 | |
1315 | endif | |
1316 | " FIXME For some reason, nomodifiable is set after :d in vim8 | |
1317 | setlocal modifiable | |
1318 | call append(b - 1, s:format_message(a:bullet, a:name, a:lines)) | |
1319 | call s:switch_out() | |
1320 | endif | |
1321 | endfunction | |
1322 | ||
1323 | function! s:update_vim() | |
1324 | let s:jobs = {} | |
1325 | ||
1326 | call s:bar() | |
1327 | call s:tick() | |
1328 | endfunction | |
1329 | ||
1330 | function! s:tick() | |
1331 | let pull = s:update.pull | |
1332 | let prog = s:progress_opt(s:nvim || s:vim8) | |
1333 | while 1 " Without TCO, Vim stack is bound to explode | |
1334 | if empty(s:update.todo) | |
1335 | if empty(s:jobs) && !s:update.fin | |
1336 | call s:update_finish() | |
1337 | let s:update.fin = 1 | |
1338 | endif | |
1339 | return | |
1340 | endif | |
1341 | ||
1342 | let name = keys(s:update.todo)[0] | |
1343 | let spec = remove(s:update.todo, name) | |
1344 | let new = empty(globpath(spec.dir, '.git', 1)) | |
1345 | ||
1346 | call s:log(new ? '+' : '*', name, pull ? 'Updating ...' : 'Installing ...') | |
1347 | redraw | |
1348 | ||
1349 | let has_tag = has_key(spec, 'tag') | |
1350 | if !new | |
1351 | let [error, _] = s:git_validate(spec, 0) | |
1352 | if empty(error) | |
1353 | if pull | |
1354 | let fetch_opt = (has_tag && !empty(globpath(spec.dir, '.git/shallow'))) ? '--depth 99999999' : '' | |
1355 | call s:spawn(name, printf('git fetch %s %s 2>&1', fetch_opt, prog), { 'dir': spec.dir }) | |
1356 | else | |
1357 | let s:jobs[name] = { 'running': 0, 'lines': ['Already installed'], 'error': 0 } | |
1358 | endif | |
1359 | else | |
1360 | let s:jobs[name] = { 'running': 0, 'lines': s:lines(error), 'error': 1 } | |
1361 | endif | |
1362 | else | |
1363 | call s:spawn(name, | |
1364 | \ printf('git clone %s %s %s %s 2>&1', | |
1365 | \ has_tag ? '' : s:clone_opt, | |
1366 | \ prog, | |
795 | 1367 | \ plug#shellescape(spec.uri, {'script': 0}), |
1368 | \ plug#shellescape(s:trim(spec.dir), {'script': 0})), { 'new': 1 }) | |
724 | 1369 | endif |
1370 | ||
1371 | if !s:jobs[name].running | |
1372 | call s:reap(name) | |
1373 | endif | |
1374 | if len(s:jobs) >= s:update.threads | |
1375 | break | |
1376 | endif | |
1377 | endwhile | |
1378 | endfunction | |
1379 | ||
1380 | function! s:update_python() | |
1381 | let py_exe = has('python') ? 'python' : 'python3' | |
1382 | execute py_exe "<< EOF" | |
1383 | import datetime | |
1384 | import functools | |
1385 | import os | |
1386 | try: | |
1387 | import queue | |
1388 | except ImportError: | |
1389 | import Queue as queue | |
1390 | import random | |
1391 | import re | |
1392 | import shutil | |
1393 | import signal | |
1394 | import subprocess | |
1395 | import tempfile | |
1396 | import threading as thr | |
1397 | import time | |
1398 | import traceback | |
1399 | import vim | |
1400 | ||
1401 | G_NVIM = vim.eval("has('nvim')") == '1' | |
1402 | G_PULL = vim.eval('s:update.pull') == '1' | |
1403 | G_RETRIES = int(vim.eval('get(g:, "plug_retries", 2)')) + 1 | |
1404 | G_TIMEOUT = int(vim.eval('get(g:, "plug_timeout", 60)')) | |
1405 | G_CLONE_OPT = vim.eval('s:clone_opt') | |
1406 | G_PROGRESS = vim.eval('s:progress_opt(1)') | |
1407 | G_LOG_PROB = 1.0 / int(vim.eval('s:update.threads')) | |
1408 | G_STOP = thr.Event() | |
1409 | G_IS_WIN = vim.eval('s:is_win') == '1' | |
1410 | ||
1411 | class PlugError(Exception): | |
1412 | def __init__(self, msg): | |
1413 | self.msg = msg | |
1414 | class CmdTimedOut(PlugError): | |
1415 | pass | |
1416 | class CmdFailed(PlugError): | |
1417 | pass | |
1418 | class InvalidURI(PlugError): | |
1419 | pass | |
1420 | class Action(object): | |
1421 | INSTALL, UPDATE, ERROR, DONE = ['+', '*', 'x', '-'] | |
1422 | ||
1423 | class Buffer(object): | |
1424 | def __init__(self, lock, num_plugs, is_pull): | |
1425 | self.bar = '' | |
1426 | self.event = 'Updating' if is_pull else 'Installing' | |
1427 | self.lock = lock | |
1428 | self.maxy = int(vim.eval('winheight(".")')) | |
1429 | self.num_plugs = num_plugs | |
1430 | ||
1431 | def __where(self, name): | |
1432 | """ Find first line with name in current buffer. Return line num. """ | |
1433 | found, lnum = False, 0 | |
1434 | matcher = re.compile('^[-+x*] {0}:'.format(name)) | |
1435 | for line in vim.current.buffer: | |
1436 | if matcher.search(line) is not None: | |
1437 | found = True | |
1438 | break | |
1439 | lnum += 1 | |
1440 | ||
1441 | if not found: | |
1442 | lnum = -1 | |
1443 | return lnum | |
1444 | ||
1445 | def header(self): | |
1446 | curbuf = vim.current.buffer | |
1447 | curbuf[0] = self.event + ' plugins ({0}/{1})'.format(len(self.bar), self.num_plugs) | |
1448 | ||
1449 | num_spaces = self.num_plugs - len(self.bar) | |
1450 | curbuf[1] = '[{0}{1}]'.format(self.bar, num_spaces * ' ') | |
1451 | ||
1452 | with self.lock: | |
1453 | vim.command('normal! 2G') | |
1454 | vim.command('redraw') | |
1455 | ||
1456 | def write(self, action, name, lines): | |
1457 | first, rest = lines[0], lines[1:] | |
1458 | msg = ['{0} {1}{2}{3}'.format(action, name, ': ' if first else '', first)] | |
1459 | msg.extend([' ' + line for line in rest]) | |
1460 | ||
1461 | try: | |
1462 | if action == Action.ERROR: | |
1463 | self.bar += 'x' | |
1464 | vim.command("call add(s:update.errors, '{0}')".format(name)) | |
1465 | elif action == Action.DONE: | |
1466 | self.bar += '=' | |
1467 | ||
1468 | curbuf = vim.current.buffer | |
1469 | lnum = self.__where(name) | |
1470 | if lnum != -1: # Found matching line num | |
1471 | del curbuf[lnum] | |
1472 | if lnum > self.maxy and action in set([Action.INSTALL, Action.UPDATE]): | |
1473 | lnum = 3 | |
1474 | else: | |
1475 | lnum = 3 | |
1476 | curbuf.append(msg, lnum) | |
1477 | ||
1478 | self.header() | |
1479 | except vim.error: | |
1480 | pass | |
1481 | ||
1482 | class Command(object): | |
1483 | CD = 'cd /d' if G_IS_WIN else 'cd' | |
1484 | ||
1485 | def __init__(self, cmd, cmd_dir=None, timeout=60, cb=None, clean=None): | |
1486 | self.cmd = cmd | |
1487 | if cmd_dir: | |
1488 | self.cmd = '{0} {1} && {2}'.format(Command.CD, cmd_dir, self.cmd) | |
1489 | self.timeout = timeout | |
1490 | self.callback = cb if cb else (lambda msg: None) | |
1491 | self.clean = clean if clean else (lambda: None) | |
1492 | self.proc = None | |
1493 | ||
1494 | @property | |
1495 | def alive(self): | |
1496 | """ Returns true only if command still running. """ | |
1497 | return self.proc and self.proc.poll() is None | |
1498 | ||
1499 | def execute(self, ntries=3): | |
1500 | """ Execute the command with ntries if CmdTimedOut. | |
1501 | Returns the output of the command if no Exception. | |
1502 | """ | |
1503 | attempt, finished, limit = 0, False, self.timeout | |
1504 | ||
1505 | while not finished: | |
1506 | try: | |
1507 | attempt += 1 | |
1508 | result = self.try_command() | |
1509 | finished = True | |
1510 | return result | |
1511 | except CmdTimedOut: | |
1512 | if attempt != ntries: | |
1513 | self.notify_retry() | |
1514 | self.timeout += limit | |
1515 | else: | |
1516 | raise | |
1517 | ||
1518 | def notify_retry(self): | |
1519 | """ Retry required for command, notify user. """ | |
1520 | for count in range(3, 0, -1): | |
1521 | if G_STOP.is_set(): | |
1522 | raise KeyboardInterrupt | |
1523 | msg = 'Timeout. Will retry in {0} second{1} ...'.format( | |
1524 | count, 's' if count != 1 else '') | |
1525 | self.callback([msg]) | |
1526 | time.sleep(1) | |
1527 | self.callback(['Retrying ...']) | |
1528 | ||
1529 | def try_command(self): | |
1530 | """ Execute a cmd & poll for callback. Returns list of output. | |
1531 | Raises CmdFailed -> return code for Popen isn't 0 | |
1532 | Raises CmdTimedOut -> command exceeded timeout without new output | |
1533 | """ | |
1534 | first_line = True | |
1535 | ||
1536 | try: | |
1537 | tfile = tempfile.NamedTemporaryFile(mode='w+b') | |
1538 | preexec_fn = not G_IS_WIN and os.setsid or None | |
1539 | self.proc = subprocess.Popen(self.cmd, stdout=tfile, | |
1540 | stderr=subprocess.STDOUT, | |
1541 | stdin=subprocess.PIPE, shell=True, | |
1542 | preexec_fn=preexec_fn) | |
1543 | thrd = thr.Thread(target=(lambda proc: proc.wait()), args=(self.proc,)) | |
1544 | thrd.start() | |
1545 | ||
1546 | thread_not_started = True | |
1547 | while thread_not_started: | |
1548 | try: | |
1549 | thrd.join(0.1) | |
1550 | thread_not_started = False | |
1551 | except RuntimeError: | |
1552 | pass | |
1553 | ||
1554 | while self.alive: | |
1555 | if G_STOP.is_set(): | |
1556 | raise KeyboardInterrupt | |
1557 | ||
1558 | if first_line or random.random() < G_LOG_PROB: | |
1559 | first_line = False | |
1560 | line = '' if G_IS_WIN else nonblock_read(tfile.name) | |
1561 | if line: | |
1562 | self.callback([line]) | |
1563 | ||
1564 | time_diff = time.time() - os.path.getmtime(tfile.name) | |
1565 | if time_diff > self.timeout: | |
1566 | raise CmdTimedOut(['Timeout!']) | |
1567 | ||
1568 | thrd.join(0.5) | |
1569 | ||
1570 | tfile.seek(0) | |
1571 | result = [line.decode('utf-8', 'replace').rstrip() for line in tfile] | |
1572 | ||
1573 | if self.proc.returncode != 0: | |
1574 | raise CmdFailed([''] + result) | |
1575 | ||
1576 | return result | |
1577 | except: | |
1578 | self.terminate() | |
1579 | raise | |
1580 | ||
1581 | def terminate(self): | |
1582 | """ Terminate process and cleanup. """ | |
1583 | if self.alive: | |
1584 | if G_IS_WIN: | |
1585 | os.kill(self.proc.pid, signal.SIGINT) | |
1586 | else: | |
1587 | os.killpg(self.proc.pid, signal.SIGTERM) | |
1588 | self.clean() | |
1589 | ||
1590 | class Plugin(object): | |
1591 | def __init__(self, name, args, buf_q, lock): | |
1592 | self.name = name | |
1593 | self.args = args | |
1594 | self.buf_q = buf_q | |
1595 | self.lock = lock | |
1596 | self.tag = args.get('tag', 0) | |
1597 | ||
1598 | def manage(self): | |
1599 | try: | |
1600 | if os.path.exists(self.args['dir']): | |
1601 | self.update() | |
1602 | else: | |
1603 | self.install() | |
1604 | with self.lock: | |
1605 | thread_vim_command("let s:update.new['{0}'] = 1".format(self.name)) | |
1606 | except PlugError as exc: | |
1607 | self.write(Action.ERROR, self.name, exc.msg) | |
1608 | except KeyboardInterrupt: | |
1609 | G_STOP.set() | |
1610 | self.write(Action.ERROR, self.name, ['Interrupted!']) | |
1611 | except: | |
1612 | # Any exception except those above print stack trace | |
1613 | msg = 'Trace:\n{0}'.format(traceback.format_exc().rstrip()) | |
1614 | self.write(Action.ERROR, self.name, msg.split('\n')) | |
1615 | raise | |
1616 | ||
1617 | def install(self): | |
1618 | target = self.args['dir'] | |
1619 | if target[-1] == '\\': | |
1620 | target = target[0:-1] | |
1621 | ||
1622 | def clean(target): | |
1623 | def _clean(): | |
1624 | try: | |
1625 | shutil.rmtree(target) | |
1626 | except OSError: | |
1627 | pass | |
1628 | return _clean | |
1629 | ||
1630 | self.write(Action.INSTALL, self.name, ['Installing ...']) | |
1631 | callback = functools.partial(self.write, Action.INSTALL, self.name) | |
1632 | cmd = 'git clone {0} {1} {2} {3} 2>&1'.format( | |
1633 | '' if self.tag else G_CLONE_OPT, G_PROGRESS, self.args['uri'], | |
1634 | esc(target)) | |
1635 | com = Command(cmd, None, G_TIMEOUT, callback, clean(target)) | |
1636 | result = com.execute(G_RETRIES) | |
1637 | self.write(Action.DONE, self.name, result[-1:]) | |
1638 | ||
1639 | def repo_uri(self): | |
1640 | cmd = 'git rev-parse --abbrev-ref HEAD 2>&1 && git config -f .git/config remote.origin.url' | |
1641 | command = Command(cmd, self.args['dir'], G_TIMEOUT,) | |
1642 | result = command.execute(G_RETRIES) | |
1643 | return result[-1] | |
1644 | ||
1645 | def update(self): | |
1646 | actual_uri = self.repo_uri() | |
1647 | expect_uri = self.args['uri'] | |
1648 | regex = re.compile(r'^(?:\w+://)?(?:[^@/]*@)?([^:/]*(?::[0-9]*)?)[:/](.*?)(?:\.git)?/?$') | |
1649 | ma = regex.match(actual_uri) | |
1650 | mb = regex.match(expect_uri) | |
1651 | if ma is None or mb is None or ma.groups() != mb.groups(): | |
1652 | msg = ['', | |
1653 | 'Invalid URI: {0}'.format(actual_uri), | |
1654 | 'Expected {0}'.format(expect_uri), | |
1655 | 'PlugClean required.'] | |
1656 | raise InvalidURI(msg) | |
1657 | ||
1658 | if G_PULL: | |
1659 | self.write(Action.UPDATE, self.name, ['Updating ...']) | |
1660 | callback = functools.partial(self.write, Action.UPDATE, self.name) | |
1661 | fetch_opt = '--depth 99999999' if self.tag and os.path.isfile(os.path.join(self.args['dir'], '.git/shallow')) else '' | |
1662 | cmd = 'git fetch {0} {1} 2>&1'.format(fetch_opt, G_PROGRESS) | |
1663 | com = Command(cmd, self.args['dir'], G_TIMEOUT, callback) | |
1664 | result = com.execute(G_RETRIES) | |
1665 | self.write(Action.DONE, self.name, result[-1:]) | |
1666 | else: | |
1667 | self.write(Action.DONE, self.name, ['Already installed']) | |
1668 | ||
1669 | def write(self, action, name, msg): | |
1670 | self.buf_q.put((action, name, msg)) | |
1671 | ||
1672 | class PlugThread(thr.Thread): | |
1673 | def __init__(self, tname, args): | |
1674 | super(PlugThread, self).__init__() | |
1675 | self.tname = tname | |
1676 | self.args = args | |
1677 | ||
1678 | def run(self): | |
1679 | thr.current_thread().name = self.tname | |
1680 | buf_q, work_q, lock = self.args | |
1681 | ||
1682 | try: | |
1683 | while not G_STOP.is_set(): | |
1684 | name, args = work_q.get_nowait() | |
1685 | plug = Plugin(name, args, buf_q, lock) | |
1686 | plug.manage() | |
1687 | work_q.task_done() | |
1688 | except queue.Empty: | |
1689 | pass | |
1690 | ||
1691 | class RefreshThread(thr.Thread): | |
1692 | def __init__(self, lock): | |
1693 | super(RefreshThread, self).__init__() | |
1694 | self.lock = lock | |
1695 | self.running = True | |
1696 | ||
1697 | def run(self): | |
1698 | while self.running: | |
1699 | with self.lock: | |
1700 | thread_vim_command('noautocmd normal! a') | |
1701 | time.sleep(0.33) | |
1702 | ||
1703 | def stop(self): | |
1704 | self.running = False | |
1705 | ||
1706 | if G_NVIM: | |
1707 | def thread_vim_command(cmd): | |
1708 | vim.session.threadsafe_call(lambda: vim.command(cmd)) | |
1709 | else: | |
1710 | def thread_vim_command(cmd): | |
1711 | vim.command(cmd) | |
1712 | ||
1713 | def esc(name): | |
1714 | return '"' + name.replace('"', '\"') + '"' | |
1715 | ||
1716 | def nonblock_read(fname): | |
1717 | """ Read a file with nonblock flag. Return the last line. """ | |
1718 | fread = os.open(fname, os.O_RDONLY | os.O_NONBLOCK) | |
1719 | buf = os.read(fread, 100000).decode('utf-8', 'replace') | |
1720 | os.close(fread) | |
1721 | ||
1722 | line = buf.rstrip('\r\n') | |
1723 | left = max(line.rfind('\r'), line.rfind('\n')) | |
1724 | if left != -1: | |
1725 | left += 1 | |
1726 | line = line[left:] | |
1727 | ||
1728 | return line | |
1729 | ||
1730 | def main(): | |
1731 | thr.current_thread().name = 'main' | |
1732 | nthreads = int(vim.eval('s:update.threads')) | |
1733 | plugs = vim.eval('s:update.todo') | |
1734 | mac_gui = vim.eval('s:mac_gui') == '1' | |
1735 | ||
1736 | lock = thr.Lock() | |
1737 | buf = Buffer(lock, len(plugs), G_PULL) | |
1738 | buf_q, work_q = queue.Queue(), queue.Queue() | |
1739 | for work in plugs.items(): | |
1740 | work_q.put(work) | |
1741 | ||
1742 | start_cnt = thr.active_count() | |
1743 | for num in range(nthreads): | |
1744 | tname = 'PlugT-{0:02}'.format(num) | |
1745 | thread = PlugThread(tname, (buf_q, work_q, lock)) | |
1746 | thread.start() | |
1747 | if mac_gui: | |
1748 | rthread = RefreshThread(lock) | |
1749 | rthread.start() | |
1750 | ||
1751 | while not buf_q.empty() or thr.active_count() != start_cnt: | |
1752 | try: | |
1753 | action, name, msg = buf_q.get(True, 0.25) | |
1754 | buf.write(action, name, ['OK'] if not msg else msg) | |
1755 | buf_q.task_done() | |
1756 | except queue.Empty: | |
1757 | pass | |
1758 | except KeyboardInterrupt: | |
1759 | G_STOP.set() | |
1760 | ||
1761 | if mac_gui: | |
1762 | rthread.stop() | |
1763 | rthread.join() | |
1764 | ||
1765 | main() | |
1766 | EOF | |
1767 | endfunction | |
1768 | ||
1769 | function! s:update_ruby() | |
1770 | ruby << EOF | |
1771 | module PlugStream | |
1772 | SEP = ["\r", "\n", nil] | |
1773 | def get_line | |
1774 | buffer = '' | |
1775 | loop do | |
1776 | char = readchar rescue return | |
1777 | if SEP.include? char.chr | |
1778 | buffer << $/ | |
1779 | break | |
1780 | else | |
1781 | buffer << char | |
1782 | end | |
1783 | end | |
1784 | buffer | |
1785 | end | |
1786 | end unless defined?(PlugStream) | |
1787 | ||
1788 | def esc arg | |
1789 | %["#{arg.gsub('"', '\"')}"] | |
1790 | end | |
1791 | ||
1792 | def killall pid | |
1793 | pids = [pid] | |
1794 | if /mswin|mingw|bccwin/ =~ RUBY_PLATFORM | |
1795 | pids.each { |pid| Process.kill 'INT', pid.to_i rescue nil } | |
1796 | else | |
1797 | unless `which pgrep 2> /dev/null`.empty? | |
1798 | children = pids | |
1799 | until children.empty? | |
1800 | children = children.map { |pid| | |
1801 | `pgrep -P #{pid}`.lines.map { |l| l.chomp } | |
1802 | }.flatten | |
1803 | pids += children | |
1804 | end | |
1805 | end | |
1806 | pids.each { |pid| Process.kill 'TERM', pid.to_i rescue nil } | |
1807 | end | |
1808 | end | |
1809 | ||
1810 | def compare_git_uri a, b | |
1811 | regex = %r{^(?:\w+://)?(?:[^@/]*@)?([^:/]*(?::[0-9]*)?)[:/](.*?)(?:\.git)?/?$} | |
1812 | regex.match(a).to_a.drop(1) == regex.match(b).to_a.drop(1) | |
1813 | end | |
1814 | ||
1815 | require 'thread' | |
1816 | require 'fileutils' | |
1817 | require 'timeout' | |
1818 | running = true | |
1819 | iswin = VIM::evaluate('s:is_win').to_i == 1 | |
1820 | pull = VIM::evaluate('s:update.pull').to_i == 1 | |
1821 | base = VIM::evaluate('g:plug_home') | |
1822 | all = VIM::evaluate('s:update.todo') | |
1823 | limit = VIM::evaluate('get(g:, "plug_timeout", 60)') | |
1824 | tries = VIM::evaluate('get(g:, "plug_retries", 2)') + 1 | |
1825 | nthr = VIM::evaluate('s:update.threads').to_i | |
1826 | maxy = VIM::evaluate('winheight(".")').to_i | |
1827 | vim7 = VIM::evaluate('v:version').to_i <= 703 && RUBY_PLATFORM =~ /darwin/ | |
1828 | cd = iswin ? 'cd /d' : 'cd' | |
1829 | tot = VIM::evaluate('len(s:update.todo)') || 0 | |
1830 | bar = '' | |
1831 | skip = 'Already installed' | |
1832 | mtx = Mutex.new | |
1833 | take1 = proc { mtx.synchronize { running && all.shift } } | |
1834 | logh = proc { | |
1835 | cnt = bar.length | |
1836 | $curbuf[1] = "#{pull ? 'Updating' : 'Installing'} plugins (#{cnt}/#{tot})" | |
1837 | $curbuf[2] = '[' + bar.ljust(tot) + ']' | |
1838 | VIM::command('normal! 2G') | |
1839 | VIM::command('redraw') | |
1840 | } | |
1841 | where = proc { |name| (1..($curbuf.length)).find { |l| $curbuf[l] =~ /^[-+x*] #{name}:/ } } | |
1842 | log = proc { |name, result, type| | |
1843 | mtx.synchronize do | |
1844 | ing = ![true, false].include?(type) | |
1845 | bar += type ? '=' : 'x' unless ing | |
1846 | b = case type | |
1847 | when :install then '+' when :update then '*' | |
1848 | when true, nil then '-' else | |
1849 | VIM::command("call add(s:update.errors, '#{name}')") | |
1850 | 'x' | |
1851 | end | |
1852 | result = | |
1853 | if type || type.nil? | |
1854 | ["#{b} #{name}: #{result.lines.to_a.last || 'OK'}"] | |
1855 | elsif result =~ /^Interrupted|^Timeout/ | |
1856 | ["#{b} #{name}: #{result}"] | |
1857 | else | |
1858 | ["#{b} #{name}"] + result.lines.map { |l| " " << l } | |
1859 | end | |
1860 | if lnum = where.call(name) | |
1861 | $curbuf.delete lnum | |
1862 | lnum = 4 if ing && lnum > maxy | |
1863 | end | |
1864 | result.each_with_index do |line, offset| | |
1865 | $curbuf.append((lnum || 4) - 1 + offset, line.gsub(/\e\[./, '').chomp) | |
1866 | end | |
1867 | logh.call | |
1868 | end | |
1869 | } | |
1870 | bt = proc { |cmd, name, type, cleanup| | |
1871 | tried = timeout = 0 | |
1872 | begin | |
1873 | tried += 1 | |
1874 | timeout += limit | |
1875 | fd = nil | |
1876 | data = '' | |
1877 | if iswin | |
1878 | Timeout::timeout(timeout) do | |
1879 | tmp = VIM::evaluate('tempname()') | |
1880 | system("(#{cmd}) > #{tmp}") | |
1881 | data = File.read(tmp).chomp | |
1882 | File.unlink tmp rescue nil | |
1883 | end | |
1884 | else | |
1885 | fd = IO.popen(cmd).extend(PlugStream) | |
1886 | first_line = true | |
1887 | log_prob = 1.0 / nthr | |
1888 | while line = Timeout::timeout(timeout) { fd.get_line } | |
1889 | data << line | |
1890 | log.call name, line.chomp, type if name && (first_line || rand < log_prob) | |
1891 | first_line = false | |
1892 | end | |
1893 | fd.close | |
1894 | end | |
1895 | [$? == 0, data.chomp] | |
1896 | rescue Timeout::Error, Interrupt => e | |
1897 | if fd && !fd.closed? | |
1898 | killall fd.pid | |
1899 | fd.close | |
1900 | end | |
1901 | cleanup.call if cleanup | |
1902 | if e.is_a?(Timeout::Error) && tried < tries | |
1903 | 3.downto(1) do |countdown| | |
1904 | s = countdown > 1 ? 's' : '' | |
1905 | log.call name, "Timeout. Will retry in #{countdown} second#{s} ...", type | |
1906 | sleep 1 | |
1907 | end | |
1908 | log.call name, 'Retrying ...', type | |
1909 | retry | |
1910 | end | |
1911 | [false, e.is_a?(Interrupt) ? "Interrupted!" : "Timeout!"] | |
1912 | end | |
1913 | } | |
1914 | main = Thread.current | |
1915 | threads = [] | |
1916 | watcher = Thread.new { | |
1917 | if vim7 | |
1918 | while VIM::evaluate('getchar(1)') | |
1919 | sleep 0.1 | |
1920 | end | |
1921 | else | |
1922 | require 'io/console' # >= Ruby 1.9 | |
1923 | nil until IO.console.getch == 3.chr | |
1924 | end | |
1925 | mtx.synchronize do | |
1926 | running = false | |
1927 | threads.each { |t| t.raise Interrupt } unless vim7 | |
1928 | end | |
1929 | threads.each { |t| t.join rescue nil } | |
1930 | main.kill | |
1931 | } | |
1932 | refresh = Thread.new { | |
1933 | while true | |
1934 | mtx.synchronize do | |
1935 | break unless running | |
1936 | VIM::command('noautocmd normal! a') | |
1937 | end | |
1938 | sleep 0.2 | |
1939 | end | |
1940 | } if VIM::evaluate('s:mac_gui') == 1 | |
1941 | ||
1942 | clone_opt = VIM::evaluate('s:clone_opt') | |
1943 | progress = VIM::evaluate('s:progress_opt(1)') | |
1944 | nthr.times do | |
1945 | mtx.synchronize do | |
1946 | threads << Thread.new { | |
1947 | while pair = take1.call | |
1948 | name = pair.first | |
1949 | dir, uri, tag = pair.last.values_at *%w[dir uri tag] | |
1950 | exists = File.directory? dir | |
1951 | ok, result = | |
1952 | if exists | |
1953 | chdir = "#{cd} #{iswin ? dir : esc(dir)}" | |
1954 | ret, data = bt.call "#{chdir} && git rev-parse --abbrev-ref HEAD 2>&1 && git config -f .git/config remote.origin.url", nil, nil, nil | |
1955 | current_uri = data.lines.to_a.last | |
1956 | if !ret | |
1957 | if data =~ /^Interrupted|^Timeout/ | |
1958 | [false, data] | |
1959 | else | |
1960 | [false, [data.chomp, "PlugClean required."].join($/)] | |
1961 | end | |
1962 | elsif !compare_git_uri(current_uri, uri) | |
1963 | [false, ["Invalid URI: #{current_uri}", | |
1964 | "Expected: #{uri}", | |
1965 | "PlugClean required."].join($/)] | |
1966 | else | |
1967 | if pull | |
1968 | log.call name, 'Updating ...', :update | |
1969 | fetch_opt = (tag && File.exist?(File.join(dir, '.git/shallow'))) ? '--depth 99999999' : '' | |
1970 | bt.call "#{chdir} && git fetch #{fetch_opt} #{progress} 2>&1", name, :update, nil | |
1971 | else | |
1972 | [true, skip] | |
1973 | end | |
1974 | end | |
1975 | else | |
1976 | d = esc dir.sub(%r{[\\/]+$}, '') | |
1977 | log.call name, 'Installing ...', :install | |
1978 | bt.call "git clone #{clone_opt unless tag} #{progress} #{uri} #{d} 2>&1", name, :install, proc { | |
1979 | FileUtils.rm_rf dir | |
1980 | } | |
1981 | end | |
1982 | mtx.synchronize { VIM::command("let s:update.new['#{name}'] = 1") } if !exists && ok | |
1983 | log.call name, result, ok | |
1984 | end | |
1985 | } if running | |
1986 | end | |
1987 | end | |
1988 | threads.each { |t| t.join rescue nil } | |
1989 | logh.call | |
1990 | refresh.kill if refresh | |
1991 | watcher.kill | |
1992 | EOF | |
1993 | endfunction | |
1994 | ||
795 | 1995 | function! s:shellesc_cmd(arg, script) |
1996 | let escaped = substitute('"'.a:arg.'"', '[&|<>()@^!"]', '^&', 'g') | |
1997 | return substitute(escaped, '%', (a:script ? '%' : '^') . '&', 'g') | |
1998 | endfunction | |
1999 | ||
2000 | function! s:shellesc_ps1(arg) | |
2001 | return "'".substitute(escape(a:arg, '\"'), "'", "''", 'g')."'" | |
724 | 2002 | endfunction |
2003 | ||
795 | 2004 | function! plug#shellescape(arg, ...) |
2005 | let opts = a:0 > 0 && type(a:1) == s:TYPE.dict ? a:1 : {} | |
2006 | let shell = get(opts, 'shell', s:is_win ? 'cmd.exe' : 'sh') | |
2007 | let script = get(opts, 'script', 1) | |
2008 | if shell =~# 'cmd\.exe$' | |
2009 | return s:shellesc_cmd(a:arg, script) | |
2010 | elseif shell =~# 'powershell\.exe$' || shell =~# 'pwsh$' | |
2011 | return s:shellesc_ps1(a:arg) | |
724 | 2012 | endif |
2013 | return shellescape(a:arg) | |
2014 | endfunction | |
2015 | ||
2016 | function! s:glob_dir(path) | |
2017 | return map(filter(s:glob(a:path, '**'), 'isdirectory(v:val)'), 's:dirpath(v:val)') | |
2018 | endfunction | |
2019 | ||
2020 | function! s:progress_bar(line, bar, total) | |
2021 | call setline(a:line, '[' . s:lpad(a:bar, a:total) . ']') | |
2022 | endfunction | |
2023 | ||
2024 | function! s:compare_git_uri(a, b) | |
2025 | " See `git help clone' | |
2026 | " https:// [user@] github.com[:port] / junegunn/vim-plug [.git] | |
2027 | " [git@] github.com[:port] : junegunn/vim-plug [.git] | |
2028 | " file:// / junegunn/vim-plug [/] | |
2029 | " / junegunn/vim-plug [/] | |
2030 | let pat = '^\%(\w\+://\)\='.'\%([^@/]*@\)\='.'\([^:/]*\%(:[0-9]*\)\=\)'.'[:/]'.'\(.\{-}\)'.'\%(\.git\)\=/\?$' | |
2031 | let ma = matchlist(a:a, pat) | |
2032 | let mb = matchlist(a:b, pat) | |
2033 | return ma[1:2] ==# mb[1:2] | |
2034 | endfunction | |
2035 | ||
2036 | function! s:format_message(bullet, name, message) | |
2037 | if a:bullet != 'x' | |
2038 | return [printf('%s %s: %s', a:bullet, a:name, s:lastline(a:message))] | |
2039 | else | |
2040 | let lines = map(s:lines(a:message), '" ".v:val') | |
2041 | return extend([printf('x %s:', a:name)], lines) | |
2042 | endif | |
2043 | endfunction | |
2044 | ||
795 | 2045 | function! s:with_cd(cmd, dir, ...) |
2046 | let script = a:0 > 0 ? a:1 : 1 | |
2047 | return printf('cd%s %s && %s', s:is_win ? ' /d' : '', plug#shellescape(a:dir, {'script': script}), a:cmd) | |
724 | 2048 | endfunction |
2049 | ||
2050 | function! s:system(cmd, ...) | |
817 | 2051 | let batchfile = '' |
724 | 2052 | try |
2053 | let [sh, shellcmdflag, shrd] = s:chsh(1) | |
2054 | let cmd = a:0 > 0 ? s:with_cd(a:cmd, a:1) : a:cmd | |
2055 | if s:is_win | |
795 | 2056 | let [batchfile, cmd] = s:batchfile(cmd) |
724 | 2057 | endif |
773 | 2058 | return system(cmd) |
724 | 2059 | finally |
2060 | let [&shell, &shellcmdflag, &shellredir] = [sh, shellcmdflag, shrd] | |
817 | 2061 | if s:is_win && filereadable(batchfile) |
724 | 2062 | call delete(batchfile) |
2063 | endif | |
2064 | endtry | |
2065 | endfunction | |
2066 | ||
2067 | function! s:system_chomp(...) | |
2068 | let ret = call('s:system', a:000) | |
2069 | return v:shell_error ? '' : substitute(ret, '\n$', '', '') | |
2070 | endfunction | |
2071 | ||
2072 | function! s:git_validate(spec, check_branch) | |
2073 | let err = '' | |
2074 | if isdirectory(a:spec.dir) | |
2075 | let result = s:lines(s:system('git rev-parse --abbrev-ref HEAD 2>&1 && git config -f .git/config remote.origin.url', a:spec.dir)) | |
2076 | let remote = result[-1] | |
2077 | if v:shell_error | |
2078 | let err = join([remote, 'PlugClean required.'], "\n") | |
2079 | elseif !s:compare_git_uri(remote, a:spec.uri) | |
2080 | let err = join(['Invalid URI: '.remote, | |
2081 | \ 'Expected: '.a:spec.uri, | |
2082 | \ 'PlugClean required.'], "\n") | |
2083 | elseif a:check_branch && has_key(a:spec, 'commit') | |
2084 | let result = s:lines(s:system('git rev-parse HEAD 2>&1', a:spec.dir)) | |
2085 | let sha = result[-1] | |
2086 | if v:shell_error | |
2087 | let err = join(add(result, 'PlugClean required.'), "\n") | |
2088 | elseif !s:hash_match(sha, a:spec.commit) | |
2089 | let err = join([printf('Invalid HEAD (expected: %s, actual: %s)', | |
2090 | \ a:spec.commit[:6], sha[:6]), | |
2091 | \ 'PlugUpdate required.'], "\n") | |
2092 | endif | |
2093 | elseif a:check_branch | |
2094 | let branch = result[0] | |
2095 | " Check tag | |
2096 | if has_key(a:spec, 'tag') | |
2097 | let tag = s:system_chomp('git describe --exact-match --tags HEAD 2>&1', a:spec.dir) | |
2098 | if a:spec.tag !=# tag && a:spec.tag !~ '\*' | |
2099 | let err = printf('Invalid tag: %s (expected: %s). Try PlugUpdate.', | |
2100 | \ (empty(tag) ? 'N/A' : tag), a:spec.tag) | |
2101 | endif | |
2102 | " Check branch | |
2103 | elseif a:spec.branch !=# branch | |
2104 | let err = printf('Invalid branch: %s (expected: %s). Try PlugUpdate.', | |
2105 | \ branch, a:spec.branch) | |
2106 | endif | |
2107 | if empty(err) | |
2108 | let [ahead, behind] = split(s:lastline(s:system(printf( | |
2109 | \ 'git rev-list --count --left-right HEAD...origin/%s', | |
2110 | \ a:spec.branch), a:spec.dir)), '\t') | |
2111 | if !v:shell_error && ahead | |
2112 | if behind | |
2113 | " Only mention PlugClean if diverged, otherwise it's likely to be | |
2114 | " pushable (and probably not that messed up). | |
2115 | let err = printf( | |
2116 | \ "Diverged from origin/%s (%d commit(s) ahead and %d commit(s) behind!\n" | |
2117 | \ .'Backup local changes and run PlugClean and PlugUpdate to reinstall it.', a:spec.branch, ahead, behind) | |
2118 | else | |
2119 | let err = printf("Ahead of origin/%s by %d commit(s).\n" | |
2120 | \ .'Cannot update until local changes are pushed.', | |
2121 | \ a:spec.branch, ahead) | |
2122 | endif | |
2123 | endif | |
2124 | endif | |
2125 | endif | |
2126 | else | |
2127 | let err = 'Not found' | |
2128 | endif | |
2129 | return [err, err =~# 'PlugClean'] | |
2130 | endfunction | |
2131 | ||
2132 | function! s:rm_rf(dir) | |
2133 | if isdirectory(a:dir) | |
795 | 2134 | call s:system((s:is_win ? 'rmdir /S /Q ' : 'rm -rf ') . plug#shellescape(a:dir)) |
724 | 2135 | endif |
2136 | endfunction | |
2137 | ||
2138 | function! s:clean(force) | |
2139 | call s:prepare() | |
2140 | call append(0, 'Searching for invalid plugins in '.g:plug_home) | |
2141 | call append(1, '') | |
2142 | ||
2143 | " List of valid directories | |
2144 | let dirs = [] | |
2145 | let errs = {} | |
2146 | let [cnt, total] = [0, len(g:plugs)] | |
2147 | for [name, spec] in items(g:plugs) | |
2148 | if !s:is_managed(name) | |
2149 | call add(dirs, spec.dir) | |
2150 | else | |
2151 | let [err, clean] = s:git_validate(spec, 1) | |
2152 | if clean | |
2153 | let errs[spec.dir] = s:lines(err)[0] | |
2154 | else | |
2155 | call add(dirs, spec.dir) | |
2156 | endif | |
2157 | endif | |
2158 | let cnt += 1 | |
2159 | call s:progress_bar(2, repeat('=', cnt), total) | |
2160 | normal! 2G | |
2161 | redraw | |
2162 | endfor | |
2163 | ||
2164 | let allowed = {} | |
2165 | for dir in dirs | |
2166 | let allowed[s:dirpath(fnamemodify(dir, ':h:h'))] = 1 | |
2167 | let allowed[dir] = 1 | |
2168 | for child in s:glob_dir(dir) | |
2169 | let allowed[child] = 1 | |
2170 | endfor | |
2171 | endfor | |
2172 | ||
2173 | let todo = [] | |
2174 | let found = sort(s:glob_dir(g:plug_home)) | |
2175 | while !empty(found) | |
2176 | let f = remove(found, 0) | |
2177 | if !has_key(allowed, f) && isdirectory(f) | |
2178 | call add(todo, f) | |
2179 | call append(line('$'), '- ' . f) | |
2180 | if has_key(errs, f) | |
2181 | call append(line('$'), ' ' . errs[f]) | |
2182 | endif | |
2183 | let found = filter(found, 'stridx(v:val, f) != 0') | |
2184 | end | |
2185 | endwhile | |
2186 | ||
2187 | 4 | |
2188 | redraw | |
2189 | if empty(todo) | |
2190 | call append(line('$'), 'Already clean.') | |
2191 | else | |
2192 | let s:clean_count = 0 | |
2193 | call append(3, ['Directories to delete:', '']) | |
2194 | redraw! | |
2195 | if a:force || s:ask_no_interrupt('Delete all directories?') | |
2196 | call s:delete([6, line('$')], 1) | |
2197 | else | |
2198 | call setline(4, 'Cancelled.') | |
2199 | nnoremap <silent> <buffer> d :set opfunc=<sid>delete_op<cr>g@ | |
2200 | nmap <silent> <buffer> dd d_ | |
2201 | xnoremap <silent> <buffer> d :<c-u>call <sid>delete_op(visualmode(), 1)<cr> | |
2202 | echo 'Delete the lines (d{motion}) to delete the corresponding directories' | |
2203 | endif | |
2204 | endif | |
2205 | 4 | |
2206 | setlocal nomodifiable | |
2207 | endfunction | |
2208 | ||
2209 | function! s:delete_op(type, ...) | |
2210 | call s:delete(a:0 ? [line("'<"), line("'>")] : [line("'["), line("']")], 0) | |
2211 | endfunction | |
2212 | ||
2213 | function! s:delete(range, force) | |
2214 | let [l1, l2] = a:range | |
2215 | let force = a:force | |
2216 | while l1 <= l2 | |
2217 | let line = getline(l1) | |
2218 | if line =~ '^- ' && isdirectory(line[2:]) | |
2219 | execute l1 | |
2220 | redraw! | |
2221 | let answer = force ? 1 : s:ask('Delete '.line[2:].'?', 1) | |
2222 | let force = force || answer > 1 | |
2223 | if answer | |
2224 | call s:rm_rf(line[2:]) | |
2225 | setlocal modifiable | |
2226 | call setline(l1, '~'.line[1:]) | |
2227 | let s:clean_count += 1 | |
2228 | call setline(4, printf('Removed %d directories.', s:clean_count)) | |
2229 | setlocal nomodifiable | |
2230 | endif | |
2231 | endif | |
2232 | let l1 += 1 | |
2233 | endwhile | |
2234 | endfunction | |
2235 | ||
2236 | function! s:upgrade() | |
2237 | echo 'Downloading the latest version of vim-plug' | |
2238 | redraw | |
2239 | let tmp = tempname() | |
2240 | let new = tmp . '/plug.vim' | |
2241 | ||
2242 | try | |
795 | 2243 | let out = s:system(printf('git clone --depth 1 %s %s', plug#shellescape(s:plug_src), plug#shellescape(tmp))) |
724 | 2244 | if v:shell_error |
2245 | return s:err('Error upgrading vim-plug: '. out) | |
2246 | endif | |
2247 | ||
2248 | if readfile(s:me) ==# readfile(new) | |
2249 | echo 'vim-plug is already up-to-date' | |
2250 | return 0 | |
2251 | else | |
2252 | call rename(s:me, s:me . '.old') | |
2253 | call rename(new, s:me) | |
2254 | unlet g:loaded_plug | |
2255 | echo 'vim-plug has been upgraded' | |
2256 | return 1 | |
2257 | endif | |
2258 | finally | |
2259 | silent! call s:rm_rf(tmp) | |
2260 | endtry | |
2261 | endfunction | |
2262 | ||
2263 | function! s:upgrade_specs() | |
2264 | for spec in values(g:plugs) | |
2265 | let spec.frozen = get(spec, 'frozen', 0) | |
2266 | endfor | |
2267 | endfunction | |
2268 | ||
2269 | function! s:status() | |
2270 | call s:prepare() | |
2271 | call append(0, 'Checking plugins') | |
2272 | call append(1, '') | |
2273 | ||
2274 | let ecnt = 0 | |
2275 | let unloaded = 0 | |
2276 | let [cnt, total] = [0, len(g:plugs)] | |
2277 | for [name, spec] in items(g:plugs) | |
2278 | let is_dir = isdirectory(spec.dir) | |
2279 | if has_key(spec, 'uri') | |
2280 | if is_dir | |
2281 | let [err, _] = s:git_validate(spec, 1) | |
2282 | let [valid, msg] = [empty(err), empty(err) ? 'OK' : err] | |
2283 | else | |
2284 | let [valid, msg] = [0, 'Not found. Try PlugInstall.'] | |
2285 | endif | |
2286 | else | |
2287 | if is_dir | |
2288 | let [valid, msg] = [1, 'OK'] | |
2289 | else | |
2290 | let [valid, msg] = [0, 'Not found.'] | |
2291 | endif | |
2292 | endif | |
2293 | let cnt += 1 | |
2294 | let ecnt += !valid | |
2295 | " `s:loaded` entry can be missing if PlugUpgraded | |
2296 | if is_dir && get(s:loaded, name, -1) == 0 | |
2297 | let unloaded = 1 | |
2298 | let msg .= ' (not loaded)' | |
2299 | endif | |
2300 | call s:progress_bar(2, repeat('=', cnt), total) | |
2301 | call append(3, s:format_message(valid ? '-' : 'x', name, msg)) | |
2302 | normal! 2G | |
2303 | redraw | |
2304 | endfor | |
2305 | call setline(1, 'Finished. '.ecnt.' error(s).') | |
2306 | normal! gg | |
2307 | setlocal nomodifiable | |
2308 | if unloaded | |
2309 | echo "Press 'L' on each line to load plugin, or 'U' to update" | |
2310 | nnoremap <silent> <buffer> L :call <SID>status_load(line('.'))<cr> | |
2311 | xnoremap <silent> <buffer> L :call <SID>status_load(line('.'))<cr> | |
2312 | end | |
2313 | endfunction | |
2314 | ||
2315 | function! s:extract_name(str, prefix, suffix) | |
2316 | return matchstr(a:str, '^'.a:prefix.' \zs[^:]\+\ze:.*'.a:suffix.'$') | |
2317 | endfunction | |
2318 | ||
2319 | function! s:status_load(lnum) | |
2320 | let line = getline(a:lnum) | |
2321 | let name = s:extract_name(line, '-', '(not loaded)') | |
2322 | if !empty(name) | |
2323 | call plug#load(name) | |
2324 | setlocal modifiable | |
2325 | call setline(a:lnum, substitute(line, ' (not loaded)$', '', '')) | |
2326 | setlocal nomodifiable | |
2327 | endif | |
2328 | endfunction | |
2329 | ||
2330 | function! s:status_update() range | |
2331 | let lines = getline(a:firstline, a:lastline) | |
2332 | let names = filter(map(lines, 's:extract_name(v:val, "[x-]", "")'), '!empty(v:val)') | |
2333 | if !empty(names) | |
2334 | echo | |
2335 | execute 'PlugUpdate' join(names) | |
2336 | endif | |
2337 | endfunction | |
2338 | ||
2339 | function! s:is_preview_window_open() | |
2340 | silent! wincmd P | |
2341 | if &previewwindow | |
2342 | wincmd p | |
2343 | return 1 | |
2344 | endif | |
2345 | endfunction | |
2346 | ||
2347 | function! s:find_name(lnum) | |
2348 | for lnum in reverse(range(1, a:lnum)) | |
2349 | let line = getline(lnum) | |
2350 | if empty(line) | |
2351 | return '' | |
2352 | endif | |
2353 | let name = s:extract_name(line, '-', '') | |
2354 | if !empty(name) | |
2355 | return name | |
2356 | endif | |
2357 | endfor | |
2358 | return '' | |
2359 | endfunction | |
2360 | ||
2361 | function! s:preview_commit() | |
2362 | if b:plug_preview < 0 | |
2363 | let b:plug_preview = !s:is_preview_window_open() | |
2364 | endif | |
2365 | ||
2366 | let sha = matchstr(getline('.'), '^ \X*\zs[0-9a-f]\{7,9}') | |
2367 | if empty(sha) | |
2368 | return | |
2369 | endif | |
2370 | ||
2371 | let name = s:find_name(line('.')) | |
2372 | if empty(name) || !has_key(g:plugs, name) || !isdirectory(g:plugs[name].dir) | |
2373 | return | |
2374 | endif | |
2375 | ||
2376 | if exists('g:plug_pwindow') && !s:is_preview_window_open() | |
2377 | execute g:plug_pwindow | |
2378 | execute 'e' sha | |
2379 | else | |
2380 | execute 'pedit' sha | |
2381 | wincmd P | |
2382 | endif | |
2383 | setlocal previewwindow filetype=git buftype=nofile nobuflisted modifiable | |
817 | 2384 | let batchfile = '' |
724 | 2385 | try |
2386 | let [sh, shellcmdflag, shrd] = s:chsh(1) | |
795 | 2387 | let cmd = 'cd '.plug#shellescape(g:plugs[name].dir).' && git show --no-color --pretty=medium '.sha |
724 | 2388 | if s:is_win |
795 | 2389 | let [batchfile, cmd] = s:batchfile(cmd) |
724 | 2390 | endif |
2391 | execute 'silent %!' cmd | |
2392 | finally | |
2393 | let [&shell, &shellcmdflag, &shellredir] = [sh, shellcmdflag, shrd] | |
817 | 2394 | if s:is_win && filereadable(batchfile) |
724 | 2395 | call delete(batchfile) |
2396 | endif | |
2397 | endtry | |
2398 | setlocal nomodifiable | |
2399 | nnoremap <silent> <buffer> q :q<cr> | |
2400 | wincmd p | |
2401 | endfunction | |
2402 | ||
2403 | function! s:section(flags) | |
2404 | call search('\(^[x-] \)\@<=[^:]\+:', a:flags) | |
2405 | endfunction | |
2406 | ||
2407 | function! s:format_git_log(line) | |
2408 | let indent = ' ' | |
2409 | let tokens = split(a:line, nr2char(1)) | |
2410 | if len(tokens) != 5 | |
2411 | return indent.substitute(a:line, '\s*$', '', '') | |
2412 | endif | |
2413 | let [graph, sha, refs, subject, date] = tokens | |
2414 | let tag = matchstr(refs, 'tag: [^,)]\+') | |
2415 | let tag = empty(tag) ? ' ' : ' ('.tag.') ' | |
2416 | return printf('%s%s%s%s%s (%s)', indent, graph, sha, tag, subject, date) | |
2417 | endfunction | |
2418 | ||
2419 | function! s:append_ul(lnum, text) | |
2420 | call append(a:lnum, ['', a:text, repeat('-', len(a:text))]) | |
2421 | endfunction | |
2422 | ||
2423 | function! s:diff() | |
2424 | call s:prepare() | |
2425 | call append(0, ['Collecting changes ...', '']) | |
2426 | let cnts = [0, 0] | |
2427 | let bar = '' | |
2428 | let total = filter(copy(g:plugs), 's:is_managed(v:key) && isdirectory(v:val.dir)') | |
2429 | call s:progress_bar(2, bar, len(total)) | |
2430 | for origin in [1, 0] | |
2431 | let plugs = reverse(sort(items(filter(copy(total), (origin ? '' : '!').'(has_key(v:val, "commit") || has_key(v:val, "tag"))')))) | |
2432 | if empty(plugs) | |
2433 | continue | |
2434 | endif | |
2435 | call s:append_ul(2, origin ? 'Pending updates:' : 'Last update:') | |
2436 | for [k, v] in plugs | |
2437 | let range = origin ? '..origin/'.v.branch : 'HEAD@{1}..' | |
795 | 2438 | let cmd = 'git log --graph --color=never '.join(map(['--pretty=format:%x01%h%x01%d%x01%s%x01%cr', range], 'plug#shellescape(v:val)')) |
724 | 2439 | if has_key(v, 'rtp') |
795 | 2440 | let cmd .= ' -- '.plug#shellescape(v.rtp) |
724 | 2441 | endif |
2442 | let diff = s:system_chomp(cmd, v.dir) | |
2443 | if !empty(diff) | |
2444 | let ref = has_key(v, 'tag') ? (' (tag: '.v.tag.')') : has_key(v, 'commit') ? (' '.v.commit) : '' | |
2445 | call append(5, extend(['', '- '.k.':'.ref], map(s:lines(diff), 's:format_git_log(v:val)'))) | |
2446 | let cnts[origin] += 1 | |
2447 | endif | |
2448 | let bar .= '=' | |
2449 | call s:progress_bar(2, bar, len(total)) | |
2450 | normal! 2G | |
2451 | redraw | |
2452 | endfor | |
2453 | if !cnts[origin] | |
2454 | call append(5, ['', 'N/A']) | |
2455 | endif | |
2456 | endfor | |
2457 | call setline(1, printf('%d plugin(s) updated.', cnts[0]) | |
2458 | \ . (cnts[1] ? printf(' %d plugin(s) have pending updates.', cnts[1]) : '')) | |
2459 | ||
2460 | if cnts[0] || cnts[1] | |
2461 | nnoremap <silent> <buffer> <plug>(plug-preview) :silent! call <SID>preview_commit()<cr> | |
2462 | if empty(maparg("\<cr>", 'n')) | |
2463 | nmap <buffer> <cr> <plug>(plug-preview) | |
2464 | endif | |
2465 | if empty(maparg('o', 'n')) | |
2466 | nmap <buffer> o <plug>(plug-preview) | |
2467 | endif | |
2468 | endif | |
2469 | if cnts[0] | |
2470 | nnoremap <silent> <buffer> X :call <SID>revert()<cr> | |
2471 | echo "Press 'X' on each block to revert the update" | |
2472 | endif | |
2473 | normal! gg | |
2474 | setlocal nomodifiable | |
2475 | endfunction | |
2476 | ||
2477 | function! s:revert() | |
2478 | if search('^Pending updates', 'bnW') | |
2479 | return | |
2480 | endif | |
2481 | ||
2482 | let name = s:find_name(line('.')) | |
2483 | if empty(name) || !has_key(g:plugs, name) || | |
2484 | \ input(printf('Revert the update of %s? (y/N) ', name)) !~? '^y' | |
2485 | return | |
2486 | endif | |
2487 | ||
2488 | call s:system('git reset --hard HEAD@{1} && git checkout '.s:esc(g:plugs[name].branch).' --', g:plugs[name].dir) | |
2489 | setlocal modifiable | |
2490 | normal! "_dap | |
2491 | setlocal nomodifiable | |
2492 | echo 'Reverted' | |
2493 | endfunction | |
2494 | ||
2495 | function! s:snapshot(force, ...) abort | |
2496 | call s:prepare() | |
2497 | setf vim | |
2498 | call append(0, ['" Generated by vim-plug', | |
2499 | \ '" '.strftime("%c"), | |
2500 | \ '" :source this file in vim to restore the snapshot', | |
2501 | \ '" or execute: vim -S snapshot.vim', | |
2502 | \ '', '', 'PlugUpdate!']) | |
2503 | 1 | |
2504 | let anchor = line('$') - 3 | |
2505 | let names = sort(keys(filter(copy(g:plugs), | |
2506 | \'has_key(v:val, "uri") && !has_key(v:val, "commit") && isdirectory(v:val.dir)'))) | |
2507 | for name in reverse(names) | |
2508 | let sha = s:system_chomp('git rev-parse --short HEAD', g:plugs[name].dir) | |
2509 | if !empty(sha) | |
2510 | call append(anchor, printf("silent! let g:plugs['%s'].commit = '%s'", name, sha)) | |
2511 | redraw | |
2512 | endif | |
2513 | endfor | |
2514 | ||
2515 | if a:0 > 0 | |
2516 | let fn = expand(a:1) | |
2517 | if filereadable(fn) && !(a:force || s:ask(a:1.' already exists. Overwrite?')) | |
2518 | return | |
2519 | endif | |
2520 | call writefile(getline(1, '$'), fn) | |
2521 | echo 'Saved as '.a:1 | |
2522 | silent execute 'e' s:esc(fn) | |
2523 | setf vim | |
2524 | endif | |
2525 | endfunction | |
2526 | ||
2527 | function! s:split_rtp() | |
2528 | return split(&rtp, '\\\@<!,') | |
2529 | endfunction | |
2530 | ||
2531 | let s:first_rtp = s:escrtp(get(s:split_rtp(), 0, '')) | |
2532 | let s:last_rtp = s:escrtp(get(s:split_rtp(), -1, '')) | |
2533 | ||
2534 | if exists('g:plugs') | |
2535 | let g:plugs_order = get(g:, 'plugs_order', keys(g:plugs)) | |
2536 | call s:upgrade_specs() | |
2537 | call s:define_commands() | |
2538 | endif | |
2539 | ||
2540 | let &cpo = s:cpo_save | |
2541 | unlet s:cpo_save |