plug.vim 73 KB

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