diff options
| -rw-r--r-- | autoload/beancount.vim | 172 | ||||
| -rw-r--r-- | ftplugin/beancount.vim | 2 |
2 files changed, 129 insertions, 45 deletions
diff --git a/autoload/beancount.vim b/autoload/beancount.vim index 7ac7f3e..e7e4901 100644 --- a/autoload/beancount.vim +++ b/autoload/beancount.vim @@ -1,3 +1,9 @@ +" Equivalent to python's startswith +" Matches based on user's ignorecase preference +function! s:startswith(string, prefix) + return strpart(a:string, 0, strlen(a:prefix)) == a:prefix +endfunction + " Align currency on decimal point. function! beancount#align_commodity(line1, line2) " Saving cursor position to adjust it if necessary. @@ -47,8 +53,12 @@ function! s:sort_accounts_by_depth(name1, name2) return depth1 == depth2 ? 0 : depth1 > depth2 ? 1 : -1 endfunction -" Complete account name. -function! beancount#complete_account(findstart, base) +let s:directives = ["open", "close", "commodity", "txn", "balance", "pad", "note", "document", "price", "event", "query", "custom"] + +" ------------------------------ +" Completion functions +" ------------------------------ +function! beancount#complete(findstart, base) if a:findstart let l:col = searchpos('\s', "bn", line("."))[1] if col == 0 @@ -58,15 +68,88 @@ function! beancount#complete_account(findstart, base) endif endif + let l:partial_line = strpart(getline("."), 0, getpos(".")[2]) + " Match directive types + if l:partial_line =~# '^\d\d\d\d\(-\|/\)\d\d\1\d\d \S*$' + return beancount#complete_basic(s:directives, a:base, '') + endif + + let l:two_tokens = searchpos('\S\+\s', "bn", line("."))[1] + let l:prev_token = strpart(getline("."), l:two_tokens, getpos(".")[2] - l:two_tokens) + " Match curriences if previous token is number + if l:prev_token =~ '^\d\+\([\.,]\d\+\)*' + call beancount#load_currencies() + return beancount#complete_basic(b:beancount_currencies, a:base, '') + endif + + let l:first = strpart(a:base, 0, 1) + let l:rest = strpart(a:base, 1) + if l:first == "#" + call beancount#load_tags() + return beancount#complete_basic(b:beancount_tags, l:rest, '#') + elseif l:first == "^" + call beancount#load_links() + return beancount#complete_basic(b:beancount_links, l:rest, '^') + elseif l:first == '"' + call beancount#load_payees() + return beancount#complete_basic(b:beancount_payees, l:rest, '"') + else + call beancount#load_accounts() + return beancount#complete_account(a:base) + endif +endfunction + +function! s:get_root() + if exists('b:beancount_root') + return b:beancount_root + endif + return expand('%') +endfunction + +function! beancount#load_accounts() if !exists('b:beancount_accounts') - if exists('b:beancount_root') - let l:root = b:beancount_root - else - let l:root = expand('%') - endif + let l:root = s:get_root() let b:beancount_accounts = beancount#find_accounts(l:root) endif +endfunction + +function! beancount#load_tags() + if !exists('b:beancount_tags') + let l:root = s:get_root() + let b:beancount_tags = beancount#find_tags(l:root) + endif +endfunction + +function! beancount#load_links() + if !exists('b:beancount_links') + let l:root = s:get_root() + let b:beancount_links = beancount#find_links(l:root) + endif +endfunction + +function! beancount#load_currencies() + if !exists('b:beancount_currencies') + let l:root = s:get_root() + let b:beancount_currencies = beancount#find_currencies(l:root) + endif +endfunction +function! beancount#load_payees() + if !exists('b:beancount_payees') + let l:root = s:get_root() + let b:beancount_payees = beancount#find_payees(l:root) + endif +endfunction + +" General completion function +function! beancount#complete_basic(input, base, prefix) + let l:matches = filter(copy(a:input), 's:startswith(v:val, a:base)') + + return map(l:matches, 'a:prefix . v:val') +endfunction + +" Complete account name. +function! beancount#complete_account(base) if g:beancount_account_completion ==? 'chunks' let l:pattern = '^\V' . substitute(a:base, ":", '\\[^:]\\*:', "g") . '\[^:]\*' else @@ -88,45 +171,46 @@ function! beancount#complete_account(findstart, base) return l:matches endfunction -" Get list of acounts. -function! beancount#find_accounts(root_file) - python << EOM -import collections -import os -import re -import sys +function! beancount#query_single(root_file, query) +python << EOF import vim +import subprocess +import os + +# We intentionally want to ignore stderr so it doesn't mess up our query processing +output = subprocess.check_output(['bean-query', vim.eval('a:root_file'), vim.eval('a:query')], stderr=open(os.devnull, 'w')).split('\n') +print(output) +output = output[2:] + +result_list = [y for y in (x.strip() for x in output) if y] + +vim.command('return [{}]'.format(','.join(repr(x) for x in sorted(result_list)))) +EOF +endfunction + +" Get list of accounts. +function! beancount#find_accounts(root_file) + return beancount#query_single(a:root_file, 'select distinct account;') +endfunction + +" Get list of tags. +function! beancount#find_tags(root_file) + return beancount#query_single(a:root_file, 'select distinct tags;') +endfunction + +" Get list of links. +function! beancount#find_links(root_file) + return beancount#query_single(a:root_file, 'select distinct links;') +endfunction + +" Get list of currencies. +function! beancount#find_currencies(root_file) + return beancount#query_single(a:root_file, 'select distinct currency;') +endfunction -RE_INCLUDE = re.compile(r'^include\s+"([^\n"]+)"') -RE_ACCOUNT = re.compile(r'^\d{4,}-\d{2}-\d{2}\s+open\s+(\S+)') - -def combine_paths(old, new): - return os.path.normpath( - new if os.path.isabs(new) else os.path.join(old, new)) - -def parse_file(fh, files, accounts): - for line in fh: - m = RE_INCLUDE.match(line) - if m: files.append(combine_paths(os.path.dirname(fh.name), m.group(1))) - m = RE_ACCOUNT.match(line) - if m: accounts.add(m.group(1)) - -files = collections.deque([vim.eval("a:root_file")]) -accounts = set() -seen = set() -while files: - current = files.popleft() - if current in seen: - continue - seen.add(current) - try: - with open(current, 'r') as fh: - parse_file(fh, files, accounts) - except IOError as err: - pass - -vim.command('return [{}]'.format(','.join(repr(x) for x in sorted(accounts)))) -EOM +" Get list of payees. +function! beancount#find_payees(root_file) + return beancount#query_single(a:root_file, 'select distinct payee;') endfunction " Call bean-doctor on the current line and dump output into a scratch buffer diff --git a/ftplugin/beancount.vim b/ftplugin/beancount.vim index dc35999..c99e2de 100644 --- a/ftplugin/beancount.vim +++ b/ftplugin/beancount.vim @@ -32,4 +32,4 @@ command! -buffer -range GetContext \ :call beancount#get_context() " Omnifunc for account completion. -setl omnifunc=beancount#complete_account +setl omnifunc=beancount#complete |