" MIT License. Copyright (c) 2013-2021 Bailey Ling et al.
" Plugin: https://github.com/petobens/poet_v
" vim: et ts=2 sts=2 sw=2

scriptencoding utf-8

let s:spc = g:airline_symbols.space

function! airline#extensions#poetv#init(ext)
  call a:ext.add_statusline_func('airline#extensions#poetv#apply')
endfunction

function! airline#extensions#poetv#apply(...)
  if &filetype =~# 'python'
    if get(g:, 'poetv_loaded', 0)
      let statusline = poetv#statusline()
    else
      let statusline = fnamemodify($VIRTUAL_ENV, ':t')
    endif
    if !empty(statusline)
      call airline#extensions#append_to_section('x',
            \ s:spc.g:airline_right_alt_sep.s:spc.statusline)
    endif
  endif
endfunction

function! airline#extensions#poetv#update()
  if &filetype =~# 'python'
    call airline#extensions#poetv#apply()
    call airline#update_statusline()
  endif
endfunction