plug.vim 67 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366
  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. let prefix = v:count ? v:count : ''
  457. let prefix .= '"'.v:register.a:prefix
  458. if mode(1) == 'no'
  459. if v:operator == 'c'
  460. let prefix = "\<esc>" . prefix
  461. endif
  462. let prefix .= v:operator
  463. endif
  464. call feedkeys(prefix, 'n')
  465. call feedkeys(substitute(a:map, '^<Plug>', "\<Plug>", '') . extra)
  466. endfunction
  467. function! plug#(repo, ...)
  468. if a:0 > 1
  469. return s:err('Invalid number of arguments (1..2)')
  470. endif
  471. try
  472. let repo = s:trim(a:repo)
  473. let opts = a:0 == 1 ? s:parse_options(a:1) : s:base_spec
  474. let name = get(opts, 'as', fnamemodify(repo, ':t:s?\.git$??'))
  475. let spec = extend(s:infer_properties(name, repo), opts)
  476. if !has_key(g:plugs, name)
  477. call add(g:plugs_order, name)
  478. endif
  479. let g:plugs[name] = spec
  480. let s:loaded[name] = get(s:loaded, name, 0)
  481. catch
  482. return s:err(v:exception)
  483. endtry
  484. endfunction
  485. function! s:parse_options(arg)
  486. let opts = copy(s:base_spec)
  487. let type = type(a:arg)
  488. if type == s:TYPE.string
  489. let opts.tag = a:arg
  490. elseif type == s:TYPE.dict
  491. call extend(opts, a:arg)
  492. if has_key(opts, 'dir')
  493. let opts.dir = s:dirpath(expand(opts.dir))
  494. endif
  495. else
  496. throw 'Invalid argument type (expected: string or dictionary)'
  497. endif
  498. return opts
  499. endfunction
  500. function! s:infer_properties(name, repo)
  501. let repo = a:repo
  502. if s:is_local_plug(repo)
  503. return { 'dir': s:dirpath(expand(repo)) }
  504. else
  505. if repo =~ ':'
  506. let uri = repo
  507. else
  508. if repo !~ '/'
  509. let repo = 'vim-scripts/'. repo
  510. endif
  511. let fmt = get(g:, 'plug_url_format', 'https://git::@github.com/%s.git')
  512. let uri = printf(fmt, repo)
  513. endif
  514. let dir = s:dirpath( fnamemodify(join([g:plug_home, a:name], '/'), ':p') )
  515. return { 'dir': dir, 'uri': uri }
  516. endif
  517. endfunction
  518. function! s:install(force, names)
  519. call s:update_impl(0, a:force, a:names)
  520. endfunction
  521. function! s:update(force, names)
  522. call s:update_impl(1, a:force, a:names)
  523. endfunction
  524. function! plug#helptags()
  525. if !exists('g:plugs')
  526. return s:err('plug#begin was not called')
  527. endif
  528. for spec in values(g:plugs)
  529. let docd = join([spec.dir, 'doc'], '/')
  530. if isdirectory(docd)
  531. silent! execute 'helptags' s:esc(docd)
  532. endif
  533. endfor
  534. return 1
  535. endfunction
  536. function! s:syntax()
  537. syntax clear
  538. syntax region plug1 start=/\%1l/ end=/\%2l/ contains=plugNumber
  539. syntax region plug2 start=/\%2l/ end=/\%3l/ contains=plugBracket,plugX
  540. syn match plugNumber /[0-9]\+[0-9.]*/ contained
  541. syn match plugBracket /[[\]]/ contained
  542. syn match plugX /x/ contained
  543. syn match plugDash /^-/
  544. syn match plugPlus /^+/
  545. syn match plugStar /^*/
  546. syn match plugMessage /\(^- \)\@<=.*/
  547. syn match plugName /\(^- \)\@<=[^ ]*:/
  548. syn match plugSha /\%(: \)\@<=[0-9a-f]\{4,}$/
  549. syn match plugTag /(tag: [^)]\+)/
  550. syn match plugInstall /\(^+ \)\@<=[^:]*/
  551. syn match plugUpdate /\(^* \)\@<=[^:]*/
  552. syn match plugCommit /^ \X*[0-9a-f]\{7} .*/ contains=plugRelDate,plugEdge,plugTag
  553. syn match plugEdge /^ \X\+$/
  554. syn match plugEdge /^ \X*/ contained nextgroup=plugSha
  555. syn match plugSha /[0-9a-f]\{7}/ contained
  556. syn match plugRelDate /([^)]*)$/ contained
  557. syn match plugNotLoaded /(not loaded)$/
  558. syn match plugError /^x.*/
  559. syn region plugDeleted start=/^\~ .*/ end=/^\ze\S/
  560. syn match plugH2 /^.*:\n-\+$/
  561. syn keyword Function PlugInstall PlugStatus PlugUpdate PlugClean
  562. hi def link plug1 Title
  563. hi def link plug2 Repeat
  564. hi def link plugH2 Type
  565. hi def link plugX Exception
  566. hi def link plugBracket Structure
  567. hi def link plugNumber Number
  568. hi def link plugDash Special
  569. hi def link plugPlus Constant
  570. hi def link plugStar Boolean
  571. hi def link plugMessage Function
  572. hi def link plugName Label
  573. hi def link plugInstall Function
  574. hi def link plugUpdate Type
  575. hi def link plugError Error
  576. hi def link plugDeleted Ignore
  577. hi def link plugRelDate Comment
  578. hi def link plugEdge PreProc
  579. hi def link plugSha Identifier
  580. hi def link plugTag Constant
  581. hi def link plugNotLoaded Comment
  582. endfunction
  583. function! s:lpad(str, len)
  584. return a:str . repeat(' ', a:len - len(a:str))
  585. endfunction
  586. function! s:lines(msg)
  587. return split(a:msg, "[\r\n]")
  588. endfunction
  589. function! s:lastline(msg)
  590. return get(s:lines(a:msg), -1, '')
  591. endfunction
  592. function! s:new_window()
  593. execute get(g:, 'plug_window', 'vertical topleft new')
  594. endfunction
  595. function! s:plug_window_exists()
  596. let buflist = tabpagebuflist(s:plug_tab)
  597. return !empty(buflist) && index(buflist, s:plug_buf) >= 0
  598. endfunction
  599. function! s:switch_in()
  600. if !s:plug_window_exists()
  601. return 0
  602. endif
  603. if winbufnr(0) != s:plug_buf
  604. let s:pos = [tabpagenr(), winnr(), winsaveview()]
  605. execute 'normal!' s:plug_tab.'gt'
  606. let winnr = bufwinnr(s:plug_buf)
  607. execute winnr.'wincmd w'
  608. call add(s:pos, winsaveview())
  609. else
  610. let s:pos = [winsaveview()]
  611. endif
  612. setlocal modifiable
  613. return 1
  614. endfunction
  615. function! s:switch_out(...)
  616. call winrestview(s:pos[-1])
  617. setlocal nomodifiable
  618. if a:0 > 0
  619. execute a:1
  620. endif
  621. if len(s:pos) > 1
  622. execute 'normal!' s:pos[0].'gt'
  623. execute s:pos[1] 'wincmd w'
  624. call winrestview(s:pos[2])
  625. endif
  626. endfunction
  627. function! s:finish_bindings()
  628. nnoremap <silent> <buffer> R :call <SID>retry()<cr>
  629. nnoremap <silent> <buffer> D :PlugDiff<cr>
  630. nnoremap <silent> <buffer> S :PlugStatus<cr>
  631. nnoremap <silent> <buffer> U :call <SID>status_update()<cr>
  632. xnoremap <silent> <buffer> U :call <SID>status_update()<cr>
  633. nnoremap <silent> <buffer> ]] :silent! call <SID>section('')<cr>
  634. nnoremap <silent> <buffer> [[ :silent! call <SID>section('b')<cr>
  635. endfunction
  636. function! s:prepare(...)
  637. if empty(getcwd())
  638. throw 'Invalid current working directory. Cannot proceed.'
  639. endif
  640. for evar in ['$GIT_DIR', '$GIT_WORK_TREE']
  641. if exists(evar)
  642. throw evar.' detected. Cannot proceed.'
  643. endif
  644. endfor
  645. call s:job_abort()
  646. if s:switch_in()
  647. if b:plug_preview == 1
  648. pc
  649. endif
  650. enew
  651. else
  652. call s:new_window()
  653. endif
  654. nnoremap <silent> <buffer> q :if b:plug_preview==1<bar>pc<bar>endif<bar>bd<cr>
  655. if a:0 == 0
  656. call s:finish_bindings()
  657. endif
  658. let b:plug_preview = -1
  659. let s:plug_tab = tabpagenr()
  660. let s:plug_buf = winbufnr(0)
  661. call s:assign_name()
  662. for k in ['<cr>', 'L', 'o', 'X', 'd', 'dd']
  663. execute 'silent! unmap <buffer>' k
  664. endfor
  665. setlocal buftype=nofile bufhidden=wipe nobuflisted noswapfile nowrap cursorline modifiable
  666. setf vim-plug
  667. if exists('g:syntax_on')
  668. call s:syntax()
  669. endif
  670. endfunction
  671. function! s:assign_name()
  672. " Assign buffer name
  673. let prefix = '[Plugins]'
  674. let name = prefix
  675. let idx = 2
  676. while bufexists(name)
  677. let name = printf('%s (%s)', prefix, idx)
  678. let idx = idx + 1
  679. endwhile
  680. silent! execute 'f' fnameescape(name)
  681. endfunction
  682. function! s:chsh(swap)
  683. let prev = [&shell, &shellredir]
  684. if !s:is_win && a:swap
  685. set shell=sh shellredir=>%s\ 2>&1
  686. endif
  687. return prev
  688. endfunction
  689. function! s:bang(cmd, ...)
  690. try
  691. let [sh, shrd] = s:chsh(a:0)
  692. " FIXME: Escaping is incomplete. We could use shellescape with eval,
  693. " but it won't work on Windows.
  694. let cmd = a:0 ? s:with_cd(a:cmd, a:1) : a:cmd
  695. let g:_plug_bang = '!'.escape(cmd, '#!%')
  696. execute "normal! :execute g:_plug_bang\<cr>\<cr>"
  697. finally
  698. unlet g:_plug_bang
  699. let [&shell, &shellredir] = [sh, shrd]
  700. endtry
  701. return v:shell_error ? 'Exit status: ' . v:shell_error : ''
  702. endfunction
  703. function! s:regress_bar()
  704. let bar = substitute(getline(2)[1:-2], '.*\zs=', 'x', '')
  705. call s:progress_bar(2, bar, len(bar))
  706. endfunction
  707. function! s:is_updated(dir)
  708. return !empty(s:system_chomp('git log --pretty=format:"%h" "HEAD...HEAD@{1}"', a:dir))
  709. endfunction
  710. function! s:do(pull, force, todo)
  711. for [name, spec] in items(a:todo)
  712. if !isdirectory(spec.dir)
  713. continue
  714. endif
  715. let installed = has_key(s:update.new, name)
  716. let updated = installed ? 0 :
  717. \ (a:pull && index(s:update.errors, name) < 0 && s:is_updated(spec.dir))
  718. if a:force || installed || updated
  719. execute 'cd' s:esc(spec.dir)
  720. call append(3, '- Post-update hook for '. name .' ... ')
  721. let error = ''
  722. let type = type(spec.do)
  723. if type == s:TYPE.string
  724. if spec.do[0] == ':'
  725. call s:load_plugin(spec)
  726. try
  727. execute spec.do[1:]
  728. catch
  729. let error = v:exception
  730. endtry
  731. if !s:plug_window_exists()
  732. cd -
  733. throw 'Warning: vim-plug was terminated by the post-update hook of '.name
  734. endif
  735. else
  736. let error = s:bang(spec.do)
  737. endif
  738. elseif type == s:TYPE.funcref
  739. try
  740. let status = installed ? 'installed' : (updated ? 'updated' : 'unchanged')
  741. call spec.do({ 'name': name, 'status': status, 'force': a:force })
  742. catch
  743. let error = v:exception
  744. endtry
  745. else
  746. let error = 'Invalid hook type'
  747. endif
  748. call s:switch_in()
  749. call setline(4, empty(error) ? (getline(4) . 'OK')
  750. \ : ('x' . getline(4)[1:] . error))
  751. if !empty(error)
  752. call add(s:update.errors, name)
  753. call s:regress_bar()
  754. endif
  755. cd -
  756. endif
  757. endfor
  758. endfunction
  759. function! s:hash_match(a, b)
  760. return stridx(a:a, a:b) == 0 || stridx(a:b, a:a) == 0
  761. endfunction
  762. function! s:checkout(spec)
  763. let sha = a:spec.commit
  764. let output = s:system('git rev-parse HEAD', a:spec.dir)
  765. if !v:shell_error && !s:hash_match(sha, s:lines(output)[0])
  766. let output = s:system(
  767. \ 'git fetch --depth 999999 && git checkout '.s:esc(sha), a:spec.dir)
  768. endif
  769. return output
  770. endfunction
  771. function! s:finish(pull)
  772. let new_frozen = len(filter(keys(s:update.new), 'g:plugs[v:val].frozen'))
  773. if new_frozen
  774. let s = new_frozen > 1 ? 's' : ''
  775. call append(3, printf('- Installed %d frozen plugin%s', new_frozen, s))
  776. endif
  777. call append(3, '- Finishing ... ') | 4
  778. redraw
  779. call plug#helptags()
  780. call plug#end()
  781. call setline(4, getline(4) . 'Done!')
  782. redraw
  783. let msgs = []
  784. if !empty(s:update.errors)
  785. call add(msgs, "Press 'R' to retry.")
  786. endif
  787. if a:pull && len(s:update.new) < len(filter(getline(5, '$'),
  788. \ "v:val =~ '^- ' && stridx(v:val, 'Already up-to-date') < 0"))
  789. call add(msgs, "Press 'D' to see the updated changes.")
  790. endif
  791. echo join(msgs, ' ')
  792. call s:finish_bindings()
  793. endfunction
  794. function! s:retry()
  795. if empty(s:update.errors)
  796. return
  797. endif
  798. echo
  799. call s:update_impl(s:update.pull, s:update.force,
  800. \ extend(copy(s:update.errors), [s:update.threads]))
  801. endfunction
  802. function! s:is_managed(name)
  803. return has_key(g:plugs[a:name], 'uri')
  804. endfunction
  805. function! s:names(...)
  806. return sort(filter(keys(g:plugs), 'stridx(v:val, a:1) == 0 && s:is_managed(v:val)'))
  807. endfunction
  808. function! s:check_ruby()
  809. silent! ruby require 'thread'; VIM::command("let g:plug_ruby = '#{RUBY_VERSION}'")
  810. if !exists('g:plug_ruby')
  811. redraw!
  812. return s:warn('echom', 'Warning: Ruby interface is broken')
  813. endif
  814. let ruby_version = split(g:plug_ruby, '\.')
  815. unlet g:plug_ruby
  816. return s:version_requirement(ruby_version, [1, 8, 7])
  817. endfunction
  818. function! s:update_impl(pull, force, args) abort
  819. let args = copy(a:args)
  820. let threads = (len(args) > 0 && args[-1] =~ '^[1-9][0-9]*$') ?
  821. \ remove(args, -1) : get(g:, 'plug_threads', 16)
  822. let managed = filter(copy(g:plugs), 's:is_managed(v:key)')
  823. let todo = empty(args) ? filter(managed, '!v:val.frozen || !isdirectory(v:val.dir)') :
  824. \ filter(managed, 'index(args, v:key) >= 0')
  825. if empty(todo)
  826. return s:warn('echo', 'No plugin to '. (a:pull ? 'update' : 'install'))
  827. endif
  828. if !s:is_win && s:git_version_requirement(2, 3)
  829. let s:git_terminal_prompt = exists('$GIT_TERMINAL_PROMPT') ? $GIT_TERMINAL_PROMPT : ''
  830. let $GIT_TERMINAL_PROMPT = 0
  831. for plug in values(todo)
  832. let plug.uri = substitute(plug.uri,
  833. \ '^https://git::@github\.com', 'https://github.com', '')
  834. endfor
  835. endif
  836. if !isdirectory(g:plug_home)
  837. try
  838. call mkdir(g:plug_home, 'p')
  839. catch
  840. return s:err(printf('Invalid plug directory: %s. '.
  841. \ 'Try to call plug#begin with a valid directory', g:plug_home))
  842. endtry
  843. endif
  844. if has('nvim') && !exists('*jobwait') && threads > 1
  845. call s:warn('echom', '[vim-plug] Update Neovim for parallel installer')
  846. endif
  847. let python = (has('python') || has('python3')) && !s:nvim
  848. let ruby = has('ruby') && !s:nvim && (v:version >= 703 || v:version == 702 && has('patch374')) && !(s:is_win && has('gui_running')) && s:check_ruby()
  849. let s:update = {
  850. \ 'start': reltime(),
  851. \ 'all': todo,
  852. \ 'todo': copy(todo),
  853. \ 'errors': [],
  854. \ 'pull': a:pull,
  855. \ 'force': a:force,
  856. \ 'new': {},
  857. \ 'threads': (python || ruby || s:nvim) ? min([len(todo), threads]) : 1,
  858. \ 'bar': '',
  859. \ 'fin': 0
  860. \ }
  861. call s:prepare(1)
  862. call append(0, ['', ''])
  863. normal! 2G
  864. silent! redraw
  865. let s:clone_opt = get(g:, 'plug_shallow', 1) ?
  866. \ '--depth 1' . (s:git_version_requirement(1, 7, 10) ? ' --no-single-branch' : '') : ''
  867. " Python version requirement (>= 2.7)
  868. if python && !has('python3') && !ruby && !s:nvim && s:update.threads > 1
  869. redir => pyv
  870. silent python import platform; print platform.python_version()
  871. redir END
  872. let python = s:version_requirement(
  873. \ map(split(split(pyv)[0], '\.'), 'str2nr(v:val)'), [2, 6])
  874. endif
  875. if (python || ruby) && s:update.threads > 1
  876. try
  877. let imd = &imd
  878. if s:mac_gui
  879. set noimd
  880. endif
  881. if ruby
  882. call s:update_ruby()
  883. else
  884. call s:update_python()
  885. endif
  886. catch
  887. let lines = getline(4, '$')
  888. let printed = {}
  889. silent! 4,$d _
  890. for line in lines
  891. let name = s:extract_name(line, '.', '')
  892. if empty(name) || !has_key(printed, name)
  893. call append('$', line)
  894. if !empty(name)
  895. let printed[name] = 1
  896. if line[0] == 'x' && index(s:update.errors, name) < 0
  897. call add(s:update.errors, name)
  898. end
  899. endif
  900. endif
  901. endfor
  902. finally
  903. let &imd = imd
  904. call s:update_finish()
  905. endtry
  906. else
  907. call s:update_vim()
  908. while s:nvim && has('vim_starting')
  909. sleep 100m
  910. if s:update.fin
  911. break
  912. endif
  913. endwhile
  914. endif
  915. endfunction
  916. function! s:log4(name, msg)
  917. call setline(4, printf('- %s (%s)', a:msg, a:name))
  918. redraw
  919. endfunction
  920. function! s:update_finish()
  921. if exists('s:git_terminal_prompt')
  922. let $GIT_TERMINAL_PROMPT = s:git_terminal_prompt
  923. endif
  924. if s:switch_in()
  925. call append(3, '- Updating ...') | 4
  926. 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))'))
  927. let pos = s:logpos(name)
  928. if !pos
  929. continue
  930. endif
  931. if has_key(spec, 'commit')
  932. call s:log4(name, 'Checking out '.spec.commit)
  933. let out = s:checkout(spec)
  934. elseif has_key(spec, 'tag')
  935. let tag = spec.tag
  936. if tag =~ '\*'
  937. let tags = s:lines(s:system('git tag --list '.string(tag).' --sort -version:refname 2>&1', spec.dir))
  938. if !v:shell_error && !empty(tags)
  939. let tag = tags[0]
  940. call s:log4(name, printf('Latest tag for %s -> %s', spec.tag, tag))
  941. call append(3, '')
  942. endif
  943. endif
  944. call s:log4(name, 'Checking out '.tag)
  945. let out = s:system('git checkout -q '.s:esc(tag).' 2>&1', spec.dir)
  946. else
  947. let branch = s:esc(get(spec, 'branch', 'master'))
  948. call s:log4(name, 'Merging origin/'.branch)
  949. let out = s:system('git checkout -q '.branch.' 2>&1'
  950. \. (has_key(s:update.new, name) ? '' : ('&& git merge --ff-only origin/'.branch.' 2>&1')), spec.dir)
  951. endif
  952. if !v:shell_error && filereadable(spec.dir.'/.gitmodules') &&
  953. \ (s:update.force || has_key(s:update.new, name) || s:is_updated(spec.dir))
  954. call s:log4(name, 'Updating submodules. This may take a while.')
  955. let out .= s:bang('git submodule update --init --recursive 2>&1', spec.dir)
  956. endif
  957. let msg = s:format_message(v:shell_error ? 'x': '-', name, out)
  958. if v:shell_error
  959. call add(s:update.errors, name)
  960. call s:regress_bar()
  961. silent execute pos 'd _'
  962. call append(4, msg) | 4
  963. elseif !empty(out)
  964. call setline(pos, msg[0])
  965. endif
  966. redraw
  967. endfor
  968. silent 4 d _
  969. try
  970. 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")'))
  971. catch
  972. call s:warn('echom', v:exception)
  973. call s:warn('echo', '')
  974. return
  975. endtry
  976. call s:finish(s:update.pull)
  977. call setline(1, 'Updated. Elapsed time: ' . split(reltimestr(reltime(s:update.start)))[0] . ' sec.')
  978. call s:switch_out('normal! gg')
  979. endif
  980. endfunction
  981. function! s:job_abort()
  982. if !s:nvim || !exists('s:jobs')
  983. return
  984. endif
  985. for [name, j] in items(s:jobs)
  986. silent! call jobstop(j.jobid)
  987. if j.new
  988. call s:system('rm -rf ' . s:shellesc(g:plugs[name].dir))
  989. endif
  990. endfor
  991. let s:jobs = {}
  992. endfunction
  993. " When a:event == 'stdout', data = list of strings
  994. " When a:event == 'exit', data = returncode
  995. function! s:job_handler(job_id, data, event) abort
  996. if !s:plug_window_exists() " plug window closed
  997. return s:job_abort()
  998. endif
  999. if a:event == 'stdout'
  1000. let complete = empty(a:data[-1])
  1001. let lines = map(filter(a:data, 'v:val =~ "[^\r\n]"'), 'split(v:val, "[\r\n]")[-1]')
  1002. call extend(self.lines, lines)
  1003. let self.result = join(self.lines, "\n")
  1004. if !complete
  1005. call remove(self.lines, -1)
  1006. endif
  1007. " To reduce the number of buffer updates
  1008. let self.tick = get(self, 'tick', -1) + 1
  1009. if self.tick % len(s:jobs) == 0
  1010. call s:log(self.new ? '+' : '*', self.name, self.result)
  1011. endif
  1012. elseif a:event == 'exit'
  1013. let self.running = 0
  1014. if a:data != 0
  1015. let self.error = 1
  1016. endif
  1017. call s:reap(self.name)
  1018. call s:tick()
  1019. endif
  1020. endfunction
  1021. function! s:spawn(name, cmd, opts)
  1022. let job = { 'name': a:name, 'running': 1, 'error': 0, 'lines': [], 'result': '',
  1023. \ 'new': get(a:opts, 'new', 0),
  1024. \ 'on_stdout': function('s:job_handler'),
  1025. \ 'on_exit' : function('s:job_handler'),
  1026. \ }
  1027. let s:jobs[a:name] = job
  1028. if s:nvim
  1029. let argv = [ 'sh', '-c',
  1030. \ (has_key(a:opts, 'dir') ? s:with_cd(a:cmd, a:opts.dir) : a:cmd) ]
  1031. let jid = jobstart(argv, job)
  1032. if jid > 0
  1033. let job.jobid = jid
  1034. else
  1035. let job.running = 0
  1036. let job.error = 1
  1037. let job.result = jid < 0 ? 'sh is not executable' :
  1038. \ 'Invalid arguments (or job table is full)'
  1039. endif
  1040. else
  1041. let params = has_key(a:opts, 'dir') ? [a:cmd, a:opts.dir] : [a:cmd]
  1042. let job.result = call('s:system', params)
  1043. let job.error = v:shell_error != 0
  1044. let job.running = 0
  1045. endif
  1046. endfunction
  1047. function! s:reap(name)
  1048. let job = s:jobs[a:name]
  1049. if job.error
  1050. call add(s:update.errors, a:name)
  1051. elseif get(job, 'new', 0)
  1052. let s:update.new[a:name] = 1
  1053. endif
  1054. let s:update.bar .= job.error ? 'x' : '='
  1055. call s:log(job.error ? 'x' : '-', a:name, empty(job.result) ? 'OK' : job.result)
  1056. call s:bar()
  1057. call remove(s:jobs, a:name)
  1058. endfunction
  1059. function! s:bar()
  1060. if s:switch_in()
  1061. let total = len(s:update.all)
  1062. call setline(1, (s:update.pull ? 'Updating' : 'Installing').
  1063. \ ' plugins ('.len(s:update.bar).'/'.total.')')
  1064. call s:progress_bar(2, s:update.bar, total)
  1065. call s:switch_out()
  1066. endif
  1067. endfunction
  1068. function! s:logpos(name)
  1069. for i in range(4, line('$'))
  1070. if getline(i) =~# '^[-+x*] '.a:name.':'
  1071. return i
  1072. endif
  1073. endfor
  1074. endfunction
  1075. function! s:log(bullet, name, lines)
  1076. if s:switch_in()
  1077. let pos = s:logpos(a:name)
  1078. if pos > 0
  1079. silent execute pos 'd _'
  1080. if pos > winheight('.')
  1081. let pos = 4
  1082. endif
  1083. else
  1084. let pos = 4
  1085. endif
  1086. call append(pos - 1, s:format_message(a:bullet, a:name, a:lines))
  1087. call s:switch_out()
  1088. endif
  1089. endfunction
  1090. function! s:update_vim()
  1091. let s:jobs = {}
  1092. call s:bar()
  1093. call s:tick()
  1094. endfunction
  1095. function! s:tick()
  1096. let pull = s:update.pull
  1097. let prog = s:progress_opt(s:nvim)
  1098. while 1 " Without TCO, Vim stack is bound to explode
  1099. if empty(s:update.todo)
  1100. if empty(s:jobs) && !s:update.fin
  1101. call s:update_finish()
  1102. let s:update.fin = 1
  1103. endif
  1104. return
  1105. endif
  1106. let name = keys(s:update.todo)[0]
  1107. let spec = remove(s:update.todo, name)
  1108. let new = !isdirectory(spec.dir)
  1109. call s:log(new ? '+' : '*', name, pull ? 'Updating ...' : 'Installing ...')
  1110. redraw
  1111. let has_tag = has_key(spec, 'tag')
  1112. if !new
  1113. let [error, _] = s:git_validate(spec, 0)
  1114. if empty(error)
  1115. if pull
  1116. let fetch_opt = (has_tag && !empty(globpath(spec.dir, '.git/shallow'))) ? '--depth 99999999' : ''
  1117. call s:spawn(name, printf('git fetch %s %s 2>&1', fetch_opt, prog), { 'dir': spec.dir })
  1118. else
  1119. let s:jobs[name] = { 'running': 0, 'result': 'Already installed', 'error': 0 }
  1120. endif
  1121. else
  1122. let s:jobs[name] = { 'running': 0, 'result': error, 'error': 1 }
  1123. endif
  1124. else
  1125. call s:spawn(name,
  1126. \ printf('git clone %s %s %s %s 2>&1',
  1127. \ has_tag ? '' : s:clone_opt,
  1128. \ prog,
  1129. \ s:shellesc(spec.uri),
  1130. \ s:shellesc(s:trim(spec.dir))), { 'new': 1 })
  1131. endif
  1132. if !s:jobs[name].running
  1133. call s:reap(name)
  1134. endif
  1135. if len(s:jobs) >= s:update.threads
  1136. break
  1137. endif
  1138. endwhile
  1139. endfunction
  1140. function! s:update_python()
  1141. let py_exe = has('python') ? 'python' : 'python3'
  1142. execute py_exe "<< EOF"
  1143. import datetime
  1144. import functools
  1145. import os
  1146. try:
  1147. import queue
  1148. except ImportError:
  1149. import Queue as queue
  1150. import random
  1151. import re
  1152. import shutil
  1153. import signal
  1154. import subprocess
  1155. import tempfile
  1156. import threading as thr
  1157. import time
  1158. import traceback
  1159. import vim
  1160. G_NVIM = vim.eval("has('nvim')") == '1'
  1161. G_PULL = vim.eval('s:update.pull') == '1'
  1162. G_RETRIES = int(vim.eval('get(g:, "plug_retries", 2)')) + 1
  1163. G_TIMEOUT = int(vim.eval('get(g:, "plug_timeout", 60)'))
  1164. G_CLONE_OPT = vim.eval('s:clone_opt')
  1165. G_PROGRESS = vim.eval('s:progress_opt(1)')
  1166. G_LOG_PROB = 1.0 / int(vim.eval('s:update.threads'))
  1167. G_STOP = thr.Event()
  1168. G_IS_WIN = vim.eval('s:is_win') == '1'
  1169. class PlugError(Exception):
  1170. def __init__(self, msg):
  1171. self.msg = msg
  1172. class CmdTimedOut(PlugError):
  1173. pass
  1174. class CmdFailed(PlugError):
  1175. pass
  1176. class InvalidURI(PlugError):
  1177. pass
  1178. class Action(object):
  1179. INSTALL, UPDATE, ERROR, DONE = ['+', '*', 'x', '-']
  1180. class Buffer(object):
  1181. def __init__(self, lock, num_plugs, is_pull):
  1182. self.bar = ''
  1183. self.event = 'Updating' if is_pull else 'Installing'
  1184. self.lock = lock
  1185. self.maxy = int(vim.eval('winheight(".")'))
  1186. self.num_plugs = num_plugs
  1187. def __where(self, name):
  1188. """ Find first line with name in current buffer. Return line num. """
  1189. found, lnum = False, 0
  1190. matcher = re.compile('^[-+x*] {0}:'.format(name))
  1191. for line in vim.current.buffer:
  1192. if matcher.search(line) is not None:
  1193. found = True
  1194. break
  1195. lnum += 1
  1196. if not found:
  1197. lnum = -1
  1198. return lnum
  1199. def header(self):
  1200. curbuf = vim.current.buffer
  1201. curbuf[0] = self.event + ' plugins ({0}/{1})'.format(len(self.bar), self.num_plugs)
  1202. num_spaces = self.num_plugs - len(self.bar)
  1203. curbuf[1] = '[{0}{1}]'.format(self.bar, num_spaces * ' ')
  1204. with self.lock:
  1205. vim.command('normal! 2G')
  1206. vim.command('redraw')
  1207. def write(self, action, name, lines):
  1208. first, rest = lines[0], lines[1:]
  1209. msg = ['{0} {1}{2}{3}'.format(action, name, ': ' if first else '', first)]
  1210. msg.extend([' ' + line for line in rest])
  1211. try:
  1212. if action == Action.ERROR:
  1213. self.bar += 'x'
  1214. vim.command("call add(s:update.errors, '{0}')".format(name))
  1215. elif action == Action.DONE:
  1216. self.bar += '='
  1217. curbuf = vim.current.buffer
  1218. lnum = self.__where(name)
  1219. if lnum != -1: # Found matching line num
  1220. del curbuf[lnum]
  1221. if lnum > self.maxy and action in set([Action.INSTALL, Action.UPDATE]):
  1222. lnum = 3
  1223. else:
  1224. lnum = 3
  1225. curbuf.append(msg, lnum)
  1226. self.header()
  1227. except vim.error:
  1228. pass
  1229. class Command(object):
  1230. CD = 'cd /d' if G_IS_WIN else 'cd'
  1231. def __init__(self, cmd, cmd_dir=None, timeout=60, cb=None, clean=None):
  1232. self.cmd = cmd
  1233. if cmd_dir:
  1234. self.cmd = '{0} {1} && {2}'.format(Command.CD, cmd_dir, self.cmd)
  1235. self.timeout = timeout
  1236. self.callback = cb if cb else (lambda msg: None)
  1237. self.clean = clean if clean else (lambda: None)
  1238. self.proc = None
  1239. @property
  1240. def alive(self):
  1241. """ Returns true only if command still running. """
  1242. return self.proc and self.proc.poll() is None
  1243. def execute(self, ntries=3):
  1244. """ Execute the command with ntries if CmdTimedOut.
  1245. Returns the output of the command if no Exception.
  1246. """
  1247. attempt, finished, limit = 0, False, self.timeout
  1248. while not finished:
  1249. try:
  1250. attempt += 1
  1251. result = self.try_command()
  1252. finished = True
  1253. return result
  1254. except CmdTimedOut:
  1255. if attempt != ntries:
  1256. self.notify_retry()
  1257. self.timeout += limit
  1258. else:
  1259. raise
  1260. def notify_retry(self):
  1261. """ Retry required for command, notify user. """
  1262. for count in range(3, 0, -1):
  1263. if G_STOP.is_set():
  1264. raise KeyboardInterrupt
  1265. msg = 'Timeout. Will retry in {0} second{1} ...'.format(
  1266. count, 's' if count != 1 else '')
  1267. self.callback([msg])
  1268. time.sleep(1)
  1269. self.callback(['Retrying ...'])
  1270. def try_command(self):
  1271. """ Execute a cmd & poll for callback. Returns list of output.
  1272. Raises CmdFailed -> return code for Popen isn't 0
  1273. Raises CmdTimedOut -> command exceeded timeout without new output
  1274. """
  1275. first_line = True
  1276. try:
  1277. tfile = tempfile.NamedTemporaryFile(mode='w+b')
  1278. preexec_fn = not G_IS_WIN and os.setsid or None
  1279. self.proc = subprocess.Popen(self.cmd, stdout=tfile,
  1280. stderr=subprocess.STDOUT,
  1281. stdin=subprocess.PIPE, shell=True,
  1282. preexec_fn=preexec_fn)
  1283. thrd = thr.Thread(target=(lambda proc: proc.wait()), args=(self.proc,))
  1284. thrd.start()
  1285. thread_not_started = True
  1286. while thread_not_started:
  1287. try:
  1288. thrd.join(0.1)
  1289. thread_not_started = False
  1290. except RuntimeError:
  1291. pass
  1292. while self.alive:
  1293. if G_STOP.is_set():
  1294. raise KeyboardInterrupt
  1295. if first_line or random.random() < G_LOG_PROB:
  1296. first_line = False
  1297. line = '' if G_IS_WIN else nonblock_read(tfile.name)
  1298. if line:
  1299. self.callback([line])
  1300. time_diff = time.time() - os.path.getmtime(tfile.name)
  1301. if time_diff > self.timeout:
  1302. raise CmdTimedOut(['Timeout!'])
  1303. thrd.join(0.5)
  1304. tfile.seek(0)
  1305. result = [line.decode('utf-8', 'replace').rstrip() for line in tfile]
  1306. if self.proc.returncode != 0:
  1307. raise CmdFailed([''] + result)
  1308. return result
  1309. except:
  1310. self.terminate()
  1311. raise
  1312. def terminate(self):
  1313. """ Terminate process and cleanup. """
  1314. if self.alive:
  1315. if G_IS_WIN:
  1316. os.kill(self.proc.pid, signal.SIGINT)
  1317. else:
  1318. os.killpg(self.proc.pid, signal.SIGTERM)
  1319. self.clean()
  1320. class Plugin(object):
  1321. def __init__(self, name, args, buf_q, lock):
  1322. self.name = name
  1323. self.args = args
  1324. self.buf_q = buf_q
  1325. self.lock = lock
  1326. self.tag = args.get('tag', 0)
  1327. def manage(self):
  1328. try:
  1329. if os.path.exists(self.args['dir']):
  1330. self.update()
  1331. else:
  1332. self.install()
  1333. with self.lock:
  1334. thread_vim_command("let s:update.new['{0}'] = 1".format(self.name))
  1335. except PlugError as exc:
  1336. self.write(Action.ERROR, self.name, exc.msg)
  1337. except KeyboardInterrupt:
  1338. G_STOP.set()
  1339. self.write(Action.ERROR, self.name, ['Interrupted!'])
  1340. except:
  1341. # Any exception except those above print stack trace
  1342. msg = 'Trace:\n{0}'.format(traceback.format_exc().rstrip())
  1343. self.write(Action.ERROR, self.name, msg.split('\n'))
  1344. raise
  1345. def install(self):
  1346. target = self.args['dir']
  1347. if target[-1] == '\\':
  1348. target = target[0:-1]
  1349. def clean(target):
  1350. def _clean():
  1351. try:
  1352. shutil.rmtree(target)
  1353. except OSError:
  1354. pass
  1355. return _clean
  1356. self.write(Action.INSTALL, self.name, ['Installing ...'])
  1357. callback = functools.partial(self.write, Action.INSTALL, self.name)
  1358. cmd = 'git clone {0} {1} {2} {3} 2>&1'.format(
  1359. '' if self.tag else G_CLONE_OPT, G_PROGRESS, self.args['uri'],
  1360. esc(target))
  1361. com = Command(cmd, None, G_TIMEOUT, callback, clean(target))
  1362. result = com.execute(G_RETRIES)
  1363. self.write(Action.DONE, self.name, result[-1:])
  1364. def repo_uri(self):
  1365. cmd = 'git rev-parse --abbrev-ref HEAD 2>&1 && git config -f .git/config remote.origin.url'
  1366. command = Command(cmd, self.args['dir'], G_TIMEOUT,)
  1367. result = command.execute(G_RETRIES)
  1368. return result[-1]
  1369. def update(self):
  1370. actual_uri = self.repo_uri()
  1371. expect_uri = self.args['uri']
  1372. regex = re.compile(r'^(?:\w+://)?(?:[^@/]*@)?([^:/]*(?::[0-9]*)?)[:/](.*?)(?:\.git)?/?$')
  1373. ma = regex.match(actual_uri)
  1374. mb = regex.match(expect_uri)
  1375. if ma is None or mb is None or ma.groups() != mb.groups():
  1376. msg = ['',
  1377. 'Invalid URI: {0}'.format(actual_uri),
  1378. 'Expected {0}'.format(expect_uri),
  1379. 'PlugClean required.']
  1380. raise InvalidURI(msg)
  1381. if G_PULL:
  1382. self.write(Action.UPDATE, self.name, ['Updating ...'])
  1383. callback = functools.partial(self.write, Action.UPDATE, self.name)
  1384. fetch_opt = '--depth 99999999' if self.tag and os.path.isfile(os.path.join(self.args['dir'], '.git/shallow')) else ''
  1385. cmd = 'git fetch {0} {1} 2>&1'.format(fetch_opt, G_PROGRESS)
  1386. com = Command(cmd, self.args['dir'], G_TIMEOUT, callback)
  1387. result = com.execute(G_RETRIES)
  1388. self.write(Action.DONE, self.name, result[-1:])
  1389. else:
  1390. self.write(Action.DONE, self.name, ['Already installed'])
  1391. def write(self, action, name, msg):
  1392. self.buf_q.put((action, name, msg))
  1393. class PlugThread(thr.Thread):
  1394. def __init__(self, tname, args):
  1395. super(PlugThread, self).__init__()
  1396. self.tname = tname
  1397. self.args = args
  1398. def run(self):
  1399. thr.current_thread().name = self.tname
  1400. buf_q, work_q, lock = self.args
  1401. try:
  1402. while not G_STOP.is_set():
  1403. name, args = work_q.get_nowait()
  1404. plug = Plugin(name, args, buf_q, lock)
  1405. plug.manage()
  1406. work_q.task_done()
  1407. except queue.Empty:
  1408. pass
  1409. class RefreshThread(thr.Thread):
  1410. def __init__(self, lock):
  1411. super(RefreshThread, self).__init__()
  1412. self.lock = lock
  1413. self.running = True
  1414. def run(self):
  1415. while self.running:
  1416. with self.lock:
  1417. thread_vim_command('noautocmd normal! a')
  1418. time.sleep(0.33)
  1419. def stop(self):
  1420. self.running = False
  1421. if G_NVIM:
  1422. def thread_vim_command(cmd):
  1423. vim.session.threadsafe_call(lambda: vim.command(cmd))
  1424. else:
  1425. def thread_vim_command(cmd):
  1426. vim.command(cmd)
  1427. def esc(name):
  1428. return '"' + name.replace('"', '\"') + '"'
  1429. def nonblock_read(fname):
  1430. """ Read a file with nonblock flag. Return the last line. """
  1431. fread = os.open(fname, os.O_RDONLY | os.O_NONBLOCK)
  1432. buf = os.read(fread, 100000).decode('utf-8', 'replace')
  1433. os.close(fread)
  1434. line = buf.rstrip('\r\n')
  1435. left = max(line.rfind('\r'), line.rfind('\n'))
  1436. if left != -1:
  1437. left += 1
  1438. line = line[left:]
  1439. return line
  1440. def main():
  1441. thr.current_thread().name = 'main'
  1442. nthreads = int(vim.eval('s:update.threads'))
  1443. plugs = vim.eval('s:update.todo')
  1444. mac_gui = vim.eval('s:mac_gui') == '1'
  1445. lock = thr.Lock()
  1446. buf = Buffer(lock, len(plugs), G_PULL)
  1447. buf_q, work_q = queue.Queue(), queue.Queue()
  1448. for work in plugs.items():
  1449. work_q.put(work)
  1450. start_cnt = thr.active_count()
  1451. for num in range(nthreads):
  1452. tname = 'PlugT-{0:02}'.format(num)
  1453. thread = PlugThread(tname, (buf_q, work_q, lock))
  1454. thread.start()
  1455. if mac_gui:
  1456. rthread = RefreshThread(lock)
  1457. rthread.start()
  1458. while not buf_q.empty() or thr.active_count() != start_cnt:
  1459. try:
  1460. action, name, msg = buf_q.get(True, 0.25)
  1461. buf.write(action, name, ['OK'] if not msg else msg)
  1462. buf_q.task_done()
  1463. except queue.Empty:
  1464. pass
  1465. except KeyboardInterrupt:
  1466. G_STOP.set()
  1467. if mac_gui:
  1468. rthread.stop()
  1469. rthread.join()
  1470. main()
  1471. EOF
  1472. endfunction
  1473. function! s:update_ruby()
  1474. ruby << EOF
  1475. module PlugStream
  1476. SEP = ["\r", "\n", nil]
  1477. def get_line
  1478. buffer = ''
  1479. loop do
  1480. char = readchar rescue return
  1481. if SEP.include? char.chr
  1482. buffer << $/
  1483. break
  1484. else
  1485. buffer << char
  1486. end
  1487. end
  1488. buffer
  1489. end
  1490. end unless defined?(PlugStream)
  1491. def esc arg
  1492. %["#{arg.gsub('"', '\"')}"]
  1493. end
  1494. def killall pid
  1495. pids = [pid]
  1496. if /mswin|mingw|bccwin/ =~ RUBY_PLATFORM
  1497. pids.each { |pid| Process.kill 'INT', pid.to_i rescue nil }
  1498. else
  1499. unless `which pgrep 2> /dev/null`.empty?
  1500. children = pids
  1501. until children.empty?
  1502. children = children.map { |pid|
  1503. `pgrep -P #{pid}`.lines.map { |l| l.chomp }
  1504. }.flatten
  1505. pids += children
  1506. end
  1507. end
  1508. pids.each { |pid| Process.kill 'TERM', pid.to_i rescue nil }
  1509. end
  1510. end
  1511. def compare_git_uri a, b
  1512. regex = %r{^(?:\w+://)?(?:[^@/]*@)?([^:/]*(?::[0-9]*)?)[:/](.*?)(?:\.git)?/?$}
  1513. regex.match(a).to_a.drop(1) == regex.match(b).to_a.drop(1)
  1514. end
  1515. require 'thread'
  1516. require 'fileutils'
  1517. require 'timeout'
  1518. running = true
  1519. iswin = VIM::evaluate('s:is_win').to_i == 1
  1520. pull = VIM::evaluate('s:update.pull').to_i == 1
  1521. base = VIM::evaluate('g:plug_home')
  1522. all = VIM::evaluate('s:update.todo')
  1523. limit = VIM::evaluate('get(g:, "plug_timeout", 60)')
  1524. tries = VIM::evaluate('get(g:, "plug_retries", 2)') + 1
  1525. nthr = VIM::evaluate('s:update.threads').to_i
  1526. maxy = VIM::evaluate('winheight(".")').to_i
  1527. cd = iswin ? 'cd /d' : 'cd'
  1528. tot = VIM::evaluate('len(s:update.todo)') || 0
  1529. bar = ''
  1530. skip = 'Already installed'
  1531. mtx = Mutex.new
  1532. take1 = proc { mtx.synchronize { running && all.shift } }
  1533. logh = proc {
  1534. cnt = bar.length
  1535. $curbuf[1] = "#{pull ? 'Updating' : 'Installing'} plugins (#{cnt}/#{tot})"
  1536. $curbuf[2] = '[' + bar.ljust(tot) + ']'
  1537. VIM::command('normal! 2G')
  1538. VIM::command('redraw')
  1539. }
  1540. where = proc { |name| (1..($curbuf.length)).find { |l| $curbuf[l] =~ /^[-+x*] #{name}:/ } }
  1541. log = proc { |name, result, type|
  1542. mtx.synchronize do
  1543. ing = ![true, false].include?(type)
  1544. bar += type ? '=' : 'x' unless ing
  1545. b = case type
  1546. when :install then '+' when :update then '*'
  1547. when true, nil then '-' else
  1548. VIM::command("call add(s:update.errors, '#{name}')")
  1549. 'x'
  1550. end
  1551. result =
  1552. if type || type.nil?
  1553. ["#{b} #{name}: #{result.lines.to_a.last || 'OK'}"]
  1554. elsif result =~ /^Interrupted|^Timeout/
  1555. ["#{b} #{name}: #{result}"]
  1556. else
  1557. ["#{b} #{name}"] + result.lines.map { |l| " " << l }
  1558. end
  1559. if lnum = where.call(name)
  1560. $curbuf.delete lnum
  1561. lnum = 4 if ing && lnum > maxy
  1562. end
  1563. result.each_with_index do |line, offset|
  1564. $curbuf.append((lnum || 4) - 1 + offset, line.gsub(/\e\[./, '').chomp)
  1565. end
  1566. logh.call
  1567. end
  1568. }
  1569. bt = proc { |cmd, name, type, cleanup|
  1570. tried = timeout = 0
  1571. begin
  1572. tried += 1
  1573. timeout += limit
  1574. fd = nil
  1575. data = ''
  1576. if iswin
  1577. Timeout::timeout(timeout) do
  1578. tmp = VIM::evaluate('tempname()')
  1579. system("(#{cmd}) > #{tmp}")
  1580. data = File.read(tmp).chomp
  1581. File.unlink tmp rescue nil
  1582. end
  1583. else
  1584. fd = IO.popen(cmd).extend(PlugStream)
  1585. first_line = true
  1586. log_prob = 1.0 / nthr
  1587. while line = Timeout::timeout(timeout) { fd.get_line }
  1588. data << line
  1589. log.call name, line.chomp, type if name && (first_line || rand < log_prob)
  1590. first_line = false
  1591. end
  1592. fd.close
  1593. end
  1594. [$? == 0, data.chomp]
  1595. rescue Timeout::Error, Interrupt => e
  1596. if fd && !fd.closed?
  1597. killall fd.pid
  1598. fd.close
  1599. end
  1600. cleanup.call if cleanup
  1601. if e.is_a?(Timeout::Error) && tried < tries
  1602. 3.downto(1) do |countdown|
  1603. s = countdown > 1 ? 's' : ''
  1604. log.call name, "Timeout. Will retry in #{countdown} second#{s} ...", type
  1605. sleep 1
  1606. end
  1607. log.call name, 'Retrying ...', type
  1608. retry
  1609. end
  1610. [false, e.is_a?(Interrupt) ? "Interrupted!" : "Timeout!"]
  1611. end
  1612. }
  1613. main = Thread.current
  1614. threads = []
  1615. watcher = Thread.new {
  1616. require 'io/console' # >= Ruby 1.9
  1617. nil until IO.console.getch == 3.chr
  1618. mtx.synchronize do
  1619. running = false
  1620. threads.each { |t| t.raise Interrupt }
  1621. end
  1622. threads.each { |t| t.join rescue nil }
  1623. main.kill
  1624. }
  1625. refresh = Thread.new {
  1626. while true
  1627. mtx.synchronize do
  1628. break unless running
  1629. VIM::command('noautocmd normal! a')
  1630. end
  1631. sleep 0.2
  1632. end
  1633. } if VIM::evaluate('s:mac_gui') == 1
  1634. clone_opt = VIM::evaluate('s:clone_opt')
  1635. progress = VIM::evaluate('s:progress_opt(1)')
  1636. nthr.times do
  1637. mtx.synchronize do
  1638. threads << Thread.new {
  1639. while pair = take1.call
  1640. name = pair.first
  1641. dir, uri, tag = pair.last.values_at *%w[dir uri tag]
  1642. exists = File.directory? dir
  1643. ok, result =
  1644. if exists
  1645. chdir = "#{cd} #{iswin ? dir : esc(dir)}"
  1646. ret, data = bt.call "#{chdir} && git rev-parse --abbrev-ref HEAD 2>&1 && git config -f .git/config remote.origin.url", nil, nil, nil
  1647. current_uri = data.lines.to_a.last
  1648. if !ret
  1649. if data =~ /^Interrupted|^Timeout/
  1650. [false, data]
  1651. else
  1652. [false, [data.chomp, "PlugClean required."].join($/)]
  1653. end
  1654. elsif !compare_git_uri(current_uri, uri)
  1655. [false, ["Invalid URI: #{current_uri}",
  1656. "Expected: #{uri}",
  1657. "PlugClean required."].join($/)]
  1658. else
  1659. if pull
  1660. log.call name, 'Updating ...', :update
  1661. fetch_opt = (tag && File.exist?(File.join(dir, '.git/shallow'))) ? '--depth 99999999' : ''
  1662. bt.call "#{chdir} && git fetch #{fetch_opt} #{progress} 2>&1", name, :update, nil
  1663. else
  1664. [true, skip]
  1665. end
  1666. end
  1667. else
  1668. d = esc dir.sub(%r{[\\/]+$}, '')
  1669. log.call name, 'Installing ...', :install
  1670. bt.call "git clone #{clone_opt unless tag} #{progress} #{uri} #{d} 2>&1", name, :install, proc {
  1671. FileUtils.rm_rf dir
  1672. }
  1673. end
  1674. mtx.synchronize { VIM::command("let s:update.new['#{name}'] = 1") } if !exists && ok
  1675. log.call name, result, ok
  1676. end
  1677. } if running
  1678. end
  1679. end
  1680. threads.each { |t| t.join rescue nil }
  1681. logh.call
  1682. refresh.kill if refresh
  1683. watcher.kill
  1684. EOF
  1685. endfunction
  1686. function! s:shellesc(arg)
  1687. return '"'.escape(a:arg, '"').'"'
  1688. endfunction
  1689. function! s:glob_dir(path)
  1690. return map(filter(s:glob(a:path, '**'), 'isdirectory(v:val)'), 's:dirpath(v:val)')
  1691. endfunction
  1692. function! s:progress_bar(line, bar, total)
  1693. call setline(a:line, '[' . s:lpad(a:bar, a:total) . ']')
  1694. endfunction
  1695. function! s:compare_git_uri(a, b)
  1696. " See `git help clone'
  1697. " https:// [user@] github.com[:port] / junegunn/vim-plug [.git]
  1698. " [git@] github.com[:port] : junegunn/vim-plug [.git]
  1699. " file:// / junegunn/vim-plug [/]
  1700. " / junegunn/vim-plug [/]
  1701. let pat = '^\%(\w\+://\)\='.'\%([^@/]*@\)\='.'\([^:/]*\%(:[0-9]*\)\=\)'.'[:/]'.'\(.\{-}\)'.'\%(\.git\)\=/\?$'
  1702. let ma = matchlist(a:a, pat)
  1703. let mb = matchlist(a:b, pat)
  1704. return ma[1:2] ==# mb[1:2]
  1705. endfunction
  1706. function! s:format_message(bullet, name, message)
  1707. if a:bullet != 'x'
  1708. return [printf('%s %s: %s', a:bullet, a:name, s:lastline(a:message))]
  1709. else
  1710. let lines = map(s:lines(a:message), '" ".v:val')
  1711. return extend([printf('x %s:', a:name)], lines)
  1712. endif
  1713. endfunction
  1714. function! s:with_cd(cmd, dir)
  1715. return printf('cd%s %s && %s', s:is_win ? ' /d' : '', s:shellesc(a:dir), a:cmd)
  1716. endfunction
  1717. function! s:system(cmd, ...)
  1718. try
  1719. let [sh, shrd] = s:chsh(1)
  1720. let cmd = a:0 > 0 ? s:with_cd(a:cmd, a:1) : a:cmd
  1721. return system(s:is_win ? '('.cmd.')' : cmd)
  1722. finally
  1723. let [&shell, &shellredir] = [sh, shrd]
  1724. endtry
  1725. endfunction
  1726. function! s:system_chomp(...)
  1727. let ret = call('s:system', a:000)
  1728. return v:shell_error ? '' : substitute(ret, '\n$', '', '')
  1729. endfunction
  1730. function! s:git_validate(spec, check_branch)
  1731. let err = ''
  1732. if isdirectory(a:spec.dir)
  1733. 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))
  1734. let remote = result[-1]
  1735. if v:shell_error
  1736. let err = join([remote, 'PlugClean required.'], "\n")
  1737. elseif !s:compare_git_uri(remote, a:spec.uri)
  1738. let err = join(['Invalid URI: '.remote,
  1739. \ 'Expected: '.a:spec.uri,
  1740. \ 'PlugClean required.'], "\n")
  1741. elseif a:check_branch && has_key(a:spec, 'commit')
  1742. let result = s:lines(s:system('git rev-parse HEAD 2>&1', a:spec.dir))
  1743. let sha = result[-1]
  1744. if v:shell_error
  1745. let err = join(add(result, 'PlugClean required.'), "\n")
  1746. elseif !s:hash_match(sha, a:spec.commit)
  1747. let err = join([printf('Invalid HEAD (expected: %s, actual: %s)',
  1748. \ a:spec.commit[:6], sha[:6]),
  1749. \ 'PlugUpdate required.'], "\n")
  1750. endif
  1751. elseif a:check_branch
  1752. let branch = result[0]
  1753. " Check tag
  1754. if has_key(a:spec, 'tag')
  1755. let tag = s:system_chomp('git describe --exact-match --tags HEAD 2>&1', a:spec.dir)
  1756. if a:spec.tag !=# tag
  1757. let err = printf('Invalid tag: %s (expected: %s). Try PlugUpdate.',
  1758. \ (empty(tag) ? 'N/A' : tag), a:spec.tag)
  1759. endif
  1760. " Check branch
  1761. elseif a:spec.branch !=# branch
  1762. let err = printf('Invalid branch: %s (expected: %s). Try PlugUpdate.',
  1763. \ branch, a:spec.branch)
  1764. endif
  1765. if empty(err)
  1766. let commits = len(s:lines(s:system(printf('git rev-list origin/%s..HEAD', a:spec.branch), a:spec.dir)))
  1767. if !v:shell_error && commits
  1768. let err = join([printf('Diverged from origin/%s by %d commit(s).', a:spec.branch, commits),
  1769. \ 'Reinstall after PlugClean.'], "\n")
  1770. endif
  1771. endif
  1772. endif
  1773. else
  1774. let err = 'Not found'
  1775. endif
  1776. return [err, err =~# 'PlugClean']
  1777. endfunction
  1778. function! s:rm_rf(dir)
  1779. if isdirectory(a:dir)
  1780. call s:system((s:is_win ? 'rmdir /S /Q ' : 'rm -rf ') . s:shellesc(a:dir))
  1781. endif
  1782. endfunction
  1783. function! s:clean(force)
  1784. call s:prepare()
  1785. call append(0, 'Searching for invalid plugins in '.g:plug_home)
  1786. call append(1, '')
  1787. " List of valid directories
  1788. let dirs = []
  1789. let errs = {}
  1790. let [cnt, total] = [0, len(g:plugs)]
  1791. for [name, spec] in items(g:plugs)
  1792. if !s:is_managed(name)
  1793. call add(dirs, spec.dir)
  1794. else
  1795. let [err, clean] = s:git_validate(spec, 1)
  1796. if clean
  1797. let errs[spec.dir] = s:lines(err)[0]
  1798. else
  1799. call add(dirs, spec.dir)
  1800. endif
  1801. endif
  1802. let cnt += 1
  1803. call s:progress_bar(2, repeat('=', cnt), total)
  1804. normal! 2G
  1805. redraw
  1806. endfor
  1807. let allowed = {}
  1808. for dir in dirs
  1809. let allowed[s:dirpath(fnamemodify(dir, ':h:h'))] = 1
  1810. let allowed[dir] = 1
  1811. for child in s:glob_dir(dir)
  1812. let allowed[child] = 1
  1813. endfor
  1814. endfor
  1815. let todo = []
  1816. let found = sort(s:glob_dir(g:plug_home))
  1817. while !empty(found)
  1818. let f = remove(found, 0)
  1819. if !has_key(allowed, f) && isdirectory(f)
  1820. call add(todo, f)
  1821. call append(line('$'), '- ' . f)
  1822. if has_key(errs, f)
  1823. call append(line('$'), ' ' . errs[f])
  1824. endif
  1825. let found = filter(found, 'stridx(v:val, f) != 0')
  1826. end
  1827. endwhile
  1828. 4
  1829. redraw
  1830. if empty(todo)
  1831. call append(line('$'), 'Already clean.')
  1832. else
  1833. let s:clean_count = 0
  1834. call append(3, ['Directories to delete:', ''])
  1835. redraw!
  1836. if a:force || s:ask_no_interrupt('Delete all directories?')
  1837. call s:delete([6, line('$')], 1)
  1838. else
  1839. call setline(4, 'Cancelled.')
  1840. nnoremap <silent> <buffer> d :set opfunc=<sid>delete_op<cr>g@
  1841. nmap <silent> <buffer> dd d_
  1842. xnoremap <silent> <buffer> d :<c-u>call <sid>delete_op(visualmode(), 1)<cr>
  1843. echo 'Delete the lines (d{motion}) to delete the corresponding directories'
  1844. endif
  1845. endif
  1846. 4
  1847. setlocal nomodifiable
  1848. endfunction
  1849. function! s:delete_op(type, ...)
  1850. call s:delete(a:0 ? [line("'<"), line("'>")] : [line("'["), line("']")], 0)
  1851. endfunction
  1852. function! s:delete(range, force)
  1853. let [l1, l2] = a:range
  1854. let force = a:force
  1855. while l1 <= l2
  1856. let line = getline(l1)
  1857. if line =~ '^- ' && isdirectory(line[2:])
  1858. execute l1
  1859. redraw!
  1860. let answer = force ? 1 : s:ask('Delete '.line[2:].'?', 1)
  1861. let force = force || answer > 1
  1862. if answer
  1863. call s:rm_rf(line[2:])
  1864. setlocal modifiable
  1865. call setline(l1, '~'.line[1:])
  1866. let s:clean_count += 1
  1867. call setline(4, printf('Removed %d directories.', s:clean_count))
  1868. setlocal nomodifiable
  1869. endif
  1870. endif
  1871. let l1 += 1
  1872. endwhile
  1873. endfunction
  1874. function! s:upgrade()
  1875. echo 'Downloading the latest version of vim-plug'
  1876. redraw
  1877. let tmp = tempname()
  1878. let new = tmp . '/plug.vim'
  1879. try
  1880. let out = s:system(printf('git clone --depth 1 %s %s', s:plug_src, tmp))
  1881. if v:shell_error
  1882. return s:err('Error upgrading vim-plug: '. out)
  1883. endif
  1884. if readfile(s:me) ==# readfile(new)
  1885. echo 'vim-plug is already up-to-date'
  1886. return 0
  1887. else
  1888. call rename(s:me, s:me . '.old')
  1889. call rename(new, s:me)
  1890. unlet g:loaded_plug
  1891. echo 'vim-plug has been upgraded'
  1892. return 1
  1893. endif
  1894. finally
  1895. silent! call s:rm_rf(tmp)
  1896. endtry
  1897. endfunction
  1898. function! s:upgrade_specs()
  1899. for spec in values(g:plugs)
  1900. let spec.frozen = get(spec, 'frozen', 0)
  1901. endfor
  1902. endfunction
  1903. function! s:status()
  1904. call s:prepare()
  1905. call append(0, 'Checking plugins')
  1906. call append(1, '')
  1907. let ecnt = 0
  1908. let unloaded = 0
  1909. let [cnt, total] = [0, len(g:plugs)]
  1910. for [name, spec] in items(g:plugs)
  1911. if has_key(spec, 'uri')
  1912. if isdirectory(spec.dir)
  1913. let [err, _] = s:git_validate(spec, 1)
  1914. let [valid, msg] = [empty(err), empty(err) ? 'OK' : err]
  1915. else
  1916. let [valid, msg] = [0, 'Not found. Try PlugInstall.']
  1917. endif
  1918. else
  1919. if isdirectory(spec.dir)
  1920. let [valid, msg] = [1, 'OK']
  1921. else
  1922. let [valid, msg] = [0, 'Not found.']
  1923. endif
  1924. endif
  1925. let cnt += 1
  1926. let ecnt += !valid
  1927. " `s:loaded` entry can be missing if PlugUpgraded
  1928. if valid && get(s:loaded, name, -1) == 0
  1929. let unloaded = 1
  1930. let msg .= ' (not loaded)'
  1931. endif
  1932. call s:progress_bar(2, repeat('=', cnt), total)
  1933. call append(3, s:format_message(valid ? '-' : 'x', name, msg))
  1934. normal! 2G
  1935. redraw
  1936. endfor
  1937. call setline(1, 'Finished. '.ecnt.' error(s).')
  1938. normal! gg
  1939. setlocal nomodifiable
  1940. if unloaded
  1941. echo "Press 'L' on each line to load plugin, or 'U' to update"
  1942. nnoremap <silent> <buffer> L :call <SID>status_load(line('.'))<cr>
  1943. xnoremap <silent> <buffer> L :call <SID>status_load(line('.'))<cr>
  1944. end
  1945. endfunction
  1946. function! s:extract_name(str, prefix, suffix)
  1947. return matchstr(a:str, '^'.a:prefix.' \zs[^:]\+\ze:.*'.a:suffix.'$')
  1948. endfunction
  1949. function! s:status_load(lnum)
  1950. let line = getline(a:lnum)
  1951. let name = s:extract_name(line, '-', '(not loaded)')
  1952. if !empty(name)
  1953. call plug#load(name)
  1954. setlocal modifiable
  1955. call setline(a:lnum, substitute(line, ' (not loaded)$', '', ''))
  1956. setlocal nomodifiable
  1957. endif
  1958. endfunction
  1959. function! s:status_update() range
  1960. let lines = getline(a:firstline, a:lastline)
  1961. let names = filter(map(lines, 's:extract_name(v:val, "[x-]", "")'), '!empty(v:val)')
  1962. if !empty(names)
  1963. echo
  1964. execute 'PlugUpdate' join(names)
  1965. endif
  1966. endfunction
  1967. function! s:is_preview_window_open()
  1968. silent! wincmd P
  1969. if &previewwindow
  1970. wincmd p
  1971. return 1
  1972. endif
  1973. endfunction
  1974. function! s:find_name(lnum)
  1975. for lnum in reverse(range(1, a:lnum))
  1976. let line = getline(lnum)
  1977. if empty(line)
  1978. return ''
  1979. endif
  1980. let name = s:extract_name(line, '-', '')
  1981. if !empty(name)
  1982. return name
  1983. endif
  1984. endfor
  1985. return ''
  1986. endfunction
  1987. function! s:preview_commit()
  1988. if b:plug_preview < 0
  1989. let b:plug_preview = !s:is_preview_window_open()
  1990. endif
  1991. let sha = matchstr(getline('.'), '^ \X*\zs[0-9a-f]\{7}')
  1992. if empty(sha)
  1993. return
  1994. endif
  1995. let name = s:find_name(line('.'))
  1996. if empty(name) || !has_key(g:plugs, name) || !isdirectory(g:plugs[name].dir)
  1997. return
  1998. endif
  1999. if exists('g:plug_pwindow') && !s:is_preview_window_open()
  2000. execute g:plug_pwindow
  2001. execute 'e' sha
  2002. else
  2003. execute 'pedit' sha
  2004. wincmd P
  2005. endif
  2006. setlocal previewwindow filetype=git buftype=nofile nobuflisted modifiable
  2007. execute 'silent %!cd' s:shellesc(g:plugs[name].dir) '&& git show --no-color --pretty=medium' sha
  2008. setlocal nomodifiable
  2009. nnoremap <silent> <buffer> q :q<cr>
  2010. wincmd p
  2011. endfunction
  2012. function! s:section(flags)
  2013. call search('\(^[x-] \)\@<=[^:]\+:', a:flags)
  2014. endfunction
  2015. function! s:format_git_log(line)
  2016. let indent = ' '
  2017. let tokens = split(a:line, nr2char(1))
  2018. if len(tokens) != 5
  2019. return indent.substitute(a:line, '\s*$', '', '')
  2020. endif
  2021. let [graph, sha, refs, subject, date] = tokens
  2022. let tag = matchstr(refs, 'tag: [^,)]\+')
  2023. let tag = empty(tag) ? ' ' : ' ('.tag.') '
  2024. return printf('%s%s%s%s%s (%s)', indent, graph, sha, tag, subject, date)
  2025. endfunction
  2026. function! s:append_ul(lnum, text)
  2027. call append(a:lnum, ['', a:text, repeat('-', len(a:text))])
  2028. endfunction
  2029. function! s:diff()
  2030. call s:prepare()
  2031. call append(0, ['Collecting changes ...', ''])
  2032. let cnts = [0, 0]
  2033. let bar = ''
  2034. let total = filter(copy(g:plugs), 's:is_managed(v:key) && isdirectory(v:val.dir)')
  2035. call s:progress_bar(2, bar, len(total))
  2036. for origin in [1, 0]
  2037. let plugs = reverse(sort(items(filter(copy(total), (origin ? '' : '!').'(has_key(v:val, "commit") || has_key(v:val, "tag"))'))))
  2038. if empty(plugs)
  2039. continue
  2040. endif
  2041. call s:append_ul(2, origin ? 'Pending updates:' : 'Last update:')
  2042. for [k, v] in plugs
  2043. let range = origin ? '..origin/'.v.branch : 'HEAD@{1}..'
  2044. 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)
  2045. if !empty(diff)
  2046. let ref = has_key(v, 'tag') ? (' (tag: '.v.tag.')') : has_key(v, 'commit') ? (' '.v.commit) : ''
  2047. call append(5, extend(['', '- '.k.':'.ref], map(s:lines(diff), 's:format_git_log(v:val)')))
  2048. let cnts[origin] += 1
  2049. endif
  2050. let bar .= '='
  2051. call s:progress_bar(2, bar, len(total))
  2052. normal! 2G
  2053. redraw
  2054. endfor
  2055. if !cnts[origin]
  2056. call append(5, ['', 'N/A'])
  2057. endif
  2058. endfor
  2059. call setline(1, printf('%d plugin(s) updated.', cnts[0])
  2060. \ . (cnts[1] ? printf(' %d plugin(s) have pending updates.', cnts[1]) : ''))
  2061. if cnts[0] || cnts[1]
  2062. nnoremap <silent> <buffer> <cr> :silent! call <SID>preview_commit()<cr>
  2063. nnoremap <silent> <buffer> o :silent! call <SID>preview_commit()<cr>
  2064. endif
  2065. if cnts[0]
  2066. nnoremap <silent> <buffer> X :call <SID>revert()<cr>
  2067. echo "Press 'X' on each block to revert the update"
  2068. endif
  2069. normal! gg
  2070. setlocal nomodifiable
  2071. endfunction
  2072. function! s:revert()
  2073. if search('^Pending updates', 'bnW')
  2074. return
  2075. endif
  2076. let name = s:find_name(line('.'))
  2077. if empty(name) || !has_key(g:plugs, name) ||
  2078. \ input(printf('Revert the update of %s? (y/N) ', name)) !~? '^y'
  2079. return
  2080. endif
  2081. call s:system('git reset --hard HEAD@{1} && git checkout '.s:esc(g:plugs[name].branch), g:plugs[name].dir)
  2082. setlocal modifiable
  2083. normal! "_dap
  2084. setlocal nomodifiable
  2085. echo 'Reverted'
  2086. endfunction
  2087. function! s:snapshot(force, ...) abort
  2088. call s:prepare()
  2089. setf vim
  2090. call append(0, ['" Generated by vim-plug',
  2091. \ '" '.strftime("%c"),
  2092. \ '" :source this file in vim to restore the snapshot',
  2093. \ '" or execute: vim -S snapshot.vim',
  2094. \ '', '', 'PlugUpdate!'])
  2095. 1
  2096. let anchor = line('$') - 3
  2097. let names = sort(keys(filter(copy(g:plugs),
  2098. \'has_key(v:val, "uri") && !has_key(v:val, "commit") && isdirectory(v:val.dir)')))
  2099. for name in reverse(names)
  2100. let sha = s:system_chomp('git rev-parse --short HEAD', g:plugs[name].dir)
  2101. if !empty(sha)
  2102. call append(anchor, printf("silent! let g:plugs['%s'].commit = '%s'", name, sha))
  2103. redraw
  2104. endif
  2105. endfor
  2106. if a:0 > 0
  2107. let fn = expand(a:1)
  2108. if filereadable(fn) && !(a:force || s:ask(a:1.' already exists. Overwrite?'))
  2109. return
  2110. endif
  2111. call writefile(getline(1, '$'), fn)
  2112. echo 'Saved as '.a:1
  2113. silent execute 'e' s:esc(fn)
  2114. setf vim
  2115. endif
  2116. endfunction
  2117. function! s:split_rtp()
  2118. return split(&rtp, '\\\@<!,')
  2119. endfunction
  2120. let s:first_rtp = s:escrtp(get(s:split_rtp(), 0, ''))
  2121. let s:last_rtp = s:escrtp(get(s:split_rtp(), -1, ''))
  2122. if exists('g:plugs')
  2123. let g:plugs_order = get(g:, 'plugs_order', keys(g:plugs))
  2124. call s:upgrade_specs()
  2125. call s:define_commands()
  2126. endif
  2127. let &cpo = s:cpo_save
  2128. unlet s:cpo_save