plug.vim 78 KB

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