Add a function to display the current line's beancount context in a scratch window

1 files changed, 9 insertions, 0 deletions

diff --git a/autoload/beancount.vim b/autoload/beancount.vim
index ba2a5cc..7ac7f3e 100644
--- a/autoload/beancount.vim
+++ b/autoload/beancount.vim
@@ -128,3 +128,12 @@ while files:
 vim.command('return [{}]'.format(','.join(repr(x) for x in sorted(accounts))))
 EOM
 endfunction
+
+" Call bean-doctor on the current line and dump output into a scratch buffer
+function! beancount#get_context()
+    let context = system('bean-doctor context ' . expand('%') . ' ' . line('.'))
+    botright new
+    setlocal buftype=nofile bufhidden=hide noswapfile
+    call append(0, split(context, '\v\n'))
+    normal! gg
+endfunction