Port to Neovim: Emulate 'out_io': 'buffer' behavior for Neovim

Author: mcepl

plugin/acme.vim

@@ -239,6 +239,58 @@ function s:GetJobBufNr(job)
 	endif
 endfunc
 
+function s:NvimOut(id, data, event) dict
+	if empty(a:data) | return | endif
+	let b = self.buf
+	if bufexists(b)
+		let mod = getbufvar(b, '&modifiable')
+		call setbufvar(b, '&modifiable', 1)
+		let data = copy(a:data)
+		let last = getbufline(b, '$')[0]
+		let data[0] = last . data[0]
+		call setbufline(b, '$', data[0])
+		if len(data) > 1
+			call appendbufline(b, '$', data[1:])
+		endif
+		call setbufvar(b, '&modifiable', mod)
+	endif
+	if has_key(self, 'callback') && !empty(self.callback)
+		call call(self.callback, [a:id, ''])
+	endif
+endfunc
+
+function s:NvimExit(id, status, event) dict
+	if has_key(s:nvim_jobs, a:id)
+		call remove(s:nvim_jobs, a:id)
+	endif
+	call s:Exited(a:id, a:status)
+endfunc
+
+function s:JobStartNvim(cmd, outb, ctxb, opts, inp)
+	let job_opts = {
+		\ 'on_exit': function('s:NvimExit'),
+		\ 'on_stdout': function('s:NvimOut'),
+		\ 'on_stderr': function('s:NvimOut'),
+		\ 'buf': a:outb,
+		\ 'callback': get(a:opts, 'callback', ''),
+		\ }
+	if has_key(a:opts, 'cwd')
+		let job_opts.cwd = a:opts.cwd
+	endif
+	let env = s:SetEnv(s:JobEnv(a:outb))
+	let job = jobstart(s:Argv(a:cmd), job_opts)
+	call s:SetEnv(env)
+	if job <= 0
+		return
+	endif
+	let s:nvim_jobs[job] = job_opts
+	call s:Started(job, s:BufWin(a:outb) != 0 ? a:outb : a:ctxb, a:cmd)
+	if a:inp != ''
+		call chansend(job, a:inp)
+		call chanclose(job, 'stdin')
+	endif
+endfunc
+
 function s:JobEnv(buf)
 	return {
 		\ 'ACMEVIMBUF': a:buf,
@@ -260,6 +312,9 @@ function s:SetEnv(env)
 endfunc
 
 function s:JobStart(cmd, outb, ctxb, opts, inp)
+	if has('nvim')
+		return s:JobStartNvim(a:cmd, a:outb, a:ctxb, a:opts, a:inp)
+	endif
 	let opts = {
 		\ 'exit_cb': 's:Exited',
 		\ 'err_io': 'out',
@@ -803,7 +858,7 @@ endfunc
 
 function s:Scroll(topline)
 	let v = winsaveview()
-	v.topline = a:topline
+	let v.topline = a:topline
 	call winrestview(v)
 endfunc
 
@@ -850,8 +905,8 @@ function s:Layout(col)
 			let s = float2nr(h / (n * (n > s:tops ? 1.75 : 1)))
 		endif
 		call win_move_statusline(win_id2win(w) - 1, winheight(w) - s)
-		h -= s
-		n -= 1
+		let h -= s
+		let n -= 1
 	endfor
 	call timer_start(0, {_ -> s:Fit(a:col)})
 endfunc
@@ -1098,7 +1153,7 @@ function s:Pty(b)
 	if !has_key(s:scratch, a:b) || w == 0
 		return
 	endif
-	s:scratch[a:b].pty = 1
+	let s:scratch[a:b].pty = 1
 	call win_execute(w, 'call s:PtyMap()')
 endfunc
 
@@ -1305,13 +1360,13 @@ if s:ctrlexe != ''
 		let s:ctrl = jobstart([s:ctrlexe], {
 			\ 'on_stdout': function('s:NvimCtrlRecv'),
 			\ 'rpc': 0,
-		\ })
+			\ })
 	else
 		let s:ctrl = job_start([s:ctrlexe], {
 			\ 'callback': 's:CtrlRecv',
 			\ 'err_io': 'null',
 			\ 'mode': 'raw',
-		\ })
+			\ })
 	endif
 	let $EDITOR = s:ctrlexe
 endif