plug.vim 72 KB

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