" MIT License. Copyright (c) 2013 Bailey Ling.
" vim: et ts=2 sts=2 sw=2

function! airline#extensions#undotree#apply(...)
  if exists('t:undotree')
    if &ft == 'undotree' && exists('*t:undotree.GetStatusLine')
      call airline#extensions#apply_left_override('undo', '%{t:undotree.GetStatusLine()}')
    else
      call airline#extensions#apply_left_override('undotree', '%f')
    endif

    if &ft == 'diff' && exists('*t:diffpanel.GetStatusLine')
      call airline#extensions#apply_left_override('diff', '%{t:diffpanel.GetStatusLine()}')
    endif
  endif
endfunction

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