floaterm.vim

" vim:sw=2:
" ============================================================================
" FileName: floaterm.vim
" Author: voldikss <dyzplus@gmail.com>
" GitHub: https://github.com/voldikss
" ============================================================================

" ----------------------------------------------------------------------------
" wrapper function for `floaterm#new()` and `floaterm#update()` since they
" share the same argument: `config`
" ----------------------------------------------------------------------------
function! floaterm#run(action, bang, rangeargs, cmdargs) abort
  let [cmd, config] = floaterm#cmdline#parse(a:cmdargs)
  if a:action == 'new'
    let [visualmode, range, line1, line2] = a:rangeargs
    if range > 0
      let lines = floaterm#util#get_selected_text(visualmode, range, line1, line2)
    endif
    let bufnr = floaterm#new(a:bang, cmd, {}, config)
    if range > 0 && !empty(lines)
      call floaterm#terminal#send(bufnr, lines)
    endif
  elseif a:action == 'update'
    call floaterm#update(config)
  endif
endfunction

" create a floaterm. return bufnr of the terminal
" argument `jobopts` is passed by user in the case using this function as API
function! floaterm#new(bang, cmd, jobopts, config) abort
  let env = floaterm#util#setenv()
  let vim_version = floaterm#util#vim_version()
  if vim_version[0] == 'nvim' && vim_version[1] <= '0.4.4'
    for [name, value] in items(env)
      call setenv(name, value)
    endfor
  else
    call floaterm#util#deep_extend(a:jobopts, {'env': env})
  endif
  if !empty(a:cmd)
    let wrappers_path = globpath(&runtimepath, 'autoload/floaterm/wrapper/*vim', 0, 1)
    let wrappers = map(wrappers_path, "substitute(fnamemodify(v:val, ':t'), '\\..\\{-}$', '', '')")
    let maybe_wrapper = split(a:cmd, '\s')[0]
    if index(wrappers, maybe_wrapper) >= 0
      try
        let [shell, shellslash, shellcmdflag, shellxquote] = floaterm#util#use_sh_or_cmd()
        let WrapFunc = function(printf('floaterm#wrapper#%s#', maybe_wrapper))
        " NOTE: a:jobopts and a:config can be changed in WrapFunc
        let [send2shell, newcmd] = WrapFunc(a:cmd, a:jobopts, a:config)
        if send2shell
          let bufnr = floaterm#terminal#open(-1, g:floaterm_shell, a:jobopts, a:config)
          call floaterm#terminal#send(bufnr, [newcmd])
        else
          let bufnr = floaterm#terminal#open(-1, newcmd, a:jobopts, a:config)
        endif
      finally
        let [&shell, &shellslash, &shellcmdflag, &shellxquote] = [shell, shellslash, shellcmdflag, shellxquote]
      endtry
    elseif a:bang
      let bufnr = floaterm#terminal#open(-1, g:floaterm_shell, a:jobopts, a:config)
      call floaterm#terminal#send(bufnr, [a:cmd])
    else
      let bufnr = floaterm#terminal#open(-1, a:cmd, a:jobopts, a:config)
    endif
  else
    let bufnr = floaterm#terminal#open(-1, g:floaterm_shell, a:jobopts, a:config)
  endif
  return bufnr
endfunction

" ----------------------------------------------------------------------------
" toggle on/off the floaterm named `name`
" ----------------------------------------------------------------------------
function! floaterm#toggle(bang, bufnr, name)  abort
  if a:bang
    let found_winnr = floaterm#window#find()
    if found_winnr > 0
      call floaterm#hide(1, 0, '')
    else
      let buffers = floaterm#buflist#gather()
      if empty(buffers)
        let curr_bufnr = floaterm#new(v:true, '', {}, {})
      else
        call floaterm#show(1, 0, '')
      endif
    endif
    return
  endif

  let bufnr = a:bufnr
  if bufnr == 0 && !empty(a:name)
    let bufnr = floaterm#terminal#get_bufnr(a:name)
  endif

  if bufnr == -1
    call floaterm#new(a:bang, '', {}, {'name': a:name})
  elseif bufnr == 0
    if &filetype == 'floaterm'
      call floaterm#window#hide(bufnr('%'))
    else
      let found_winnr = floaterm#window#find()
      if found_winnr > 0
        execute found_winnr . 'wincmd w'
      else
        call floaterm#curr()
      endif
    endif
  elseif getbufvar(bufnr, 'floaterm_winid', -1) != -1
    if bufnr == bufnr('%')
      call floaterm#window#hide(bufnr)
    elseif bufwinnr(bufnr) > -1
      execute bufwinnr(bufnr) . 'wincmd w'
    else
      call floaterm#terminal#open_existing(bufnr)
    endif
  else
    call floaterm#util#show_msg('No floaterms with the bufnr or name', 'error')
  endif
endfunction

" ----------------------------------------------------------------------------
" toggle fullscreen on/off the floaterm
" ----------------------------------------------------------------------------
function! floaterm#toggle_fullscreen()  abort
  let bufnr = bufnr('%')
  call floaterm#update({'fullscreen': !getbufvar(bufnr, 'floaterm_fullscreen', v:false)})
endfunction

" ----------------------------------------------------------------------------
" toggle border on/off the floaterm
" ----------------------------------------------------------------------------
function! floaterm#toggle_border()  abort
  let bufnr = bufnr('%')
  call floaterm#update({'show_border': !getbufvar(bufnr, 'floaterm_show_border', v:false)})
