plug.vim 77 KB

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