plug.vim 66 KB

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