plug.vim 58 KB

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