plug.vim 73 KB

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