plug.vim 64 KB

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