plug.vim 70 KB

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