From 28c154081b41ac10ef9684ac9a8b3393bcd1b1b3 Mon Sep 17 00:00:00 2001
From: Nick Jensen <nickspoon@gmail.com>
Date: Sun, 3 Jul 2022 23:28:14 +1200
Subject: [PATCH] Add foldtext for testrunner folds

---
 autoload/OmniSharp/testrunner.vim    | 54 +++++++++++++++++++++++-----
 ftplugin/omnisharptest/OmniSharp.vim |  2 ++
 2 files changed, 47 insertions(+), 9 deletions(-)

diff --git a/autoload/OmniSharp/testrunner.vim b/autoload/OmniSharp/testrunner.vim
index 630ca4d0e..0d8a07716 100644
--- a/autoload/OmniSharp/testrunner.vim
+++ b/autoload/OmniSharp/testrunner.vim
@@ -33,6 +33,42 @@ function! OmniSharp#testrunner#Init(buffers) abort
 endfunction
 
 
+function! OmniSharp#testrunner#FoldText() abort
+  let line = getline(v:foldstart)
+  if line =~# '^;'
+    " Project
+    let projectkey = matchlist(line, '^\S\+')[0]
+    let [assembly, _] = split(projectkey, ';')
+    let ntests = 0
+    for filename in keys(s:tests[projectkey].files)
+      let ntests += len(s:tests[projectkey].files[filename].tests)
+    endfor
+    let err = match(line, '; ERROR$') == -1 ? '' : ' ERROR'
+    return printf('%s [%d]%s', assembly, ntests, err)
+  elseif line =~# '^    \f'
+    " File
+    let filename = trim(line)
+    let fullpath = fnamemodify(filename, ':p')
+    let displayname = matchlist(filename, '^\f\{-}\([^/\\]\+\)\.csx\?$')[1]
+    " Position the cursor so that search() is relative to the fold, not the
+    " actual cursor position
+    let winview = winsaveview()
+    call cursor(v:foldstart, 0)
+    let projectline = search('^;', 'bcnWz')
+    call winrestview(winview)
+    let projectkey = matchlist(getline(projectline), '^\S\+')[0]
+    let ntests = len(s:tests[projectkey].files[fullpath].tests)
+    return printf('    %s [%d]', displayname, ntests)
+  elseif line =~# '^<'
+    return printf('  Error details (%d lines)', v:foldend - v:foldstart + 1)
+  elseif line =~# '^>'
+    return printf('            Results (%d lines)', v:foldend - v:foldstart + 1)
+  elseif line =~# '^//'
+    return printf('          Output (%d lines)', v:foldend - v:foldstart + 1)
+  endif
+  return printf('%s (%d lines)', line, v:foldend - v:foldstart + 1)
+endfunction
+
 function! OmniSharp#testrunner#Log(message) abort
   call extend(s:current.log, a:message)
 endfunction
@@ -43,9 +79,9 @@ function! OmniSharp#testrunner#Run() abort
   let line = getline('.')
   if line =~# '^;'
     " Project selected - run all tests
-    let projectname = matchlist(getline('.'), '^\S\+')[0]
-    let filenames = filter(keys(s:tests[projectname].files),
-    \ {_,f -> s:tests[projectname].files[f].visible})
+    let projectkey = matchlist(getline('.'), '^\S\+')[0]
+    let filenames = filter(keys(s:tests[projectkey].files),
+    \ {_,f -> s:tests[projectkey].files[f].visible})
     call OmniSharp#actions#test#RunInFile(1, filenames)
   elseif line =~# '^    \f'
     " File selected
@@ -66,14 +102,14 @@ function! OmniSharp#testrunner#Remove() abort
   let line = getline('.')
   if line =~# '^;'
     " Project selected - run all tests
-    let projectname = matchlist(getline('.'), '^\S\+')[0]
-    let s:tests[projectname].visible = 0
+    let projectkey = matchlist(getline('.'), '^\S\+')[0]
+    let s:tests[projectkey].visible = 0
   elseif line =~# '^    \f'
     " File selected
     let filename = fnamemodify(trim(line), ':p')
     let projectline = search('^;', 'bcnWz')
-    let projectname = matchlist(getline(projectline), '^\S\+')[0]
-    let s:tests[projectname].files[filename].visible = 0
+    let projectkey = matchlist(getline(projectline), '^\S\+')[0]
+    let s:tests[projectkey].files[filename].visible = 0
   else
     let test = s:utils.findTest()
     let test.state = 'hidden'
@@ -487,11 +523,11 @@ function! s:utils.findTest() abort
   if testline > 0
     let testname = matchlist(getline(testline), '[-|*!]        \zs.*$')[0]
     let projectline = search('^;', 'bcnWz')
-    let projectname = matchlist(getline(projectline), '^\S\+')[0]
+    let projectkey = matchlist(getline(projectline), '^\S\+')[0]
     let fileline = search('^    \f', 'bcnWz')
     let filename = matchlist(getline(fileline), '^    \zs.*$')[0]
     let filename = fnamemodify(filename, ':p')
-    return s:tests[projectname].files[filename].tests[testname]
+    return s:tests[projectkey].files[filename].tests[testname]
   endif
   return {}
 endfunction
diff --git a/ftplugin/omnisharptest/OmniSharp.vim b/ftplugin/omnisharptest/OmniSharp.vim
index fa8919be0..0c4e0b860 100644
--- a/ftplugin/omnisharptest/OmniSharp.vim
+++ b/ftplugin/omnisharptest/OmniSharp.vim
@@ -26,3 +26,5 @@ call s:map('n', '<F6>', 'omnisharp_testrunner_debug')
 call s:map('n', '<F9>', 'omnisharp_testrunner_set_breakpoints')
 call s:map('n', '<CR>', 'omnisharp_testrunner_navigate')
 call s:map('n', 'dd', 'omnisharp_testrunner_remove')
+
+set foldtext=OmniSharp#testrunner#FoldText()