plug.vim 64 KB

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