plug.vim 64 KB

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