plug.vim 81 KB

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