plug.vim 57 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039
  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. " Plug 'junegunn/seoul256.vim'
  15. " Plug 'junegunn/vim-easy-align'
  16. "
  17. " " On-demand loading
  18. " Plug 'scrooloose/nerdtree', { 'on': 'NERDTreeToggle' }
  19. " Plug 'tpope/vim-fireplace', { 'for': 'clojure' }
  20. "
  21. " " Using git URL
  22. " Plug 'https://github.com/junegunn/vim-github-dashboard.git'
  23. "
  24. " " Plugin options
  25. " Plug 'nsf/gocode', { 'tag': 'v.20150303', 'rtp': 'vim' }
  26. "
  27. " " Plugin outside ~/.vim/plugged with post-update hook
  28. " Plug 'junegunn/fzf', { 'dir': '~/.fzf', 'do': 'yes \| ./install' }
  29. "
  30. " " Unmanaged plugin (manually installed and updated)
  31. " Plug '~/my-prototype-plugin'
  32. "
  33. " call plug#end()
  34. "
  35. " Then reload .vimrc and :PlugInstall to install plugins.
  36. " Visit https://github.com/junegunn/vim-plug for more information.
  37. "
  38. "
  39. " Copyright (c) 2015 Junegunn Choi
  40. "
  41. " MIT License
  42. "
  43. " Permission is hereby granted, free of charge, to any person obtaining
  44. " a copy of this software and associated documentation files (the
  45. " "Software"), to deal in the Software without restriction, including
  46. " without limitation the rights to use, copy, modify, merge, publish,
  47. " distribute, sublicense, and/or sell copies of the Software, and to
  48. " permit persons to whom the Software is furnished to do so, subject to
  49. " the following conditions:
  50. "
  51. " The above copyright notice and this permission notice shall be
  52. " included in all copies or substantial portions of the Software.
  53. "
  54. " THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  55. " EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  56. " MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  57. " NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  58. " LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  59. " OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  60. " WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  61. if exists('g:loaded_plug')
  62. finish
  63. endif
  64. let g:loaded_plug = 1
  65. let s:cpo_save = &cpo
  66. set cpo&vim
  67. let s:plug_src = 'https://github.com/junegunn/vim-plug.git'
  68. let s:plug_tab = get(s:, 'plug_tab', -1)
  69. let s:plug_buf = get(s:, 'plug_buf', -1)
  70. let s:mac_gui = has('gui_macvim') && has('gui_running')
  71. let s:is_win = has('win32') || has('win64')
  72. let s:nvim = has('nvim') && exists('*jobwait') && !s:is_win
  73. let s:me = resolve(expand('<sfile>:p'))
  74. let s:base_spec = { 'branch': 'master', 'frozen': 0 }
  75. let s:TYPE = {
  76. \ 'string': type(''),
  77. \ 'list': type([]),
  78. \ 'dict': type({}),
  79. \ 'funcref': type(function('call'))
  80. \ }
  81. let s:loaded = get(s:, 'loaded', {})
  82. let s:triggers = get(s:, 'triggers', {})
  83. function! plug#begin(...)
  84. if a:0 > 0
  85. let s:plug_home_org = a:1
  86. let home = s:path(fnamemodify(expand(a:1), ':p'))
  87. elseif exists('g:plug_home')
  88. let home = s:path(g:plug_home)
  89. elseif !empty(&rtp)
  90. let home = s:path(split(&rtp, ',')[0]) . '/plugged'
  91. else
  92. return s:err('Unable to determine plug home. Try calling plug#begin() with a path argument.')
  93. endif
  94. let g:plug_home = home
  95. let g:plugs = {}
  96. let g:plugs_order = []
  97. let s:triggers = {}
  98. call s:define_commands()
  99. return 1
  100. endfunction
  101. function! s:define_commands()
  102. command! -nargs=+ -bar Plug call s:add(<args>)
  103. if !executable('git')
  104. return s:err('`git` executable not found. vim-plug requires git.')
  105. endif
  106. command! -nargs=* -bar -bang -complete=customlist,s:names PlugInstall call s:install('<bang>' == '!', [<f-args>])
  107. command! -nargs=* -bar -bang -complete=customlist,s:names PlugUpdate call s:update('<bang>' == '!', [<f-args>])
  108. command! -nargs=0 -bar -bang PlugClean call s:clean('<bang>' == '!')
  109. command! -nargs=0 -bar PlugUpgrade if s:upgrade() | execute 'source' s:esc(s:me) | endif
  110. command! -nargs=0 -bar PlugStatus call s:status()
  111. command! -nargs=0 -bar PlugDiff call s:diff()
  112. command! -nargs=? -bar PlugSnapshot call s:snapshot(<f-args>)
  113. endfunction
  114. function! s:to_a(v)
  115. return type(a:v) == s:TYPE.list ? a:v : [a:v]
  116. endfunction
  117. function! s:to_s(v)
  118. return type(a:v) == s:TYPE.string ? a:v : join(a:v, "\n") . "\n"
  119. endfunction
  120. function! s:source(from, ...)
  121. for pattern in a:000
  122. for vim in s:lines(globpath(a:from, pattern))
  123. execute 'source' s:esc(vim)
  124. endfor
  125. endfor
  126. endfunction
  127. function! s:assoc(dict, key, val)
  128. let a:dict[a:key] = add(get(a:dict, a:key, []), a:val)
  129. endfunction
  130. function! plug#end()
  131. if !exists('g:plugs')
  132. return s:err('Call plug#begin() first')
  133. endif
  134. if exists('#PlugLOD')
  135. augroup PlugLOD
  136. autocmd!
  137. augroup END
  138. augroup! PlugLOD
  139. endif
  140. let lod = { 'ft': {}, 'map': {}, 'cmd': {} }
  141. filetype off
  142. for name in g:plugs_order
  143. let plug = g:plugs[name]
  144. if get(s:loaded, name, 0) || !has_key(plug, 'on') && !has_key(plug, 'for')
  145. let s:loaded[name] = 1
  146. continue
  147. endif
  148. if has_key(plug, 'on')
  149. let s:triggers[name] = { 'map': [], 'cmd': [] }
  150. for cmd in s:to_a(plug.on)
  151. if cmd =~? '^<Plug>.\+'
  152. if empty(mapcheck(cmd)) && empty(mapcheck(cmd, 'i'))
  153. call s:assoc(lod.map, cmd, name)
  154. endif
  155. call add(s:triggers[name].map, cmd)
  156. elseif cmd =~ '^[A-Z]'
  157. if exists(':'.cmd) != 2
  158. call s:assoc(lod.cmd, cmd, name)
  159. endif
  160. call add(s:triggers[name].cmd, cmd)
  161. endif
  162. endfor
  163. endif
  164. if has_key(plug, 'for')
  165. let types = s:to_a(plug.for)
  166. if !empty(types)
  167. call s:source(s:rtp(plug), 'ftdetect/**/*.vim', 'after/ftdetect/**/*.vim')
  168. endif
  169. for type in types
  170. call s:assoc(lod.ft, type, name)
  171. endfor
  172. endif
  173. endfor
  174. for [cmd, names] in items(lod.cmd)
  175. execute printf(
  176. \ 'command! -nargs=* -range -bang %s call s:lod_cmd(%s, "<bang>", <line1>, <line2>, <q-args>, %s)',
  177. \ cmd, string(cmd), string(names))
  178. endfor
  179. for [map, names] in items(lod.map)
  180. for [mode, map_prefix, key_prefix] in
  181. \ [['i', '<C-O>', ''], ['n', '', ''], ['v', '', 'gv'], ['o', '', '']]
  182. execute printf(
  183. \ '%snoremap <silent> %s %s:<C-U>call <SID>lod_map(%s, %s, "%s")<CR>',
  184. \ mode, map, map_prefix, string(map), string(names), key_prefix)
  185. endfor
  186. endfor
  187. for [ft, names] in items(lod.ft)
  188. augroup PlugLOD
  189. execute printf('autocmd FileType %s call <SID>lod_ft(%s, %s)',
  190. \ ft, string(ft), string(names))
  191. augroup END
  192. endfor
  193. call s:reorg_rtp()
  194. filetype plugin indent on
  195. if has('vim_starting')
  196. syntax enable
  197. else
  198. call s:reload()
  199. endif
  200. endfunction
  201. function! s:loaded_names()
  202. return filter(copy(g:plugs_order), 'get(s:loaded, v:val, 0)')
  203. endfunction
  204. function! s:reload()
  205. for name in s:loaded_names()
  206. call s:source(s:rtp(g:plugs[name]), 'plugin/**/*.vim', 'after/plugin/**/*.vim')
  207. endfor
  208. endfunction
  209. function! s:trim(str)
  210. return substitute(a:str, '[\/]\+$', '', '')
  211. endfunction
  212. function! s:version_requirement(val, min)
  213. for idx in range(0, len(a:min) - 1)
  214. let v = get(a:val, idx, 0)
  215. if v < a:min[idx] | return 0
  216. elseif v > a:min[idx] | return 1
  217. endif
  218. endfor
  219. return 1
  220. endfunction
  221. function! s:git_version_requirement(...)
  222. let s:git_version = get(s:, 'git_version',
  223. \ map(split(split(s:system('git --version'))[-1], '\.'), 'str2nr(v:val)'))
  224. return s:version_requirement(s:git_version, a:000)
  225. endfunction
  226. function! s:progress_opt(base)
  227. return a:base && !s:is_win &&
  228. \ s:git_version_requirement(1, 7, 1) ? '--progress' : ''
  229. endfunction
  230. if s:is_win
  231. function! s:rtp(spec)
  232. return s:path(a:spec.dir . get(a:spec, 'rtp', ''))
  233. endfunction
  234. function! s:path(path)
  235. return s:trim(substitute(a:path, '/', '\', 'g'))
  236. endfunction
  237. function! s:dirpath(path)
  238. return s:path(a:path) . '\'
  239. endfunction
  240. function! s:is_local_plug(repo)
  241. return a:repo =~? '^[a-z]:\|^[%~]'
  242. endfunction
  243. else
  244. function! s:rtp(spec)
  245. return s:dirpath(a:spec.dir . get(a:spec, 'rtp', ''))
  246. endfunction
  247. function! s:path(path)
  248. return s:trim(a:path)
  249. endfunction
  250. function! s:dirpath(path)
  251. return substitute(a:path, '[/\\]*$', '/', '')
  252. endfunction
  253. function! s:is_local_plug(repo)
  254. return a:repo[0] =~ '[/$~]'
  255. endfunction
  256. endif
  257. function! s:err(msg)
  258. echohl ErrorMsg
  259. echom a:msg
  260. echohl None
  261. return 0
  262. endfunction
  263. function! s:esc(path)
  264. return escape(a:path, ' ')
  265. endfunction
  266. function! s:escrtp(path)
  267. return escape(a:path, ' ,')
  268. endfunction
  269. function! s:remove_rtp()
  270. for name in s:loaded_names()
  271. let rtp = s:rtp(g:plugs[name])
  272. execute 'set rtp-='.s:escrtp(rtp)
  273. let after = globpath(rtp, 'after')
  274. if isdirectory(after)
  275. execute 'set rtp-='.s:escrtp(after)
  276. endif
  277. endfor
  278. endfunction
  279. function! s:reorg_rtp()
  280. if !empty(s:first_rtp)
  281. execute 'set rtp-='.s:first_rtp
  282. execute 'set rtp-='.s:last_rtp
  283. endif
  284. " &rtp is modified from outside
  285. if exists('s:prtp') && s:prtp !=# &rtp
  286. call s:remove_rtp()
  287. unlet! s:middle
  288. endif
  289. let s:middle = get(s:, 'middle', &rtp)
  290. let rtps = map(s:loaded_names(), 's:rtp(g:plugs[v:val])')
  291. let afters = filter(map(copy(rtps), 'globpath(v:val, "after")'), 'isdirectory(v:val)')
  292. let rtp = join(map(rtps, 'escape(v:val, ",")'), ',')
  293. \ . ','.s:middle.','
  294. \ . join(map(afters, 'escape(v:val, ",")'), ',')
  295. let &rtp = substitute(substitute(rtp, ',,*', ',', 'g'), '^,\|,$', '', 'g')
  296. let s:prtp = &rtp
  297. if !empty(s:first_rtp)
  298. execute 'set rtp^='.s:first_rtp
  299. execute 'set rtp+='.s:last_rtp
  300. endif
  301. endfunction
  302. function! plug#load(...)
  303. if a:0 == 0
  304. return s:err('Argument missing: plugin name(s) required')
  305. endif
  306. if !exists('g:plugs')
  307. return s:err('plug#begin was not called')
  308. endif
  309. let unknowns = filter(copy(a:000), '!has_key(g:plugs, v:val)')
  310. if !empty(unknowns)
  311. let s = len(unknowns) > 1 ? 's' : ''
  312. return s:err(printf('Unknown plugin%s: %s', s, join(unknowns, ', ')))
  313. end
  314. for name in a:000
  315. call s:lod([name], ['ftdetect', 'after/ftdetect', 'plugin', 'after/plugin'])
  316. endfor
  317. doautocmd BufRead
  318. return 1
  319. endfunction
  320. function! s:remove_triggers(name)
  321. if !has_key(s:triggers, a:name)
  322. return
  323. endif
  324. for cmd in s:triggers[a:name].cmd
  325. execute 'silent! delc' cmd
  326. endfor
  327. for map in s:triggers[a:name].map
  328. execute 'silent! unmap' map
  329. execute 'silent! iunmap' map
  330. endfor
  331. call remove(s:triggers, a:name)
  332. endfunction
  333. function! s:lod(names, types)
  334. for name in a:names
  335. call s:remove_triggers(name)
  336. let s:loaded[name] = 1
  337. endfor
  338. call s:reorg_rtp()
  339. for name in a:names
  340. let rtp = s:rtp(g:plugs[name])
  341. for dir in a:types
  342. call s:source(rtp, dir.'/**/*.vim')
  343. endfor
  344. endfor
  345. endfunction
  346. function! s:lod_ft(pat, names)
  347. call s:lod(a:names, ['plugin', 'after/plugin'])
  348. execute 'autocmd! PlugLOD FileType' a:pat
  349. doautocmd filetypeplugin FileType
  350. doautocmd filetypeindent FileType
  351. endfunction
  352. function! s:lod_cmd(cmd, bang, l1, l2, args, names)
  353. call s:lod(a:names, ['ftdetect', 'after/ftdetect', 'plugin', 'after/plugin'])
  354. execute printf('%s%s%s %s', (a:l1 == a:l2 ? '' : (a:l1.','.a:l2)), a:cmd, a:bang, a:args)
  355. endfunction
  356. function! s:lod_map(map, names, prefix)
  357. call s:lod(a:names, ['ftdetect', 'after/ftdetect', 'plugin', 'after/plugin'])
  358. let extra = ''
  359. while 1
  360. let c = getchar(0)
  361. if c == 0
  362. break
  363. endif
  364. let extra .= nr2char(c)
  365. endwhile
  366. call feedkeys(a:prefix . substitute(a:map, '^<Plug>', "\<Plug>", '') . extra)
  367. endfunction
  368. function! s:add(repo, ...)
  369. if a:0 > 1
  370. return s:err('Invalid number of arguments (1..2)')
  371. endif
  372. try
  373. let repo = s:trim(a:repo)
  374. let name = fnamemodify(repo, ':t:s?\.git$??')
  375. let spec = extend(s:infer_properties(name, repo),
  376. \ a:0 == 1 ? s:parse_options(a:1) : s:base_spec)
  377. if !has_key(g:plugs, name)
  378. call add(g:plugs_order, name)
  379. endif
  380. let g:plugs[name] = spec
  381. let s:loaded[name] = 0
  382. catch
  383. return s:err(v:exception)
  384. endtry
  385. endfunction
  386. function! s:parse_options(arg)
  387. let opts = copy(s:base_spec)
  388. let type = type(a:arg)
  389. if type == s:TYPE.string
  390. let opts.tag = a:arg
  391. elseif type == s:TYPE.dict
  392. call extend(opts, a:arg)
  393. if has_key(opts, 'dir')
  394. let opts.dir = s:dirpath(expand(opts.dir))
  395. endif
  396. else
  397. throw 'Invalid argument type (expected: string or dictionary)'
  398. endif
  399. return opts
  400. endfunction
  401. function! s:infer_properties(name, repo)
  402. let repo = a:repo
  403. if s:is_local_plug(repo)
  404. return { 'dir': s:dirpath(expand(repo)) }
  405. else
  406. if repo =~ ':'
  407. let uri = repo
  408. else
  409. if repo !~ '/'
  410. let repo = 'vim-scripts/'. repo
  411. endif
  412. let fmt = get(g:, 'plug_url_format', 'https://git::@github.com/%s.git')
  413. let uri = printf(fmt, repo)
  414. endif
  415. let dir = s:dirpath( fnamemodify(join([g:plug_home, a:name], '/'), ':p') )
  416. return { 'dir': dir, 'uri': uri }
  417. endif
  418. endfunction
  419. function! s:install(force, names)
  420. call s:update_impl(0, a:force, a:names)
  421. endfunction
  422. function! s:update(force, names)
  423. call s:update_impl(1, a:force, a:names)
  424. endfunction
  425. function! plug#helptags()
  426. if !exists('g:plugs')
  427. return s:err('plug#begin was not called')
  428. endif
  429. for spec in values(g:plugs)
  430. let docd = join([spec.dir, 'doc'], '/')
  431. if isdirectory(docd)
  432. silent! execute 'helptags' s:esc(docd)
  433. endif
  434. endfor
  435. return 1
  436. endfunction
  437. function! s:syntax()
  438. syntax clear
  439. syntax region plug1 start=/\%1l/ end=/\%2l/ contains=plugNumber
  440. syntax region plug2 start=/\%2l/ end=/\%3l/ contains=plugBracket,plugX
  441. syn match plugNumber /[0-9]\+[0-9.]*/ contained
  442. syn match plugBracket /[[\]]/ contained
  443. syn match plugX /x/ contained
  444. syn match plugDash /^-/
  445. syn match plugPlus /^+/
  446. syn match plugStar /^*/
  447. syn match plugMessage /\(^- \)\@<=.*/
  448. syn match plugName /\(^- \)\@<=[^ ]*:/
  449. syn match plugInstall /\(^+ \)\@<=[^:]*/
  450. syn match plugUpdate /\(^* \)\@<=[^:]*/
  451. syn match plugCommit /^ [0-9a-z]\{7} .*/ contains=plugRelDate,plugSha
  452. syn match plugSha /\(^ \)\@<=[0-9a-z]\{7}/ contained
  453. syn match plugRelDate /([^)]*)$/ contained
  454. syn match plugNotLoaded /(not loaded)$/
  455. syn match plugError /^x.*/
  456. syn keyword Function PlugInstall PlugStatus PlugUpdate PlugClean
  457. hi def link plug1 Title
  458. hi def link plug2 Repeat
  459. hi def link plugX Exception
  460. hi def link plugBracket Structure
  461. hi def link plugNumber Number
  462. hi def link plugDash Special
  463. hi def link plugPlus Constant
  464. hi def link plugStar Boolean
  465. hi def link plugMessage Function
  466. hi def link plugName Label
  467. hi def link plugInstall Function
  468. hi def link plugUpdate Type
  469. hi def link plugError Error
  470. hi def link plugRelDate Comment
  471. hi def link plugSha Identifier
  472. hi def link plugNotLoaded Comment
  473. endfunction
  474. function! s:lpad(str, len)
  475. return a:str . repeat(' ', a:len - len(a:str))
  476. endfunction
  477. function! s:lines(msg)
  478. return split(a:msg, "[\r\n]")
  479. endfunction
  480. function! s:lastline(msg)
  481. return get(s:lines(a:msg), -1, '')
  482. endfunction
  483. function! s:new_window()
  484. execute get(g:, 'plug_window', 'vertical topleft new')
  485. endfunction
  486. function! s:plug_window_exists()
  487. let buflist = tabpagebuflist(s:plug_tab)
  488. return !empty(buflist) && index(buflist, s:plug_buf) >= 0
  489. endfunction
  490. function! s:switch_in()
  491. if !s:plug_window_exists()
  492. return 0
  493. endif
  494. if winbufnr(0) != s:plug_buf
  495. let s:pos = [tabpagenr(), winnr(), winsaveview()]
  496. execute 'normal!' s:plug_tab.'gt'
  497. let winnr = bufwinnr(s:plug_buf)
  498. execute winnr.'wincmd w'
  499. call add(s:pos, winsaveview())
  500. else
  501. let s:pos = [winsaveview()]
  502. endif
  503. setlocal modifiable
  504. return 1
  505. endfunction
  506. function! s:switch_out(...)
  507. call winrestview(s:pos[-1])
  508. setlocal nomodifiable
  509. if a:0 > 0
  510. execute a:1
  511. endif
  512. if len(s:pos) > 1
  513. execute 'normal!' s:pos[0].'gt'
  514. execute s:pos[1] 'wincmd w'
  515. call winrestview(s:pos[2])
  516. endif
  517. endfunction
  518. function! s:prepare()
  519. call s:job_abort()
  520. if s:switch_in()
  521. silent %d _
  522. else
  523. call s:new_window()
  524. nnoremap <silent> <buffer> q :if b:plug_preview==1<bar>pc<bar>endif<bar>echo<bar>q<cr>
  525. nnoremap <silent> <buffer> R :silent! call <SID>retry()<cr>
  526. nnoremap <silent> <buffer> D :PlugDiff<cr>
  527. nnoremap <silent> <buffer> S :PlugStatus<cr>
  528. nnoremap <silent> <buffer> U :call <SID>status_update()<cr>
  529. xnoremap <silent> <buffer> U :call <SID>status_update()<cr>
  530. nnoremap <silent> <buffer> ]] :silent! call <SID>section('')<cr>
  531. nnoremap <silent> <buffer> [[ :silent! call <SID>section('b')<cr>
  532. let b:plug_preview = -1
  533. let s:plug_tab = tabpagenr()
  534. let s:plug_buf = winbufnr(0)
  535. call s:assign_name()
  536. endif
  537. silent! unmap <buffer> <cr>
  538. silent! unmap <buffer> L
  539. silent! unmap <buffer> o
  540. silent! unmap <buffer> X
  541. setlocal buftype=nofile bufhidden=wipe nobuflisted noswapfile nowrap cursorline modifiable
  542. setf vim-plug
  543. call s:syntax()
  544. endfunction
  545. function! s:assign_name()
  546. " Assign buffer name
  547. let prefix = '[Plugins]'
  548. let name = prefix
  549. let idx = 2
  550. while bufexists(name)
  551. let name = printf('%s (%s)', prefix, idx)
  552. let idx = idx + 1
  553. endwhile
  554. silent! execute 'f' fnameescape(name)
  555. endfunction
  556. function! s:do(pull, force, todo)
  557. for [name, spec] in items(a:todo)
  558. if !isdirectory(spec.dir)
  559. continue
  560. endif
  561. let installed = has_key(s:update.new, name)
  562. let updated = installed ? 0 :
  563. \ (a:pull && !empty(s:system_chomp('git log --pretty=format:"%h" "HEAD...HEAD@{1}"', spec.dir)))
  564. if a:force || installed || updated
  565. execute 'cd' s:esc(spec.dir)
  566. call append(3, '- Post-update hook for '. name .' ... ')
  567. let type = type(spec.do)
  568. if type == s:TYPE.string
  569. try
  570. " FIXME: Escaping is incomplete. We could use shellescape with eval,
  571. " but it won't work on Windows.
  572. let g:_plug_do = '!'.escape(spec.do, '#!%')
  573. execute "normal! :execute g:_plug_do\<cr>\<cr>"
  574. finally
  575. let result = v:shell_error ? ('Exit status: '.v:shell_error) : 'Done!'
  576. unlet g:_plug_do
  577. endtry
  578. elseif type == s:TYPE.funcref
  579. try
  580. let status = installed ? 'installed' : (updated ? 'updated' : 'unchanged')
  581. call spec.do({ 'name': name, 'status': status, 'force': a:force })
  582. let result = 'Done!'
  583. catch
  584. let result = 'Error: ' . v:exception
  585. endtry
  586. else
  587. let result = 'Error: Invalid type!'
  588. endif
  589. call setline(4, getline(4) . result)
  590. cd -
  591. endif
  592. endfor
  593. endfunction
  594. function! s:finish(pull)
  595. let new_frozen = len(filter(keys(s:update.new), 'g:plugs[v:val].frozen'))
  596. if new_frozen
  597. let s = new_frozen > 1 ? 's' : ''
  598. call append(3, printf('- Installed %d frozen plugin%s', new_frozen, s))
  599. endif
  600. call append(3, '- Finishing ... ')
  601. redraw
  602. call plug#helptags()
  603. call plug#end()
  604. call setline(4, getline(4) . 'Done!')
  605. redraw
  606. let msgs = []
  607. if !empty(s:update.errors)
  608. call add(msgs, "Press 'R' to retry.")
  609. endif
  610. if a:pull && len(s:update.new) < len(filter(getline(5, '$'),
  611. \ "v:val =~ '^- ' && stridx(v:val, 'Already up-to-date') < 0"))
  612. call add(msgs, "Press 'D' to see the updated changes.")
  613. endif
  614. echo join(msgs, ' ')
  615. endfunction
  616. function! s:retry()
  617. if empty(s:update.errors)
  618. return
  619. endif
  620. call s:update_impl(s:update.pull, s:update.force,
  621. \ extend(copy(s:update.errors), [s:update.threads]))
  622. endfunction
  623. function! s:is_managed(name)
  624. return has_key(g:plugs[a:name], 'uri')
  625. endfunction
  626. function! s:names(...)
  627. return sort(filter(keys(g:plugs), 'stridx(v:val, a:1) == 0 && s:is_managed(v:val)'))
  628. endfunction
  629. function! s:update_impl(pull, force, args) abort
  630. let args = copy(a:args)
  631. let threads = (len(args) > 0 && args[-1] =~ '^[1-9][0-9]*$') ?
  632. \ remove(args, -1) : get(g:, 'plug_threads', s:is_win ? 1 : 16)
  633. let managed = filter(copy(g:plugs), 's:is_managed(v:key)')
  634. let todo = empty(args) ? filter(managed, '!v:val.frozen || !isdirectory(v:val.dir)') :
  635. \ filter(managed, 'index(args, v:key) >= 0')
  636. if empty(todo)
  637. echohl WarningMsg
  638. echo 'No plugin to '. (a:pull ? 'update' : 'install') . '.'
  639. echohl None
  640. return
  641. endif
  642. if !s:is_win && s:git_version_requirement(2, 3)
  643. let s:git_terminal_prompt = exists('$GIT_TERMINAL_PROMPT') ? $GIT_TERMINAL_PROMPT : ''
  644. let $GIT_TERMINAL_PROMPT = 0
  645. for plug in values(todo)
  646. let plug.uri = substitute(plug.uri,
  647. \ '^https://git::@github\.com', 'https://github.com', '')
  648. endfor
  649. endif
  650. if !isdirectory(g:plug_home)
  651. try
  652. call mkdir(g:plug_home, 'p')
  653. catch
  654. return s:err(printf('Invalid plug directory: %s. '.
  655. \ 'Try to call plug#begin with a valid directory', g:plug_home))
  656. endtry
  657. endif
  658. if has('nvim') && !exists('*jobwait') && threads > 1
  659. echohl WarningMsg
  660. echomsg 'vim-plug: update Neovim for parallel installer'
  661. echohl None
  662. endif
  663. let python = (has('python') || has('python3')) && !s:is_win && !has('win32unix')
  664. \ && (!s:nvim || has('vim_starting'))
  665. let ruby = has('ruby') && !s:nvim && (v:version >= 703 || v:version == 702 && has('patch374'))
  666. let s:update = {
  667. \ 'start': reltime(),
  668. \ 'all': todo,
  669. \ 'todo': copy(todo),
  670. \ 'errors': [],
  671. \ 'pull': a:pull,
  672. \ 'force': a:force,
  673. \ 'new': {},
  674. \ 'threads': (python || ruby || s:nvim) ? min([len(todo), threads]) : 1,
  675. \ 'bar': '',
  676. \ 'fin': 0
  677. \ }
  678. call s:prepare()
  679. call append(0, ['', ''])
  680. normal! 2G
  681. silent! redraw
  682. let s:clone_opt = get(g:, 'plug_shallow', 1) ?
  683. \ '--depth 1' . (s:git_version_requirement(1, 7, 10) ? ' --no-single-branch' : '') : ''
  684. " Python version requirement (>= 2.7)
  685. if python && !has('python3') && !ruby && !s:nvim && s:update.threads > 1
  686. redir => pyv
  687. silent python import platform; print(platform.python_version())
  688. redir END
  689. let python = s:version_requirement(
  690. \ map(split(split(pyv)[0], '\.'), 'str2nr(v:val)'), [2, 6])
  691. endif
  692. if (python || ruby) && s:update.threads > 1
  693. try
  694. let imd = &imd
  695. if s:mac_gui
  696. set noimd
  697. endif
  698. if ruby
  699. call s:update_ruby()
  700. else
  701. call s:update_python()
  702. endif
  703. catch
  704. let lines = getline(4, '$')
  705. let printed = {}
  706. silent! 4,$d _
  707. for line in lines
  708. let name = s:extract_name(line, '.', '')
  709. if empty(name) || !has_key(printed, name)
  710. call append('$', line)
  711. if !empty(name)
  712. let printed[name] = 1
  713. if line[0] == 'x' && index(s:update.errors, name) < 0
  714. call add(s:update.errors, name)
  715. end
  716. endif
  717. endif
  718. endfor
  719. finally
  720. let &imd = imd
  721. call s:update_finish()
  722. endtry
  723. else
  724. call s:update_vim()
  725. endif
  726. endfunction
  727. function! s:update_finish()
  728. if exists('s:git_terminal_prompt')
  729. let $GIT_TERMINAL_PROMPT = s:git_terminal_prompt
  730. endif
  731. if s:switch_in()
  732. call s:do(s:update.pull, s:update.force, filter(copy(s:update.all), 'has_key(v:val, "do")'))
  733. call s:finish(s:update.pull)
  734. call setline(1, 'Updated. Elapsed time: ' . split(reltimestr(reltime(s:update.start)))[0] . ' sec.')
  735. call s:switch_out('normal! gg')
  736. endif
  737. endfunction
  738. function! s:job_abort()
  739. if !s:nvim || !exists('s:jobs')
  740. return
  741. endif
  742. for [name, j] in items(s:jobs)
  743. silent! call jobstop(j.jobid)
  744. if j.new
  745. call s:system('rm -rf ' . s:shellesc(g:plugs[name].dir))
  746. endif
  747. endfor
  748. let s:jobs = {}
  749. endfunction
  750. " When a:event == 'stdout', data = list of strings
  751. " When a:event == 'exit', data = returncode
  752. function! s:job_handler(job_id, data, event) abort
  753. if !s:plug_window_exists() " plug window closed
  754. return s:job_abort()
  755. endif
  756. if a:event == 'stdout'
  757. let self.result .= substitute(s:to_s(a:data), '[\r\n]', '', 'g') . "\n"
  758. " To reduce the number of buffer updates
  759. let self.tick = get(self, 'tick', -1) + 1
  760. if self.tick % len(s:jobs) == 0
  761. call s:log(self.new ? '+' : '*', self.name, self.result)
  762. endif
  763. elseif a:event == 'exit'
  764. let self.running = 0
  765. if a:data != 0
  766. let self.error = 1
  767. endif
  768. call s:reap(self.name)
  769. call s:tick()
  770. endif
  771. endfunction
  772. function! s:spawn(name, cmd, opts)
  773. let job = { 'name': a:name, 'running': 1, 'error': 0, 'result': '',
  774. \ 'new': get(a:opts, 'new', 0),
  775. \ 'on_stdout': function('s:job_handler'),
  776. \ 'on_exit' : function('s:job_handler'),
  777. \ }
  778. let s:jobs[a:name] = job
  779. if s:nvim
  780. let argv = [ 'sh', '-c',
  781. \ (has_key(a:opts, 'dir') ? s:with_cd(a:cmd, a:opts.dir) : a:cmd) ]
  782. let jid = jobstart(argv, job)
  783. if jid > 0
  784. let job.jobid = jid
  785. else
  786. let job.running = 0
  787. let job.error = 1
  788. let job.result = jid < 0 ? 'sh is not executable' :
  789. \ 'Invalid arguments (or job table is full)'
  790. endif
  791. else
  792. let params = has_key(a:opts, 'dir') ? [a:cmd, a:opts.dir] : [a:cmd]
  793. let job.result = call('s:system', params)
  794. let job.error = v:shell_error != 0
  795. let job.running = 0
  796. endif
  797. endfunction
  798. function! s:reap(name)
  799. let job = s:jobs[a:name]
  800. if job.error
  801. call add(s:update.errors, a:name)
  802. elseif get(job, 'new', 0)
  803. let s:update.new[a:name] = 1
  804. endif
  805. let s:update.bar .= job.error ? 'x' : '='
  806. call s:log(job.error ? 'x' : '-', a:name, job.result)
  807. call s:bar()
  808. call remove(s:jobs, a:name)
  809. endfunction
  810. function! s:bar()
  811. if s:switch_in()
  812. let total = len(s:update.all)
  813. call setline(1, (s:update.pull ? 'Updating' : 'Installing').
  814. \ ' plugins ('.len(s:update.bar).'/'.total.')')
  815. call s:progress_bar(2, s:update.bar, total)
  816. call s:switch_out()
  817. endif
  818. endfunction
  819. function! s:logpos(name)
  820. for i in range(1, line('$'))
  821. if getline(i) =~# '^[-+x*] '.a:name.':'
  822. return i
  823. endif
  824. endfor
  825. return 0
  826. endfunction
  827. function! s:log(bullet, name, lines)
  828. if s:switch_in()
  829. let pos = s:logpos(a:name)
  830. if pos > 0
  831. execute pos 'd _'
  832. if pos > winheight('.')
  833. let pos = 4
  834. endif
  835. else
  836. let pos = 4
  837. endif
  838. call append(pos - 1, s:format_message(a:bullet, a:name, a:lines))
  839. call s:switch_out()
  840. endif
  841. endfunction
  842. function! s:update_vim()
  843. let s:jobs = {}
  844. call s:bar()
  845. call s:tick()
  846. endfunction
  847. function! s:tick()
  848. let pull = s:update.pull
  849. let prog = s:progress_opt(s:nvim)
  850. while 1 " Without TCO, Vim stack is bound to explode
  851. if empty(s:update.todo)
  852. if empty(s:jobs) && !s:update.fin
  853. let s:update.fin = 1
  854. call s:update_finish()
  855. endif
  856. return
  857. endif
  858. let name = keys(s:update.todo)[0]
  859. let spec = remove(s:update.todo, name)
  860. let new = !isdirectory(spec.dir)
  861. call s:log(new ? '+' : '*', name, pull ? 'Updating ...' : 'Installing ...')
  862. redraw
  863. let has_tag = has_key(spec, 'tag')
  864. let checkout = s:shellesc(has_tag ? spec.tag : spec.branch)
  865. let merge = s:shellesc(has_tag ? spec.tag : 'origin/'.spec.branch)
  866. if !new
  867. let [valid, msg] = s:git_valid(spec, 0)
  868. if valid
  869. if pull
  870. let fetch_opt = (has_tag && !empty(globpath(spec.dir, '.git/shallow'))) ? '--depth 99999999' : ''
  871. call s:spawn(name,
  872. \ printf('(git fetch %s %s 2>&1 && git checkout -q %s 2>&1 && git merge --ff-only %s 2>&1 && git submodule update --init --recursive 2>&1)',
  873. \ fetch_opt, prog, checkout, merge), { 'dir': spec.dir })
  874. else
  875. let s:jobs[name] = { 'running': 0, 'result': 'Already installed', 'error': 0 }
  876. endif
  877. else
  878. let s:jobs[name] = { 'running': 0, 'result': msg, 'error': 1 }
  879. endif
  880. else
  881. call s:spawn(name,
  882. \ printf('git clone %s %s --recursive %s -b %s %s 2>&1',
  883. \ has_tag ? '' : s:clone_opt,
  884. \ prog,
  885. \ s:shellesc(spec.uri),
  886. \ checkout,
  887. \ s:shellesc(s:trim(spec.dir))), { 'new': 1 })
  888. endif
  889. if !s:jobs[name].running
  890. call s:reap(name)
  891. endif
  892. if len(s:jobs) >= s:update.threads
  893. break
  894. endif
  895. endwhile
  896. endfunction
  897. function! s:update_python()
  898. let py_exe = has('python3') ? 'python3' : 'python'
  899. execute py_exe "<< EOF"
  900. """ Due to use of signals this function is POSIX only. """
  901. import datetime
  902. import functools
  903. import os
  904. try:
  905. import queue
  906. except ImportError:
  907. import Queue as queue
  908. import random
  909. import re
  910. import shutil
  911. import signal
  912. import subprocess
  913. import tempfile
  914. import threading as thr
  915. import time
  916. import traceback
  917. import vim
  918. G_NVIM = vim.eval("has('nvim')") == '1'
  919. G_PULL = vim.eval('s:update.pull') == '1'
  920. G_RETRIES = int(vim.eval('get(g:, "plug_retries", 2)')) + 1
  921. G_TIMEOUT = int(vim.eval('get(g:, "plug_timeout", 60)'))
  922. G_CLONE_OPT = vim.eval('s:clone_opt')
  923. G_PROGRESS = vim.eval('s:progress_opt(1)')
  924. G_LOG_PROB = 1.0 / int(vim.eval('s:update.threads'))
  925. G_STOP = thr.Event()
  926. G_THREADS = {}
  927. class BaseExc(Exception):
  928. def __init__(self, msg):
  929. self._msg = msg
  930. @property
  931. def msg(self):
  932. return self._msg
  933. class CmdTimedOut(BaseExc):
  934. pass
  935. class CmdFailed(BaseExc):
  936. pass
  937. class InvalidURI(BaseExc):
  938. pass
  939. class Action(object):
  940. INSTALL, UPDATE, ERROR, DONE = ['+', '*', 'x', '-']
  941. class GLog(object):
  942. ON = None
  943. LOGDIR = None
  944. @classmethod
  945. def write(cls, msg):
  946. if cls.ON is None:
  947. cls.ON = int(vim.eval('get(g:, "plug_log_on", 0)'))
  948. cls.LOGDIR = os.path.expanduser(vim.eval('get(g:, "plug_logs", "~/plug_logs")'))
  949. if cls.ON:
  950. if not os.path.exists(cls.LOGDIR):
  951. os.makedirs(cls.LOGDIR)
  952. cls._write(msg)
  953. @classmethod
  954. def _write(cls, msg):
  955. name = thr.current_thread().name
  956. fname = cls.LOGDIR + os.path.sep + name
  957. with open(fname, 'ab') as flog:
  958. ltime = datetime.datetime.now().strftime("%H:%M:%S.%f")
  959. msg = '[{0},{1}] {2}{3}'.format(name, ltime, msg, '\n')
  960. flog.write(msg.encode())
  961. class Buffer(object):
  962. def __init__(self, lock, num_plugs, is_pull, is_win):
  963. self.bar = ''
  964. self.event = 'Updating' if is_pull else 'Installing'
  965. self.is_win = is_win
  966. self.lock = lock
  967. self.maxy = int(vim.eval('winheight(".")'))
  968. self.num_plugs = num_plugs
  969. def _where(self, name):
  970. """ Find first line with name in current buffer. Return line num. """
  971. found, lnum = False, 0
  972. matcher = re.compile('^[-+x*] {0}:'.format(name))
  973. for line in vim.current.buffer:
  974. if matcher.search(line) is not None:
  975. found = True
  976. break
  977. lnum += 1
  978. if not found:
  979. lnum = -1
  980. return lnum
  981. def header(self):
  982. curbuf = vim.current.buffer
  983. curbuf[0] = self.event + ' plugins ({0}/{1})'.format(len(self.bar), self.num_plugs)
  984. num_spaces = self.num_plugs - len(self.bar)
  985. curbuf[1] = '[{0}{1}]'.format(self.bar, num_spaces * ' ')
  986. with self.lock:
  987. vim.command('normal! 2G')
  988. if not self.is_win:
  989. vim.command('redraw')
  990. def write(self, action, name, lines):
  991. first, rest = lines[0], lines[1:]
  992. msg = ['{0} {1}{2}{3}'.format(action, name, ': ' if first else '', first)]
  993. padded_rest = [' ' + line for line in rest]
  994. msg.extend(padded_rest)
  995. try:
  996. if action == Action.ERROR:
  997. self.bar += 'x'
  998. vim.command("call add(s:update.errors, '{0}')".format(name))
  999. elif action == Action.DONE:
  1000. self.bar += '='
  1001. curbuf = vim.current.buffer
  1002. lnum = self._where(name)
  1003. if lnum != -1: # Found matching line num
  1004. del curbuf[lnum]
  1005. if lnum > self.maxy and action in set([Action.INSTALL, Action.UPDATE]):
  1006. lnum = 3
  1007. else:
  1008. lnum = 3
  1009. curbuf.append(msg, lnum)
  1010. self.header()
  1011. except vim.error:
  1012. GLog.write('Buffer Update FAILED.')
  1013. class Command(object):
  1014. def __init__(self, cmd, cmd_dir=None, timeout=60, ntries=3, cb=None, clean=None):
  1015. self.cmd = cmd
  1016. self.cmd_dir = cmd_dir
  1017. self.timeout = timeout
  1018. self.ntries = ntries
  1019. self.callback = cb if cb else (lambda msg: None)
  1020. self.clean = clean
  1021. def attempt_cmd(self):
  1022. """ Tries to run the command, returns result if no exceptions. """
  1023. attempt = 0
  1024. finished = False
  1025. limit = self.timeout
  1026. while not finished:
  1027. try:
  1028. attempt += 1
  1029. result = self.timeout_cmd()
  1030. finished = True
  1031. except CmdTimedOut:
  1032. if attempt != self.ntries:
  1033. for count in range(3, 0, -1):
  1034. if G_STOP.is_set():
  1035. raise KeyboardInterrupt
  1036. msg = 'Timeout. Will retry in {0} second{1} ...'.format(
  1037. count, 's' if count != 1 else '')
  1038. self.callback([msg])
  1039. time.sleep(1)
  1040. self.timeout += limit
  1041. self.callback(['Retrying ...'])
  1042. else:
  1043. raise
  1044. return result
  1045. def timeout_cmd(self):
  1046. """ Execute a cmd & poll for callback. Returns list of output.
  1047. Raises CmdFailed -> return code for Popen isn't 0
  1048. Raises CmdTimedOut -> command exceeded timeout without new output
  1049. """
  1050. proc = None
  1051. first_line = True
  1052. try:
  1053. tfile = tempfile.NamedTemporaryFile(mode='w+b', delete=False)
  1054. proc = subprocess.Popen(self.cmd, cwd=self.cmd_dir, stdout=tfile,
  1055. stderr=subprocess.STDOUT, shell=True, preexec_fn=os.setsid)
  1056. while proc.poll() is None:
  1057. # Yield this thread
  1058. time.sleep(0.2)
  1059. if G_STOP.is_set():
  1060. raise KeyboardInterrupt
  1061. if first_line or random.random() < G_LOG_PROB:
  1062. first_line = False
  1063. line = nonblock_read(tfile.name)
  1064. if line:
  1065. self.callback([line])
  1066. time_diff = time.time() - os.path.getmtime(tfile.name)
  1067. if time_diff > self.timeout:
  1068. raise CmdTimedOut(['Timeout!'])
  1069. tfile.seek(0)
  1070. result = [line.decode().rstrip() for line in tfile]
  1071. if proc.returncode != 0:
  1072. msg = ['']
  1073. msg.extend(result)
  1074. raise CmdFailed(msg)
  1075. except:
  1076. if proc and proc.poll() is None:
  1077. os.killpg(proc.pid, signal.SIGTERM)
  1078. if self.clean:
  1079. self.clean()
  1080. raise
  1081. finally:
  1082. os.remove(tfile.name)
  1083. return result
  1084. class Plugin(object):
  1085. def __init__(self, name, args, buf_q, lock):
  1086. self.name = name
  1087. self.args = args
  1088. self.buf_q = buf_q
  1089. self.lock = lock
  1090. tag = args.get('tag', 0)
  1091. self.checkout = esc(tag if tag else args['branch'])
  1092. self.merge = esc(tag if tag else 'origin/' + args['branch'])
  1093. self.tag = tag
  1094. def manage(self):
  1095. try:
  1096. if os.path.exists(self.args['dir']):
  1097. self.update()
  1098. else:
  1099. self.install()
  1100. with self.lock:
  1101. thread_vim_command("let s:update.new['{0}'] = 1".format(self.name))
  1102. except (CmdTimedOut, CmdFailed, InvalidURI) as exc:
  1103. self.write(Action.ERROR, self.name, exc.msg)
  1104. except KeyboardInterrupt:
  1105. G_STOP.set()
  1106. self.write(Action.ERROR, self.name, ['Interrupted!'])
  1107. except:
  1108. # Any exception except those above print stack trace
  1109. msg = 'Trace:\n{0}'.format(traceback.format_exc().rstrip())
  1110. self.write(Action.ERROR, self.name, msg.split('\n'))
  1111. raise
  1112. def install(self):
  1113. target = self.args['dir']
  1114. def clean(target):
  1115. def _clean():
  1116. try:
  1117. shutil.rmtree(target)
  1118. except OSError:
  1119. pass
  1120. return _clean
  1121. self.write(Action.INSTALL, self.name, ['Installing ...'])
  1122. callback = functools.partial(self.write, Action.INSTALL, self.name)
  1123. cmd = 'git clone {0} {1} --recursive {2} -b {3} {4} 2>&1'.format(
  1124. '' if self.tag else G_CLONE_OPT, G_PROGRESS, self.args['uri'], self.checkout, esc(target))
  1125. com = Command(cmd, None, G_TIMEOUT, G_RETRIES, callback, clean(target))
  1126. result = com.attempt_cmd()
  1127. self.write(Action.DONE, self.name, result[-1:])
  1128. def update(self):
  1129. match = re.compile(r'git::?@')
  1130. actual_uri = re.sub(match, '', self.repo_uri())
  1131. expect_uri = re.sub(match, '', self.args['uri'])
  1132. if actual_uri != expect_uri:
  1133. msg = ['',
  1134. 'Invalid URI: {0}'.format(actual_uri),
  1135. 'Expected {0}'.format(expect_uri),
  1136. 'PlugClean required.']
  1137. raise InvalidURI(msg)
  1138. if G_PULL:
  1139. self.write(Action.UPDATE, self.name, ['Updating ...'])
  1140. callback = functools.partial(self.write, Action.UPDATE, self.name)
  1141. fetch_opt = '--depth 99999999' if self.tag and os.path.isfile(os.path.join(self.args['dir'], '.git/shallow')) else ''
  1142. cmds = ['git fetch {0} {1}'.format(fetch_opt, G_PROGRESS),
  1143. 'git checkout -q {0}'.format(self.checkout),
  1144. 'git merge --ff-only {0}'.format(self.merge),
  1145. 'git submodule update --init --recursive']
  1146. cmd = ' 2>&1 && '.join(cmds)
  1147. GLog.write(cmd)
  1148. com = Command(cmd, self.args['dir'], G_TIMEOUT, G_RETRIES, callback)
  1149. result = com.attempt_cmd()
  1150. GLog.write(result)
  1151. self.write(Action.DONE, self.name, result[-1:])
  1152. else:
  1153. self.write(Action.DONE, self.name, ['Already installed'])
  1154. def repo_uri(self):
  1155. cmd = 'git rev-parse --abbrev-ref HEAD 2>&1 && git config remote.origin.url'
  1156. command = Command(cmd, self.args['dir'], G_TIMEOUT, G_RETRIES)
  1157. result = command.attempt_cmd()
  1158. return result[-1]
  1159. def write(self, action, name, msg):
  1160. GLog.write('{0} {1}: {2}'.format(action, name, '\n'.join(msg)))
  1161. self.buf_q.put((action, name, msg))
  1162. class PlugThread(thr.Thread):
  1163. def __init__(self, tname, args):
  1164. super(PlugThread, self).__init__()
  1165. self.tname = tname
  1166. self.args = args
  1167. def run(self):
  1168. thr.current_thread().name = self.tname
  1169. buf_q, work_q, lock = self.args
  1170. try:
  1171. while not G_STOP.is_set():
  1172. name, args = work_q.get_nowait()
  1173. GLog.write('{0}: Dir {1}'.format(name, args['dir']))
  1174. plug = Plugin(name, args, buf_q, lock)
  1175. plug.manage()
  1176. work_q.task_done()
  1177. except queue.Empty:
  1178. GLog.write('Queue now empty.')
  1179. finally:
  1180. global G_THREADS
  1181. with lock:
  1182. del G_THREADS[thr.current_thread().name]
  1183. class RefreshThread(thr.Thread):
  1184. def __init__(self, lock):
  1185. super(RefreshThread, self).__init__()
  1186. self.lock = lock
  1187. self.running = True
  1188. def run(self):
  1189. while self.running:
  1190. with self.lock:
  1191. thread_vim_command('noautocmd normal! a')
  1192. time.sleep(0.2)
  1193. def stop(self):
  1194. self.running = False
  1195. if G_NVIM:
  1196. def thread_vim_command(cmd):
  1197. vim.session.threadsafe_call(lambda: vim.command(cmd))
  1198. else:
  1199. def thread_vim_command(cmd):
  1200. vim.command(cmd)
  1201. def esc(name):
  1202. return '"' + name.replace('"', '\"') + '"'
  1203. def nonblock_read(fname):
  1204. """ Read a file with nonblock flag. Return the last line. """
  1205. fread = os.open(fname, os.O_RDONLY | os.O_NONBLOCK)
  1206. buf = os.read(fread, 100000).decode()
  1207. os.close(fread)
  1208. line = buf.rstrip('\r\n')
  1209. left = max(line.rfind('\r'), line.rfind('\n'))
  1210. if left != -1:
  1211. left += 1
  1212. line = line[left:]
  1213. return line
  1214. def main():
  1215. thr.current_thread().name = 'main'
  1216. GLog.write('')
  1217. if GLog.ON and os.path.exists(GLog.LOGDIR):
  1218. shutil.rmtree(GLog.LOGDIR)
  1219. nthreads = int(vim.eval('s:update.threads'))
  1220. plugs = vim.eval('s:update.todo')
  1221. mac_gui = vim.eval('s:mac_gui') == '1'
  1222. is_win = vim.eval('s:is_win') == '1'
  1223. GLog.write('Plugs: {0}'.format(plugs))
  1224. GLog.write('PULL: {0}, WIN: {1}, MAC: {2}'.format(G_PULL, is_win, mac_gui))
  1225. GLog.write('Num Threads: {0}'.format(nthreads))
  1226. lock = thr.Lock()
  1227. buf = Buffer(lock, len(plugs), G_PULL, is_win)
  1228. buf_q, work_q = queue.Queue(), queue.Queue()
  1229. for work in plugs.items():
  1230. work_q.put(work)
  1231. GLog.write('Starting Threads')
  1232. global G_THREADS
  1233. for num in range(nthreads):
  1234. tname = 'PlugT-{0:02}'.format(num)
  1235. thread = PlugThread(tname, (buf_q, work_q, lock))
  1236. thread.start()
  1237. G_THREADS[tname] = thread
  1238. if mac_gui:
  1239. rthread = RefreshThread(lock)
  1240. rthread.start()
  1241. GLog.write('Buffer Writing Loop')
  1242. while not buf_q.empty() or len(G_THREADS) != 0:
  1243. try:
  1244. action, name, msg = buf_q.get(True, 0.25)
  1245. buf.write(action, name, msg)
  1246. buf_q.task_done()
  1247. except queue.Empty:
  1248. pass
  1249. except KeyboardInterrupt:
  1250. G_STOP.set()
  1251. if mac_gui:
  1252. rthread.stop()
  1253. rthread.join()
  1254. GLog.write('Cleanly Exited Main')
  1255. main()
  1256. EOF
  1257. endfunction
  1258. function! s:update_ruby()
  1259. ruby << EOF
  1260. module PlugStream
  1261. SEP = ["\r", "\n", nil]
  1262. def get_line
  1263. buffer = ''
  1264. loop do
  1265. char = readchar rescue return
  1266. if SEP.include? char.chr
  1267. buffer << $/
  1268. break
  1269. else
  1270. buffer << char
  1271. end
  1272. end
  1273. buffer
  1274. end
  1275. end unless defined?(PlugStream)
  1276. def esc arg
  1277. %["#{arg.gsub('"', '\"')}"]
  1278. end
  1279. def killall pid
  1280. pids = [pid]
  1281. unless `which pgrep 2> /dev/null`.empty?
  1282. children = pids
  1283. until children.empty?
  1284. children = children.map { |pid|
  1285. `pgrep -P #{pid}`.lines.map { |l| l.chomp }
  1286. }.flatten
  1287. pids += children
  1288. end
  1289. end
  1290. pids.each { |pid| Process.kill 'TERM', pid.to_i rescue nil }
  1291. end
  1292. require 'thread'
  1293. require 'fileutils'
  1294. require 'timeout'
  1295. running = true
  1296. iswin = VIM::evaluate('s:is_win').to_i == 1
  1297. pull = VIM::evaluate('s:update.pull').to_i == 1
  1298. base = VIM::evaluate('g:plug_home')
  1299. all = VIM::evaluate('s:update.todo')
  1300. limit = VIM::evaluate('get(g:, "plug_timeout", 60)')
  1301. tries = VIM::evaluate('get(g:, "plug_retries", 2)') + 1
  1302. nthr = VIM::evaluate('s:update.threads').to_i
  1303. maxy = VIM::evaluate('winheight(".")').to_i
  1304. cd = iswin ? 'cd /d' : 'cd'
  1305. tot = VIM::evaluate('len(s:update.todo)') || 0
  1306. bar = ''
  1307. skip = 'Already installed'
  1308. mtx = Mutex.new
  1309. take1 = proc { mtx.synchronize { running && all.shift } }
  1310. logh = proc {
  1311. cnt = bar.length
  1312. $curbuf[1] = "#{pull ? 'Updating' : 'Installing'} plugins (#{cnt}/#{tot})"
  1313. $curbuf[2] = '[' + bar.ljust(tot) + ']'
  1314. VIM::command('normal! 2G')
  1315. VIM::command('redraw') unless iswin
  1316. }
  1317. where = proc { |name| (1..($curbuf.length)).find { |l| $curbuf[l] =~ /^[-+x*] #{name}:/ } }
  1318. log = proc { |name, result, type|
  1319. mtx.synchronize do
  1320. ing = ![true, false].include?(type)
  1321. bar += type ? '=' : 'x' unless ing
  1322. b = case type
  1323. when :install then '+' when :update then '*'
  1324. when true, nil then '-' else
  1325. VIM::command("call add(s:update.errors, '#{name}')")
  1326. 'x'
  1327. end
  1328. result =
  1329. if type || type.nil?
  1330. ["#{b} #{name}: #{result.lines.to_a.last}"]
  1331. elsif result =~ /^Interrupted|^Timeout/
  1332. ["#{b} #{name}: #{result}"]
  1333. else
  1334. ["#{b} #{name}"] + result.lines.map { |l| " " << l }
  1335. end
  1336. if lnum = where.call(name)
  1337. $curbuf.delete lnum
  1338. lnum = 4 if ing && lnum > maxy
  1339. end
  1340. result.each_with_index do |line, offset|
  1341. $curbuf.append((lnum || 4) - 1 + offset, line.gsub(/\e\[./, '').chomp)
  1342. end
  1343. logh.call
  1344. end
  1345. }
  1346. bt = proc { |cmd, name, type, cleanup|
  1347. tried = timeout = 0
  1348. begin
  1349. tried += 1
  1350. timeout += limit
  1351. fd = nil
  1352. data = ''
  1353. if iswin
  1354. Timeout::timeout(timeout) do
  1355. tmp = VIM::evaluate('tempname()')
  1356. system("(#{cmd}) > #{tmp}")
  1357. data = File.read(tmp).chomp
  1358. File.unlink tmp rescue nil
  1359. end
  1360. else
  1361. fd = IO.popen(cmd).extend(PlugStream)
  1362. first_line = true
  1363. log_prob = 1.0 / nthr
  1364. while line = Timeout::timeout(timeout) { fd.get_line }
  1365. data << line
  1366. log.call name, line.chomp, type if name && (first_line || rand < log_prob)
  1367. first_line = false
  1368. end
  1369. fd.close
  1370. end
  1371. [$? == 0, data.chomp]
  1372. rescue Timeout::Error, Interrupt => e
  1373. if fd && !fd.closed?
  1374. killall fd.pid
  1375. fd.close
  1376. end
  1377. cleanup.call if cleanup
  1378. if e.is_a?(Timeout::Error) && tried < tries
  1379. 3.downto(1) do |countdown|
  1380. s = countdown > 1 ? 's' : ''
  1381. log.call name, "Timeout. Will retry in #{countdown} second#{s} ...", type
  1382. sleep 1
  1383. end
  1384. log.call name, 'Retrying ...', type
  1385. retry
  1386. end
  1387. [false, e.is_a?(Interrupt) ? "Interrupted!" : "Timeout!"]
  1388. end
  1389. }
  1390. main = Thread.current
  1391. threads = []
  1392. watcher = Thread.new {
  1393. while VIM::evaluate('getchar(1)')
  1394. sleep 0.1
  1395. end
  1396. mtx.synchronize do
  1397. running = false
  1398. threads.each { |t| t.raise Interrupt }
  1399. end
  1400. threads.each { |t| t.join rescue nil }
  1401. main.kill
  1402. }
  1403. refresh = Thread.new {
  1404. while true
  1405. mtx.synchronize do
  1406. break unless running
  1407. VIM::command('noautocmd normal! a')
  1408. end
  1409. sleep 0.2
  1410. end
  1411. } if VIM::evaluate('s:mac_gui') == 1
  1412. clone_opt = VIM::evaluate('s:clone_opt')
  1413. progress = VIM::evaluate('s:progress_opt(1)')
  1414. nthr.times do
  1415. mtx.synchronize do
  1416. threads << Thread.new {
  1417. while pair = take1.call
  1418. name = pair.first
  1419. dir, uri, branch, tag = pair.last.values_at *%w[dir uri branch tag]
  1420. checkout = esc(tag ? tag : branch)
  1421. merge = esc(tag ? tag : "origin/#{branch}")
  1422. subm = "git submodule update --init --recursive 2>&1"
  1423. exists = File.directory? dir
  1424. ok, result =
  1425. if exists
  1426. dir = iswin ? dir : esc(dir)
  1427. ret, data = bt.call "#{cd} #{dir} && git rev-parse --abbrev-ref HEAD 2>&1 && git config remote.origin.url", nil, nil, nil
  1428. current_uri = data.lines.to_a.last
  1429. if !ret
  1430. if data =~ /^Interrupted|^Timeout/
  1431. [false, data]
  1432. else
  1433. [false, [data.chomp, "PlugClean required."].join($/)]
  1434. end
  1435. elsif current_uri.sub(/git::?@/, '') != uri.sub(/git::?@/, '')
  1436. [false, ["Invalid URI: #{current_uri}",
  1437. "Expected: #{uri}",
  1438. "PlugClean required."].join($/)]
  1439. else
  1440. if pull
  1441. log.call name, 'Updating ...', :update
  1442. fetch_opt = (tag && File.exist?(File.join(dir, '.git/shallow'))) ? '--depth 99999999' : ''
  1443. bt.call "#{cd} #{dir} && git fetch #{fetch_opt} #{progress} 2>&1 && git checkout -q #{checkout} 2>&1 && git merge --ff-only #{merge} 2>&1 && #{subm}", name, :update, nil
  1444. else
  1445. [true, skip]
  1446. end
  1447. end
  1448. else
  1449. d = esc dir.sub(%r{[\\/]+$}, '')
  1450. log.call name, 'Installing ...', :install
  1451. bt.call "git clone #{clone_opt unless tag} #{progress} --recursive #{uri} -b #{checkout} #{d} 2>&1", name, :install, proc {
  1452. FileUtils.rm_rf dir
  1453. }
  1454. end
  1455. mtx.synchronize { VIM::command("let s:update.new['#{name}'] = 1") } if !exists && ok
  1456. log.call name, result, ok
  1457. end
  1458. } if running
  1459. end
  1460. end
  1461. threads.each { |t| t.join rescue nil }
  1462. logh.call
  1463. refresh.kill if refresh
  1464. watcher.kill
  1465. EOF
  1466. endfunction
  1467. function! s:shellesc(arg)
  1468. return '"'.escape(a:arg, '"').'"'
  1469. endfunction
  1470. function! s:glob_dir(path)
  1471. return map(filter(s:lines(globpath(a:path, '**')), 'isdirectory(v:val)'), 's:dirpath(v:val)')
  1472. endfunction
  1473. function! s:progress_bar(line, bar, total)
  1474. call setline(a:line, '[' . s:lpad(a:bar, a:total) . ']')
  1475. endfunction
  1476. function! s:compare_git_uri(a, b)
  1477. let a = substitute(a:a, 'git:\{1,2}@', '', '')
  1478. let b = substitute(a:b, 'git:\{1,2}@', '', '')
  1479. return a ==# b
  1480. endfunction
  1481. function! s:format_message(bullet, name, message)
  1482. if a:bullet != 'x'
  1483. return [printf('%s %s: %s', a:bullet, a:name, s:lastline(a:message))]
  1484. else
  1485. let lines = map(s:lines(a:message), '" ".v:val')
  1486. return extend([printf('x %s:', a:name)], lines)
  1487. endif
  1488. endfunction
  1489. function! s:with_cd(cmd, dir)
  1490. return printf('cd%s %s && %s', s:is_win ? ' /d' : '', s:shellesc(a:dir), a:cmd)
  1491. endfunction
  1492. function! s:system(cmd, ...)
  1493. try
  1494. let sh = &shell
  1495. if !s:is_win
  1496. set shell=sh
  1497. endif
  1498. let cmd = a:0 > 0 ? s:with_cd(a:cmd, a:1) : a:cmd
  1499. return system(s:is_win ? '('.cmd.')' : cmd)
  1500. finally
  1501. let &shell = sh
  1502. endtry
  1503. endfunction
  1504. function! s:system_chomp(...)
  1505. let ret = call('s:system', a:000)
  1506. return v:shell_error ? '' : substitute(ret, '\n$', '', '')
  1507. endfunction
  1508. function! s:git_valid(spec, check_branch)
  1509. let ret = 1
  1510. let msg = 'OK'
  1511. if isdirectory(a:spec.dir)
  1512. let result = s:lines(s:system('git rev-parse --abbrev-ref HEAD 2>&1 && git config remote.origin.url', a:spec.dir))
  1513. let remote = result[-1]
  1514. if v:shell_error
  1515. let msg = join([remote, 'PlugClean required.'], "\n")
  1516. let ret = 0
  1517. elseif !s:compare_git_uri(remote, a:spec.uri)
  1518. let msg = join(['Invalid URI: '.remote,
  1519. \ 'Expected: '.a:spec.uri,
  1520. \ 'PlugClean required.'], "\n")
  1521. let ret = 0
  1522. elseif a:check_branch
  1523. let branch = result[0]
  1524. " Check tag
  1525. if has_key(a:spec, 'tag')
  1526. let tag = s:system_chomp('git describe --exact-match --tags HEAD 2>&1', a:spec.dir)
  1527. if a:spec.tag !=# tag
  1528. let msg = printf('Invalid tag: %s (expected: %s). Try PlugUpdate.',
  1529. \ (empty(tag) ? 'N/A' : tag), a:spec.tag)
  1530. let ret = 0
  1531. endif
  1532. " Check branch
  1533. elseif a:spec.branch !=# branch
  1534. let msg = printf('Invalid branch: %s (expected: %s). Try PlugUpdate.',
  1535. \ branch, a:spec.branch)
  1536. let ret = 0
  1537. endif
  1538. endif
  1539. else
  1540. let msg = 'Not found'
  1541. let ret = 0
  1542. endif
  1543. return [ret, msg]
  1544. endfunction
  1545. function! s:rm_rf(dir)
  1546. if isdirectory(a:dir)
  1547. call s:system((s:is_win ? 'rmdir /S /Q ' : 'rm -rf ') . s:shellesc(a:dir))
  1548. endif
  1549. endfunction
  1550. function! s:clean(force)
  1551. call s:prepare()
  1552. call append(0, 'Searching for unused plugins in '.g:plug_home)
  1553. call append(1, '')
  1554. " List of valid directories
  1555. let dirs = []
  1556. let [cnt, total] = [0, len(g:plugs)]
  1557. for [name, spec] in items(g:plugs)
  1558. if !s:is_managed(name) || s:git_valid(spec, 0)[0]
  1559. call add(dirs, spec.dir)
  1560. endif
  1561. let cnt += 1
  1562. call s:progress_bar(2, repeat('=', cnt), total)
  1563. normal! 2G
  1564. redraw
  1565. endfor
  1566. let allowed = {}
  1567. for dir in dirs
  1568. let allowed[s:dirpath(fnamemodify(dir, ':h:h'))] = 1
  1569. let allowed[dir] = 1
  1570. for child in s:glob_dir(dir)
  1571. let allowed[child] = 1
  1572. endfor
  1573. endfor
  1574. let todo = []
  1575. let found = sort(s:glob_dir(g:plug_home))
  1576. while !empty(found)
  1577. let f = remove(found, 0)
  1578. if !has_key(allowed, f) && isdirectory(f)
  1579. call add(todo, f)
  1580. call append(line('$'), '- ' . f)
  1581. let found = filter(found, 'stridx(v:val, f) != 0')
  1582. end
  1583. endwhile
  1584. normal! G
  1585. redraw
  1586. if empty(todo)
  1587. call append(line('$'), 'Already clean.')
  1588. else
  1589. call inputsave()
  1590. let yes = a:force || (input('Proceed? (y/N) ') =~? '^y')
  1591. call inputrestore()
  1592. if yes
  1593. for dir in todo
  1594. call s:rm_rf(dir)
  1595. endfor
  1596. call append(line('$'), 'Removed.')
  1597. else
  1598. call append(line('$'), 'Cancelled.')
  1599. endif
  1600. endif
  1601. normal! G
  1602. endfunction
  1603. function! s:upgrade()
  1604. echo 'Downloading the latest version of vim-plug'
  1605. redraw
  1606. let tmp = tempname()
  1607. let new = tmp . '/plug.vim'
  1608. try
  1609. let out = s:system(printf('git clone --depth 1 %s %s', s:plug_src, tmp))
  1610. if v:shell_error
  1611. return s:err('Error upgrading vim-plug: '. out)
  1612. endif
  1613. if readfile(s:me) ==# readfile(new)
  1614. echo 'vim-plug is already up-to-date'
  1615. return 0
  1616. else
  1617. call rename(s:me, s:me . '.old')
  1618. call rename(new, s:me)
  1619. unlet g:loaded_plug
  1620. echo 'vim-plug has been upgraded'
  1621. return 1
  1622. endif
  1623. finally
  1624. silent! call s:rm_rf(tmp)
  1625. endtry
  1626. endfunction
  1627. function! s:upgrade_specs()
  1628. for spec in values(g:plugs)
  1629. let spec.frozen = get(spec, 'frozen', 0)
  1630. endfor
  1631. endfunction
  1632. function! s:status()
  1633. call s:prepare()
  1634. call append(0, 'Checking plugins')
  1635. call append(1, '')
  1636. let ecnt = 0
  1637. let unloaded = 0
  1638. let [cnt, total] = [0, len(g:plugs)]
  1639. for [name, spec] in items(g:plugs)
  1640. if has_key(spec, 'uri')
  1641. if isdirectory(spec.dir)
  1642. let [valid, msg] = s:git_valid(spec, 1)
  1643. else
  1644. let [valid, msg] = [0, 'Not found. Try PlugInstall.']
  1645. endif
  1646. else
  1647. if isdirectory(spec.dir)
  1648. let [valid, msg] = [1, 'OK']
  1649. else
  1650. let [valid, msg] = [0, 'Not found.']
  1651. endif
  1652. endif
  1653. let cnt += 1
  1654. let ecnt += !valid
  1655. " `s:loaded` entry can be missing if PlugUpgraded
  1656. if valid && get(s:loaded, name, -1) == 0
  1657. let unloaded = 1
  1658. let msg .= ' (not loaded)'
  1659. endif
  1660. call s:progress_bar(2, repeat('=', cnt), total)
  1661. call append(3, s:format_message(valid ? '-' : 'x', name, msg))
  1662. normal! 2G
  1663. redraw
  1664. endfor
  1665. call setline(1, 'Finished. '.ecnt.' error(s).')
  1666. normal! gg
  1667. setlocal nomodifiable
  1668. if unloaded
  1669. echo "Press 'L' on each line to load plugin, or 'U' to update"
  1670. nnoremap <silent> <buffer> L :call <SID>status_load(line('.'))<cr>
  1671. xnoremap <silent> <buffer> L :call <SID>status_load(line('.'))<cr>
  1672. end
  1673. endfunction
  1674. function! s:extract_name(str, prefix, suffix)
  1675. return matchstr(a:str, '^'.a:prefix.' \zs[^:]\+\ze:.*'.a:suffix.'$')
  1676. endfunction
  1677. function! s:status_load(lnum)
  1678. let line = getline(a:lnum)
  1679. let name = s:extract_name(line, '-', '(not loaded)')
  1680. if !empty(name)
  1681. call plug#load(name)
  1682. setlocal modifiable
  1683. call setline(a:lnum, substitute(line, ' (not loaded)$', '', ''))
  1684. setlocal nomodifiable
  1685. endif
  1686. endfunction
  1687. function! s:status_update() range
  1688. let lines = getline(a:firstline, a:lastline)
  1689. let names = filter(map(lines, 's:extract_name(v:val, "[x-]", "")'), '!empty(v:val)')
  1690. if !empty(names)
  1691. echo
  1692. execute 'PlugUpdate' join(names)
  1693. endif
  1694. endfunction
  1695. function! s:is_preview_window_open()
  1696. silent! wincmd P
  1697. if &previewwindow
  1698. wincmd p
  1699. return 1
  1700. endif
  1701. return 0
  1702. endfunction
  1703. function! s:find_name(lnum)
  1704. for lnum in reverse(range(1, a:lnum))
  1705. let line = getline(lnum)
  1706. if empty(line)
  1707. return ''
  1708. endif
  1709. let name = s:extract_name(line, '-', '')
  1710. if !empty(name)
  1711. return name
  1712. endif
  1713. endfor
  1714. return ''
  1715. endfunction
  1716. function! s:preview_commit()
  1717. if b:plug_preview < 0
  1718. let b:plug_preview = !s:is_preview_window_open()
  1719. endif
  1720. let sha = matchstr(getline('.'), '\(^ \)\@<=[0-9a-z]\{7}')
  1721. if empty(sha)
  1722. return
  1723. endif
  1724. let name = s:find_name(line('.'))
  1725. if empty(name) || !has_key(g:plugs, name) || !isdirectory(g:plugs[name].dir)
  1726. return
  1727. endif
  1728. execute 'pedit' sha
  1729. wincmd P
  1730. setlocal filetype=git buftype=nofile nobuflisted
  1731. execute 'silent read !cd' s:shellesc(g:plugs[name].dir) '&& git show' sha
  1732. normal! gg"_dd
  1733. wincmd p
  1734. endfunction
  1735. function! s:section(flags)
  1736. call search('\(^[x-] \)\@<=[^:]\+:', a:flags)
  1737. endfunction
  1738. function! s:diff()
  1739. call s:prepare()
  1740. call append(0, 'Collecting updated changes ...')
  1741. normal! gg
  1742. redraw
  1743. let cnt = 0
  1744. for [k, v] in items(g:plugs)
  1745. if !isdirectory(v.dir) || !s:is_managed(k)
  1746. continue
  1747. endif
  1748. let diff = s:system_chomp('git log --pretty=format:"%h %s (%cr)" "HEAD...HEAD@{1}"', v.dir)
  1749. if !empty(diff)
  1750. call append(1, '')
  1751. call append(2, '- '.k.':')
  1752. call append(3, map(s:lines(diff), '" ". v:val'))
  1753. let cnt += 1
  1754. normal! gg
  1755. redraw
  1756. endif
  1757. endfor
  1758. call setline(1, cnt == 0 ? 'No updates.' : 'Last update:')
  1759. nnoremap <silent> <buffer> <cr> :silent! call <SID>preview_commit()<cr>
  1760. nnoremap <silent> <buffer> o :silent! call <SID>preview_commit()<cr>
  1761. nnoremap <silent> <buffer> X :call <SID>revert()<cr>
  1762. normal! gg
  1763. setlocal nomodifiable
  1764. if cnt > 0
  1765. echo "Press 'X' on each block to revert the update"
  1766. endif
  1767. endfunction
  1768. function! s:revert()
  1769. let name = s:find_name(line('.'))
  1770. if empty(name) || !has_key(g:plugs, name) ||
  1771. \ input(printf('Revert the update of %s? (y/N) ', name)) !~? '^y'
  1772. return
  1773. endif
  1774. call s:system('git reset --hard HEAD@{1} && git checkout '.s:esc(g:plugs[name].branch), g:plugs[name].dir)
  1775. setlocal modifiable
  1776. normal! "_dap
  1777. setlocal nomodifiable
  1778. echo 'Reverted.'
  1779. endfunction
  1780. function! s:snapshot(...) abort
  1781. let home = get(s:, 'plug_home_org', g:plug_home)
  1782. let [type, var, header] = s:is_win ?
  1783. \ ['dosbatch', '%PLUG_HOME%',
  1784. \ ['@echo off', ':: Generated by vim-plug', ':: '.strftime("%c"), '',
  1785. \ ':: Make sure to PlugUpdate first', '', 'set PLUG_HOME='.home]] :
  1786. \ ['sh', '$PLUG_HOME',
  1787. \ ['#!/bin/sh', '# Generated by vim-plug', '# '.strftime("%c"), '',
  1788. \ 'vim +PlugUpdate +qa', '', 'PLUG_HOME='.s:esc(home)]]
  1789. call s:prepare()
  1790. execute 'setf' type
  1791. call append(0, header)
  1792. call append('$', '')
  1793. 1
  1794. redraw
  1795. let dirs = sort(map(values(filter(copy(g:plugs),
  1796. \'has_key(v:val, "uri") && isdirectory(v:val.dir)')), 'v:val.dir'))
  1797. let anchor = line('$') - 1
  1798. for dir in reverse(dirs)
  1799. let sha = s:system_chomp('git rev-parse --short HEAD', dir)
  1800. if !empty(sha)
  1801. call append(anchor, printf('cd %s && git reset --hard %s',
  1802. \ substitute(dir, '^\V'.escape(g:plug_home, '\'), var, ''), sha))
  1803. redraw
  1804. endif
  1805. endfor
  1806. if a:0 > 0
  1807. let fn = expand(a:1)
  1808. let fne = s:esc(fn)
  1809. call writefile(getline(1, '$'), fn)
  1810. if !s:is_win | call s:system('chmod +x ' . fne) | endif
  1811. echo 'Saved to '.a:1
  1812. silent execute 'e' fne
  1813. endif
  1814. endfunction
  1815. function! s:split_rtp()
  1816. return split(&rtp, '\\\@<!,')
  1817. endfunction
  1818. let s:first_rtp = s:escrtp(get(s:split_rtp(), 0, ''))
  1819. let s:last_rtp = s:escrtp(get(s:split_rtp(), -1, ''))
  1820. if exists('g:plugs')
  1821. let g:plugs_order = get(g:, 'plugs_order', keys(g:plugs))
  1822. call s:upgrade_specs()
  1823. call s:define_commands()
  1824. endif
  1825. let &cpo = s:cpo_save
  1826. unlet s:cpo_save