" TODO " handle multiwidth characters " handle multiple windows " support neovim (extmark) function! s:getlines_except_folded(from, to) let lines = [] let from = type(a:from) == v:t_number ? a:from : line(a:from) let to = type(a:to) == v:t_number ? a:to : line(a:to) for lnum in range(from, to) let fold_end = foldclosedend(lnum) if fold_end > -1 let lnum = fold_end else call add(lines, {'lnum': lnum, 'line': getline(lnum)}) endif endfor return lines endfunction function! s:double(num) abort return a:num * a:num endfunction function! s:winids() abort return map(getwininfo(), 'v:val.winid') endfunction function! s:separate(str, idx) abort return [slice(a:str, 0, a:idx), slice(a:str, a:idx)] endfunction function! s:sep3(str, from, to) abort let [tmp, a3] = s:separate(a:str, a:to) let [a1, a2] = s:separate(tmp, a:from) return [a1, a2, a3] endfunction " let text='hello world super owo end one two three' function! s:recursive_fuzzy_match_in_line(line, query, off = 0) abort let line = a:line let [match, cols, scores] = matchfuzzypos([line], a:query) if empty(match) return [] endif let cols = map(cols[0], 'v:val+1') let from = cols[0] - 1 let to = cols[-1] let len = to - from let [pre_matched, matched, post_matched] = s:sep3(line, from, to) let matches = [] " 日本語混じりのときに壊れないか確認が必要 " 1行が長いとそれだけでスコアが下がるため、 " マッチ範囲に再度マッチさせてスコアを再計算する " これだと必ず単語頭マッチになってしまってよくないか？ let score = matchfuzzypos([matched], a:query)[2][0] call add(matches, {'score': score, 'cols': map(cols, 'v:val+' .. a:off), 'matched': matched, 'len': len}) " マッチ前後の範囲で再帰的にマッチを探す call extend(matches, s:recursive_fuzzy_match_in_line(pre_matched, a:query, a:off)) call extend(matches, s:recursive_fuzzy_match_in_line(post_matched, a:query, to + a:off)) return matches endfunction " super long long sentence that has query at once owo super long long sentence that has query at once super long long sentence that has query at once let s:prop_type = 'prop_type' let s:prop_type_primary = 'prop_type_primary' highlight FuzzyJumpMarker term=standout gui=underline,bold guifg=#262626 guibg=#ffd700 highlight FuzzyJumpMarkerPrimary term=standout gui=underline,bold guifg=#aa2626 guibg=#ffd700 let s:jump_markers = 'ABCDEFGHIJKLMNOPQRSTUVWXYZ' function! s:compare_matches(a, b) abort " short length is strong let len_diff = a:a.len - a:b.len if len_diff != 0 return len_diff endif " nearby cursor is strong let lnum_diff = s:double(s:cursor_pos[0] - a:a.lnum) - s:double(s:cursor_pos[0] - a:b.lnum) if lnum_diff != 0 return lnum_diff endif " high score is strong return a:b.score - a:a.score endfunction function! s:gen_fuzzy_match_specs(query) abort let matches = [] if a:query == '' return matches endif for line_spec in s:getlines_except_folded('w0', 'w$') let match = s:recursive_fuzzy_match_in_line(line_spec.line, a:query) if !empty(match) call extend(matches, map(match, {_,v ->extend(v, {'lnum': line_spec.lnum}) } )) endif endfor " rank: short length is strong, high score is strong in same length call sort(matches, funcref('s:compare_matches')) let matches = matches[:strlen(s:jump_markers)-1] let idx = 0 for spec in matches let type = idx == 0 ? s:prop_type_primary : s:prop_type let spec.prop = {'text': s:jump_markers[idx], 'type': type} let idx += 1 endfor return matches endfunction function! s:add_highlights() abort let s:match_ids = [] for lnum in range(line('w0'), line('w$')) call add(s:match_ids, matchaddpos('Comment', [lnum])) endfor for match_spec in s:match_specs if match_spec.len > 1 call add(s:match_ids, matchaddpos('Visual', [[match_spec.lnum, match_spec.cols[0]+1, match_spec.len-1]])) endif for col in match_spec.cols[1:] call add(s:match_ids, matchaddpos('IncSearch', [[match_spec.lnum, col]])) endfor call add(s:match_ids, matchaddpos('Conceal', [[match_spec.lnum, match_spec.cols[0]]])) call prop_add(match_spec.lnum, match_spec.cols[0], match_spec.prop) endfor endfunction function! s:clear_highlights() abort silent! call prop_remove({'all': 1, 'type': s:prop_type}, line('w0'), line('w$')) silent! call prop_remove({'all': 1, 'type': s:prop_type_primary}, line('w0'), line('w$')) for id in get(s:, 'match_ids', []) silent! call matchdelete(id) endfor endfunction function! s:on_input() abort call s:clear_highlights() let query = getcmdline() let s:last_query = query let last_char = query[-1:] if last_char =~# '\u' let keys = "\" if s:jump_markers =~# last_char let s:selected_marker = last_char let keys ..= "\" endif call feedkeys(keys, 'n') else let s:match_specs = s:gen_fuzzy_match_specs(query) call s:add_highlights() endif redraw endfunction function! s:on_leave() abort call s:clear_highlights() autocmd! fuzzy_jump_augroup if exists('s:save_conceal') let &conceallevel = s:save_conceal.level let &concealcursor = s:save_conceal.cursor unlet! s:save_conceal endif endfunction function! s:on_enter() abort let s:cursor_pos = getpos('.')[1:2] let s:save_conceal = {} let s:save_conceal.level = &conceallevel let s:save_conceal.cursor = &concealcursor set conceallevel=2 set concealcursor=nc call prop_type_delete(s:prop_type, {}) call prop_type_delete(s:prop_type_primary, {}) call prop_type_add(s:prop_type, {'highlight': 'FuzzyJumpMarker'}) call prop_type_add(s:prop_type_primary, {'highlight': 'FuzzyJumpMarkerPrimary'}) call s:on_input() endfunction function! s:start_fuzzy_match(repeat = 0) abort augroup fuzzy_jump_augroup autocmd CmdlineEnter @ call s:on_enter() autocmd CmdlineChanged @ call s:on_input() augroup END autocmd CmdlineLeave @ ++once call s:on_leave() let s:selected_marker = '' let q = a:repeat ? get(s:, 'last_query', '') : '' let last_query = input('fuzzy match > ', q) if last_query == '' return endif let s:last_query = last_query if s:selected_marker == '' let s:selected_marker = s:jump_markers[0] endif if exists('s:match_specs') let match_spec = s:match_specs[stridx(s:jump_markers, s:selected_marker)] call cursor(match_spec.lnum, match_spec.cols[0]) unlet! s:match_specs endif endfunction nnoremap f call start_fuzzy_match() nnoremap f call start_fuzzy_match(1)