endfunction

" ----------------------------------------------------------------------------
" toggle title on/off the floaterm
" ----------------------------------------------------------------------------
function! floaterm#toggle_title()  abort
  let bufnr = bufnr('%')
  call floaterm#update({'show_title': !getbufvar(bufnr, 'floaterm_show_title', v:false)})
endfunction

" ----------------------------------------------------------------------------
" update the attributes of a floaterm
" ----------------------------------------------------------------------------
function! floaterm#update(config) abort
  if &filetype !=# 'floaterm'
    call floaterm#util#show_msg('You have to be in a floaterm window to change window config.', 'error')
    return
  endif

  let bufnr = bufnr('%')
  call floaterm#window#hide(bufnr)
  call floaterm#config#set_all(bufnr, a:config)
  call floaterm#terminal#open_existing(bufnr)
endfunction

function! floaterm#next()  abort
  let next_bufnr = floaterm#buflist#next()
  if next_bufnr == -1
    let msg = 'No more floaterms'
    call floaterm#util#show_msg(msg, 'warning')
  else
    call floaterm#terminal#open_existing(next_bufnr)
  endif
endfunction

function! floaterm#prev()  abort
  let prev_bufnr = floaterm#buflist#prev()
  if prev_bufnr == -1
    let msg = 'No more floaterms'
    call floaterm#util#show_msg(msg, 'warning')
  else
    call floaterm#terminal#open_existing(prev_bufnr)
  endif
endfunction

function! floaterm#curr() abort
  let curr_bufnr = floaterm#buflist#curr()
  if curr_bufnr == -1
    let curr_bufnr = floaterm#new(v:true, '', {}, {})
  else
    call floaterm#terminal#open_existing(curr_bufnr)
  endif
  return curr_bufnr
endfunction

function! floaterm#first() abort
  let first_bufnr = floaterm#buflist#first()
  if first_bufnr == -1
    call floaterm#util#show_msg('No more floaterms', 'warning')
  else
    call floaterm#terminal#open_existing(first_bufnr)
  endif
endfunction

function! floaterm#last() abort
  let last_bufnr = floaterm#buflist#last()
  if last_bufnr == -1
    call floaterm#util#show_msg('No more floaterms', 'warning')
  else
    call floaterm#terminal#open_existing(last_bufnr)
  endif
endfunction

function! floaterm#kill(bang, bufnr, name) abort
  if a:bang
    for bufnr in floaterm#buflist#gather()
      call floaterm#terminal#kill(bufnr)
    endfor
    return
  endif

  let bufnr = a:bufnr
  if bufnr == 0 && !empty(a:name)
    let bufnr = floaterm#terminal#get_bufnr(a:name)
  endif
  if bufnr == 0 || bufnr == -1
    let bufnr = floaterm#buflist#curr()
  endif

  if bufnr > 0
    call floaterm#terminal#kill(bufnr)
  else
    call floaterm#util#show_msg('No floaterms with the bufnr or name', 'error')
  endif
endfunction

function! floaterm#show(bang, bufnr, name) abort
  if a:bang
    call floaterm#hide(1, 0, '')
    for bufnr in floaterm#buflist#gather()
      call floaterm#terminal#open_existing(bufnr)
    endfor
    return
  endif

  let bufnr = a:bufnr
  if bufnr == 0 && !empty(a:name)
    let bufnr = floaterm#terminal#get_bufnr(a:name)
  endif
  if bufnr == 0 || bufnr == -1
    let bufnr = floaterm#buflist#curr()
  endif

  if bufnr > 0
    call floaterm#terminal#open_existing(bufnr)
  else
    call floaterm#util#show_msg('No floaterms with the bufnr or name', 'error')
  endif
endfunction

function! floaterm#hide(bang, bufnr, name) abort
  if a:bang
    for bufnr in floaterm#buflist#gather()
      call floaterm#window#hide(bufnr)
    endfor
    return
  endif

  let bufnr = a:bufnr
  if bufnr == 0 && !empty(a:name)
    let bufnr = floaterm#terminal#get_bufnr(a:name)
  endif
  if bufnr == 0 || bufnr == -1
    let bufnr = bufnr('%')
  endif

  if bufnr > 0
    call floaterm#window#hide(bufnr)
  else
    call floaterm#util#show_msg('No floaterms with the bufnr or name', 'error')
  endif
endfunction

function! floaterm#send(bang, visualmode, range, line1, line2, argstr) abort
  let [cmd, config] = floaterm#cmdline#parse(a:argstr)
  let termname = get(config, 'name', '')
  if !empty(termname)
    let bufnr = floaterm#terminal#get_bufnr(termname)
    if bufnr == -1
      call floaterm#util#show_msg('No floaterm found with name: ' . termname, 'error')
      return
    endif
  else
    let bufnr = floaterm#buflist#curr()
    if bufnr == -1
      call floaterm#util#show_msg('No more floaterms', 'warning')
      return
    endif
  endif
  if !empty(cmd)
    call floaterm#terminal#send(bufnr, [cmd])
    return
  endif

  let lines = floaterm#util#get_selected_text(a:visualmode, a:range, a:line1, a:line2)
  if empty(lines)
    call floaterm#util#show_msg('No lines were selected', 'error')
    return
  endif

  if a:bang
    let lines = floaterm#util#leftalign_lines(lines)
  endif
  call floaterm#terminal#send(bufnr, lines)
endfunction