plug.vim 55 KB

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